From a6ae677d8fbe0badf34b29323467ff00db6dce7f Mon Sep 17 00:00:00 2001 From: fanmin shi Date: Fri, 14 Jul 2017 16:45:27 -0700 Subject: [PATCH] proxy: support HashKV in grpcproxy --- 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 84c56f4ea..66aaabba1 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) HashKV(ctx context.Context, r *pb.HashKVRequest, opts ...grpc.CallOption) (*pb.HashKVResponse, error) { + return s.mts.HashKV(ctx, r) +} + func (s *mts2mtc) MoveLeader(ctx context.Context, r *pb.MoveLeaderRequest, opts ...grpc.CallOption) (*pb.MoveLeaderResponse, error) { return s.mts.MoveLeader(ctx, r) } diff --git a/proxy/grpcproxy/maintenance.go b/proxy/grpcproxy/maintenance.go index 0a852c350..1696bfb83 100644 --- a/proxy/grpcproxy/maintenance.go +++ b/proxy/grpcproxy/maintenance.go @@ -63,6 +63,11 @@ func (mp *maintenanceProxy) Hash(ctx context.Context, r *pb.HashRequest) (*pb.Ha return pb.NewMaintenanceClient(conn).Hash(ctx, r) } +func (mp *maintenanceProxy) HashKV(ctx context.Context, r *pb.HashKVRequest) (*pb.HashKVResponse, error) { + conn := mp.client.ActiveConnection() + return pb.NewMaintenanceClient(conn).HashKV(ctx, r) +} + func (mp *maintenanceProxy) Alarm(ctx context.Context, r *pb.AlarmRequest) (*pb.AlarmResponse, error) { conn := mp.client.ActiveConnection() return pb.NewMaintenanceClient(conn).Alarm(ctx, r)