mirror of https://github.com/linka-cloud/grpc.git
137 lines
2.6 KiB
Go
137 lines
2.6 KiB
Go
package client
|
|
|
|
import (
|
|
"crypto/tls"
|
|
|
|
"google.golang.org/grpc"
|
|
|
|
"go.linka.cloud/grpc-toolkit/interceptors"
|
|
"go.linka.cloud/grpc-toolkit/registry"
|
|
)
|
|
|
|
type Options interface {
|
|
Name() string
|
|
Version() string
|
|
Address() string
|
|
Secure() bool
|
|
Registry() registry.Registry
|
|
TLSConfig() *tls.Config
|
|
DialOptions() []grpc.DialOption
|
|
UnaryInterceptors() []grpc.UnaryClientInterceptor
|
|
StreamInterceptors() []grpc.StreamClientInterceptor
|
|
}
|
|
|
|
type Option func(*options)
|
|
|
|
func WithRegistry(registry registry.Registry) Option {
|
|
return func(o *options) {
|
|
o.registry = registry
|
|
}
|
|
}
|
|
|
|
func WithName(name string) Option {
|
|
return func(o *options) {
|
|
o.name = name
|
|
}
|
|
}
|
|
|
|
func WithVersion(version string) Option {
|
|
return func(o *options) {
|
|
o.version = version
|
|
}
|
|
}
|
|
|
|
func WithAddress(address string) Option {
|
|
return func(o *options) {
|
|
o.addr = address
|
|
}
|
|
}
|
|
|
|
func WithTLSConfig(conf *tls.Config) Option {
|
|
return func(o *options) {
|
|
o.tlsConfig = conf
|
|
}
|
|
}
|
|
|
|
func WithSecure(s bool) Option {
|
|
return func(o *options) {
|
|
o.secure = s
|
|
}
|
|
}
|
|
|
|
func WithDialOptions(opts ...grpc.DialOption) Option {
|
|
return func(o *options) {
|
|
o.dialOptions = opts
|
|
}
|
|
}
|
|
|
|
func WithInterceptors(i ...interceptors.ClientInterceptors) Option {
|
|
return func(o *options) {
|
|
for _, v := range i {
|
|
o.unaryInterceptors = append(o.unaryInterceptors, v.UnaryClientInterceptor())
|
|
o.streamInterceptors = append(o.streamInterceptors, v.StreamClientInterceptor())
|
|
}
|
|
}
|
|
}
|
|
|
|
func WithUnaryInterceptors(i ...grpc.UnaryClientInterceptor) Option {
|
|
return func(o *options) {
|
|
o.unaryInterceptors = append(o.unaryInterceptors, i...)
|
|
}
|
|
}
|
|
|
|
func WithStreamInterceptors(i ...grpc.StreamClientInterceptor) Option {
|
|
return func(o *options) {
|
|
o.streamInterceptors = append(o.streamInterceptors, i...)
|
|
}
|
|
}
|
|
|
|
type options struct {
|
|
registry registry.Registry
|
|
name string
|
|
version string
|
|
addr string
|
|
tlsConfig *tls.Config
|
|
secure bool
|
|
dialOptions []grpc.DialOption
|
|
|
|
unaryInterceptors []grpc.UnaryClientInterceptor
|
|
streamInterceptors []grpc.StreamClientInterceptor
|
|
}
|
|
|
|
func (o *options) Name() string {
|
|
return o.name
|
|
}
|
|
|
|
func (o *options) Version() string {
|
|
return o.version
|
|
}
|
|
|
|
func (o *options) Address() string {
|
|
return o.addr
|
|
}
|
|
|
|
func (o *options) Registry() registry.Registry {
|
|
return o.registry
|
|
}
|
|
|
|
func (o *options) TLSConfig() *tls.Config {
|
|
return o.tlsConfig
|
|
}
|
|
|
|
func (o *options) Secure() bool {
|
|
return o.secure
|
|
}
|
|
|
|
func (o *options) DialOptions() []grpc.DialOption {
|
|
return o.dialOptions
|
|
}
|
|
|
|
func (o *options) UnaryInterceptors() []grpc.UnaryClientInterceptor {
|
|
return o.unaryInterceptors
|
|
}
|
|
|
|
func (o *options) StreamInterceptors() []grpc.StreamClientInterceptor {
|
|
return o.streamInterceptors
|
|
}
|