rbd: Add capability and register service

Signed-off-by: Yug Gupta <[email protected]>
ceph · Dec 22, 2021 · f7da55e · f7da55e
1 parent 592c9a5
commit f7da55e
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/internal/csi-addons/rbd/identity.go b/internal/csi-addons/rbd/identity.go
@@ -75,6 +75,12 @@ func (is *IdentityServer) GetCapabilities(
 						Type: identity.Capability_Service_CONTROLLER_SERVICE,
 					},
 				},
+			}, &identity.Capability{
+				Type: &identity.Capability_NetworkFence_{
+					NetworkFence: &identity.Capability_NetworkFence{
+						Type: identity.Capability_NetworkFence_NETWORK_FENCE,
+					},
+				},
 			})
 	}
 

diff --git a/internal/rbd/driver/driver.go b/internal/rbd/driver/driver.go
@@ -213,6 +213,11 @@ func (r *Driver) setupCSIAddonsServer(conf *util.Config) error {
 	is := casrbd.NewIdentityServer(conf)
 	r.cas.RegisterService(is)
 
+	if conf.IsControllerServer {
+		fcs := casrbd.NewFenceControllerServer()
+		r.cas.RegisterService(fcs)
+	}
+
 	// start the server, this does not block, it runs a new go-routine
 	err = r.cas.Start()
 	if err != nil {