rogee/quyun
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: update

Browse Source
This commit is contained in:
Rogee
2025-05-23 21:39:01 +08:00
parent f370887a71
commit f13ef4388e
20 changed files with 114 additions and 159 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
backend/app/http/admin/medias.go
View File

@@ -19,8 +19,8 @@ type medias struct {
// @Bind pagination query // @Bind pagination query
// @Bind query query // @Bind query query
func (ctl *medias) List(ctx fiber.Ctx, pagination *requests.Pagination, query *ListQuery) (*requests.Pager, error) { func (ctl *medias) List(ctx fiber.Ctx, pagination *requests.Pagination, query *ListQuery) (*requests.Pager, error) {
cond := model.MediasModel.BuildConditionWithKey(query.Keyword) cond := model.MediasModel().BuildConditionWithKey(query.Keyword)
return model.MediasModel.List(ctx.Context(), pagination, cond) return model.MediasModel().List(ctx.Context(), pagination, cond)
} }
// Show media // Show media
@@ -28,7 +28,7 @@ func (ctl *medias) List(ctx fiber.Ctx, pagination *requests.Pagination, query *L
// @Router /admin/medias/:id [get] // @Router /admin/medias/:id [get]
// @Bind id path // @Bind id path
func (ctl *medias) Show(ctx fiber.Ctx, id int64) error { func (ctl *medias) Show(ctx fiber.Ctx, id int64) error {
media, err := model.MediasModel.GetByID(ctx.Context(), id) media, err := model.MediasModel().GetByID(ctx.Context(), id)
if err != nil { if err != nil {
return ctx.SendString("Media not found") return ctx.SendString("Media not found")
} }
@@ -46,7 +46,7 @@ func (ctl *medias) Show(ctx fiber.Ctx, id int64) error {
// @Router /admin/medias/:id [delete] // @Router /admin/medias/:id [delete]
// @Bind id path // @Bind id path
func (ctl *medias) Delete(ctx fiber.Ctx, id int64) error { func (ctl *medias) Delete(ctx fiber.Ctx, id int64) error {
media, err := model.MediasModel.GetByID(ctx.Context(), id) media, err := model.MediasModel().GetByID(ctx.Context(), id)
if err != nil { if err != nil {
return ctx.SendString("Media not found") return ctx.SendString("Media not found")
} }
@@ -55,7 +55,7 @@ func (ctl *medias) Delete(ctx fiber.Ctx, id int64) error {
return err return err
} }
if err := model.MediasModel.Delete(ctx.Context(), id); err != nil { if err := model.MediasModel().Delete(ctx.Context(), id); err != nil {
return err return err
} }
return ctx.SendStatus(fiber.StatusNoContent) return ctx.SendStatus(fiber.StatusNoContent)

14
backend/app/http/admin/orders.go
View File

@@ -28,25 +28,25 @@ type orders struct {
// @Bind pagination query // @Bind pagination query
// @Bind query query // @Bind query query
func (ctl *orders) List(ctx fiber.Ctx, pagination *requests.Pagination, query *OrderListQuery) (*requests.Pager, error) { func (ctl *orders) List(ctx fiber.Ctx, pagination *requests.Pagination, query *OrderListQuery) (*requests.Pager, error) {
cond := model.OrdersModel.BuildConditionWithKey(query.OrderNumber, query.UserID) cond := model.OrdersModel().BuildConditionWithKey(query.OrderNumber, query.UserID)
return model.OrdersModel.List(ctx.Context(), pagination, cond) return model.OrdersModel().List(ctx.Context(), pagination, cond)
} }
// Refund // Refund
// @Router /admin/orders/:id/refund [post] // @Router /admin/orders/:id/refund [post]
// @Bind id path // @Bind id path
func (ctl *orders) Refund(ctx fiber.Ctx, id int64) error { func (ctl *orders) Refund(ctx fiber.Ctx, id int64) error {
order, err := model.OrdersModel.GetByID(ctx.Context(), id) order, err := model.OrdersModel().GetByID(ctx.Context(), id)
if err != nil { if err != nil {
return err return err
} }
user, err := model.UsersModel.GetByID(ctx.Context(), order.UserID) user, err := model.UsersModel().GetByID(ctx.Context(), order.UserID)
if err != nil { if err != nil {
return err return err
} }
post, err := model.PostsModel.GetByID(ctx.Context(), order.PostID) post, err := model.PostsModel().GetByID(ctx.Context(), order.PostID)
if err != nil { if err != nil {
return err return err
} }
@@ -61,7 +61,7 @@ func (ctl *orders) Refund(ctx fiber.Ctx, id int64) error {
} }
order.Status = fields.OrderStatusRefundSuccess order.Status = fields.OrderStatusRefundSuccess
if err := model.OrdersModel.Update(ctx.Context(), order); err != nil { if err := order.Update(ctx.Context()); err != nil {
return errors.Wrap(err, "update order failed") return errors.Wrap(err, "update order failed")
} }
@@ -87,7 +87,7 @@ func (ctl *orders) Refund(ctx fiber.Ctx, id int64) error {
order.RefundTransactionID = resp.RefundId order.RefundTransactionID = resp.RefundId
order.Status = fields.OrderStatusRefundProcessing order.Status = fields.OrderStatusRefundProcessing
if err := model.OrdersModel.Update(ctx.Context(), order); err != nil { if err := order.Update(ctx.Context()); err != nil {
return err return err
} }

20
backend/app/http/admin/posts.go
View File

@@ -28,7 +28,7 @@ func (ctl *posts) List(ctx fiber.Ctx, pagination *requests.Pagination, query *Li
// conds.Post_Status(fields.PostStatusPublished), // conds.Post_Status(fields.PostStatusPublished),
conds.Post_Like(query.Keyword), conds.Post_Like(query.Keyword),
} }
pager, err := model.PostsModel.List(ctx.Context(), pagination, conds...) pager, err := model.PostsModel().List(ctx.Context(), pagination, conds...)
if err != nil { if err != nil {
return nil, err return nil, err
} }
@@ -37,7 +37,7 @@ func (ctl *posts) List(ctx fiber.Ctx, pagination *requests.Pagination, query *Li
return item.ID return item.ID
}) })
if len(postIds) > 0 { if len(postIds) > 0 {
postCntMap, err := model.PostsModel.BoughtStatistics(ctx.Context(), postIds) postCntMap, err := model.PostsModel().BoughtStatistics(ctx.Context(), postIds)
if err != nil { if err != nil {
return pager, err return pager, err
} }
@@ -85,7 +85,7 @@ func (ctl *posts) Create(ctx fiber.Ctx, form *PostForm) error {
} }
if form.Medias != nil { if form.Medias != nil {
medias, err := model.MediasModel.GetByIds(ctx.Context(), form.Medias) medias, err := model.MediasModel().GetByIds(ctx.Context(), form.Medias)
if err != nil { if err != nil {
return err return err
} }
@@ -111,7 +111,7 @@ func (ctl *posts) Create(ctx fiber.Ctx, form *PostForm) error {
// @Bind id path // @Bind id path
// @Bind form body // @Bind form body
func (ctl *posts) Update(ctx fiber.Ctx, id int64, form *PostForm) error { func (ctl *posts) Update(ctx fiber.Ctx, id int64, form *PostForm) error {
post, err := model.PostsModel.GetByID(ctx.Context(), id) post, err := model.PostsModel().GetByID(ctx.Context(), id)
if err != nil { if err != nil {
return err return err
} }
@@ -125,7 +125,7 @@ func (ctl *posts) Update(ctx fiber.Ctx, id int64, form *PostForm) error {
post.Tags = fields.Json[[]string]{} post.Tags = fields.Json[[]string]{}
if form.Medias != nil { if form.Medias != nil {
medias, err := model.MediasModel.GetByIds(ctx.Context(), form.Medias) medias, err := model.MediasModel().GetByIds(ctx.Context(), form.Medias)
if err != nil { if err != nil {
return err return err
} }
@@ -150,7 +150,7 @@ func (ctl *posts) Update(ctx fiber.Ctx, id int64, form *PostForm) error {
// @Router /admin/posts/:id [delete] // @Router /admin/posts/:id [delete]
// @Bind id path // @Bind id path
func (ctl *posts) Delete(ctx fiber.Ctx, id int64) error { func (ctl *posts) Delete(ctx fiber.Ctx, id int64) error {
post, err := model.PostsModel.GetByID(ctx.Context(), id) post, err := model.PostsModel().GetByID(ctx.Context(), id)
if err != nil { if err != nil {
return err return err
} }
@@ -175,12 +175,12 @@ type PostItem struct {
// @Router /admin/posts/:id [get] // @Router /admin/posts/:id [get]
// @Bind id path // @Bind id path
func (ctl *posts) Show(ctx fiber.Ctx, id int64) (*PostItem, error) { func (ctl *posts) Show(ctx fiber.Ctx, id int64) (*PostItem, error) {
post, err := model.PostsModel.GetByID(ctx.Context(), id) post, err := model.PostsModel().GetByID(ctx.Context(), id)
if err != nil { if err != nil {
return nil, err return nil, err
} }
medias, err := model.MediasModel.GetByIds(ctx.Context(), lo.Map(post.Assets.Data, func(asset fields.MediaAsset, _ int) int64 { medias, err := model.MediasModel().GetByIds(ctx.Context(), lo.Map(post.Assets.Data, func(asset fields.MediaAsset, _ int) int64 {
return asset.Media return asset.Media
})) }))
if err != nil { if err != nil {
@@ -198,12 +198,12 @@ func (ctl *posts) Show(ctx fiber.Ctx, id int64) (*PostItem, error) {
// @Bind id path // @Bind id path
// @Bind userId path // @Bind userId path
func (ctl *posts) SendTo(ctx fiber.Ctx, id, userId int64) error { func (ctl *posts) SendTo(ctx fiber.Ctx, id, userId int64) error {
post, err := model.PostsModel.GetByID(ctx.Context(), id) post, err := model.PostsModel().GetByID(ctx.Context(), id)
if err != nil { if err != nil {
return err return err
} }
user, err := model.UsersModel.GetByID(ctx.Context(), userId) user, err := model.UsersModel().GetByID(ctx.Context(), userId)
if err != nil { if err != nil {
return err return err
} }

12
backend/app/http/admin/statistics.go
View File

@@ -29,31 +29,31 @@ func (s *statistics) statistics(ctx fiber.Ctx) (*StatisticsResponse, error) {
var err error var err error
statistics.PostDraft, err = model.PostsModel.Count(ctx.Context(), table.Posts.Status.EQ(Int(int64(fields.PostStatusDraft)))) statistics.PostDraft, err = model.PostsModel().Count(ctx.Context(), table.Posts.Status.EQ(Int(int64(fields.PostStatusDraft))))
if err != nil { if err != nil {
return nil, err return nil, err
} }
statistics.PostPublished, err = model.PostsModel.Count(ctx.Context(), table.Posts.Status.EQ(Int(int64(fields.PostStatusPublished)))) statistics.PostPublished, err = model.PostsModel().Count(ctx.Context(), table.Posts.Status.EQ(Int(int64(fields.PostStatusPublished))))
if err != nil { if err != nil {
return nil, err return nil, err
} }
statistics.Media, err = model.MediasModel.Count(ctx.Context()) statistics.Media, err = model.MediasModel().Count(ctx.Context())
if err != nil { if err != nil {
return nil, err return nil, err
} }
statistics.Order, err = model.OrdersModel.Count(ctx.Context(), table.Orders.Status.EQ(Int(int64(fields.OrderStatusCompleted)))) statistics.Order, err = model.OrdersModel().Count(ctx.Context(), table.Orders.Status.EQ(Int(int64(fields.OrderStatusCompleted))))
if err != nil { if err != nil {
return nil, err return nil, err
} }
statistics.User, err = model.UsersModel.Count(ctx.Context(), BoolExp(Bool(true))) statistics.User, err = model.UsersModel().Count(ctx.Context(), BoolExp(Bool(true)))
if err != nil { if err != nil {
return nil, err return nil, err
} }
statistics.Amount, err = model.OrdersModel.SumAmount(ctx.Context()) statistics.Amount, err = model.OrdersModel().SumAmount(ctx.Context())
if err != nil { if err != nil {
return nil, err return nil, err
} }

6
backend/app/http/admin/uploads.go
View File

@@ -38,7 +38,7 @@ type PreCheckResp struct {
// @Bind ext path // @Bind ext path
// @Bind mime query // @Bind mime query
func (up *uploads) PreUploadCheck(ctx fiber.Ctx, md5, ext, mime string) (*PreCheckResp, error) { func (up *uploads) PreUploadCheck(ctx fiber.Ctx, md5, ext, mime string) (*PreCheckResp, error) {
_, err := model.MediasModel.GetByHash(ctx.Context(), md5) _, err := model.MediasModel().GetByHash(ctx.Context(), md5)
if err != nil && errors.Is(err, qrm.ErrNoRows) { if err != nil && errors.Is(err, qrm.ErrNoRows) {
preSign, err := up.oss.PreSignUpload(ctx.Context(), fmt.Sprintf("%s.%s", md5, ext), mime) preSign, err := up.oss.PreSignUpload(ctx.Context(), fmt.Sprintf("%s.%s", md5, ext), mime)
if err != nil { if err != nil {
@@ -63,7 +63,7 @@ type PostUploadedForm struct {
// @Router /admin/uploads/post-uploaded-action [post] // @Router /admin/uploads/post-uploaded-action [post]
// @Bind body body // @Bind body body
func (up *uploads) PostUploadedAction(ctx fiber.Ctx, body *PostUploadedForm) error { func (up *uploads) PostUploadedAction(ctx fiber.Ctx, body *PostUploadedForm) error {
m, err := model.MediasModel.GetByHash(ctx.Context(), body.Md5) m, err := model.MediasModel().GetByHash(ctx.Context(), body.Md5)
if err != nil && !errors.Is(err, qrm.ErrNoRows) { if err != nil && !errors.Is(err, qrm.ErrNoRows) {
return err return err
} }
@@ -75,7 +75,7 @@ func (up *uploads) PostUploadedAction(ctx fiber.Ctx, body *PostUploadedForm) err
Hash: body.Md5, Hash: body.Md5,
Path: filepath.Join(UPLOAD_PATH, body.Md5+filepath.Ext(body.OriginalName)), Path: filepath.Join(UPLOAD_PATH, body.Md5+filepath.Ext(body.OriginalName)),
} }
if err := model.MediasModel.Create(ctx.Context(), m); err != nil { if err := model.MediasModel().Create(ctx.Context(), m); err != nil {
return err return err
} }

10
backend/app/http/admin/users.go
View File

@@ -20,8 +20,8 @@ type users struct{}
// @Bind pagination query // @Bind pagination query
// @Bind query query // @Bind query query
func (ctl *users) List(ctx fiber.Ctx, pagination *requests.Pagination, query *UserListQuery) (*requests.Pager, error) { func (ctl *users) List(ctx fiber.Ctx, pagination *requests.Pagination, query *UserListQuery) (*requests.Pager, error) {
cond := model.UsersModel.BuildConditionWithKey(query.Keyword) cond := model.UsersModel().BuildConditionWithKey(query.Keyword)
return model.UsersModel.List(ctx.Context(), pagination, cond) return model.UsersModel().List(ctx.Context(), pagination, cond)
} }
// Show user // Show user
@@ -29,7 +29,7 @@ func (ctl *users) List(ctx fiber.Ctx, pagination *requests.Pagination, query *Us
// @Router /admin/users/:id [get] // @Router /admin/users/:id [get]
// @Bind id path // @Bind id path
func (ctl *users) Show(ctx fiber.Ctx, id int64) (*model.Users, error) { func (ctl *users) Show(ctx fiber.Ctx, id int64) (*model.Users, error) {
return model.UsersModel.GetByID(ctx.Context(), id) return model.UsersModel().GetByID(ctx.Context(), id)
} }
// Articles show user bought articles // Articles show user bought articles
@@ -38,7 +38,7 @@ func (ctl *users) Show(ctx fiber.Ctx, id int64) (*model.Users, error) {
// @Bind id path // @Bind id path
// @Bind pagination query // @Bind pagination query
func (ctl *users) Articles(ctx fiber.Ctx, id int64, pagination *requests.Pagination) (*requests.Pager, error) { func (ctl *users) Articles(ctx fiber.Ctx, id int64, pagination *requests.Pagination) (*requests.Pager, error) {
return model.PostsModel.Bought(ctx.Context(), id, pagination) return model.PostsModel().Bought(ctx.Context(), id, pagination)
} }
type UserBalance struct { type UserBalance struct {
@@ -51,7 +51,7 @@ type UserBalance struct {
// @Bind id path // @Bind id path
// @Bind balance body // @Bind balance body
func (ctl *users) Balance(ctx fiber.Ctx, id int64, balance *UserBalance) error { func (ctl *users) Balance(ctx fiber.Ctx, id int64, balance *UserBalance) error {
user, err := model.UsersModel.GetByID(ctx.Context(), id) user, err := model.UsersModel().GetByID(ctx.Context(), id)
if err != nil { if err != nil {
return err return err
} }

2
backend/app/http/auth.go
View File

@@ -89,7 +89,7 @@ func (ctl *auth) Login(ctx fiber.Ctx, code, state, redirect string) error {
Scope: token.Scope, Scope: token.Scope,
}), }),
} }
user, err := model.UsersModel.GetUserByOpenIDOrCreate(ctx.Context(), token.GetOpenID(), userModel) user, err := model.UsersModel().GetUserByOpenIDOrCreate(ctx.Context(), token.GetOpenID(), userModel)
if err != nil { if err != nil {
return errors.Wrap(err, "failed to get user by openid") return errors.Wrap(err, "failed to get user by openid")
} }

24
backend/app/http/posts.go
View File

@@ -45,7 +45,7 @@ func (ctl *posts) List(ctx fiber.Ctx, pagination *requests.Pagination, query *Li
conds.Post_Like(query.Keyword), conds.Post_Like(query.Keyword),
} }
pager, err := model.PostsModel.List(ctx.Context(), pagination, conds...) pager, err := model.PostsModel().List(ctx.Context(), pagination, conds...)
if err != nil { if err != nil {
log.WithError(err).Errorf("post list err: %v", err) log.WithError(err).Errorf("post list err: %v", err)
return nil, err return nil, err
@@ -59,7 +59,7 @@ func (ctl *posts) List(ctx fiber.Ctx, pagination *requests.Pagination, query *Li
} }
items := lo.FilterMap(pager.Items.([]model.Posts), func(item model.Posts, _ int) (PostItem, bool) { items := lo.FilterMap(pager.Items.([]model.Posts), func(item model.Posts, _ int) (PostItem, bool) {
medias, err := model.PostsModel.GetMediaByIds(ctx.Context(), item.HeadImages.Data) medias, err := model.PostsModel().GetMediaByIds(ctx.Context(), item.HeadImages.Data)
if err != nil { if err != nil {
log.Errorf("GetMediaByIds err: %v", err) log.Errorf("GetMediaByIds err: %v", err)
return PostItem{}, false return PostItem{}, false
@@ -118,7 +118,7 @@ type PostItem struct {
func (ctl *posts) Show(ctx fiber.Ctx, id int64, user *model.Users) (*PostItem, error) { func (ctl *posts) Show(ctx fiber.Ctx, id int64, user *model.Users) (*PostItem, error) {
log.Infof("Fetching post with ID: %d", id) log.Infof("Fetching post with ID: %d", id)
post, err := model.PostsModel.GetByID(ctx.Context(), id, conds.Post_NotDeleted(), conds.Post_Status(fields.PostStatusPublished)) post, err := model.PostsModel().GetByID(ctx.Context(), id, conds.Post_NotDeleted(), conds.Post_Status(fields.PostStatusPublished))
if err != nil { if err != nil {
log.WithError(err).Errorf("GetByID err: %v", err) log.WithError(err).Errorf("GetByID err: %v", err)
return nil, err return nil, err
@@ -129,7 +129,7 @@ func (ctl *posts) Show(ctx fiber.Ctx, id int64, user *model.Users) (*PostItem, e
return nil, err return nil, err
} }
medias, err := model.PostsModel.GetMediaByIds(ctx.Context(), post.HeadImages.Data) medias, err := model.PostsModel().GetMediaByIds(ctx.Context(), post.HeadImages.Data)
if err != nil { if err != nil {
return nil, err return nil, err
} }
@@ -179,7 +179,7 @@ func (ctl *posts) Play(ctx fiber.Ctx, id int64, user *model.Users) (*PlayUrl, er
} }
log.Infof("Fetching play URL for post ID: %d", id) log.Infof("Fetching play URL for post ID: %d", id)
post, err := model.PostsModel.GetByID(ctx.Context(), id) post, err := model.PostsModel().GetByID(ctx.Context(), id)
if err != nil { if err != nil {
log.WithError(err).Errorf("GetByID err: %v", err) log.WithError(err).Errorf("GetByID err: %v", err)
return nil, err return nil, err
@@ -188,7 +188,7 @@ func (ctl *posts) Play(ctx fiber.Ctx, id int64, user *model.Users) (*PlayUrl, er
for _, asset := range post.Assets.Data { for _, asset := range post.Assets.Data {
if asset.Type == "video/mp4" && asset.Metas != nil && asset.Metas.Short == preview { if asset.Type == "video/mp4" && asset.Metas != nil && asset.Metas.Short == preview {
media, err := model.MediasModel.GetByID(ctx.Context(), asset.Media) media, err := model.MediasModel().GetByID(ctx.Context(), asset.Media)
if err != nil { if err != nil {
log.WithError(err).Errorf("medias GetByID err: %v", err) log.WithError(err).Errorf("medias GetByID err: %v", err)
return nil, err return nil, err
@@ -222,7 +222,7 @@ func (ctl *posts) Mine(ctx fiber.Ctx, pagination *requests.Pagination, query *Li
conds.Post_Status(fields.PostStatusPublished), conds.Post_Status(fields.PostStatusPublished),
conds.Post_Like(query.Keyword), conds.Post_Like(query.Keyword),
} }
pager, err := model.UsersModel.PostList(ctx.Context(), user.ID, pagination, conds...) pager, err := model.UsersModel().PostList(ctx.Context(), user.ID, pagination, conds...)
if err != nil { if err != nil {
log.WithError(err).Errorf("post list err: %v", err) log.WithError(err).Errorf("post list err: %v", err)
return nil, err return nil, err
@@ -231,7 +231,7 @@ func (ctl *posts) Mine(ctx fiber.Ctx, pagination *requests.Pagination, query *Li
postIds := lo.Map(pager.Items.([]model.Posts), func(item model.Posts, _ int) int64 { return item.ID }) postIds := lo.Map(pager.Items.([]model.Posts), func(item model.Posts, _ int) int64 { return item.ID })
if len(postIds) > 0 { if len(postIds) > 0 {
items := lo.FilterMap(pager.Items.([]model.Posts), func(item model.Posts, _ int) (PostItem, bool) { items := lo.FilterMap(pager.Items.([]model.Posts), func(item model.Posts, _ int) (PostItem, bool) {
medias, err := model.PostsModel.GetMediaByIds(ctx.Context(), item.HeadImages.Data) medias, err := model.PostsModel().GetMediaByIds(ctx.Context(), item.HeadImages.Data)
if err != nil { if err != nil {
log.Errorf("GetMediaByIds err: %v", err) log.Errorf("GetMediaByIds err: %v", err)
return PostItem{}, false return PostItem{}, false
@@ -279,19 +279,19 @@ func (ctl *posts) Buy(ctx fiber.Ctx, id int64, user *model.Users) (*wechat.JSAPI
return nil, errors.New("已经购买过了") return nil, errors.New("已经购买过了")
} }
post, err := model.PostsModel.GetByID(ctx.Context(), id) post, err := model.PostsModel().GetByID(ctx.Context(), id)
if err != nil { if err != nil {
return nil, errors.Wrapf(err, " failed to get post: %d", id) return nil, errors.Wrapf(err, " failed to get post: %d", id)
} }
payPrice := post.Price * int64(post.Discount) / 100 payPrice := post.Price * int64(post.Discount) / 100
order, err := model.OrdersModel.Create(ctx.Context(), user.ID, post.ID) order, err := model.OrdersModel().Create(ctx.Context(), user.ID, post.ID)
if err != nil { if err != nil {
return nil, errors.Wrap(err, "订单创建失败") return nil, errors.Wrap(err, "订单创建失败")
} }
if user.Balance >= payPrice { if user.Balance >= payPrice {
if err := model.OrdersModel.SetMeta(ctx.Context(), order.ID, func(om fields.OrderMeta) fields.OrderMeta { if err := order.SetMeta(ctx.Context(), func(om fields.OrderMeta) fields.OrderMeta {
om.CostBalance = payPrice om.CostBalance = payPrice
return om return om
}); err != nil { }); err != nil {
@@ -309,7 +309,7 @@ func (ctl *posts) Buy(ctx fiber.Ctx, id int64, user *model.Users) (*wechat.JSAPI
} }
payPrice = payPrice - user.Balance payPrice = payPrice - user.Balance
if err := model.OrdersModel.SetMeta(ctx.Context(), order.ID, func(om fields.OrderMeta) fields.OrderMeta { if err := order.SetMeta(ctx.Context(), func(om fields.OrderMeta) fields.OrderMeta {
om.CostBalance = user.Balance om.CostBalance = user.Balance
return om return om
}); err != nil { }); err != nil {

6
backend/app/jobs/balance_pay_notify.go
View File

@@ -45,7 +45,7 @@ func (w *BalancePayNotifyWorker) Work(ctx context.Context, job *Job[BalancePayNo
log.Infof("[Start] Working on job with strings: %+v", job.Args) log.Infof("[Start] Working on job with strings: %+v", job.Args)
defer log.Infof("[End] Finished %s", job.Args.Kind()) defer log.Infof("[End] Finished %s", job.Args.Kind())
order, err := model.OrdersModel.GetByOrderNo(context.Background(), job.Args.OrderNo) order, err := model.OrdersModel().GetByOrderNo(context.Background(), job.Args.OrderNo)
if err != nil { if err != nil {
log.Errorf("GetByOrderNo error:%v", err) log.Errorf("GetByOrderNo error:%v", err)
return err return err
@@ -56,7 +56,7 @@ func (w *BalancePayNotifyWorker) Work(ctx context.Context, job *Job[BalancePayNo
return JobCancel(fmt.Errorf("Order already paid, currently status: %d", order.Status)) return JobCancel(fmt.Errorf("Order already paid, currently status: %d", order.Status))
} }
user, err := model.UsersModel.GetByID(context.Background(), order.UserID) user, err := model.UsersModel().GetByID(context.Background(), order.UserID)
if err != nil { if err != nil {
log.Errorf("GetByID error:%v", err) log.Errorf("GetByID error:%v", err)
return errors.Wrap(err, "get user error") return errors.Wrap(err, "get user error")
@@ -93,7 +93,7 @@ func (w *BalancePayNotifyWorker) Work(ctx context.Context, job *Job[BalancePayNo
return errors.Wrap(err, "BuyPosts error") return errors.Wrap(err, "BuyPosts error")
} }
if err := model.OrdersModel.Update(context.Background(), order); err != nil { if err := order.Update(context.Background()); err != nil {
log.Errorf("Update order error:%v", err) log.Errorf("Update order error:%v", err)
return errors.Wrap(err, "Update order error") return errors.Wrap(err, "Update order error")
} }

2
backend/app/jobs/download_from_alioss.go
View File

@@ -55,7 +55,7 @@ func (w *DownloadFromAliOSSWorker) Work(ctx context.Context, job *Job[DownloadFr
log.Infof("[Start] Working on job with strings: %+v", job.Args) log.Infof("[Start] Working on job with strings: %+v", job.Args)
defer log.Infof("[End] Finished %s", job.Args.Kind()) defer log.Infof("[End] Finished %s", job.Args.Kind())
media, err := model.MediasModel.GetByHash(ctx, job.Args.MediaHash) media, err := model.MediasModel().GetByHash(ctx, job.Args.MediaHash)
if err != nil { if err != nil {
log.Errorf("Error getting media by ID: %v", err) log.Errorf("Error getting media by ID: %v", err)
return JobCancel(err) return JobCancel(err)

4
backend/app/jobs/publish_draft_posts.go
View File

@@ -56,13 +56,13 @@ func (w *PublishDraftPostsWorker) Work(ctx context.Context, job *Job[PublishDraf
log.Infof("[Start] Working on job with strings: %+v", job.Args) log.Infof("[Start] Working on job with strings: %+v", job.Args)
defer log.Infof("[End] Finished %s", job.Args.Kind()) defer log.Infof("[End] Finished %s", job.Args.Kind())
media, err := model.MediasModel.GetByHash(ctx, job.Args.MediaHash) media, err := model.MediasModel().GetByHash(ctx, job.Args.MediaHash)
if err != nil { if err != nil {
log.Errorf("Error getting media by ID: %v", err) log.Errorf("Error getting media by ID: %v", err)
return JobCancel(err) return JobCancel(err)
} }
relationMedias, err := model.MediasModel.GetRelations(ctx, media.Hash) relationMedias, err := model.MediasModel().GetRelations(ctx, media.Hash)
if err != nil { if err != nil {
log.Errorf("Error getting relation medias: %v", err) log.Errorf("Error getting relation medias: %v", err)
return JobCancel(err) return JobCancel(err)

4
backend/app/jobs/video_cut.go
View File

@@ -54,7 +54,7 @@ func (w *VideoCutWorker) Work(ctx context.Context, job *Job[VideoCut]) error {
log.Infof("[Start] Working on job with strings: %+v", job.Args) log.Infof("[Start] Working on job with strings: %+v", job.Args)
defer log.Infof("[End] Finished %s", job.Args.Kind()) defer log.Infof("[End] Finished %s", job.Args.Kind())
media, err := model.MediasModel.GetByHash(ctx, job.Args.MediaHash) media, err := model.MediasModel().GetByHash(ctx, job.Args.MediaHash)
if err != nil { if err != nil {
log.Errorf("Error getting media by ID: %v", err) log.Errorf("Error getting media by ID: %v", err)
return JobCancel(err) return JobCancel(err)
@@ -81,7 +81,7 @@ func (w *VideoCutWorker) Work(ctx context.Context, job *Job[VideoCut]) error {
Short: false, Short: false,
Duration: duration, Duration: duration,
} }
if err := model.MediasModel.UpdateMetas(ctx, media.ID, metas); err != nil { if err := model.MediasModel().UpdateMetas(ctx, media.ID, metas); err != nil {
log.Errorf("Error updating media metas: %v", err) log.Errorf("Error updating media metas: %v", err)
return errors.Wrap(err, "update media metas") return errors.Wrap(err, "update media metas")
} }

4
backend/app/jobs/video_extract_head_image.go
View File

@@ -57,7 +57,7 @@ func (w *VideoExtractHeadImageWorker) Work(ctx context.Context, job *Job[VideoEx
log.Infof("[Start] Working on job with strings: %+v", job.Args) log.Infof("[Start] Working on job with strings: %+v", job.Args)
defer log.Infof("[End] Finished %s", job.Args.Kind()) defer log.Infof("[End] Finished %s", job.Args.Kind())
media, err := model.MediasModel.GetByHash(ctx, job.Args.MediaHash) media, err := model.MediasModel().GetByHash(ctx, job.Args.MediaHash)
if err != nil { if err != nil {
log.Errorf("Error getting media by ID: %v", err) log.Errorf("Error getting media by ID: %v", err)
return JobCancel(err) return JobCancel(err)
@@ -110,7 +110,7 @@ func (w *VideoExtractHeadImageWorker) Work(ctx context.Context, job *Job[VideoEx
log.Errorf("Error removing original file: %v", err) log.Errorf("Error removing original file: %v", err)
} }
if err := model.MediasModel.Create(ctx, imageMedia); err != nil { if err := model.MediasModel().Create(ctx, imageMedia); err != nil {
log.Errorf("Error creating media record: %v", err) log.Errorf("Error creating media record: %v", err)
return errors.Wrap(err, "failed to create media record") return errors.Wrap(err, "failed to create media record")
} }

4
backend/app/jobs/video_store_short.go
View File

@@ -57,7 +57,7 @@ func (w *VideoStoreShortWorker) Work(ctx context.Context, job *Job[VideoStoreSho
log.Infof("[Start] Working on job with strings: %+v", job.Args) log.Infof("[Start] Working on job with strings: %+v", job.Args)
defer log.Infof("[End] Finished %s", job.Args.Kind()) defer log.Infof("[End] Finished %s", job.Args.Kind())
media, err := model.MediasModel.GetByHash(ctx, job.Args.MediaHash) media, err := model.MediasModel().GetByHash(ctx, job.Args.MediaHash)
if err != nil { if err != nil {
log.Errorf("Error getting media by ID: %v", err) log.Errorf("Error getting media by ID: %v", err)
return JobCancel(err) return JobCancel(err)
@@ -112,7 +112,7 @@ func (w *VideoStoreShortWorker) Work(ctx context.Context, job *Job[VideoStoreSho
} }
log.Infof("pending create media record %s", job.Args.FilePath) log.Infof("pending create media record %s", job.Args.FilePath)
if err := model.MediasModel.Create(ctx, mediaModel); err != nil { if err := model.MediasModel().Create(ctx, mediaModel); err != nil {
log.Errorf("Error saving media record: %v data: %+v", err, mediaModel) log.Errorf("Error saving media record: %v data: %+v", err, mediaModel)
return err return err
} }

6
backend/app/jobs/wechat_pay_notify.go
View File

@@ -53,7 +53,7 @@ func (w *WechatPayNotifyWorker) Work(ctx context.Context, job *Job[WechatPayNoti
return JobCancel(errors.New("TradeState is not SUCCESS")) return JobCancel(errors.New("TradeState is not SUCCESS"))
} }
order, err := model.OrdersModel.GetByOrderNo(context.Background(), notify.OutTradeNo) order, err := model.OrdersModel().GetByOrderNo(context.Background(), notify.OutTradeNo)
if err != nil { if err != nil {
log.Errorf("GetByOrderNo error:%v", err) log.Errorf("GetByOrderNo error:%v", err)
return err return err
@@ -64,7 +64,7 @@ func (w *WechatPayNotifyWorker) Work(ctx context.Context, job *Job[WechatPayNoti
return JobCancel(fmt.Errorf("Order already paid, currently status: %d", order.Status)) return JobCancel(fmt.Errorf("Order already paid, currently status: %d", order.Status))
} }
user, err := model.UsersModel.GetByID(context.Background(), order.UserID) user, err := model.UsersModel().GetByID(context.Background(), order.UserID)
if err != nil { if err != nil {
return errors.Wrapf(err, "get user by id(%d) failed", order.UserID) return errors.Wrapf(err, "get user by id(%d) failed", order.UserID)
} }
@@ -103,7 +103,7 @@ func (w *WechatPayNotifyWorker) Work(ctx context.Context, job *Job[WechatPayNoti
return errors.Wrap(err, "BuyPosts error") return errors.Wrap(err, "BuyPosts error")
} }
if err := model.OrdersModel.Update(context.Background(), order); err != nil { if err := order.Update(context.Background()); err != nil {
log.Errorf("Update order error:%v", err) log.Errorf("Update order error:%v", err)
return errors.Wrap(err, "Update order error") return errors.Wrap(err, "Update order error")
} }

6
backend/app/jobs/wechat_refund_notify.go
View File

@@ -47,13 +47,13 @@ func (w *WechatRefundNotifyWorker) Work(ctx context.Context, job *Job[WechatRefu
notify := job.Args.Notify notify := job.Args.Notify
order, err := model.OrdersModel.GetByOrderNo(context.Background(), notify.OutTradeNo) order, err := model.OrdersModel().GetByOrderNo(context.Background(), notify.OutTradeNo)
if err != nil { if err != nil {
log.Errorf("GetByOrderNo error:%v", err) log.Errorf("GetByOrderNo error:%v", err)
return err return err
} }
user, err := model.UsersModel.GetByID(context.Background(), order.UserID) user, err := model.UsersModel().GetByID(context.Background(), order.UserID)
if err != nil { if err != nil {
log.Errorf("GetByID error:%v", err) log.Errorf("GetByID error:%v", err)
return err return err
@@ -88,7 +88,7 @@ func (w *WechatRefundNotifyWorker) Work(ctx context.Context, job *Job[WechatRefu
} }
} }
if err := model.OrdersModel.Update(context.Background(), order); err != nil { if err := order.Update(context.Background()); err != nil {
log.Errorf("Update order error:%v", err) log.Errorf("Update order error:%v", err)
return errors.Wrap(err, "Update order error") return errors.Wrap(err, "Update order error")
} }

4
backend/app/middlewares/mid_auth.go
View File

@@ -26,7 +26,7 @@ func (f *Middlewares) Auth(ctx fiber.Ctx) error {
} }
if f.app.IsDevMode() && false { if f.app.IsDevMode() && false {
user, err := model.UsersModel.GetByID(ctx.Context(), 1) user, err := model.UsersModel().GetByID(ctx.Context(), 1)
if err != nil { if err != nil {
return ctx.Send([]byte("User not found")) return ctx.Send([]byte("User not found"))
} }
@@ -66,7 +66,7 @@ func (f *Middlewares) Auth(ctx fiber.Ctx) error {
return ctx.Redirect().To(fullUrl) return ctx.Redirect().To(fullUrl)
} }
user, err := model.UsersModel.GetByID(ctx.Context(), jwt.UserID) user, err := model.UsersModel().GetByID(ctx.Context(), jwt.UserID)
if err != nil { if err != nil {
// remove cookie // remove cookie
ctx.ClearCookie("token") ctx.ClearCookie("token")

22
backend/app/model/medias_test.go
View File

@@ -52,7 +52,7 @@ func (s *MediasTestSuite) Test_countByCondition() {
Convey("no cond", func() { Convey("no cond", func() {
database.Truncate(context.Background(), db, table.Medias.TableName()) database.Truncate(context.Background(), db, table.Medias.TableName())
cnt, err := MediasModel.countByCondition(context.Background(), nil) cnt, err := MediasModel().countByCondition(context.Background(), nil)
Convey("should not return an error", func() { Convey("should not return an error", func() {
So(err, ShouldBeNil) So(err, ShouldBeNil)
}) })
@@ -110,7 +110,7 @@ func (s *MediasTestSuite) Test_BatchCreate() {
count := 10 count := 10
for i := 0; i < count; i++ { for i := 0; i < count; i++ {
err := MediasModel.BatchCreate(context.Background(), models) err := MediasModel().BatchCreate(context.Background(), models)
Convey("Create should not return an error: "+fmt.Sprintf("%d", i), func() { Convey("Create should not return an error: "+fmt.Sprintf("%d", i), func() {
So(err, ShouldBeNil) So(err, ShouldBeNil)
}) })
@@ -132,12 +132,12 @@ func (s *MediasTestSuite) Test_Create() {
Path: "path/to/media.pdf", Path: "path/to/media.pdf",
} }
err := MediasModel.Create(context.Background(), model) err := MediasModel().Create(context.Background(), model)
Convey("Create should not return an error", func() { Convey("Create should not return an error", func() {
So(err, ShouldBeNil) So(err, ShouldBeNil)
}) })
cnt, err := MediasModel.countByCondition(context.Background(), nil) cnt, err := MediasModel().countByCondition(context.Background(), nil)
Convey("Count should not return an error", func() { Convey("Count should not return an error", func() {
So(err, ShouldBeNil) So(err, ShouldBeNil)
}) })
@@ -165,31 +165,31 @@ func (s *MediasTestSuite) Test_Page() {
Path: "path/to/media.pdf", Path: "path/to/media.pdf",
} }
err := MediasModel.Create(context.Background(), model) err := MediasModel().Create(context.Background(), model)
So(err, ShouldBeNil) So(err, ShouldBeNil)
} }
cnt, err := MediasModel.countByCondition(context.Background(), nil) cnt, err := MediasModel().countByCondition(context.Background(), nil)
So(err, ShouldBeNil) So(err, ShouldBeNil)
So(cnt, ShouldEqual, 20) So(cnt, ShouldEqual, 20)
}) })
Convey("Page", func() { Convey("Page", func() {
Convey("page 1", func() { Convey("page 1", func() {
pager, err := MediasModel.List(context.Background(), &requests.Pagination{Page: 1, Limit: 10}, nil) pager, err := MediasModel().List(context.Background(), &requests.Pagination{Page: 1, Limit: 10}, nil)
So(err, ShouldBeNil) So(err, ShouldBeNil)
So(pager.Total, ShouldEqual, 20) So(pager.Total, ShouldEqual, 20)
So(pager.Items, ShouldHaveLength, 10) So(pager.Items, ShouldHaveLength, 10)
}) })
Convey("page 2", func() { Convey("page 2", func() {
pager, err := MediasModel.List(context.Background(), &requests.Pagination{Page: 2, Limit: 10}, nil) pager, err := MediasModel().List(context.Background(), &requests.Pagination{Page: 2, Limit: 10}, nil)
So(err, ShouldBeNil) So(err, ShouldBeNil)
So(pager.Total, ShouldEqual, 20) So(pager.Total, ShouldEqual, 20)
So(pager.Items, ShouldHaveLength, 10) So(pager.Items, ShouldHaveLength, 10)
}) })
Convey("page 3", func() { Convey("page 3", func() {
pager, err := MediasModel.List(context.Background(), &requests.Pagination{Page: 3, Limit: 10}, nil) pager, err := MediasModel().List(context.Background(), &requests.Pagination{Page: 3, Limit: 10}, nil)
So(err, ShouldBeNil) So(err, ShouldBeNil)
So(pager.Total, ShouldEqual, 20) So(pager.Total, ShouldEqual, 20)
So(pager.Items, ShouldBeEmpty) So(pager.Items, ShouldBeEmpty)
@@ -208,7 +208,7 @@ func (s *MediasTestSuite) Test_CreateGetID() {
Path: "path/to/media.pdf", Path: "path/to/media.pdf",
} }
err := MediasModel.Create(context.Background(), model) err := MediasModel().Create(context.Background(), model)
So(err, ShouldBeNil) So(err, ShouldBeNil)
So(model.ID, ShouldNotBeEmpty) So(model.ID, ShouldNotBeEmpty)
@@ -219,7 +219,7 @@ func (s *MediasTestSuite) Test_CreateGetID() {
func (s *MediasTestSuite) Test_GetRelations() { func (s *MediasTestSuite) Test_GetRelations() {
Convey("GetByHash", s.T(), func() { Convey("GetByHash", s.T(), func() {
hash := "ce4cd071128cef282cf315dda75bdab4" hash := "ce4cd071128cef282cf315dda75bdab4"
media, err := MediasModel.GetRelations(context.Background(), hash) media, err := MediasModel().GetRelations(context.Background(), hash)
So(err, ShouldBeNil) So(err, ShouldBeNil)
So(media, ShouldNotBeNil) So(media, ShouldNotBeNil)

54
backend/app/model/orders.go
View File

@@ -99,7 +99,7 @@ func (m *Orders) List(ctx context.Context, pagination *requests.Pagination, cond
return nil, err return nil, err
} }
postsMap, err := PostsModel.GetPostsMapByIDs(ctx, lo.Map(orders, func(order Orders, _ int) int64 { postsMap, err := PostsModel().GetPostsMapByIDs(ctx, lo.Map(orders, func(order Orders, _ int) int64 {
return order.PostID return order.PostID
})) }))
if err != nil { if err != nil {
@@ -107,7 +107,7 @@ func (m *Orders) List(ctx context.Context, pagination *requests.Pagination, cond
return nil, err return nil, err
} }
userMap, err := UsersModel.GetUsersMapByIDs(ctx, lo.Map(orders, func(order Orders, _ int) int64 { userMap, err := UsersModel().GetUsersMapByIDs(ctx, lo.Map(orders, func(order Orders, _ int) int64 {
return order.UserID return order.UserID
})) }))
if err != nil { if err != nil {
@@ -151,7 +151,7 @@ func (m *Orders) List(ctx context.Context, pagination *requests.Pagination, cond
// Create creates a new order // Create creates a new order
func (o *Orders) Create(ctx context.Context, userId, postId int64) (*Orders, error) { func (o *Orders) Create(ctx context.Context, userId, postId int64) (*Orders, error) {
post, err := PostsModel.GetByID(ctx, postId) post, err := PostsModel().GetByID(ctx, postId)
if err != nil { if err != nil {
return nil, errors.Wrap(err, "failed to get post") return nil, errors.Wrap(err, "failed to get post")
} }
@@ -172,48 +172,43 @@ func (o *Orders) Create(ctx context.Context, userId, postId int64) (*Orders, err
stmt := tbl.INSERT(tbl.MutableColumns).MODEL(m).RETURNING(tbl.AllColumns) stmt := tbl.INSERT(tbl.MutableColumns).MODEL(m).RETURNING(tbl.AllColumns)
o.log().Infof("sql: %s", stmt.DebugSql()) o.log().Infof("sql: %s", stmt.DebugSql())
var order Orders if err := stmt.QueryContext(ctx, db, o); err != nil {
if err := stmt.QueryContext(ctx, db, &order); err != nil {
o.log().Errorf("error creating order: %v", err) o.log().Errorf("error creating order: %v", err)
return nil, err return nil, err
} }
return &order, nil return o, nil
} }
func (m *Orders) SetMeta(ctx context.Context, id int64, metaFunc func(fields.OrderMeta) fields.OrderMeta) error { func (m *Orders) SetMeta(ctx context.Context, metaFunc func(fields.OrderMeta) fields.OrderMeta) error {
order, err := m.GetByID(ctx, id)
if err != nil {
return errors.Wrap(err, "failed to get order")
}
tbl := table.Orders tbl := table.Orders
stmt := tbl. stmt := tbl.
UPDATE(tbl.Meta). UPDATE(tbl.Meta).
SET(fields.ToJson(metaFunc(order.Meta.Data))). SET(fields.ToJson(metaFunc(m.Meta.Data))).
WHERE( WHERE(
tbl.ID.EQ(Int64(id)), tbl.ID.EQ(Int64(m.ID)),
) ).
RETURNING(tbl.AllColumns)
m.log().Infof("sql: %s", stmt.DebugSql()) m.log().Infof("sql: %s", stmt.DebugSql())
if _, err := stmt.ExecContext(ctx, db); err != nil { if err := stmt.QueryContext(ctx, db, m); err != nil {
m.log().Errorf("error set order meta: %v", err) m.log().Errorf("error set order meta: %v", err)
return err return err
} }
return nil return nil
} }
// DeleteByID soft deletes an order by ID func (m *Orders) SetStatus(ctx context.Context, status fields.OrderStatus) error {
func (m *Orders) SetStatus(ctx context.Context, orderNo string, status fields.OrderStatus) error {
tbl := table.Orders tbl := table.Orders
stmt := tbl. stmt := tbl.
UPDATE(tbl.Status). UPDATE(tbl.Status).
SET(status). SET(status).
WHERE( WHERE(
tbl.OrderNo.EQ(String(orderNo)), tbl.ID.EQ(Int64(m.ID)),
) ).
RETURNING(tbl.AllColumns)
m.log().Infof("sql: %s", stmt.DebugSql()) m.log().Infof("sql: %s", stmt.DebugSql())
if _, err := stmt.ExecContext(ctx, db); err != nil { if err := stmt.QueryContext(ctx, db, m); err != nil {
m.log().Errorf("error set order status: %v", err) m.log().Errorf("error set order status: %v", err)
return err return err
} }
@@ -280,14 +275,14 @@ func (m *Orders) GetByOrderNo(ctx context.Context, orderNo string) (*Orders, err
return &order, nil return &order, nil
} }
// SetTranscationID // SetTransactionID
func (m *Orders) SetTranscationID(ctx context.Context, id int64, transactionID string) error { func (m *Orders) SetTransactionID(ctx context.Context, transactionID string) error {
tbl := table.Orders tbl := table.Orders
stmt := tbl. stmt := tbl.
UPDATE(tbl.TransactionID). UPDATE(tbl.TransactionID).
SET(String(transactionID)). SET(String(transactionID)).
WHERE( WHERE(
tbl.ID.EQ(Int64(id)), tbl.ID.EQ(Int64(m.ID)),
) )
m.log().Infof("sql: %s", stmt.DebugSql()) m.log().Infof("sql: %s", stmt.DebugSql())
@@ -299,7 +294,7 @@ func (m *Orders) SetTranscationID(ctx context.Context, id int64, transactionID s
} }
// Update // Update
func (m *Orders) Update(ctx context.Context, order *Orders) error { func (m *Orders) Update(ctx context.Context) error {
tbl := table.Orders tbl := table.Orders
stmt := tbl. stmt := tbl.
UPDATE( UPDATE(
@@ -312,13 +307,14 @@ func (m *Orders) Update(ctx context.Context, order *Orders) error {
tbl.UserID, tbl.UserID,
), ),
). ).
MODEL(order). MODEL(m).
WHERE( WHERE(
tbl.ID.EQ(Int64(order.ID)), tbl.ID.EQ(Int64(m.ID)),
) ).
RETURNING(tbl.AllColumns)
m.log().Infof("sql: %s", stmt.DebugSql()) m.log().Infof("sql: %s", stmt.DebugSql())
if _, err := stmt.ExecContext(ctx, db); err != nil { if err := stmt.QueryContext(ctx, db, m); err != nil {
m.log().Errorf("error updating order: %v", err) m.log().Errorf("error updating order: %v", err)
return err return err
} }

59
backend/app/model/provider.gen.go
View File

@@ -14,60 +14,19 @@ import (
) )
var db *sql.DB var db *sql.DB
var MediasModel *Medias
var OrdersModel *Orders
var PostsModel *Posts
var UsersModel *Users
func Transaction(ctx context.Context) (*sql.Tx, error) { func MediasModel() *Medias { return &Medias{} }
return db.Begin() func OrdersModel() *Orders { return &Orders{} }
} func PostsModel() *Posts { return &Posts{} }
func DB() *sql.DB { func UsersModel() *Users { return &Users{} }
return db
} func Transaction(ctx context.Context) (*sql.Tx, error) { return db.Begin() }
func DB() *sql.DB { return db }
func Provide(opts ...opt.Option) error { func Provide(opts ...opt.Option) error {
if err := container.Container.Provide(func() (*Medias, error) { if err := container.Container.Provide(func(_db *sql.DB) (contracts.Initial, error) {
obj := &Medias{}
return obj, nil
}); err != nil {
return err
}
if err := container.Container.Provide(func() (*Orders, error) {
obj := &Orders{}
return obj, nil
}); err != nil {
return err
}
if err := container.Container.Provide(func() (*Posts, error) {
obj := &Posts{}
return obj, nil
}); err != nil {
return err
}
if err := container.Container.Provide(func() (*Users, error) {
obj := &Users{}
return obj, nil
}); err != nil {
return err
}
if err := container.Container.Provide(func(
_db *sql.DB,
medias *Medias,
orders *Orders,
posts *Posts,
users *Users,
) (contracts.Initial, error) {
db = _db db = _db
MediasModel = medias
OrdersModel = orders
PostsModel = posts
UsersModel = users
return nil, nil return nil, nil
}, atom.GroupInitial); err != nil { }, atom.GroupInitial); err != nil {
return err return err