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

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

pkg/queue/client_test.go

@@ -0,0 +1,17 @@
+package queue
+
+import (
+	"context"
+	"testing"
+
+	"go.uber.org/zap"
+)
+
+func TestEnqueueTaskRejectsPreSerializedBytesBeforeQueueAccess(t *testing.T) {
+	t.Parallel()
+
+	client := &Client{logger: zap.NewNop()}
+	if err := client.EnqueueTask(context.Background(), "test:bytes", []byte(`{"task_id":1}`)); err == nil {
+		t.Fatal("预序列化 []byte 载荷必须在访问 Asynq 前被拒绝")
+	}
+}