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)),
|
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")
|
||||||
|
|||||||
@@ -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
|
||||||
|
|||||||
@@ -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)
|
||||||
}
|
}
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user