mirror of https://github.com/linka-cloud/grpc.git
116 lines
2.0 KiB
Go
116 lines
2.0 KiB
Go
package registry
|
|
|
|
import (
|
|
"context"
|
|
"crypto/tls"
|
|
"time"
|
|
)
|
|
|
|
type Options struct {
|
|
Addrs []string
|
|
Timeout time.Duration
|
|
Secure bool
|
|
TLSConfig *tls.Config
|
|
// Other options for implementations of the interface
|
|
// can be stored in a context
|
|
Context context.Context
|
|
}
|
|
|
|
type RegisterOptions struct {
|
|
TTL time.Duration
|
|
// Other options for implementations of the interface
|
|
// can be stored in a context
|
|
Context context.Context
|
|
}
|
|
|
|
type WatchOptions struct {
|
|
// Specify a service to watch
|
|
// If blank, the watch is for all services
|
|
Service string
|
|
// Other options for implementations of the interface
|
|
// can be stored in a context
|
|
Context context.Context
|
|
}
|
|
|
|
type DeregisterOptions struct {
|
|
Context context.Context
|
|
}
|
|
|
|
type GetOptions struct {
|
|
Context context.Context
|
|
}
|
|
|
|
type ListOptions struct {
|
|
Context context.Context
|
|
}
|
|
|
|
// Addrs is the registry addresses to use
|
|
func Addrs(addrs ...string) Option {
|
|
return func(o *Options) {
|
|
o.Addrs = addrs
|
|
}
|
|
}
|
|
|
|
func Timeout(t time.Duration) Option {
|
|
return func(o *Options) {
|
|
o.Timeout = t
|
|
}
|
|
}
|
|
|
|
// Secure communication with the registry
|
|
func Secure(b bool) Option {
|
|
return func(o *Options) {
|
|
o.Secure = b
|
|
}
|
|
}
|
|
|
|
// Specify TLS Config
|
|
func TLSConfig(t *tls.Config) Option {
|
|
return func(o *Options) {
|
|
o.TLSConfig = t
|
|
}
|
|
}
|
|
|
|
func RegisterTTL(t time.Duration) RegisterOption {
|
|
return func(o *RegisterOptions) {
|
|
o.TTL = t
|
|
}
|
|
}
|
|
|
|
func RegisterContext(ctx context.Context) RegisterOption {
|
|
return func(o *RegisterOptions) {
|
|
o.Context = ctx
|
|
}
|
|
}
|
|
|
|
// Watch a service
|
|
func WatchService(name string) WatchOption {
|
|
return func(o *WatchOptions) {
|
|
o.Service = name
|
|
}
|
|
}
|
|
|
|
func WatchContext(ctx context.Context) WatchOption {
|
|
return func(o *WatchOptions) {
|
|
o.Context = ctx
|
|
}
|
|
}
|
|
|
|
func DeregisterContext(ctx context.Context) DeregisterOption {
|
|
return func(o *DeregisterOptions) {
|
|
o.Context = ctx
|
|
}
|
|
}
|
|
|
|
func GetContext(ctx context.Context) GetOption {
|
|
return func(o *GetOptions) {
|
|
o.Context = ctx
|
|
}
|
|
}
|
|
|
|
func ListContext(ctx context.Context) ListOption {
|
|
return func(o *ListOptions) {
|
|
o.Context = ctx
|
|
}
|
|
}
|