Ops/gluster-dir-csi
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Make parameters configurable via envoronment variables

Browse Source
This commit is contained in:
Fabian Stamm 2022-08-14 19:39:10 +02:00
parent 0be4bce96f
commit 2e6826a414
1 changed files with 34 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
main.py
View File

@ -11,19 +11,22 @@ import csi_pb2_grpc
import subprocess
#csi_sock_full_path = "/Users/andre/Projects/opensource/gluster-dir-csi/csi.sock"
csi_sock_full_path = "/csi/csi.sock"
gluster_host = "10.0.114.218"
gluster_volume = "vol_main"
gluster_host = os.environ["GLUSTER_HOST"]
gluster_volume = os.environ["GLUSTER_VOLUME"]
csi_sock_full_path = os.environ.get("CSI_PATH", "/csi/csi.sock")
def unmount(path):
print("Unmounting", path)
return subprocess.run(["umount", path])
def mount(volume, path):
print("Mounting", volume, "at", path)
p = subprocess.run(["mount","-t", "glusterfs", "%s:/%s/%s" % (gluster_host, gluster_volume, volume), path])
p = subprocess.run(["mount", "-t", "glusterfs", "%s:/%s/%s" %
(gluster_host, gluster_volume, volume), path])
if p.returncode == 0:
return True
@ -45,15 +48,18 @@ class Identity(csi_pb2_grpc.Identity):
return csi_pb2.GetPluginCapabilitiesResponse(
capabilities=[
csi_pb2.PluginCapability(
service=csi_pb2.PluginCapability.Service(type=csi_pb2.PluginCapability.Service.Type.CONTROLLER_SERVICE)
service=csi_pb2.PluginCapability.Service(
type=csi_pb2.PluginCapability.Service.Type.CONTROLLER_SERVICE)
)
])
class Controller(csi_pb2_grpc.Controller):
def ControllerGetCapabilities(self, request, context):
return csi_pb2.ControllerGetCapabilitiesResponse(capabilities=[
csi_pb2.ControllerServiceCapability(rpc = csi_pb2.ControllerServiceCapability.RPC(type=csi_pb2.ControllerServiceCapability.RPC.Type.CREATE_DELETE_VOLUME)),
csi_pb2.ControllerServiceCapability(rpc=csi_pb2.ControllerServiceCapability.RPC(
type=csi_pb2.ControllerServiceCapability.RPC.Type.CREATE_DELETE_VOLUME)),
])
def CreateVolume(self, request, context):
@ -61,7 +67,8 @@ class Controller(csi_pb2_grpc.Controller):
print("CreateVolume", name)
p = subprocess.run(["mkdir", "-p", "/mnt/main/%s" % name])
volume = csi_pb2.Volume(volume_id=name)
return csi_pb2.CreateVolumeResponse(volume=volume);
return csi_pb2.CreateVolumeResponse(volume=volume)
class Node(csi_pb2_grpc.Node):
@ -71,8 +78,9 @@ class Node(csi_pb2_grpc.Node):
def NodeGetCapabilities(self, request, context):
return csi_pb2.NodeGetCapabilitiesResponse(capabilities=[
csi_pb2.NodeServiceCapability(rpc= csi_pb2.NodeServiceCapability.RPC(type = csi_pb2.NodeServiceCapability.RPC.Type.STAGE_UNSTAGE_VOLUME))
]);
csi_pb2.NodeServiceCapability(rpc=csi_pb2.NodeServiceCapability.RPC(
type=csi_pb2.NodeServiceCapability.RPC.Type.STAGE_UNSTAGE_VOLUME))
])
def NodePublishVolume(self, request, context):
volume_id = request.volume_id
@ -81,14 +89,14 @@ class Node(csi_pb2_grpc.Node):
p = subprocess.run(["mkdir", "-p", path], check=True)
res = mount(volume_id, path)
if res is True:
return csi_pb2.NodePublishVolumeResponse();
return csi_pb2.NodePublishVolumeResponse()
print(res)
def NodeUnpublishVolume(self, request, context):
path = request.target_path
print("NodeUnpublishVolume", path)
p = subprocess.run(["umount", path])
return csi_pb2.NodeUnpublishVolumeResponse();
return csi_pb2.NodeUnpublishVolumeResponse()
def NodeStageVolume(self, request, context):
volume_id = request.volume_id
@ -118,9 +126,11 @@ def serve():
print("Waiting for termination")
server.wait_for_termination()
if __name__ == '__main__':
p = subprocess.run(["mkdir", "-p", "/mnt/main"])
p = subprocess.run(["mount", "-t", "glusterfs", "%s:/%s" % (gluster_host, gluster_volume), "/mnt/main"],check=True)
p = subprocess.run(["mount", "-t", "glusterfs", "%s:/%s" %
(gluster_host, gluster_volume), "/mnt/main"], check=True)
logging.basicConfig()
serve()