publisher sample

pstest_publisher.go

package pubsub

import (
	"context"

	"cloud.google.com/go/pubsub"
	"google.golang.org/api/option"
)

type Publisher struct {
	client *pubsub.Client
}

func NewPublisher(ctx context.Context, projectID string, opts ...option.ClientOption) (*Publisher, error) {
	client, err := pubsub.NewClient(ctx, projectID, opts...)
	if err != nil {
		return nil, err
	}
	return &Publisher{
		client: client,
	}, nil
}

func (p *Publisher) Publish(ctx context.Context, topicID string, msg string) error {
	topic := p.client.Topic(topicID)
	_, err := topic.Publish(ctx, &pubsub.Message{
		Data: []byte(msg),
	}).Get(ctx)

	return err
}