initial import

core/rpc/client.go

@@ -0,0 +1,71 @@
+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
+}