Register the GroupControllerServer

nixpanic · Aug 7, 2023 · 04a8ae0 · 04a8ae0
1 parent 7e9bc5b
commit 04a8ae0
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/pkg/hostpath/hostpath.go b/pkg/hostpath/hostpath.go
@@ -123,7 +123,7 @@ func NewHostPathDriver(cfg Config) (*hostPath, error) {
 func (hp *hostPath) Run() error {
 	s := NewNonBlockingGRPCServer()
 	// hp itself implements ControllerServer, NodeServer, and IdentityServer.
-	s.Start(hp.config.Endpoint, hp, hp, hp)
+	s.Start(hp.config.Endpoint, hp, hp, hp, hp)
 	s.Wait()
 
 	return nil

diff --git a/pkg/hostpath/server.go b/pkg/hostpath/server.go
@@ -40,11 +40,11 @@ type nonBlockingGRPCServer struct {
 	cleanup func()
 }
 
-func (s *nonBlockingGRPCServer) Start(endpoint string, ids csi.IdentityServer, cs csi.ControllerServer, ns csi.NodeServer) {
+func (s *nonBlockingGRPCServer) Start(endpoint string, ids csi.IdentityServer, cs csi.ControllerServer, ns csi.NodeServer, gcs csi.GroupControllerServer) {
 
 	s.wg.Add(1)
 
-	go s.serve(endpoint, ids, cs, ns)
+	go s.serve(endpoint, ids, cs, ns, gcs)
 
 	return
 }
@@ -63,7 +63,7 @@ func (s *nonBlockingGRPCServer) ForceStop() {
 	s.cleanup()
 }
 
-func (s *nonBlockingGRPCServer) serve(ep string, ids csi.IdentityServer, cs csi.ControllerServer, ns csi.NodeServer) {
+func (s *nonBlockingGRPCServer) serve(ep string, ids csi.IdentityServer, cs csi.ControllerServer, ns csi.NodeServer, gcs csi.GroupControllerServer) {
 	listener, cleanup, err := endpoint.Listen(ep)
 	if err != nil {
 		glog.Fatalf("Failed to listen: %v", err)
@@ -85,6 +85,9 @@ func (s *nonBlockingGRPCServer) serve(ep string, ids csi.IdentityServer, cs csi.
 	if ns != nil {
 		csi.RegisterNodeServer(server, ns)
 	}
+	if gcs != nil {
+		csi.RegisterGroupControllerServer(server, gcs)
+	}
 
 	glog.Infof("Listening for connections on address: %#v", listener.Addr())