From bd9e4df23156e0727ecf1f896ebc66b250ef7a37 Mon Sep 17 00:00:00 2001 From: Rogee Date: Tue, 13 May 2025 14:31:44 +0800 Subject: [PATCH] feat: update provider --- backend/app/http/provider.gen.go | 2 ++ backend/app/jobs/provider.gen.go | 12 ++++++++++++ backend/app/models/users.go | 4 ++-- 3 files changed, 16 insertions(+), 2 deletions(-) diff --git a/backend/app/http/provider.gen.go b/backend/app/http/provider.gen.go index 6d3b0bb..ca4ec69 100755 --- a/backend/app/http/provider.gen.go +++ b/backend/app/http/provider.gen.go @@ -41,10 +41,12 @@ func Provide(opts ...opt.Option) error { return err } if err := container.Container.Provide(func( + job *job.Job, oss *ali.OSSClient, wepay *wepay.Client, ) (*posts, error) { obj := &posts{ + job: job, oss: oss, wepay: wepay, } diff --git a/backend/app/jobs/provider.gen.go b/backend/app/jobs/provider.gen.go index cde7267..a95aa01 100755 --- a/backend/app/jobs/provider.gen.go +++ b/backend/app/jobs/provider.gen.go @@ -13,6 +13,18 @@ import ( ) func Provide(opts ...opt.Option) error { + if err := container.Container.Provide(func( + __job *job.Job, + ) (contracts.Initial, error) { + obj := &BalancePayNotifyWorker{} + if err := river.AddWorkerSafely(__job.Workers, obj); err != nil { + return nil, err + } + + return obj, nil + }, atom.GroupInitial); err != nil { + return err + } if err := container.Container.Provide(func( __job *job.Job, ) (contracts.Initial, error) { diff --git a/backend/app/models/users.go b/backend/app/models/users.go index 3b98d0e..7cf883f 100644 --- a/backend/app/models/users.go +++ b/backend/app/models/users.go @@ -476,7 +476,7 @@ func (m *usersModel) RevokePosts(ctx context.Context, userID, postID int64) erro } // SetBalance -func (m *usersModel) SetBalance(ctx context.Context, id int64, balance int64) error { +func (m *usersModel) SetBalance(ctx context.Context, id, balance int64) error { tbl := table.Users stmt := tbl. UPDATE(tbl.Balance). @@ -494,7 +494,7 @@ func (m *usersModel) SetBalance(ctx context.Context, id int64, balance int64) er } // AddBalance adds the given amount to the user's balance -func (m *usersModel) AddBalance(ctx context.Context, id int64, amount int64) error { +func (m *usersModel) AddBalance(ctx context.Context, id, amount int64) error { tbl := table.Users stmt := tbl. UPDATE(tbl.Balance).