实现七月迭代公共技术基础

pkg/queue/client.go

@@ -4,6 +4,7 @@ import (
 	"context"
 	"fmt"
 
+	"github.com/break/junhong_cmp_fiber/pkg/asynctask"
 	"github.com/break/junhong_cmp_fiber/pkg/config"
 	"github.com/break/junhong_cmp_fiber/pkg/constants"
 	"github.com/bytedance/sonic"
@@ -39,10 +40,10 @@ func NewClient(redisClient *redis.Client, logger *zap.Logger) *Client {
 // payload 必须传入 struct 或 map，禁止传入 []byte。
 // 传入 []byte 会导致 sonic.Marshal 将其 base64 编码，handler 解析时类型不匹配。
 func (c *Client) EnqueueTask(ctx context.Context, taskType string, payload interface{}, opts ...asynq.Option) error {
-	if _, ok := payload.([]byte); ok {
+	if err := asynctask.ValidatePayload(payload); err != nil {
 		c.logger.Error("任务载荷类型错误，禁止传入 []byte",
-			zap.String("task_type", taskType))
-		return fmt.Errorf("task payload must be struct or map, got []byte")
+			zap.String("task_type", taskType), zap.Error(err))
+		return err
 	}
 
 	// 内部统一序列化，调用方无需预先 Marshal