优化

backend/internal/modules/wallet/service.go

@@ -1,6 +1,9 @@
 package wallet
 
-import "errors"
+import (
+	"context"
+	"errors"
+)
 
 var (
 	ErrDependencyUnavailable = errors.New("dependency unavailable")
@@ -21,28 +24,28 @@ func NewService(repo *Repository) *Service {
 	return &Service{repo: repo}
 }
 
-func (s *Service) Account(userID uint64) (*AccountDTO, error) {
+func (s *Service) Account(ctx context.Context, userID uint64) (*AccountDTO, error) {
 	if s.repo == nil {
 		return nil, ErrDependencyUnavailable
 	}
-	return s.repo.Account(userID)
+	return s.repo.Account(ctx, userID)
 }
 
-func (s *Service) Ledger(userID uint64, page, pageSize int) (*PaginatedResult, error) {
+func (s *Service) Ledger(ctx context.Context, userID uint64, page, pageSize int) (*PaginatedResult, error) {
 	if s.repo == nil {
 		return nil, ErrDependencyUnavailable
 	}
-	return s.repo.Ledger(userID, page, pageSize)
+	return s.repo.Ledger(ctx, userID, page, pageSize)
 }
 
-func (s *Service) Recharge(userID uint64, req RechargeRequest) (*AccountDTO, error) {
+func (s *Service) Recharge(ctx context.Context, userID uint64, req RechargeRequest) (*AccountDTO, error) {
 	if s.repo == nil {
 		return nil, ErrDependencyUnavailable
 	}
 	return nil, ErrRechargeDisabled
 }
 
-func (s *Service) Withdraw(userID uint64, req WithdrawRequest) (*AccountDTO, error) {
+func (s *Service) Withdraw(ctx context.Context, userID uint64, req WithdrawRequest) (*AccountDTO, error) {
 	if s.repo == nil {
 		return nil, ErrDependencyUnavailable
 	}
@@ -51,9 +54,9 @@ func (s *Service) Withdraw(userID uint64, req WithdrawRequest) (*AccountDTO, err
 	return nil, ErrFeaturePending
 }
 
-func (s *Service) AdminLedger(query AdminLedgerQuery) (*PaginatedResult, error) {
+func (s *Service) AdminLedger(ctx context.Context, query AdminLedgerQuery) (*PaginatedResult, error) {
 	if s.repo == nil {
 		return nil, ErrDependencyUnavailable
 	}
-	return s.repo.AdminLedger(query)
+	return s.repo.AdminLedger(ctx, query)
 }