规划消费者

delay/redis_dispatch_consumer.go

@@ -0,0 +1,96 @@
+// Package delay ...
+//
+// Description : delay ...
+//
+// Author : go_developer@163.com<白茶清欢>
+//
+// Date : 2022-07-07 10:32
+package delay
+
+import (
+	"context"
+	"encoding/json"
+	"errors"
+	"time"
+
+	"github.com/go-redis/redis/v8"
+)
+
+// NewRedisConsumer redis延迟队列消费者
+//
+// Author : go_developer@163.com<白茶清欢>
+//
+// Date : 14:10 2022/7/8
+func NewRedisConsumer(redisInstance *redis.Client, cfg *ConsumerConfig) IConsumer {
+	return &redisConsumer{
+		redisInstance: redisInstance,
+		cfg:           cfg,
+	}
+}
+
+type redisConsumer struct {
+	redisInstance *redis.Client
+	cfg           *ConsumerConfig
+}
+
+func (r *redisConsumer) Consume(ctx context.Context) ([]*ZRangeData, error) {
+	if nil == ctx {
+		ctx = context.Background()
+	}
+	zRangeResult := r.redisInstance.ZRange(ctx, r.cfg.QueueName, 0, time.Now().UnixNano()/1e6)
+	if err := zRangeResult.Err(); nil != err {
+		return make([]*ZRangeData, 0), err
+	}
+	// 格式化数据
+	var (
+		result []*ZRangeData
+	)
+	if err := json.Unmarshal([]byte(zRangeResult.String()), &result); nil != err {
+		return make([]*ZRangeData, 0), err
+	}
+	return result, nil
+}
+
+func (r *redisConsumer) ConsumeWithHandler(ctx context.Context, handler IHandler) error {
+	if nil == handler {
+		return errors.New("handler instance is nil")
+	}
+	var (
+		msgList []*ZRangeData
+		err     error
+	)
+
+	if msgList, err = r.Consume(ctx); nil != err {
+		return err
+	}
+
+	// 未订阅到消息
+	if len(msgList) == 0 {
+		return nil
+	}
+	return handler.Handle(msgList)
+	/* wg := &sync.WaitGroup{}
+	wg.Add(len(msgList))
+	for _, item := range msgList {
+
+		go func(msgData *ZRangeData) {
+			defer wg.Done()
+			hashValue, exist := msgData.Data.Data[r.cfg.HashKey]
+			if !exist || hashValue == nil {
+				hashValue = msgData.Data.MsgID
+			}
+			shard := util.Hash.GetHashIDMod(hashValue, r.cfg.SonQueueCnt)
+			realQueue := fmt.Sprintf(r.cfg.QueueName+"_%d", shard)
+			r.redisInstance.LPush()
+		}(item)
+	}
+	wg.Wait()
+	return nil*/
+}
+
+type redisConsumerMsgHandler struct {
+}
+
+func (r redisConsumerMsgHandler) Handle(queData []*ZRangeData) error {
+	panic("implement me")
+}