feat: grpc scheduler client dial options (#1792)
Signed-off-by: Gaius <gaius.qi@gmail.com>
This commit is contained in:
parent
536b039462
commit
04bc6ae493
|
|
@ -51,7 +51,6 @@ const (
|
||||||
backoffWaitBetween = 500 * time.Millisecond
|
backoffWaitBetween = 500 * time.Millisecond
|
||||||
)
|
)
|
||||||
|
|
||||||
// GetClient get scheduler clients using resolver and balancer,
|
|
||||||
func GetClient(ctx context.Context, dynconfig config.Dynconfig, opts ...grpc.DialOption) (Client, error) {
|
func GetClient(ctx context.Context, dynconfig config.Dynconfig, opts ...grpc.DialOption) (Client, error) {
|
||||||
// Register resolver and balancer.
|
// Register resolver and balancer.
|
||||||
resolver.RegisterScheduler(dynconfig)
|
resolver.RegisterScheduler(dynconfig)
|
||||||
|
|
@ -90,6 +89,7 @@ func GetClient(ctx context.Context, dynconfig config.Dynconfig, opts ...grpc.Dia
|
||||||
SchedulerClient: schedulerv1.NewSchedulerClient(conn),
|
SchedulerClient: schedulerv1.NewSchedulerClient(conn),
|
||||||
ClientConn: conn,
|
ClientConn: conn,
|
||||||
Dynconfig: dynconfig,
|
Dynconfig: dynconfig,
|
||||||
|
dialOptions: opts,
|
||||||
}, nil
|
}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
@ -125,6 +125,7 @@ type client struct {
|
||||||
schedulerv1.SchedulerClient
|
schedulerv1.SchedulerClient
|
||||||
*grpc.ClientConn
|
*grpc.ClientConn
|
||||||
config.Dynconfig
|
config.Dynconfig
|
||||||
|
dialOptions []grpc.DialOption
|
||||||
}
|
}
|
||||||
|
|
||||||
// RegisterPeerTask registers a peer into task.
|
// RegisterPeerTask registers a peer into task.
|
||||||
|
|
@ -208,7 +209,7 @@ func (c *client) leaveHost(ctx context.Context, addr string, req *schedulerv1.Le
|
||||||
conn, err := grpc.DialContext(
|
conn, err := grpc.DialContext(
|
||||||
ctx,
|
ctx,
|
||||||
addr,
|
addr,
|
||||||
[]grpc.DialOption{
|
append([]grpc.DialOption{
|
||||||
grpc.WithUnaryInterceptor(grpc_middleware.ChainUnaryClient(
|
grpc.WithUnaryInterceptor(grpc_middleware.ChainUnaryClient(
|
||||||
rpc.ConvertErrorUnaryClientInterceptor,
|
rpc.ConvertErrorUnaryClientInterceptor,
|
||||||
otelgrpc.UnaryClientInterceptor(),
|
otelgrpc.UnaryClientInterceptor(),
|
||||||
|
|
@ -219,7 +220,7 @@ func (c *client) leaveHost(ctx context.Context, addr string, req *schedulerv1.Le
|
||||||
grpc_retry.WithBackoff(grpc_retry.BackoffLinear(backoffWaitBetween)),
|
grpc_retry.WithBackoff(grpc_retry.BackoffLinear(backoffWaitBetween)),
|
||||||
),
|
),
|
||||||
)),
|
)),
|
||||||
}...,
|
}, c.dialOptions...)...,
|
||||||
)
|
)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue