workerstd

type App struct {
	Broker          *Broker
	Logger          *zap.SugaredLogger
	TaskHandler     TaskHandler
	ShutdownTimeout time.Duration

	// ReceiveTaskFn is called to receive a task from the Sub client. Ideally this is not necessary, but because
	// proto.Message is a pointer type, we can't instantiate the struct without knowing what protobuf message we want to
	// unmarshal to.
	ReceiveTaskFn func(ctx context.Context, subClt *SubClient) (proto.Message, *pubsub.Message, error)

	// CloseFn is called on close. contain Any additional close routine should be handled in the custom close function passed in here.
	CloseFn func() error
}

type Broker struct {
	// Engine is the engine of the message broker. Must be one of azuresb or rabbitmq.
	Engine string `mapstructure:"engine"`

	// TopicName is the message queue topic where messages are published. This corresponds to the exchange when using
	// rabbitmq.
	TopicName string `mapstructure:"topic"`

	// ConnectionString is the connection string for connecting to the specific broker. For AzureSB, this is the
	// ServiceBus Namespace FQDN (NAMESPACE.servicebus.windows.net), while for RabbitMQ this is the server URL in the
	// format USERNAME:PASSWORD@HOST:PORT.
	ConnectionString string `mapstructure:"connstring"`

	// ServiceBusSubscriptionName is the Azure ServiceBus Topic Subscription that the worker should consume as. If
	// blank, assume that the topic is an Azure ServiceBus Queue instead of a Topic. This is only used with Azure
	// ServiceBus.
	ServiceBusSubscriptionName string `mapstructure:"subscription"`
}

type PubClient struct {
	// contains filtered or unexported fields
}

type SubClient struct {
	// contains filtered or unexported fields
}

type TaskHandler interface {
	HandleTaskMsg(proto.Message, *pubsub.Message) error
}