pm/services/user/token.go

package user

import (
	"time"
	"zhiyuan/models"
	"zhiyuan/pkg/config"
	"zhiyuan/pkg/db"
	"zhiyuan/pkg/errcode"
	"zhiyuan/pkg/redis"
	"zhiyuan/pkg/utils"
)

const tokenPrefix = "user_token_"

func CheckToken(token string) (*models.User, errcode.Err) {
	result, err := redis.Get(tokenPrefix + token)
	var user *models.User
	if err != nil {
		user, err = GetInfoByToken(token, nil, nil)
		if err != nil || user == nil {
			return nil, errcode.TokenInvalid
		}
		redis.Set(tokenPrefix+"_"+token, user, 3600)
		redis.Set(tokenPrefix+"_"+utils.ToStr(user.ID), tokenPrefix+token, 3600)
	} else {
		utils.JsonDecode(result).To(&user)
	}

	if user == nil || time.Now().Unix() >= utils.StrTo(utils.ToStr(user.AccessTokenExipred)).MustInt64() {
		return nil, errcode.TokenExpired
	}
	return user, errcode.Err{}
}

func UpdateToken(id int) (string, error) {
	token := utils.MD5(utils.ToStr(time.Now().UnixNano()) + string(utils.RandomCreateBytes(10, utils.RAND_KIND_ALL)))
	tokenExpired := int(time.Now().Unix()) + config.Cfg.App.TokenExpire
	_, err := db.Update(User.TableName(), map[string]interface{}{"id": id}, map[string]interface{}{
		"access_token":         token,
		"access_token_expired": tokenExpired})
	if err != nil {
		return token, err
	}
	preToken, err := redis.Get(tokenPrefix + "_" + utils.ToStr(id))
	if preToken != "" {
		redis.Del(preToken)
	}
	return token, nil
}

func ClearToken(id int) error {
	_, err := db.Update(User.TableName(), map[string]interface{}{"id": id}, map[string]interface{}{"access_token": ""})
	preToken, err := redis.Get(tokenPrefix + "_" + utils.ToStr(id))
	if preToken != "" {
		redis.Del(preToken)
	}
	return err
}