feat: update models

This commit is contained in:
Rogee
2025-05-23 20:49:10 +08:00
parent 57cb0a750b
commit 18808200c2
14 changed files with 69 additions and 73 deletions

View File

@@ -41,17 +41,22 @@ func (ctl *orders) Refund(ctx fiber.Ctx, id int64) error {
return err
}
user, err := model.UsersModel.GetByID(ctx.Context(), order.UserID)
if err != nil {
return err
}
post, err := model.PostsModel.GetByID(ctx.Context(), order.PostID)
if err != nil {
return err
}
if order.PaymentMethod == "balance" {
if err := model.UsersModel.AddBalance(ctx.Context(), order.UserID, order.Meta.Data.CostBalance); err != nil {
if err := user.AddBalance(ctx.Context(), order.Meta.Data.CostBalance); err != nil {
return errors.Wrap(err, "add balance failed")
}
if err := model.UsersModel.RevokePosts(ctx.Context(), order.UserID, order.PostID); err != nil {
if err := user.RevokePosts(ctx.Context(), order.PostID); err != nil {
return errors.Wrap(err, "revoke posts failed")
}

View File

@@ -51,5 +51,9 @@ type UserBalance struct {
// @Bind id path
// @Bind balance body
func (ctl *users) Balance(ctx fiber.Ctx, id int64, balance *UserBalance) error {
return model.UsersModel.AddBalance(ctx.Context(), id, balance.Balance)
user, err := model.UsersModel.GetByID(ctx.Context(), id)
if err != nil {
return err
}
return user.SetBalance(ctx.Context(), balance.Balance)
}