github_mirrors/go-zero
1
0
Fork 0
mirror of https://github.com/zeromicro/go-zero.git synced 2025-02-02 16:28:39 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor rpcx, export WithDialOption and WithTimeout

Browse Source
This commit is contained in:
kevin 2020-08-11 14:33:10 +08:00 committed by kingxt
parent d1529fced8
commit 10cd6053bc
7 changed files with 27 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
example/graceful/dns/api/svc/servicecontext.go
View File

@ -3,5 +3,5 @@ package svc
import "github.com/tal-tech/go-zero/rpcx"
type ServiceContext struct {
Client *rpcx.RpcClient
Client rpcx.Client
}

2
example/graceful/etcd/api/svc/servicecontext.go
View File

@ -3,5 +3,5 @@ package svc
import "github.com/tal-tech/go-zero/rpcx"
type ServiceContext struct {
Client *rpcx.RpcClient
Client rpcx.Client
}

2
example/tracing/edge/main.go
View File

@ -15,7 +15,7 @@ import (
var (
configFile = flag.String("f", "config.json", "the config file")
client *rpcx.RpcClient
client rpcx.Client
)
func handle(w http.ResponseWriter, r *http.Request) {

4
example/tracing/portal/server.go
View File

@ -20,11 +20,11 @@ type (
}
PortalServer struct {
userRpc *rpcx.RpcClient
userRpc rpcx.Client
}
)
func NewPortalServer(client *rpcx.RpcClient) *PortalServer {
func NewPortalServer(client rpcx.Client) *PortalServer {
return &PortalServer{
userRpc: client,
}

29
rpcx/client.go
View File

@ -10,11 +10,22 @@ import (
"google.golang.org/grpc"
)
type RpcClient struct {
client internal.Client
}
var (
WithDialOption = internal.WithDialOption
WithTimeout = internal.WithTimeout
)
func MustNewClient(c RpcClientConf, options ...internal.ClientOption) *RpcClient {
type (
Client interface {
Conn() *grpc.ClientConn
}
RpcClient struct {
client Client
}
)
func MustNewClient(c RpcClientConf, options ...internal.ClientOption) Client {
cli, err := NewClient(c, options...)
if err != nil {
log.Fatal(err)
@ -23,20 +34,20 @@ func MustNewClient(c RpcClientConf, options ...internal.ClientOption) *RpcClient
return cli
}
func NewClient(c RpcClientConf, options ...internal.ClientOption) (*RpcClient, error) {
func NewClient(c RpcClientConf, options ...internal.ClientOption) (Client, error) {
var opts []internal.ClientOption
if c.HasCredential() {
opts = append(opts, internal.WithDialOption(grpc.WithPerRPCCredentials(&auth.Credential{
opts = append(opts, WithDialOption(grpc.WithPerRPCCredentials(&auth.Credential{
App: c.App,
Token: c.Token,
})))
}
if c.Timeout > 0 {
opts = append(opts, internal.WithTimeout(time.Duration(c.Timeout)*time.Millisecond))
opts = append(opts, WithTimeout(time.Duration(c.Timeout)*time.Millisecond))
}
opts = append(opts, options...)
var client internal.Client
var client Client
var err error
if len(c.Server) > 0 {
client, err = internal.NewDirectClient(c.Server, opts...)
@ -52,7 +63,7 @@ func NewClient(c RpcClientConf, options ...internal.ClientOption) (*RpcClient, e
}, nil
}
func NewClientNoAuth(c discov.EtcdConf) (*RpcClient, error) {
func NewClientNoAuth(c discov.EtcdConf) (Client, error) {
client, err := internal.NewDiscovClient(c.Hosts, c.Key)
if err != nil {
return nil, err

4
rpcx/internal/client.go
View File

@ -18,10 +18,6 @@ type (
}
ClientOption func(options *ClientOptions)
Client interface {
Conn() *grpc.ClientConn
}
)
func WithDialOption(opt grpc.DialOption) ClientOption {

4
rpcx/proxy.go
View File

@ -12,7 +12,7 @@ import (
type RpcProxy struct {
backend string
clients map[string]*RpcClient
clients map[string]Client
options []internal.ClientOption
sharedCalls syncx.SharedCalls
lock sync.Mutex
@ -21,7 +21,7 @@ type RpcProxy struct {
func NewRpcProxy(backend string, opts ...internal.ClientOption) *RpcProxy {
return &RpcProxy{
backend: backend,
clients: make(map[string]*RpcClient),
clients: make(map[string]Client),
options: opts,
sharedCalls: syncx.NewSharedCalls(),
}