From 47907d6da2770605e98d51a644084264979770a7 Mon Sep 17 00:00:00 2001
From: Aliaksandr Mianzhynski <amenzhinsky@gmail.com>
Date: Thu, 3 Jun 2021 11:47:09 +0300
Subject: [PATCH] Add GetServiceInfo to xds.GRPCServer

---
 xds/server.go      | 7 +++++++
 xds/server_test.go | 4 ++++
 2 files changed, 11 insertions(+)

diff --git a/xds/server.go b/xds/server.go
index b83a073ac1c5..2e1ef4806637 100644
--- a/xds/server.go
+++ b/xds/server.go
@@ -73,6 +73,7 @@ type grpcServerInterface interface {
 	Serve(net.Listener) error
 	Stop()
 	GracefulStop()
+	GetServiceInfo() map[string]grpc.ServiceInfo
 }
 
 // GRPCServer wraps a gRPC server and provides server-side xDS functionality, by
@@ -145,6 +146,12 @@ func (s *GRPCServer) RegisterService(sd *grpc.ServiceDesc, ss interface{}) {
 	s.gs.RegisterService(sd, ss)
 }
 
+// GetServiceInfo returns a map from service names to ServiceInfo.
+// Service names include the package names, in the form of <package>.<service>.
+func (s *GRPCServer) GetServiceInfo() map[string]grpc.ServiceInfo {
+	return s.gs.GetServiceInfo()
+}
+
 // initXDSClient creates a new xdsClient if there is no existing one available.
 func (s *GRPCServer) initXDSClient() error {
 	s.clientMu.Lock()
diff --git a/xds/server_test.go b/xds/server_test.go
index e16ac36b01f2..1f8bc977626b 100644
--- a/xds/server_test.go
+++ b/xds/server_test.go
@@ -86,6 +86,10 @@ func (f *fakeGRPCServer) GracefulStop() {
 	f.gracefulStopCh.Send(nil)
 }
 
+func (f *fakeGRPCServer) GetServiceInfo() map[string]grpc.ServiceInfo {
+	return map[string]grpc.ServiceInfo{}
+}
+
 func newFakeGRPCServer() *fakeGRPCServer {
 	return &fakeGRPCServer{
 		done:              make(chan struct{}),