slow_wild_queue/mqClient/tweetTransfer.go

package mqClient

import (
	"context"
	"encoding/json"
	"github.com/zeromicro/go-queue/kq"
	"paopaoimtask/mq"
)

type TweetTransferClient interface {
	Push(msg *mq.MsgTweetData) error
}

type tweetTransferClient struct {
	pusher *kq.Pusher
}

func NewTweetTransferClient(addr []string, topic string, opts ...kq.Pusher) TweetTransferClient {
	return &tweetTransferClient{pusher: kq.NewPusher(addr, topic)}
}

func (m *tweetTransferClient) Push(msg *mq.MsgTweetData) error {
	data, err := json.Marshal(msg)
	if err != nil {
		return err
	}
	return m.pusher.Push(context.Background(), string(data))
}