feat: update

This commit is contained in:
yanghao05
2025-04-29 20:44:28 +08:00
parent e029327b0d
commit 751bbea51c
3 changed files with 12 additions and 14 deletions

View File

@@ -61,13 +61,10 @@ func (ctl *auth) Login(ctx fiber.Ctx, code, state, redirect string) error {
Username: fmt.Sprintf("u_%s", gonanoid.MustGenerate(salt, 8)), Username: fmt.Sprintf("u_%s", gonanoid.MustGenerate(salt, 8)),
Avatar: nil, Avatar: nil,
} }
if err := models.Users.Create(ctx.Context(), model); err != nil {
return errors.Wrap(err, "failed to create user")
}
user, err = models.Users.GetUserByOpenID(ctx.Context(), token.Openid) user, err = models.Users.Create(ctx.Context(), model)
if err != nil { if err != nil {
return errors.Wrap(err, "failed to get user") return errors.Wrap(err, "failed to create user")
} }
} else { } else {
return errors.Wrap(err, "failed to get user") return errors.Wrap(err, "failed to get user")

View File

@@ -135,20 +135,21 @@ func (m *usersModel) List(ctx context.Context, pagination *requests.Pagination,
} }
// Create creates a new user // Create creates a new user
func (m *usersModel) Create(ctx context.Context, model *model.Users) error { func (m *usersModel) Create(ctx context.Context, userModel *model.Users) (*model.Users, error) {
model.CreatedAt = time.Now() userModel.CreatedAt = time.Now()
model.UpdatedAt = time.Now() userModel.UpdatedAt = time.Now()
tbl := table.Users tbl := table.Users
stmt := tbl.INSERT(tbl.MutableColumns).MODEL(model) stmt := tbl.INSERT(tbl.MutableColumns).MODEL(userModel).RETURNING(tbl.AllColumns)
m.log.Infof("sql: %s", stmt.DebugSql()) m.log.Infof("sql: %s", stmt.DebugSql())
_, err := stmt.ExecContext(ctx, db) var createdUser model.Users
err := stmt.QueryContext(ctx, db, &createdUser)
if err != nil { if err != nil {
m.log.Errorf("error creating user: %v", err) m.log.Errorf("error creating user: %v", err)
return err return nil, err
} }
return nil return &createdUser, nil
} }
// Update updates an existing user // Update updates an existing user

View File

@@ -56,9 +56,9 @@ func (s *UsersTestSuite) Test_BatchInsert() {
Username: "Username_" + openID, Username: "Username_" + openID,
Avatar: lo.ToPtr("Avatar_" + openID), Avatar: lo.ToPtr("Avatar_" + openID),
} }
err = Users.Create(context.Background(), user) m, err := Users.Create(context.Background(), user)
So(err, ShouldBeNil) So(err, ShouldBeNil)
So(m, ShouldNotBeNil)
} }
}) })
} }