You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
72 lines
1.5 KiB
Go
72 lines
1.5 KiB
Go
package rpc
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"time"
|
|
|
|
"zero/core/rpc/clientinterceptors"
|
|
|
|
"google.golang.org/grpc"
|
|
)
|
|
|
|
const dialTimeout = time.Second * 3
|
|
|
|
type (
|
|
ClientOptions struct {
|
|
Timeout time.Duration
|
|
DialOptions []grpc.DialOption
|
|
}
|
|
|
|
ClientOption func(options *ClientOptions)
|
|
|
|
Client interface {
|
|
Next() (*grpc.ClientConn, bool)
|
|
}
|
|
)
|
|
|
|
func WithDialOption(opt grpc.DialOption) ClientOption {
|
|
return func(options *ClientOptions) {
|
|
options.DialOptions = append(options.DialOptions, opt)
|
|
}
|
|
}
|
|
|
|
func WithTimeout(timeout time.Duration) ClientOption {
|
|
return func(options *ClientOptions) {
|
|
options.Timeout = timeout
|
|
}
|
|
}
|
|
|
|
func buildDialOptions(opts ...ClientOption) []grpc.DialOption {
|
|
var clientOptions ClientOptions
|
|
for _, opt := range opts {
|
|
opt(&clientOptions)
|
|
}
|
|
|
|
options := []grpc.DialOption{
|
|
grpc.WithInsecure(),
|
|
grpc.WithBlock(),
|
|
WithUnaryClientInterceptors(
|
|
clientinterceptors.BreakerInterceptor,
|
|
clientinterceptors.DurationInterceptor,
|
|
clientinterceptors.PromMetricInterceptor,
|
|
clientinterceptors.TimeoutInterceptor(clientOptions.Timeout),
|
|
clientinterceptors.TracingInterceptor,
|
|
),
|
|
}
|
|
|
|
return append(options, clientOptions.DialOptions...)
|
|
}
|
|
|
|
func dial(server string, opts ...ClientOption) (*grpc.ClientConn, error) {
|
|
options := buildDialOptions(opts...)
|
|
timeCtx, cancel := context.WithTimeout(context.Background(), dialTimeout)
|
|
defer cancel()
|
|
conn, err := grpc.DialContext(timeCtx, server, options...)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("rpc dial: %s, error: %s", server, err.Error())
|
|
}
|
|
|
|
return conn, nil
|
|
}
|