MTP20_IF/mtpcache/useraccount.go

/**
* @Author: zou.yingbin
* @Create  : 2021/1/13 14:07
* @Modify  : 2021/1/13 14:07
 */

package mtpcache

import (
	"mtp2_if/db"
	"mtp2_if/logger"
	"sync"
	"time"
)

var vUsrMgr userAccountMgr

// 用户缓存
type userAccountMgr struct {
	mtx  sync.RWMutex         //锁
	data map[int64]usrAccData //数据[userId]accountName
	last time.Time
}

type usrAccData struct {
	USERID       int64  `json:"userid"  xorm:"'USERID'"`             //用户ID
	USERTYPE     int32  `json:"usertype"  xorm:"'USERTYPE'"`         //账户类型 -  1:交易所 2:机构 3:会员子机构 4:经纪人 5:投资者 6:客户
	ACCOUNTNAME  string `json:"accountname"  xorm:"'ACCOUNTNAME'"`   //账户名称(机构名称)
	PARENTUSERID int32  `json:"parentuserid"  xorm:"'PARENTUSERID'"` //所属机构ID
	MEMBERUSERID int32  `json:"memberuserid"  xorm:"'MEMBERUSERID'"` //所属会员ID
	NICKNAME     string `json:"nickname"  xorm:"'NICKNAME'"`         // 昵称
}

// 加载数据
func (r *userAccountMgr) load() {
	now := time.Now()
	// 间隔3秒以上才重新加载
	if d := now.Sub(r.last); d.Seconds() < 3 {
		return
	}

	r.mtx.Lock()
	defer r.mtx.Unlock()
	e := db.GetEngine()
	if e == nil {
		return
	}

	sqlId := "select t.USERID," +
		"       t.USERTYPE," +
		"       t.ACCOUNTNAME," +
		"       t.PARENTUSERID," +
		"       t.MEMBERUSERID," +
		"       u.nickname" +
		"  from useraccount t" +
		"  left join userinfo u" +
		"    on t.userid = u.userid" +
		" where 1 = 1"

	sData := make([]usrAccData, 0)
	s := e.SQL(sqlId)
	if err := s.Find(&sData); err != nil {
		logger.GetLogger().Errorf("cache load useraccount, %v", err)
	} else {
		r.data = make(map[int64]usrAccData)
		for i := range sData {
			r.data[sData[i].USERID] = sData[i]
		}
		r.last = time.Now()
	}
}

func (r *userAccountMgr) get(userId int64) string {
	r.mtx.RLock()
	defer r.mtx.RUnlock()

	if r.data == nil {
		return ""
	}

	now := time.Now()
	// 用户名可能会修改, 间隔5分钟失效, 返回空, 需要重新加载
	if d := now.Sub(r.last); d.Seconds() > 300 {
		return ""
	}

	if v, ok := r.data[userId]; ok {
		return v.ACCOUNTNAME
	}
	return ""
}

// getEx 返回 accountname, nickname
func (r *userAccountMgr) getEx(userId int64) (string, string) {
	r.mtx.RLock()
	defer r.mtx.RUnlock()

	if r.data == nil {
		return "", ""
	}

	now := time.Now()
	// 用户名可能会修改, 间隔5分钟失效, 返回空, 需要重新加载
	if d := now.Sub(r.last); d.Seconds() > 300 {
		return "", ""
	}

	if v, ok := r.data[userId]; ok {
		return v.ACCOUNTNAME, v.NICKNAME
	}
	return "", ""
}

// GetUserNameByUserId 根据UserId获取名称
func GetUserNameByUserId(userId int64) string {
	if v := vUsrMgr.get(userId); v != "" {
		return v
	} else {
		vUsrMgr.load()
	}

	return vUsrMgr.get(userId)
}

// GetUserNameAndNickName 获取名称和昵称
func GetUserNameAndNickName(userId int64) (string, string) {
	if userName, nickName := vUsrMgr.getEx(userId); userName != "" {
		return userName, nickName
	} else {
		vUsrMgr.load()
	}

	return vUsrMgr.getEx(userId)
}