Implement GetAllStores API (tikv#937)

oh-my-tidb · Jan 29, 2018 · b998669 · b998669
1 parent f995d7b
commit b998669
Show file tree

Hide file tree

Showing 3 changed files with 613 additions and 224 deletions.
diff --git a/Gopkg.lock b/Gopkg.lock
diff --git a/server/grpc_service.go b/server/grpc_service.go
@@ -207,6 +207,23 @@ func (s *Server) PutStore(ctx context.Context, request *pdpb.PutStoreRequest) (*
 	}, nil
 }
 
+// GetAllStores implements gRPC PDServer.
+func (s *Server) GetAllStores(ctx context.Context, request *pdpb.GetAllStoresRequest) (*pdpb.GetAllStoresResponse, error) {
+	if err := s.validateRequest(request.GetHeader()); err != nil {
+		return nil, errors.Trace(err)
+	}
+
+	cluster := s.GetRaftCluster()
+	if cluster == nil {
+		return &pdpb.GetAllStoresResponse{Header: s.notBootstrappedHeader()}, nil
+	}
+
+	return &pdpb.GetAllStoresResponse{
+		Header: s.header(),
+		Stores: cluster.GetStores(),
+	}, nil
+}
+
 // StoreHeartbeat implements gRPC PDServer.
 func (s *Server) StoreHeartbeat(ctx context.Context, request *pdpb.StoreHeartbeatRequest) (*pdpb.StoreHeartbeatResponse, error) {
 	if err := s.validateRequest(request.GetHeader()); err != nil {