[infra-proxy-service] add service call before reflection

Signed-off-by: Vivek Singh <[email protected]>
chef · Feb 17, 2020 · 9c11c0a · 9c11c0a
1 parent bf87308
commit 9c11c0a
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/components/automate-gateway/gateway/services.go b/components/automate-gateway/gateway/services.go
@@ -317,6 +317,12 @@ func (s *Server) RegisterGRPCServices(grpcServer *grpc.Server) error {
 	)
 	pb_data_lifecycle.RegisterDataLifecycleServer(grpcServer, dataLifecycleServer)
 
+	infraProxyClient, err := clients.InfraProxyClient()
+	if err != nil {
+		return errors.Wrap(err, "create client for infra proxy service")
+	}
+	pb_infra_proxy.RegisterInfraProxyServer(grpcServer, handler_infra_proxy.NewInfraProxyHandler(infraProxyClient))
+
 	// Reflection to be able to make grpcurl calls
 	reflection.Register(grpcServer)
 
@@ -326,12 +332,6 @@ func (s *Server) RegisterGRPCServices(grpcServer *grpc.Server) error {
 	}
 	pb_data_feed.RegisterDatafeedServiceServer(grpcServer, handler.NewDatafeedHandler(datafeedClient))
 
-	infraProxyClient, err := clients.InfraProxyClient()
-	if err != nil {
-		return errors.Wrap(err, "create client for infra proxy service")
-	}
-	pb_infra_proxy.RegisterInfraProxyServer(grpcServer, handler_infra_proxy.NewInfraProxyHandler(infraProxyClient))
-
 	grpc_prometheus.Register(grpcServer)
 
 	return nil