huang
03a0960c4d
All checks were successful
构建并部署到测试环境（无 SSH） / build-and-deploy (push) Successful in 7m2s
- 移除 RegisterDataPermissionCallback 和 SkipDataPermission 机制 - 在 Auth 中间件预计算 SubordinateShopIDs 并注入 Context - 新增 ApplyShopFilter/ApplyEnterpriseFilter/ApplyOwnerShopFilter 等 Helper 函数 - 所有 Store 层查询方法显式调用数据权限过滤函数 - 权限检查函数 CanManageShop/CanManageEnterprise 改为从 Context 获取数据 Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
93 lines
2.5 KiB
Go
93 lines
2.5 KiB
Go
package task
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/bytedance/sonic"
|
|
"github.com/hibiken/asynq"
|
|
"github.com/redis/go-redis/v9"
|
|
"go.uber.org/zap"
|
|
|
|
"github.com/break/junhong_cmp_fiber/internal/store/postgres"
|
|
"github.com/break/junhong_cmp_fiber/pkg/constants"
|
|
)
|
|
|
|
type CommissionStatsUpdatePayload struct {
|
|
AllocationID uint `json:"allocation_id"`
|
|
SalesCount int64 `json:"sales_count"`
|
|
SalesAmount int64 `json:"sales_amount"`
|
|
}
|
|
|
|
type CommissionStatsUpdateHandler struct {
|
|
redis *redis.Client
|
|
statsStore *postgres.ShopSeriesCommissionStatsStore
|
|
allocationStore *postgres.ShopPackageAllocationStore
|
|
logger *zap.Logger
|
|
}
|
|
|
|
func NewCommissionStatsUpdateHandler(
|
|
redis *redis.Client,
|
|
statsStore *postgres.ShopSeriesCommissionStatsStore,
|
|
allocationStore *postgres.ShopPackageAllocationStore,
|
|
logger *zap.Logger,
|
|
) *CommissionStatsUpdateHandler {
|
|
return &CommissionStatsUpdateHandler{
|
|
redis: redis,
|
|
statsStore: statsStore,
|
|
allocationStore: allocationStore,
|
|
logger: logger,
|
|
}
|
|
}
|
|
|
|
func (h *CommissionStatsUpdateHandler) HandleCommissionStatsUpdate(ctx context.Context, task *asynq.Task) error {
|
|
var payload CommissionStatsUpdatePayload
|
|
if err := sonic.Unmarshal(task.Payload(), &payload); err != nil {
|
|
h.logger.Error("解析统计更新任务载荷失败",
|
|
zap.Error(err),
|
|
zap.String("task_id", task.ResultWriter().TaskID()),
|
|
)
|
|
return asynq.SkipRetry
|
|
}
|
|
|
|
now := time.Now()
|
|
period := getCurrentPeriod(now)
|
|
redisKey := constants.RedisCommissionStatsKey(payload.AllocationID, period)
|
|
|
|
pipe := h.redis.Pipeline()
|
|
pipe.HIncrBy(ctx, redisKey, "total_count", payload.SalesCount)
|
|
pipe.HIncrBy(ctx, redisKey, "total_amount", payload.SalesAmount)
|
|
|
|
periodEnd := getPeriodEnd(now)
|
|
expireAt := periodEnd.AddDate(0, 0, 7)
|
|
pipe.ExpireAt(ctx, redisKey, expireAt)
|
|
|
|
if _, err := pipe.Exec(ctx); err != nil {
|
|
h.logger.Error("更新 Redis 统计失败",
|
|
zap.Uint("allocation_id", payload.AllocationID),
|
|
zap.String("period", period),
|
|
zap.Error(err),
|
|
)
|
|
return err
|
|
}
|
|
|
|
h.logger.Info("统计更新成功",
|
|
zap.Uint("allocation_id", payload.AllocationID),
|
|
zap.String("period", period),
|
|
zap.Int64("sales_count", payload.SalesCount),
|
|
zap.Int64("sales_amount", payload.SalesAmount),
|
|
)
|
|
|
|
return nil
|
|
}
|
|
|
|
func getCurrentPeriod(t time.Time) string {
|
|
return t.Format("2006-01")
|
|
}
|
|
|
|
func getPeriodEnd(t time.Time) time.Time {
|
|
year, month, _ := t.Date()
|
|
nextMonth := time.Date(year, month+1, 1, 0, 0, 0, 0, t.Location())
|
|
return nextMonth.Add(-time.Second)
|
|
}
|