fix

model/gmodel/fs_user_logic.go

@@ -2,8 +2,11 @@ package gmodel
 
 import (
 	"context"
+	"encoding/json"
 	"fmt"
+
 	"fusenapi/utils/auth"
+	"log"
 	"time"
 
 	"gorm.io/gorm"
@@ -47,11 +50,20 @@ func (u *FsUserModel) Transaction(ctx context.Context, fc func(tx *gorm.DB) erro
 }
 
 // 继承guest_id的资源表
-func InheritGuestIdResource(tx *gorm.DB, userId, guestId int64) error {
+func InheritGuestIdResource(tx *gorm.DB, userId, guestId int64, afterDo func(txResouce *gorm.DB, txUserMaterial *gorm.DB, txUserInfo *gorm.DB) error) error {
 	var err error
 	if guestId != 0 {
 		// 继承guest_id的资源表
-		err = tx.Model(&FsResource{}).
+		txRes := tx.Model(&FsResource{})
+		err = txRes.
+			Where("guest_id = ?", guestId).
+			UpdateColumn("user_id", userId).Error
+
+		if err != nil {
+			return err
+		}
+		txUserMaterial := tx.Model(&FsUserMaterial{})
+		err = txUserMaterial.
 			Where("guest_id = ?", guestId).
 			UpdateColumn("user_id", userId).Error
 
@@ -59,19 +71,16 @@ func InheritGuestIdResource(tx *gorm.DB, userId, guestId int64) error {
 			return err
 		}
 
-		err = tx.Model(&FsUserMaterial{}).
+		txUserInfo := tx.Model(&FsUserInfo{})
+		err = txUserInfo.
 			Where("guest_id = ?", guestId).
 			UpdateColumn("user_id", userId).Error
-
 		if err != nil {
 			return err
 		}
 
-		err = tx.Model(&FsUserInfo{}).
-			Where("guest_id = ?", guestId).
-			UpdateColumn("user_id", userId).Error
-		if err != nil {
-			return err
+		if afterDo != nil {
+			return afterDo(txRes, txUserMaterial, txUserInfo)
 		}
 	}
 	return fmt.Errorf("guest_id must not be 0")
@@ -82,15 +91,17 @@ func (u *FsUserModel) RegisterByGoogleOAuth(ctx context.Context, token *auth.Reg
 	user := &FsUser{}
 
 	err := u.db.WithContext(ctx).Transaction(func(tx *gorm.DB) error {
+		googleId := token.Extend["google_id"].(int64)
 
 		err := tx.Model(user).Where("email = ?", token.Email).Take(user).Error
 		if err != nil {
 			// 没有找到在数据库就创建注册
 			if err == gorm.ErrRecordNotFound {
+
 				createAt := time.Now().UTC().Unix()
 				user.Email = &token.Email
 				user.CreatedAt = &createAt
-				user.GoogleId = &token.Id
+				user.GoogleId = &googleId
 				user.PasswordHash = &token.Password
 				err = tx.Model(user).Create(user).Error
 				if err != nil {
@@ -99,7 +110,7 @@ func (u *FsUserModel) RegisterByGoogleOAuth(ctx context.Context, token *auth.Reg
 
 				if token.GuestId != 0 {
 					// 继承guest_id的资源表
-					return InheritGuestIdResource(tx, user.Id, token.GuestId)
+					return InheritGuestIdResource(tx, user.Id, token.GuestId, nil)
 				}
 				return err
 			}
@@ -108,7 +119,7 @@ func (u *FsUserModel) RegisterByGoogleOAuth(ctx context.Context, token *auth.Reg
 		}
 
 		// 如果已经存在,把谷歌id 加入到用户信息里
-		user.GoogleId = &token.Id
+		user.GoogleId = &googleId
 		return tx.Model(user).Update("google_id", user).Error
 	})
 
@@ -119,27 +130,64 @@ func (u *FsUserModel) RegisterByGoogleOAuth(ctx context.Context, token *auth.Reg
 	return user, nil
 }
 
+type UserProfile struct {
+	FirstName   string `json:"first_name"`
+	LastName    string `json:"last_name"`
+	Resetaurant string `json:"resetaurant"`
+}
+
 // 自平台的注册流程
 func (u *FsUserModel) RegisterByFusen(ctx context.Context, token *auth.RegisterToken) (*FsUser, error) {
 	user := &FsUser{}
 
 	err := u.db.WithContext(ctx).Transaction(func(tx *gorm.DB) error {
 
-		err := tx.Model(user).Where("email = ?", token.Email).Take(user).Error
+		UserTx := tx.Model(user)
+		err := UserTx.Where("email = ?", token.Email).Take(user).Error
 		if err != nil {
 			// 没有找到在数据库就创建注册
 			if err == gorm.ErrRecordNotFound {
+				FirstName := token.Extend["first_name"].(string)
+				LastName := token.Extend["last_name"].(string)
+				Resetaurant := token.Extend["resetaurant"].(string)
+
 				createAt := time.Now().UTC().Unix()
 				user.Email = &token.Email
 				user.CreatedAt = &createAt
 				user.PasswordHash = &token.Password
-				err = tx.Model(user).Create(user).Error
+				user.FirstName = &FirstName
+				user.LastName = &LastName
+
+				err = UserTx.Create(user).Error
+				log.Println("create")
 				if err != nil {
 					return err
 				}
+
 				if token.GuestId != 0 {
+
 					// 继承guest_id的资源表
-					return InheritGuestIdResource(tx, user.Id, token.GuestId)
+					return InheritGuestIdResource(tx, user.Id, token.GuestId, func(txResouce, txUserMaterial, txUserInfo *gorm.DB) error {
+						userProfile := &UserProfile{
+							FirstName:   FirstName,
+							LastName:    LastName,
+							Resetaurant: Resetaurant,
+						}
+						metadata, err := json.Marshal(userProfile)
+						if err != nil {
+							return err
+						}
+						now := time.Now()
+						uinfo := &FsUserInfo{
+							Module:   FsString("profile"),
+							UserId:   &user.Id,
+							GuestId:  &token.GuestId,
+							Metadata: &metadata,
+							Ctime:    &now,
+							Utime:    &now,
+						}
+						return txUserInfo.Create(uinfo).Error
+					})
 				}
 				return err
 			}

model/gmodel/var.go

@@ -0,0 +1,37 @@
+package gmodel
+
+func FsString(v string) *string {
+	return &v
+}
+
+func FsInt(v int) *int {
+	return &v
+}
+
+func FsInt32(v int32) *int32 {
+	return &v
+}
+
+func FsInt64(v int64) *int64 {
+	return &v
+}
+
+func FsUint(v uint) *uint {
+	return &v
+}
+
+func FsUint32(v uint32) *uint32 {
+	return &v
+}
+
+func FsUint64(v uint64) *uint64 {
+	return &v
+}
+
+func FsFloat(v float64) *float64 {
+	return &v
+}
+
+func FsBool(v bool) *bool {
+	return &v
+}