From 09dc84aaa932cb55786d84a10ff9a0aeb86f3135 Mon Sep 17 00:00:00 2001 From: Jeff Layton Date: Wed, 19 Dec 2018 10:21:40 -0500 Subject: [PATCH] mgr/rook: add decorator for add_stateless_service Signed-off-by: Jeff Layton --- src/pybind/mgr/rook/module.py | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/src/pybind/mgr/rook/module.py b/src/pybind/mgr/rook/module.py index 0ed0f0936c5..60129540a18 100644 --- a/src/pybind/mgr/rook/module.py +++ b/src/pybind/mgr/rook/module.py @@ -361,19 +361,19 @@ class RookOrchestrator(MgrModule, orchestrator.Orchestrator): return result + def _service_add_decorate(self, typename, spec, func): + return RookWriteCompletion(lambda: func(spec), None, + "Creating {0} services for {1}".format(typename, spec.name)) + def add_stateless_service(self, service_type, spec): # assert isinstance(spec, orchestrator.StatelessServiceSpec) - if service_type == "mds": - return RookWriteCompletion( - lambda: self.rook_cluster.add_filesystem(spec), None, - "Creating Filesystem services for {0}".format(spec.name)) + return self._service_add_decorate("Filesystem", spec, + self.rook_cluster.add_filesystem) elif service_type == "rgw" : - return RookWriteCompletion( - lambda: self.rook_cluster.add_objectstore(spec), None, - "Creating RGW services for {0}".format(spec.name)) + return self._service_add_decorate("RGW", spec, + self.rook_cluster.add_objectstore) else: - # TODO: RGW, NFS raise NotImplementedError(service_type) def remove_stateless_service(self, service_type, service_id): -- 2.39.5