double.huang 5 سال پیش
والد
کامیت
5256e7a192
3فایلهای تغییر یافته به همراه10 افزوده شده و 4 حذف شده
  1. 5 3
      engine/auth/auth.go
  2. 2 1
      engine/auth/auth_client.go
  3. 3 0
      entitys/token.go

+ 5 - 3
engine/auth/auth.go

@@ -2,11 +2,13 @@ package auth
 
 import (
 	"errors"
+	"strconv"
+	"sync"
+	"time"
+
 	"git.qianqiusoft.com/qianqiusoft/light-apiengine/entitys"
 	sysmodel "git.qianqiusoft.com/qianqiusoft/light-apiengine/models"
 	sysutils "git.qianqiusoft.com/qianqiusoft/light-apiengine/utils"
-	"strconv"
-	"time"
 )
 
 type IAuth interface {
@@ -61,7 +63,7 @@ func GetAuth(t string) IAuth {
  * @param1 userInfo: 用户信息
  */
 func AddToGlobalTokenStore(c *entitys.CtrlContext, userInfo *sysmodel.SysUser) (*sysmodel.LoginReturnInfo, error) {
-	token := &entitys.Token{}
+	token := &entitys.Token{Lock: new(sync.RWMutex)}
 	timestamp := uint64(time.Now().UnixNano())
 	timestamp_str := strconv.FormatUint(timestamp, 10)
 	sec_tooken := sysutils.GenerateToken(userInfo.LoginId + timestamp_str)

+ 2 - 1
engine/auth/auth_client.go

@@ -9,6 +9,7 @@ import (
 	"fmt"
 	"net"
 	"strconv"
+	"sync"
 	"time"
 
 	"git.qianqiusoft.com/library/glog"
@@ -366,7 +367,7 @@ func uint64ToBytes(v int) []byte {
 
 // 转token
 func bytesToToken(content []byte) (*entitys.Token, error) {
-	token := &entitys.Token{}
+	token := &entitys.Token{Lock: new(sync.RWMutex)}
 	var index int = 0
 	var size int
 	var err error = nil

+ 3 - 0
entitys/token.go

@@ -1,6 +1,9 @@
 package entitys
 
+import "sync"
+
 type Token struct {
+	Lock         *sync.RWMutex
 	Result       int                    `json:"-"`
 	UserId       string                 `json:"user_id"`
 	AccessToken  string                 `json:"access_token"`