Merge branch 'develop' of gitee.com:fusenpack/fusenapi into develop

This commit is contained in:
laodaming 2023-08-29 15:04:20 +08:00
commit 60255b5c45
3 changed files with 73 additions and 68 deletions

View File

@ -6,9 +6,9 @@ import (
"fmt" "fmt"
"fusenapi/utils/auth" "fusenapi/utils/auth"
"log"
"time" "time"
"github.com/zeromicro/go-zero/core/logx"
"gorm.io/gorm" "gorm.io/gorm"
) )
@ -52,37 +52,38 @@ func (u *FsUserModel) Transaction(ctx context.Context, fc func(tx *gorm.DB) erro
// 继承guest_id的资源表 // 继承guest_id的资源表
func InheritGuestIdResource(tx *gorm.DB, userId, guestId int64, afterDo func(txResouce *gorm.DB, txUserMaterial *gorm.DB, txUserInfo *gorm.DB) error) 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 var err error
txRes := tx.Model(&FsResource{})
txUserMaterial := tx.Model(&FsUserMaterial{})
txUserInfo := tx.Model(&FsUserInfo{})
if guestId != 0 { if guestId != 0 {
// 继承guest_id的资源表 // 继承guest_id的资源表
txRes := tx.Model(&FsResource{})
err = txRes. err = txRes.
Where("guest_id = ?", guestId). Where("guest_id = ?", guestId).
UpdateColumn("user_id", userId).Error UpdateColumn("user_id", userId).Error
if err != nil { if err != nil && err != gorm.ErrRecordNotFound {
return err return err
} }
txUserMaterial := tx.Model(&FsUserMaterial{})
err = txUserMaterial. err = txUserMaterial.
Where("guest_id = ?", guestId). Where("guest_id = ?", guestId).
UpdateColumn("user_id", userId).Error UpdateColumn("user_id", userId).Error
if err != nil { if err != nil && err != gorm.ErrRecordNotFound {
return err return err
} }
txUserInfo := tx.Model(&FsUserInfo{})
err = txUserInfo. err = txUserInfo.
Where("guest_id = ?", guestId). Where("guest_id = ?", guestId).
UpdateColumn("user_id", userId).Error UpdateColumn("user_id", userId).Error
if err != nil { if err != nil && err != gorm.ErrRecordNotFound {
return err return err
} }
}
if afterDo != nil { if afterDo != nil {
return afterDo(txRes, txUserMaterial, txUserInfo) return afterDo(txRes, txUserMaterial, txUserInfo)
} }
}
return fmt.Errorf("guest_id must not be 0") return fmt.Errorf("guest_id must not be 0")
} }
@ -108,12 +109,9 @@ func (u *FsUserModel) RegisterByGoogleOAuth(ctx context.Context, token *auth.Reg
return err return err
} }
if token.GuestId != 0 {
// 继承guest_id的资源表 // 继承guest_id的资源表
return InheritGuestIdResource(tx, user.Id, token.GuestId, nil) return InheritGuestIdResource(tx, user.Id, token.GuestId, nil)
} }
return err
}
return err return err
} }
@ -140,13 +138,17 @@ type UserProfile struct {
func (u *FsUserModel) RegisterByFusen(ctx context.Context, token *auth.RegisterToken) (*FsUser, error) { func (u *FsUserModel) RegisterByFusen(ctx context.Context, token *auth.RegisterToken) (*FsUser, error) {
user := &FsUser{} user := &FsUser{}
err := u.db.WithContext(ctx).Transaction(func(tx *gorm.DB) error { logcontent := ""
defer func() {
logx.Info("aaaa:", logcontent)
}()
err := u.db.WithContext(ctx).Transaction(func(tx *gorm.DB) error {
var err error
UserTx := tx.Model(user) UserTx := tx.Model(user)
err := UserTx.Where("email = ?", token.Email).Take(user).Error err = UserTx.Where("email = ?", token.Email).Take(user).Error
if err != nil {
// 没有找到在数据库就创建注册
if err == gorm.ErrRecordNotFound { if err == gorm.ErrRecordNotFound {
FirstName := token.Extend["first_name"].(string) FirstName := token.Extend["first_name"].(string)
LastName := token.Extend["last_name"].(string) LastName := token.Extend["last_name"].(string)
Resetaurant := token.Extend["resetaurant"].(string) Resetaurant := token.Extend["resetaurant"].(string)
@ -159,15 +161,16 @@ func (u *FsUserModel) RegisterByFusen(ctx context.Context, token *auth.RegisterT
user.LastName = &LastName user.LastName = &LastName
err = UserTx.Create(user).Error err = UserTx.Create(user).Error
log.Println("create")
if err != nil { if err != nil {
logx.Error(err)
return err return err
} }
if token.GuestId != 0 { logcontent += "create."
// 继承guest_id的资源表 // 继承guest_id的资源表
return InheritGuestIdResource(tx, user.Id, token.GuestId, func(txResouce, txUserMaterial, txUserInfo *gorm.DB) error { err = InheritGuestIdResource(tx, user.Id, token.GuestId, func(txResouce, txUserMaterial, txUserInfo *gorm.DB) error {
userProfile := &UserProfile{ userProfile := &UserProfile{
FirstName: FirstName, FirstName: FirstName,
LastName: LastName, LastName: LastName,
@ -186,21 +189,22 @@ func (u *FsUserModel) RegisterByFusen(ctx context.Context, token *auth.RegisterT
Ctime: &now, Ctime: &now,
Utime: &now, Utime: &now,
} }
logcontent += "profile."
return txUserInfo.Create(uinfo).Error return txUserInfo.Create(uinfo).Error
}) })
}
return err if err == gorm.ErrRecordNotFound {
} return nil
return err
} }
if err == nil {
return fmt.Errorf("the email had registered")
}
return err return err
}
return fmt.Errorf("the email had registered")
}) })
if err != nil { if err != nil && err != gorm.ErrRecordNotFound {
return nil, err return nil, err
} }

View File

@ -85,12 +85,13 @@ func main() {
indexHtmlPath := vueBuild + "/index.html" indexHtmlPath := vueBuild + "/index.html"
mux.Handle("/", http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { mux.Handle("/", http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
if strings.HasPrefix(r.URL.Path, "/api/") {
err := r.ParseMultipartForm(100 << 20) err := r.ParseMultipartForm(100 << 20)
if err != nil { if err != nil {
log.Println(err) log.Println(err)
} }
if strings.HasPrefix(r.URL.Path, "/api/") {
// 对/api开头的请求进行反向代理 // 对/api开头的请求进行反向代理
proxy := httputil.NewSingleHostReverseProxy(apiURL) proxy := httputil.NewSingleHostReverseProxy(apiURL)
proxy.ServeHTTP(w, r) proxy.ServeHTTP(w, r)

View File

@ -118,7 +118,7 @@ func (l *UserEmailConfirmationLogic) UserEmailConfirmation(req *types.RequestEma
// log.Println("aaaa", token) // log.Println("aaaa", token)
user, err := l.svcCtx.AllModels.FsUser.RegisterByFusen(l.ctx, token) user, err := l.svcCtx.AllModels.FsUser.RegisterByFusen(l.ctx, token)
if err != nil { if err != nil {
logx.Error(err, token.TraceId) logx.Error(err, ":", token.TraceId)
return resp.SetStatus(basic.CodeDbSqlErr) return resp.SetStatus(basic.CodeDbSqlErr)
} }