66 lines
2.6 KiB
Go
66 lines
2.6 KiB
Go
package grpc
|
|
|
|
import (
|
|
"context"
|
|
|
|
walletv1 "hyapp.local/api/proto/wallet/v1"
|
|
"hyapp/pkg/appcode"
|
|
"hyapp/pkg/xerr"
|
|
"hyapp/services/wallet-service/internal/domain/ledger"
|
|
walletservice "hyapp/services/wallet-service/internal/service/wallet"
|
|
)
|
|
|
|
// CronServer exposes wallet-service owned settlement batches to cron-service.
|
|
type CronServer struct {
|
|
walletv1.UnimplementedWalletCronServiceServer
|
|
|
|
svc *walletservice.Service
|
|
}
|
|
|
|
// NewCronServer 创建只给 cron-service 调用的批处理适配器；真实账务事务仍在 wallet repository 内完成。
|
|
func NewCronServer(svc *walletservice.Service) *CronServer {
|
|
return &CronServer{svc: svc}
|
|
}
|
|
|
|
func (s *CronServer) ProcessHostSalaryDailySettlementBatch(ctx context.Context, req *walletv1.CronBatchRequest) (*walletv1.CronBatchResponse, error) {
|
|
return s.processHostSalarySettlementBatch(ctx, req, ledger.HostSalarySettlementModeDaily)
|
|
}
|
|
|
|
func (s *CronServer) ProcessHostSalaryHalfMonthSettlementBatch(ctx context.Context, req *walletv1.CronBatchRequest) (*walletv1.CronBatchResponse, error) {
|
|
return s.processHostSalarySettlementBatch(ctx, req, ledger.HostSalarySettlementModeHalfMonth)
|
|
}
|
|
|
|
func (s *CronServer) ProcessHostSalaryMonthEndBatch(ctx context.Context, req *walletv1.CronBatchRequest) (*walletv1.CronBatchResponse, error) {
|
|
return s.processHostSalarySettlementBatch(ctx, req, ledger.HostSalarySettlementTypeMonthEnd)
|
|
}
|
|
|
|
func (s *CronServer) processHostSalarySettlementBatch(ctx context.Context, req *walletv1.CronBatchRequest, settlementType string) (*walletv1.CronBatchResponse, error) {
|
|
if s.svc == nil {
|
|
return nil, xerr.ToGRPCError(xerr.New(xerr.Unavailable, "wallet service is not configured"))
|
|
}
|
|
appCode := appcode.Normalize(req.GetAppCode())
|
|
ctx = appcode.WithContext(ctx, appCode)
|
|
// cron-service 只决定触发哪类批次；周期、当前时间、幂等交易和余额入账全部由 wallet-service 统一裁决。
|
|
result, err := s.svc.ProcessHostSalarySettlementBatch(ctx, ledger.HostSalarySettlementBatchCommand{
|
|
AppCode: appCode,
|
|
RunID: req.GetRunId(),
|
|
WorkerID: req.GetWorkerId(),
|
|
BatchSize: int(req.GetBatchSize()),
|
|
SettlementType: settlementType,
|
|
TriggerMode: req.GetTriggerMode(),
|
|
SettlementRole: req.GetSettlementRole(),
|
|
CycleKey: req.GetCycleKey(),
|
|
UserIDs: append([]int64(nil), req.GetUserIds()...),
|
|
})
|
|
if err != nil {
|
|
return nil, xerr.ToGRPCError(err)
|
|
}
|
|
return &walletv1.CronBatchResponse{
|
|
ClaimedCount: int32(result.ClaimedCount),
|
|
ProcessedCount: int32(result.ProcessedCount),
|
|
SuccessCount: int32(result.SuccessCount),
|
|
FailureCount: int32(result.FailureCount),
|
|
HasMore: result.HasMore,
|
|
}, nil
|
|
}
|