tele-recv/utils/pulsar.go

package utils

import (
	"context"

	"github.com/apache/pulsar-client-go/pulsar"
	"go.uber.org/zap"
)

var (
	pulsarClient pulsar.Client
	producer     pulsar.Producer
	pulsarUrl    string
	pulsarTopic  string
	clientName   string
)

func CreateProducer(url string, topic string, name string) {
	pulsarUrl = url
	pulsarTopic = topic
	clientName = name
	create()
}

func create() {
	var err error
	pulsarClient, err = pulsar.NewClient(pulsar.ClientOptions{
		URL: pulsarUrl,
	})
	if err != nil {
		Log.Error("error in create pulsar client ", zap.Error(err))
	}
	producer, err = pulsarClient.CreateProducer(pulsar.ProducerOptions{
		Topic: pulsarTopic,
		Name:  clientName,
	})
	if err != nil {
		Log.Error("error in create pulsar producer ", zap.Error(err))
	}
}

func resetPulsarProducer() {
	if producer != nil {
		producer.Close()
	}
	if pulsarClient != nil {
		pulsarClient.Close()
	}
	create()
}

func PulsarSend(data string) error {
	var err error
	var id pulsar.MessageID
	for i := 0; i < 5; i++ {
		id, err = producer.Send(context.Background(), &pulsar.ProducerMessage{
			Payload: []byte(data),
		})
		if err == nil {
			Log.Info("send ", zap.Any("id", id))
			break
		}
		resetPulsarProducer()
	}
	return err
}

func ClosePulsar() {
	if producer != nil {
		producer.Close()
	}
	if pulsarClient != nil {
		pulsarClient.Close()
	}
}