70 lines
1.8 KiB
Go
70 lines
1.8 KiB
Go
package stan
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/micro/go-micro/broker"
|
|
"github.com/micro/go-micro/server"
|
|
stan "github.com/nats-io/go-nats-streaming"
|
|
)
|
|
|
|
type optionsKey struct{}
|
|
|
|
// Options accepts stan.Options
|
|
func Options(opts stan.Options) broker.Option {
|
|
return setBrokerOption(optionsKey{}, opts)
|
|
}
|
|
|
|
type clusterIDKey struct{}
|
|
|
|
// ClusterID specify cluster id to connect
|
|
func ClusterID(clusterID string) broker.Option {
|
|
return setBrokerOption(clusterIDKey{}, clusterID)
|
|
}
|
|
|
|
type clientIDKey struct{}
|
|
|
|
// ClientID specify client id to connect
|
|
func ClientID(clientID string) broker.Option {
|
|
return setBrokerOption(clientIDKey{}, clientID)
|
|
}
|
|
|
|
type subscribeOptionKey struct{}
|
|
|
|
func SubscribeOption(opts ...stan.SubscriptionOption) broker.SubscribeOption {
|
|
return setSubscribeOption(subscribeOptionKey{}, opts)
|
|
}
|
|
|
|
func ServerSubscriberOption(opts ...stan.SubscriptionOption) server.SubscriberOption {
|
|
return setServerSubscriberOption(subscribeOptionKey{}, opts)
|
|
}
|
|
|
|
type subscribeContextKey struct{}
|
|
|
|
// SubscribeContext set the context for broker.SubscribeOption
|
|
func SubscribeContext(ctx context.Context) broker.SubscribeOption {
|
|
return setSubscribeOption(subscribeContextKey{}, ctx)
|
|
}
|
|
|
|
type ackSuccessKey struct{}
|
|
|
|
// AckOnSuccess will automatically acknowledge messages when no error is returned
|
|
func AckOnSuccess() broker.SubscribeOption {
|
|
return setSubscribeOption(ackSuccessKey{}, true)
|
|
}
|
|
|
|
type connectTimeoutKey struct{}
|
|
|
|
// ConnectTimeout timeout for connecting to broker -1 infinitive or time.Duration value
|
|
func ConnectTimeout(td time.Duration) broker.Option {
|
|
return setBrokerOption(connectTimeoutKey{}, td)
|
|
}
|
|
|
|
type connectRetryKey struct{}
|
|
|
|
// ConnectRetry reconnect to broker in case of errors
|
|
func ConnectRetry(v bool) broker.Option {
|
|
return setBrokerOption(connectRetryKey{}, v)
|
|
}
|