feat: update
This commit is contained in:
@@ -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)),
|
||||
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 {
|
||||
return errors.Wrap(err, "failed to get user")
|
||||
return errors.Wrap(err, "failed to create user")
|
||||
}
|
||||
} else {
|
||||
return errors.Wrap(err, "failed to get user")
|
||||
|
||||
@@ -135,20 +135,21 @@ func (m *usersModel) List(ctx context.Context, pagination *requests.Pagination,
|
||||
}
|
||||
|
||||
// Create creates a new user
|
||||
func (m *usersModel) Create(ctx context.Context, model *model.Users) error {
|
||||
model.CreatedAt = time.Now()
|
||||
model.UpdatedAt = time.Now()
|
||||
func (m *usersModel) Create(ctx context.Context, userModel *model.Users) (*model.Users, error) {
|
||||
userModel.CreatedAt = time.Now()
|
||||
userModel.UpdatedAt = time.Now()
|
||||
|
||||
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())
|
||||
|
||||
_, err := stmt.ExecContext(ctx, db)
|
||||
var createdUser model.Users
|
||||
err := stmt.QueryContext(ctx, db, &createdUser)
|
||||
if err != nil {
|
||||
m.log.Errorf("error creating user: %v", err)
|
||||
return err
|
||||
return nil, err
|
||||
}
|
||||
return nil
|
||||
return &createdUser, nil
|
||||
}
|
||||
|
||||
// Update updates an existing user
|
||||
|
||||
@@ -56,9 +56,9 @@ func (s *UsersTestSuite) Test_BatchInsert() {
|
||||
Username: "Username_" + openID,
|
||||
Avatar: lo.ToPtr("Avatar_" + openID),
|
||||
}
|
||||
err = Users.Create(context.Background(), user)
|
||||
m, err := Users.Create(context.Background(), user)
|
||||
So(err, ShouldBeNil)
|
||||
|
||||
So(m, ShouldNotBeNil)
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user