From 3e263d5a4daec9d53d03a348e57d6f119bd6bb8d Mon Sep 17 00:00:00 2001 From: Gyu-Ho Lee Date: Fri, 23 Jun 2017 12:49:02 -0700 Subject: [PATCH] proxy/*: add 'MoveLeader' RPC Signed-off-by: Gyu-Ho Lee --- proxy/grpcproxy/adapter/maintenance_client_adapter.go | 4 ++++ proxy/grpcproxy/maintenance.go | 5 +++++ 2 files changed, 9 insertions(+) diff --git a/proxy/grpcproxy/adapter/maintenance_client_adapter.go b/proxy/grpcproxy/adapter/maintenance_client_adapter.go index 9b21bf257..84c56f4ea 100644 --- a/proxy/grpcproxy/adapter/maintenance_client_adapter.go +++ b/proxy/grpcproxy/adapter/maintenance_client_adapter.go @@ -43,6 +43,10 @@ func (s *mts2mtc) Hash(ctx context.Context, r *pb.HashRequest, opts ...grpc.Call return s.mts.Hash(ctx, r) } +func (s *mts2mtc) MoveLeader(ctx context.Context, r *pb.MoveLeaderRequest, opts ...grpc.CallOption) (*pb.MoveLeaderResponse, error) { + return s.mts.MoveLeader(ctx, r) +} + func (s *mts2mtc) Snapshot(ctx context.Context, in *pb.SnapshotRequest, opts ...grpc.CallOption) (pb.Maintenance_SnapshotClient, error) { cs := newPipeStream(ctx, func(ss chanServerStream) error { return s.mts.Snapshot(in, &ss2scServerStream{ss}) diff --git a/proxy/grpcproxy/maintenance.go b/proxy/grpcproxy/maintenance.go index 209dc94a7..0a852c350 100644 --- a/proxy/grpcproxy/maintenance.go +++ b/proxy/grpcproxy/maintenance.go @@ -72,3 +72,8 @@ func (mp *maintenanceProxy) Status(ctx context.Context, r *pb.StatusRequest) (*p conn := mp.client.ActiveConnection() return pb.NewMaintenanceClient(conn).Status(ctx, r) } + +func (mp *maintenanceProxy) MoveLeader(ctx context.Context, r *pb.MoveLeaderRequest) (*pb.MoveLeaderResponse, error) { + conn := mp.client.ActiveConnection() + return pb.NewMaintenanceClient(conn).MoveLeader(ctx, r) +}