finish: gorm序列化

utils/auth/user.go

@@ -8,17 +8,41 @@ import (
 	"net/http"
 
 	"github.com/golang-jwt/jwt"
-	"github.com/google/uuid"
 	"github.com/zeromicro/go-zero/core/logx"
 )
 
+type IDTYPE int
+
+const (
+	// 白板用户, 以观众身份命名, 没有接收Cookie, 没有拿到guest_id的用户
+	IDTYPE_Onlooker IDTYPE = 0
+	// 登录用户
+	IDTYPE_User IDTYPE = 1
+	// 游客 接收授权拿到guest_id的用户
+	IDTYPE_Guest IDTYPE = 2
+)
+
 type UserInfo struct {
-	UserId int64 `json:"userid"`
+	UserId  int64 `json:"user_id"`
+	GuestId int64 `json:"guest_id"`
+}
+
+// GetIdType 用户确认用户身份类型
+func (info *UserInfo) GetIdType() IDTYPE {
+	if info.UserId != 0 {
+		return IDTYPE_User
+	}
+
+	if info.GuestId != 0 {
+		return IDTYPE_Guest
+	}
+
+	return IDTYPE_Onlooker
 }
 
 // 获取登录信息
 func GetUserInfoFormCtx(ctx context.Context) UserInfo {
-	uid, err := ctx.Value("userid").(json.Number).Int64()
+	uid, err := ctx.Value("user_id").(json.Number).Int64()
 	if err != nil {
 		logx.Error("parse uid form context err:", err.Error())
 		return UserInfo{}
@@ -28,43 +52,60 @@ func GetUserInfoFormCtx(ctx context.Context) UserInfo {
 
 // 获取登录信息
 func GetUserInfoFormMapClaims(claims jwt.MapClaims) (*UserInfo, error) {
-	if userid, ok := claims["userid"]; ok {
+	userinfo := &UserInfo{}
+	if userid, ok := claims["user_id"]; ok {
 		uid, ok := userid.(float64)
 		if !ok {
 			err := errors.New(fmt.Sprint("parse uid form context err:", userid))
 			logx.Error("parse uid form context err:", err)
 			return nil, err
 		}
-
-		return &UserInfo{UserId: int64(uid)}, nil
+		userinfo.UserId = int64(uid)
 	} else {
 		err := errors.New(`userid not in claims`)
 		logx.Error(`userid not in claims`)
 		return nil, err
 	}
+
+	if guestid, ok := claims["guest_id"]; ok {
+		gid, ok := guestid.(float64)
+		if !ok {
+			err := errors.New(fmt.Sprint("parse guestid form context err:", guestid))
+			logx.Error("parse guestid form context err:", err)
+			return nil, err
+		}
+		userinfo.GuestId = int64(gid)
+	} else {
+		err := errors.New(`userid not in claims`)
+		logx.Error(`userid not in claims`)
+		return nil, err
+	}
+
+	return userinfo, nil
 }
 
-func GenerateJwtToken(accessSecret string, accessExpire, nowSec int64, userid int) (string, error) {
+func GenerateJwtToken(accessSecret *string, accessExpire, nowSec int64, userid int64, guestid int64) (string, error) {
 	claims := make(jwt.MapClaims)
 	claims["exp"] = nowSec + accessExpire
 	claims["iat"] = nowSec
-	claims["userid"] = userid
-	if userid == 0 {
-		u, err := uuid.NewUUID()
-		if err != nil {
-			logx.Error(err)
-			return "", err
-		}
-		claims["guestid"] = u.String() // TODO: 未完成
+
+	if userid == 0 && guestid == 0 {
+		err := errors.New("userid and guestid cannot be 0 at the same time")
+		logx.Error(err)
+		return "", err
+
 	}
+	claims["user_id"] = userid
+	claims["guest_id"] = guestid
+
 	token := jwt.New(jwt.SigningMethodHS256)
 	token.Claims = claims
-	return token.SignedString([]byte(accessSecret))
+	return token.SignedString([]byte(*accessSecret))
 }
 
 func ParseJwtToken(w http.ResponseWriter, r *http.Request, AccessSecret *string) (*UserInfo, error) {
 	// 解析jwtToken
-	claims, err := getJwtClaims(r, AccessSecret)
+	claims, err := getJwtClaimsFromRequest(r, AccessSecret)
 	// 如果解析出错，则返回未授权的JSON响应并记录错误消息
 	if err != nil {
 		// httpx.OkJsonCtx(r.Context(), w, &basic.Response{
@@ -89,12 +130,17 @@ func ParseJwtToken(w http.ResponseWriter, r *http.Request, AccessSecret *string)
 	return userinfo, err
 }
 
-func getJwtClaims(r *http.Request, AccessSecret *string) (jwt.MapClaims, error) {
+func getJwtClaimsFromRequest(r *http.Request, AccessSecret *string) (jwt.MapClaims, error) {
 	AuthKey := r.Header.Get("Authorization")
 	if len(AuthKey) <= 50 {
 		return nil, errors.New(fmt.Sprint("Error parsing token, len:", len(AuthKey)))
 	}
 
+	return getJwtClaims(AuthKey, AccessSecret)
+}
+
+func getJwtClaims(AuthKey string, AccessSecret *string) (jwt.MapClaims, error) {
+
 	token, err := jwt.Parse(AuthKey, func(token *jwt.Token) (interface{}, error) {
 		// 检查签名方法是否为 HS256
 		if _, ok := token.Method.(*jwt.SigningMethodHMAC); !ok {

utils/auth/user_test.go

@@ -0,0 +1,30 @@
+package auth
+
+import (
+	"testing"
+	"time"
+)
+
+// TestGenJwt 测试jwt序列化
+func TestGenJwt(t *testing.T) {
+	now := time.Now().Unix()
+	secret := "fusen123"
+	a, err := GenerateJwtToken(&secret, 3600, now, 123, 1234)
+	if err != nil {
+		t.Error(err)
+	}
+	// log.Println(a)
+
+	claims, err := getJwtClaims(a, &secret)
+	if err != nil {
+		t.Error(err)
+	}
+	userinfo, err := GetUserInfoFormMapClaims(claims)
+	if err != nil {
+		t.Error(err)
+	}
+	if userinfo.UserId != 123 || userinfo.GuestId != 1234 {
+		t.Error(userinfo)
+	}
+	// log.Println(claims)
+}