package gmodel import ( "context" "fusenapi/utils/auth" "time" "gorm.io/gorm" ) // TODO: 使用model的属性做你想做的 type UserBasicInfoForSave struct { ID uint `gorm:"primary_key" json:"id"` FirstName string `gorm:"-" json:"first_name"` LastName string `gorm:"-" json:"last_name"` Mobile string `gorm:"-" json:"mobile"` Company string `gorm:"-" json:"company"` IsOrderStatusEmail int64 `gorm:"-" json:"is_order_status_email"` IsEmailAdvertisement int64 `gorm:"-" json:"is_email_advertisement"` IsOrderStatusPhone int64 `gorm:"-" json:"is_order_status_phone"` IsPhoneAdvertisement int64 `gorm:"-" json:"is_phone_advertisement"` Type int64 `gorm:"-" json:"type"` IsOpenRender int64 `gorm:"-" json:"is_open_render"` IsLowRendering int64 `gorm:"-" json:"is_low_rendering"` IsRemoveBg int64 `gorm:"-" json:"is_remove_bg"` } func (u *FsUserModel) FindUserByEmail(ctx context.Context, emailname string) (resp FsUser, err error) { err = u.db.WithContext(ctx).Model(&FsUser{}).Where("`email` = ?", emailname).Take(&resp).Error return resp, err } func (u *FsUserModel) FindUserById(ctx context.Context, Id int64) (resp FsUser, err error) { err = u.db.WithContext(ctx).Model(&FsUser{}).Where("`id` = ? and is_del = ?", Id, 0).Take(&resp).Error return resp, err } func (u *FsUserModel) FindUserByGoogleId(ctx context.Context, Id int64) (resp *FsUser, err error) { err = u.db.WithContext(ctx).Model(resp).Where("`google_id` = ? and is_del = ?", Id, 0).Take(resp).Error return resp, err } func (u *FsUserModel) Transaction(ctx context.Context, fc func(tx *gorm.DB) error) (err error) { return u.db.WithContext(ctx).Transaction(fc) } // 谷歌平台的注册流程 func (u *FsUserModel) RegisterByGoogleOAuth(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 if err != nil { // 没有找到在数据库就创建注册 if err == gorm.ErrRecordNotFound { createAt := time.Now().Unix() user.Email = &token.Email user.CreatedAt = &createAt user.GoogleId = &token.Id user.PasswordHash = &token.Password err = tx.Model(user).Create(user).Error if err != nil { return err } if token.GuestId != 0 { // 继承guest_id的资源表 return tx.Model(&FsResource{}). Where("guest_id = ?", token.GuestId). UpdateColumn("user_id", user.Id).Error } return err } return err } // 如果已经存在,把谷歌id 加入到用户信息里 user.GoogleId = &token.Id return tx.Model(user).Update("google_id", user).Error }) if err != nil { return nil, err } return user, nil } func (u *FsUserModel) UpdateUserBasicInfoById(ctx context.Context, Id int64, user *UserBasicInfoForSave) (err error) { err = u.db.WithContext(ctx).Model(&FsUser{}). Where("`id` = ? and is_del = ? and status = ?", Id, 0, 1). Updates(map[string]interface{}{ "first_name": user.FirstName, "last_name": user.LastName, "mobile": user.Mobile, "company": user.Company, "is_order_status_email": user.IsOrderStatusEmail, "is_email_advertisement": user.IsEmailAdvertisement, "is_order_status_phone": user.IsOrderStatusPhone, "is_phone_advertisement": user.IsPhoneAdvertisement, "type": user.Type, "is_open_render": user.IsOpenRender, "is_low_rendering": user.IsLowRendering, "is_remove_bg": user.IsRemoveBg, }).Error return err }