From 95e48a892d6c51e95d2aa77742da72c2df14dc28 Mon Sep 17 00:00:00 2001
From: Aliaksandr Mianzhynski <amenzhinsky@gmail.com>
Date: Wed, 9 Jun 2021 21:05:17 +0300
Subject: [PATCH] Add GetServiceInfo to xds.GRPCServer (#4507)

---
 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 cfbea1a1bca..aad05d81d11 100644
--- a/xds/server.go
+++ b/xds/server.go
@@ -64,6 +64,7 @@ type grpcServer interface {
 	Serve(net.Listener) error
 	Stop()
 	GracefulStop()
+	GetServiceInfo() map[string]grpc.ServiceInfo
 }
 
 // GRPCServer wraps a gRPC server and provides server-side xDS functionality, by
@@ -136,6 +137,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 45df8b76fca..68e0d85c692 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 {
+	panic("implement me")
+}
+
 func newFakeGRPCServer() *fakeGRPCServer {
 	return &fakeGRPCServer{
 		done:              make(chan struct{}),