natsio/pubsub.go

package natsio

import (
	"errors"
	"fmt"

	"github.com/nats-io/nats.go"
)

// Publish ...
func Publish(stream, subject string, payload []byte) error {
	channel := combineStreamAndSubjectName(stream, subject)

	_, err := natsJS.PublishAsync(channel, payload)
	if err != nil {
		msg := fmt.Sprintf("publish message to subject %s error: %s", channel, err.Error())
		return errors.New(msg)
	}
	return nil
}

// PublishWithResponse ...
func PublishWithResponse(stream, subject string, payload []byte) (*nats.Msg, error) {
	channel := combineStreamAndSubjectName(stream, subject)

	resp, err := natsJS.PublishAsync(channel, payload)
	if err != nil {
		msg := fmt.Sprintf("publish message to subject %s error: %s", channel, err.Error())
		return nil, errors.New(msg)
	}
	return resp.Msg(), nil
}

// Subscribe ...
func Subscribe(stream, subject string, cb nats.MsgHandler) (*nats.Subscription, error) {
	channel := combineStreamAndSubjectName(stream, subject)

	sub, err := natsJS.Subscribe(channel, cb)
	if err != nil {
		msg := fmt.Sprintf("subscribe subject %s error: %s", channel, err.Error())
		return nil, errors.New(msg)
	}
	return sub, nil
}