# Conflicts: # apps/fushouxian-server/config/dbconfig.json # apps/fushouxian-server/config/togocdn.client.config.json # init/Category.godev
RIceWqy
2 years ago
commit
eee26f52e4
@ -1 +1 @@
|
||||
{"pid":42691}
|
||||
{"pid":28155}
|
||||
@ -0,0 +1,7 @@
|
||||
package init
|
||||
|
||||
import "src/module/usercenter"
|
||||
|
||||
func init() {
|
||||
usercenter.InitManageApi()
|
||||
}
|
||||
@ -0,0 +1,51 @@
|
||||
package usercenter
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"crypto/rand"
|
||||
"math/big"
|
||||
"sync"
|
||||
"time"
|
||||
)
|
||||
|
||||
var regSMSVerificationCode sync.Map
|
||||
|
||||
// 登记注册用验证码
|
||||
func StoreRegSMSVerificationCode(mobile string) (string, error) {
|
||||
codeInterface, ok := regSMSVerificationCode.Load(mobile)
|
||||
if ok {
|
||||
return codeInterface.(string), nil
|
||||
}
|
||||
verificationCode := randCharNumber(6)
|
||||
regSMSVerificationCode.Store(mobile, verificationCode)
|
||||
go func(mobile string) {
|
||||
time.Sleep(time.Second * 5 * 60)
|
||||
regSMSVerificationCode.Delete(mobile)
|
||||
}(mobile)
|
||||
return verificationCode, nil
|
||||
}
|
||||
|
||||
// 验证注册验证码
|
||||
func RegSMSVerification(mobile, verificationCode string) bool {
|
||||
codeInterface, ok := regSMSVerificationCode.LoadAndDelete(mobile)
|
||||
if !ok {
|
||||
return false
|
||||
}
|
||||
if codeInterface.(string) == verificationCode {
|
||||
return true
|
||||
} else {
|
||||
return false
|
||||
}
|
||||
}
|
||||
|
||||
// 随机验证码
|
||||
func randCharNumber(size int) string {
|
||||
char := "0123456789"
|
||||
len64 := int64(len(char))
|
||||
var s bytes.Buffer
|
||||
for i := 0; i < size; i++ {
|
||||
in, _ := rand.Int(rand.Reader, big.NewInt(len64))
|
||||
s.WriteByte(char[in.Int64()])
|
||||
}
|
||||
return s.String()
|
||||
}
|
||||
@ -0,0 +1,293 @@
|
||||
package usercenter
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"regexp"
|
||||
"src/module/tencent"
|
||||
|
||||
"github.com/towgo/towgo/towgo"
|
||||
)
|
||||
|
||||
func InitManageApi() {
|
||||
|
||||
//初始化API加载器
|
||||
initLoader()
|
||||
|
||||
//注册JSON-RPC服务处理器method路由
|
||||
//账户登录 F
|
||||
towgo.SetFunc(_methodHead+"/user/login", userLogin)
|
||||
|
||||
//获取自己的账户信息
|
||||
towgo.SetFunc(_methodHead+"/user/myinfo", userMyinfo)
|
||||
|
||||
//账户注销 F
|
||||
towgo.SetFunc(_methodHead+"/user/logoff", userLogoff)
|
||||
|
||||
//账户注册移动用户注册
|
||||
towgo.SetFunc(_methodHead+"/user/regByMobile", userRegByMobile)
|
||||
|
||||
//获取注册短信验证码
|
||||
towgo.SetFunc(_methodHead+"/user/getRegSMSVerificationCode", getRegSMSVerificationCode)
|
||||
|
||||
//修改密码 F
|
||||
towgo.SetFunc(_methodHead+"/user/changepassword", userChangepassword)
|
||||
|
||||
}
|
||||
|
||||
func isPhoneNumber(input string) bool {
|
||||
// 中国手机号码正则表达式
|
||||
// 13[0-9], 14[5,7,9], 15[0-3,5-9], 16[6], 17[0-8], 18[0-9], 19[1,8,9]
|
||||
phoneNumberPattern := `^1([38][0-9]|14[579]|5[^4]|6[6]|7[0-8]|9[189])\d{8}$`
|
||||
reg := regexp.MustCompile(phoneNumberPattern)
|
||||
return reg.MatchString(input)
|
||||
}
|
||||
|
||||
func getRegSMSVerificationCode(rpcConn towgo.JsonRpcConnection) {
|
||||
var params struct {
|
||||
Mobile string `json:"mobile"`
|
||||
}
|
||||
rpcConn.ReadParams(¶ms)
|
||||
|
||||
if !isPhoneNumber(params.Mobile) {
|
||||
rpcConn.WriteError(500, "手机号码非法")
|
||||
return
|
||||
}
|
||||
|
||||
code, err := StoreRegSMSVerificationCode(params.Mobile)
|
||||
|
||||
err = tencent.SendSMSVerificationCode("2030693", "蕊鑫信息科技", params.Mobile, code)
|
||||
if err != nil {
|
||||
rpcConn.WriteError(500, err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
rpcConn.WriteResult("验证码已经发送,请查收")
|
||||
|
||||
}
|
||||
|
||||
// 注册用户
|
||||
func userRegByMobile(rpcConn towgo.JsonRpcConnection) {
|
||||
result := map[string]interface{}{} //初始化结果参数
|
||||
|
||||
var params struct {
|
||||
Mobile string `json:"mobile"`
|
||||
VerificationCode string `json:"verification_code"`
|
||||
}
|
||||
|
||||
rpcConn.ReadParams(¶ms)
|
||||
|
||||
if params.Mobile == "" {
|
||||
rpcConn.WriteError(500, "手机号码不能为空")
|
||||
return
|
||||
}
|
||||
if params.VerificationCode == "" {
|
||||
rpcConn.WriteError(500, "验证码不能为空")
|
||||
return
|
||||
}
|
||||
|
||||
if !RegSMSVerification(params.Mobile, params.VerificationCode) {
|
||||
rpcConn.WriteError(500, "验证码错误")
|
||||
return
|
||||
}
|
||||
|
||||
var user User
|
||||
user.Username = params.Mobile
|
||||
user.Password = randCharNumber(8)
|
||||
|
||||
/*
|
||||
user := user{}
|
||||
user.Nickname = jsonObj.Params.Nickname
|
||||
user.Email = jsonObj.Params.Email
|
||||
*/
|
||||
Err := user.Reg(user.Username, user.Password)
|
||||
if Err != nil {
|
||||
rpcConn.WriteError(500, Err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
//拼装结果返回
|
||||
result["id"] = user.ID
|
||||
result["username"] = user.Username
|
||||
rpcConn.WriteResult(result)
|
||||
}
|
||||
|
||||
// 用户登陆
|
||||
func userLogin(rpcConn towgo.JsonRpcConnection) {
|
||||
result := map[string]interface{}{} //初始化结果参数
|
||||
var err error
|
||||
|
||||
rpcResponse := rpcConn.GetRpcResponse()
|
||||
jsonObj := struct {
|
||||
Params struct {
|
||||
Username string `json:"username"`
|
||||
Password string `json:"password"`
|
||||
} `json:"params"`
|
||||
}{}
|
||||
err = json.Unmarshal([]byte(rpcConn.Read()), &jsonObj)
|
||||
if err != nil {
|
||||
rpcResponse.Error.Set(1, err.Error())
|
||||
rpcConn.Write()
|
||||
return
|
||||
}
|
||||
if jsonObj.Params.Username == "" {
|
||||
rpcResponse.Error.Set(1001, "")
|
||||
rpcConn.Write()
|
||||
return
|
||||
}
|
||||
if jsonObj.Params.Password == "" {
|
||||
rpcResponse.Error.Set(1002, "")
|
||||
rpcConn.Write()
|
||||
return
|
||||
}
|
||||
|
||||
user := User{}
|
||||
loginErr := user.Login(jsonObj.Params.Username, jsonObj.Params.Password)
|
||||
|
||||
if loginErr != nil { //模型层登陆成功
|
||||
//dblog.Write("user:info", fmt.Sprintf("%s@%s 登录失败！ 错误信息:%s", user.Username, rpcConn.GetRemoteAddr(), loginErr.Error()))
|
||||
rpcResponse.Error.Set(1, "用户名或密码错误")
|
||||
rpcConn.Write()
|
||||
return
|
||||
}
|
||||
|
||||
result["id"] = user.ID
|
||||
result["username"] = user.Username
|
||||
|
||||
result["token"] = user.UserToken.TokenKey
|
||||
|
||||
//dblog.Write("user:info", fmt.Sprintf("%s@%s 登录成功！", user.Username, rpcConn.GetRemoteAddr()))
|
||||
rpcConn.WriteResult(result)
|
||||
|
||||
}
|
||||
|
||||
// token check
|
||||
func userTokenCheck(rpcConn towgo.JsonRpcConnection) {
|
||||
result := map[string]interface{}{} //初始化结果参数
|
||||
var err error
|
||||
|
||||
rpcResponse := rpcConn.GetRpcResponse()
|
||||
jsonObj := struct {
|
||||
Session string `json:"session"`
|
||||
Params struct {
|
||||
Username string `json:"username"`
|
||||
Userid int `json:"userid"`
|
||||
Token string `json:"token"`
|
||||
} `json:"params"`
|
||||
}{}
|
||||
err = json.Unmarshal([]byte(rpcConn.Read()), &jsonObj)
|
||||
if err != nil {
|
||||
rpcResponse.Error.Set(1, err.Error())
|
||||
rpcConn.Write()
|
||||
return
|
||||
}
|
||||
|
||||
var user *User
|
||||
user, err = user.LoginByToken(jsonObj.Params.Token)
|
||||
if err != nil {
|
||||
result["valid"] = false
|
||||
rpcConn.WriteResult(result)
|
||||
return
|
||||
}
|
||||
if user.ID == 0 {
|
||||
result["valid"] = false
|
||||
rpcConn.WriteResult(result)
|
||||
return
|
||||
}
|
||||
|
||||
if !user.UserToken.Valid() {
|
||||
result["valid"] = false
|
||||
rpcConn.WriteResult(result)
|
||||
return
|
||||
}
|
||||
|
||||
result["valid"] = true
|
||||
rpcConn.WriteResult(result)
|
||||
}
|
||||
|
||||
// 用户注销
|
||||
func userLogoff(rpcConn towgo.JsonRpcConnection) {
|
||||
//result := map[string]interface{}{} //初始化结果参数
|
||||
var err error
|
||||
|
||||
rpcResponse := rpcConn.GetRpcResponse()
|
||||
jsonObj := struct {
|
||||
Session string `json:"session"`
|
||||
}{}
|
||||
err = json.Unmarshal([]byte(rpcConn.Read()), &jsonObj)
|
||||
if err != nil {
|
||||
rpcResponse.Error.Set(1, err.Error())
|
||||
rpcConn.Write()
|
||||
return
|
||||
}
|
||||
|
||||
var user *User
|
||||
user, err = user.LoginByToken(jsonObj.Session)
|
||||
if err != nil {
|
||||
rpcConn.WriteResult(map[string]string{"success": "ok"})
|
||||
return
|
||||
}
|
||||
if user.ID > 0 {
|
||||
user.Logoff()
|
||||
}
|
||||
|
||||
rpcConn.WriteResult(map[string]string{"success": "ok"})
|
||||
|
||||
}
|
||||
|
||||
func userMyinfo(rpcConn towgo.JsonRpcConnection) {
|
||||
userSession, err := LoginByToken(rpcConn.GetRpcRequest().Session)
|
||||
userSession.Token = rpcConn.GetRpcRequest().Session
|
||||
if err != nil {
|
||||
rpcConn.GetRpcResponse().Error.Set(401, err.Error())
|
||||
rpcConn.Write()
|
||||
return
|
||||
}
|
||||
rpcConn.WriteResult(userSession)
|
||||
}
|
||||
|
||||
func userChangepassword(rpcConn towgo.JsonRpcConnection) {
|
||||
result := map[string]interface{}{} //初始化结果参数
|
||||
var err error
|
||||
|
||||
rpcResponse := rpcConn.GetRpcResponse()
|
||||
jsonObj := struct {
|
||||
Session string `json:"session"`
|
||||
Params struct {
|
||||
Oldpassword string `json:"oldpassword"`
|
||||
Newpassword string `json:"newpassword"`
|
||||
} `json:"params"`
|
||||
}{}
|
||||
err = json.Unmarshal([]byte(rpcConn.Read()), &jsonObj)
|
||||
|
||||
if err != nil {
|
||||
rpcResponse.Error.Set(1, err.Error())
|
||||
rpcConn.WriteResult(result)
|
||||
return
|
||||
}
|
||||
|
||||
var user *User
|
||||
user, err = user.LoginByToken(jsonObj.Session)
|
||||
if err != nil {
|
||||
rpcResponse.Error.Set(401, err.Error())
|
||||
rpcConn.WriteResult(result)
|
||||
return
|
||||
}
|
||||
|
||||
if user.ID == 0 {
|
||||
rpcResponse.Error.Set(1003, "")
|
||||
rpcConn.WriteResult(result)
|
||||
return
|
||||
}
|
||||
|
||||
err = user.Changepassword(jsonObj.Params.Oldpassword, jsonObj.Params.Newpassword)
|
||||
if err != nil {
|
||||
rpcResponse.Error.Set(1, err.Error())
|
||||
rpcConn.WriteResult(result)
|
||||
return
|
||||
}
|
||||
|
||||
rpcConn.WriteResult(struct {
|
||||
Success string `json:"success"`
|
||||
}{Success: "ok"})
|
||||
|
||||
}
|
||||
@ -0,0 +1,25 @@
|
||||
package usercenter
|
||||
|
||||
import (
|
||||
"github.com/towgo/towgo/dao/ormDriver/xormDriver"
|
||||
)
|
||||
|
||||
var _methodHead string
|
||||
var _tableHead string
|
||||
|
||||
func SetMethodHead(methodHead string) {
|
||||
_methodHead = methodHead
|
||||
}
|
||||
|
||||
func SetTableHead(tableHead string) {
|
||||
_tableHead = tableHead
|
||||
}
|
||||
|
||||
func initLoader() {
|
||||
|
||||
//token任务
|
||||
InitTokenTask()
|
||||
|
||||
xormDriver.Sync2(new(User), new(UserToken))
|
||||
|
||||
}
|
||||
@ -0,0 +1,280 @@
|
||||
package usercenter
|
||||
|
||||
import (
|
||||
"errors"
|
||||
|
||||
"github.com/towgo/towgo/dao/basedboperat"
|
||||
"github.com/towgo/towgo/lib/system"
|
||||
)
|
||||
|
||||
func (User) TableName() string {
|
||||
return _tableHead + "users"
|
||||
}
|
||||
|
||||
func (*User) CacheExpire() int64 {
|
||||
return 5000
|
||||
}
|
||||
|
||||
// 账户对象 关联账户信息
|
||||
type User struct {
|
||||
ID int64 `json:"id"`
|
||||
Username string `json:"username"`
|
||||
Nickname string `json:"nickname"`
|
||||
Password string `json:"password"`
|
||||
Salt string `json:"-"` //密码加盐
|
||||
Email string `json:"email"`
|
||||
Mobile string `json:"mobile"`
|
||||
CanDelete bool `json:"-"`
|
||||
AccessToken string `json:"-"`
|
||||
Token string `json:"token" gorm:"-" xorm:"-"`
|
||||
UserToken *UserToken `json:"-" gorm:"-" xorm:"-"`
|
||||
CreatedAt int64 `json:"created_at"` //创建时间
|
||||
UpdatedAt int64 `json:"updated_at"` //更新时间
|
||||
}
|
||||
|
||||
// 注册
|
||||
func (a *User) Reg(username, password string) error {
|
||||
|
||||
e := a.CheckForInput(username, password)
|
||||
if e != nil {
|
||||
return e
|
||||
}
|
||||
|
||||
//数据库查询出用户信息
|
||||
|
||||
finduser := User{}
|
||||
basedboperat.Get(&finduser, nil, "username = ?", username)
|
||||
|
||||
//检查用户名是否存在
|
||||
if username == finduser.Username {
|
||||
return errors.New("账户已经存在")
|
||||
}
|
||||
|
||||
//生成密码
|
||||
a.NewPassword(password)
|
||||
|
||||
a.Username = username
|
||||
a.CanDelete = true
|
||||
|
||||
_, err := basedboperat.Create(a) // 通过数据的指针来创建
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
_, err = a.CreateRelation()
|
||||
return err
|
||||
}
|
||||
|
||||
// 检查输入参数
|
||||
func (a *User) CheckForInput(username, password string) error {
|
||||
|
||||
if username == "" {
|
||||
return errors.New("用户名不能为空")
|
||||
}
|
||||
|
||||
if password == "" {
|
||||
return errors.New("密码不能为空")
|
||||
}
|
||||
|
||||
//防sql注入
|
||||
if system.FilteredSQLInject(username) {
|
||||
return errors.New("用户名存在系统保留或非法的字符")
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (a *User) NewPassword(newpassword string) {
|
||||
if newpassword == "" {
|
||||
return
|
||||
}
|
||||
//加密密码
|
||||
|
||||
password := system.MD5(newpassword)
|
||||
|
||||
//生成salt
|
||||
salt := system.RandCharCrypto(6)
|
||||
|
||||
//密码加盐
|
||||
password = password + salt
|
||||
|
||||
//混合加密
|
||||
password = system.MD5(password)
|
||||
|
||||
a.Password = password
|
||||
a.Salt = salt
|
||||
}
|
||||
|
||||
// 用户登陆
|
||||
func (a *User) Login(username, password string) error {
|
||||
|
||||
erro := a.CheckForInput(username, password)
|
||||
if erro != nil {
|
||||
return erro
|
||||
}
|
||||
|
||||
//通过用户名查询用户数据
|
||||
err := basedboperat.Get(a, nil, "username = ?", username)
|
||||
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
//检查用户名是否存在
|
||||
|
||||
//判断用户是否存在
|
||||
if a.Username == "" {
|
||||
return errors.New("用户名不存在")
|
||||
}
|
||||
|
||||
//加密密码
|
||||
upassword := system.MD5(password)
|
||||
|
||||
//撒盐
|
||||
upassword = upassword + a.Salt
|
||||
|
||||
//混合加密
|
||||
upassword = system.MD5(upassword)
|
||||
|
||||
//判断密码是否一致
|
||||
if a.Password != upassword {
|
||||
//不一致:返回错误
|
||||
return errors.New("密码错误")
|
||||
}
|
||||
|
||||
//验证通过
|
||||
|
||||
//生成用户信息
|
||||
a.UserToken = NewToken(a)
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
// 用户注销
|
||||
func (a *User) Logoff() {
|
||||
DeleteToken(a.UserToken.TokenKey)
|
||||
}
|
||||
|
||||
func LoginByToken(tokenKey string) (*User, error) {
|
||||
userToken, err := GetToken(tokenKey)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
sessionuser := userToken.Payload.(*User)
|
||||
sessionuser.UserToken = userToken
|
||||
return sessionuser, nil
|
||||
}
|
||||
|
||||
func (a *User) LoginByToken(tokenKey string) (*User, error) {
|
||||
userToken, err := GetToken(tokenKey)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
sessionuser := userToken.Payload.(*User)
|
||||
sessionuser.UserToken = userToken
|
||||
return sessionuser, nil
|
||||
}
|
||||
|
||||
func (a *User) CheckToken(s string) bool {
|
||||
//判断token是否正确
|
||||
if s != a.UserToken.TokenKey {
|
||||
return false
|
||||
}
|
||||
//再判断token是否过期
|
||||
return a.UserToken.Valid()
|
||||
}
|
||||
func (a *User) Get() error {
|
||||
|
||||
if a.ID > 0 {
|
||||
return basedboperat.Get(a, nil, "id = ?", a.ID)
|
||||
}
|
||||
if a.Username != "" {
|
||||
return basedboperat.Get(a, nil, "username = ?", a.Username)
|
||||
}
|
||||
return errors.New("id或username不能为空")
|
||||
}
|
||||
|
||||
// 修改密码
|
||||
func (a *User) Changepassword(oldpassword, newpassword string) error {
|
||||
|
||||
//通过用户名查询用户数据
|
||||
|
||||
err := basedboperat.Get(a, nil, "id = ?", a.ID)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
//检查用户名是否存在
|
||||
|
||||
//判断用户是否存在
|
||||
if a.Username == "" {
|
||||
return errors.New("用户名不存在")
|
||||
}
|
||||
|
||||
//加密密码
|
||||
upassword := system.MD5(oldpassword)
|
||||
|
||||
//撒盐
|
||||
upassword = upassword + a.Salt
|
||||
|
||||
//混合加密
|
||||
upassword = system.MD5(upassword)
|
||||
|
||||
//判断密码是否一致
|
||||
if a.Password != upassword {
|
||||
//不一致:返回错误
|
||||
return errors.New("原始密码错误")
|
||||
}
|
||||
|
||||
a.NewPassword(newpassword)
|
||||
|
||||
basedboperat.Update(a, []string{"password", "salt"}, "id = ?", a.ID)
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (a *User) Update() error {
|
||||
var findModel User
|
||||
basedboperat.Get(&findModel, nil, "id = ?", a.ID)
|
||||
if findModel.ID <= 0 {
|
||||
return errors.New("记录不存在")
|
||||
}
|
||||
a.DeleteRelation()
|
||||
_, err := a.CreateRelation()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
basedboperat.Update(a, []string{"nickname", "email"}, "id = ?", a.ID)
|
||||
return nil
|
||||
}
|
||||
|
||||
func (a *User) Delete() (int64, error) {
|
||||
var findModel User
|
||||
basedboperat.Get(&findModel, nil, "id = ?", a.ID)
|
||||
if !findModel.CanDelete {
|
||||
return 0, errors.New("无法删除系统用户")
|
||||
}
|
||||
a.DeleteRelation()
|
||||
return basedboperat.Delete(a, a.ID, nil)
|
||||
}
|
||||
|
||||
// 删除关联数据
|
||||
func (a *User) DeleteRelation() {
|
||||
if a.ID == 0 {
|
||||
return
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
// 创建关联数据
|
||||
func (a *User) CreateRelation() (int64, error) {
|
||||
if a.ID == 0 {
|
||||
return 0, nil
|
||||
}
|
||||
var rowsAffected int64
|
||||
|
||||
return rowsAffected, nil
|
||||
}
|
||||
|
||||
func (a *User) AfterQuery() {
|
||||
if a.ID == 0 {
|
||||
return
|
||||
}
|
||||
|
||||
}
|
||||
@ -0,0 +1,227 @@
|
||||
package usercenter
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
"log"
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
"github.com/towgo/towgo/dao/basedboperat"
|
||||
"github.com/towgo/towgo/lib/system"
|
||||
)
|
||||
|
||||
// 缓存有效期
|
||||
var memCacheTimer int64 = 60 * 10
|
||||
|
||||
// token有效期 秒单位计算
|
||||
var expirationLimit int64 = 86400 * 20
|
||||
|
||||
var autoClearLimit int64 = 60 * 10 //10分钟清理一次过期的token
|
||||
|
||||
// var expirationLimit int64 = 60
|
||||
var memCache *MemCache
|
||||
|
||||
// UserToken结构体
|
||||
func (UserToken) TableName() string {
|
||||
return _tableHead + "users_token"
|
||||
}
|
||||
|
||||
type UserToken struct {
|
||||
TokenKey string
|
||||
Uid int64
|
||||
Payload any `gorm:"-" xorm:"-"`
|
||||
Expiration int64
|
||||
UpdatedAt int64
|
||||
CreatedAt int64
|
||||
}
|
||||
|
||||
func InitTokenTask() {
|
||||
memCache = &MemCache{}
|
||||
autoTimeToClear()
|
||||
}
|
||||
|
||||
type MemCache struct {
|
||||
timers sync.Map
|
||||
cacheObject sync.Map
|
||||
cancels sync.Map
|
||||
}
|
||||
|
||||
func (mc *MemCache) CreateTimerToDelete(tokenKey string) {
|
||||
timer := time.NewTimer(time.Second * time.Duration(memCacheTimer))
|
||||
mc.timers.Store(tokenKey, timer)
|
||||
ctx, cancel := context.WithCancel(context.Background())
|
||||
|
||||
mc.cancels.Store(tokenKey, cancel)
|
||||
go mc.DeleteWhenTimeOut(ctx, tokenKey, timer)
|
||||
}
|
||||
|
||||
func (mc *MemCache) DeleteWhenTimeOut(ctx context.Context, tokenKey string, timer *time.Timer) {
|
||||
select {
|
||||
case <-timer.C:
|
||||
mc.timers.Delete(tokenKey)
|
||||
mc.cacheObject.Delete(tokenKey)
|
||||
mc.cancels.Delete(tokenKey)
|
||||
return
|
||||
case <-ctx.Done():
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
func (mc *MemCache) ResetTimer(tokenKey string) {
|
||||
timerInterface, ok := mc.timers.Load(tokenKey)
|
||||
if ok {
|
||||
timer := timerInterface.(*time.Timer)
|
||||
timer.Reset(time.Second * time.Duration(memCacheTimer))
|
||||
}
|
||||
}
|
||||
|
||||
func (mc *MemCache) Add(tokenKey string, value any) {
|
||||
mc.cacheObject.Store(tokenKey, value)
|
||||
mc.CreateTimerToDelete(tokenKey)
|
||||
}
|
||||
|
||||
func (mc *MemCache) Del(tokenKey string) {
|
||||
timerInterface, ok := mc.timers.Load(tokenKey)
|
||||
if ok {
|
||||
timer := timerInterface.(*time.Timer)
|
||||
timer.Stop() //关闭定时器
|
||||
cancel_any, isLoaded := mc.cancels.LoadAndDelete(tokenKey) //关闭定时器线程
|
||||
if isLoaded {
|
||||
cancel := cancel_any.(context.CancelFunc)
|
||||
if cancel != nil {
|
||||
cancel()
|
||||
}
|
||||
}
|
||||
mc.timers.Delete(tokenKey) //清除定时器委托
|
||||
}
|
||||
mc.cacheObject.Delete(tokenKey) //清除内存
|
||||
}
|
||||
|
||||
func (mc *MemCache) Get(tokenKey string) (any, bool) {
|
||||
return mc.cacheObject.Load(tokenKey)
|
||||
}
|
||||
|
||||
// 自动清理过期token定时器
|
||||
func autoTimeToClear() {
|
||||
go func() {
|
||||
defer func() {
|
||||
err := recover()
|
||||
if err != nil {
|
||||
log.Print(err)
|
||||
}
|
||||
autoTimeToClear()
|
||||
}()
|
||||
var userToken UserToken
|
||||
for {
|
||||
time.Sleep(time.Second * time.Duration(autoClearLimit))
|
||||
basedboperat.SqlExec("delete from "+userToken.TableName()+" where expiration < ?", time.Now().Unix())
|
||||
}
|
||||
}()
|
||||
}
|
||||
|
||||
func (t *UserToken) NewTokenGUID(salt string) {
|
||||
guid := system.GetGUID().Hex()
|
||||
saltEncode := system.MD5(salt)
|
||||
tokenCode := system.MD5(guid + saltEncode)
|
||||
t.TokenKey = tokenCode
|
||||
}
|
||||
|
||||
// 返回一个唯一标识的token令牌
|
||||
func (t *UserToken) String() string {
|
||||
return t.TokenKey
|
||||
}
|
||||
|
||||
// token是否有效 检查有效期
|
||||
// 有效返回true
|
||||
// 无效返回false
|
||||
func (t *UserToken) Valid() bool {
|
||||
return time.Now().Unix() < t.Expiration
|
||||
}
|
||||
|
||||
func (t *UserToken) Check(tokenKey string) bool {
|
||||
return tokenKey == t.TokenKey
|
||||
}
|
||||
|
||||
// 更新token有效期
|
||||
func (t *UserToken) Update(expiration int64) {
|
||||
if expiration > 0 {
|
||||
t.Expiration = time.Now().Unix() + expiration
|
||||
} else {
|
||||
t.Expiration = time.Now().Unix() + expirationLimit
|
||||
}
|
||||
}
|
||||
|
||||
// 新建一个token
|
||||
func NewToken(user *User) *UserToken {
|
||||
timenow := time.Now().Unix()
|
||||
token := &UserToken{
|
||||
CreatedAt: timenow,
|
||||
Uid: user.ID,
|
||||
Payload: user,
|
||||
Expiration: timenow + expirationLimit,
|
||||
}
|
||||
user.Token = token.TokenKey
|
||||
token.NewTokenGUID(user.Salt)
|
||||
basedboperat.Create(token)
|
||||
memCache.Add(token.TokenKey, token)
|
||||
return token
|
||||
}
|
||||
|
||||
func GetToken(tokenKey string) (*UserToken, error) {
|
||||
var userToken *UserToken = &UserToken{}
|
||||
|
||||
//缓存查询
|
||||
userTokenInterface, ok := memCache.Get(tokenKey)
|
||||
if ok {
|
||||
//缓存命中
|
||||
userToken = userTokenInterface.(*UserToken)
|
||||
//缓存过期 清理
|
||||
if !userToken.Valid() {
|
||||
memCache.Del(tokenKey)
|
||||
return nil, errors.New("token过期(登录失效,请重新登录)")
|
||||
}
|
||||
} else {
|
||||
//数据库查询
|
||||
|
||||
//查询持久化数据
|
||||
err := basedboperat.Get(userToken, nil, "token_key = ?", tokenKey)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if userToken.Uid == 0 {
|
||||
return nil, errors.New("token不存在(登录失效,请重新登录)") //数据不存在
|
||||
}
|
||||
|
||||
//token过期
|
||||
if !userToken.Valid() {
|
||||
return nil, errors.New("token过期(登录失效,请重新登录)")
|
||||
}
|
||||
|
||||
//查询token关联的用户
|
||||
var user *User = &User{}
|
||||
basedboperat.Get(user, nil, "id = ?", userToken.Uid)
|
||||
if user.ID == 0 {
|
||||
return nil, errors.New("token关联用户不存在(登录失效,请重新登录)") //用户不存在
|
||||
}
|
||||
|
||||
//写入缓存
|
||||
userToken.Payload = user
|
||||
memCache.Add(userToken.TokenKey, userToken)
|
||||
}
|
||||
|
||||
return userToken, nil
|
||||
}
|
||||
|
||||
func DeleteToken(tokenKey string) {
|
||||
memCache.Del(tokenKey)
|
||||
var userToken *UserToken = &UserToken{}
|
||||
userToken.TokenKey = tokenKey
|
||||
|
||||
basedboperat.Delete(userToken, nil, "token_key = ?", tokenKey)
|
||||
|
||||
}
|
||||
|
||||
func SetExpiration(hour int64) {
|
||||
expirationLimit = 3600 * hour
|
||||
}
|
||||
Loading…
Reference in new issue