grpc added node service file and service action

This commit is contained in:
bharnden 2019-03-16 20:53:26 -07:00
parent f24376d66c
commit a62a03e6b9
3 changed files with 107 additions and 0 deletions

View file

@ -278,6 +278,22 @@ class CoreApiClient(object):
request.service = service
return self.stub.GetNodeService(request)
def get_node_service_file(self, session, _id, service, file_name):
request = core_pb2.GetNodeServiceFileRequest()
request.session = session
request.id = _id
request.service = service
request.file = file_name
return self.stub.GetNodeServiceFile(request)
def service_action(self, session, _id, service, action):
request = core_pb2.ServiceActionRequest()
request.session = session
request.id = _id
request.service = service
request.action = action
return self.stub.ServiceAction(request)
def get_wlan_config(self, session, _id):
request = core_pb2.GetWlanConfigRequest()
request.session = session

View file

@ -633,6 +633,64 @@ class CoreApiServer(core_pb2_grpc.CoreApiServicer):
response.service.meta = service.meta
return response
def GetNodeServiceFile(self, request, context):
session = self.coreemu.sessions.get(request.session)
if not session:
raise Exception("no session found")
node = session.get_object(request.id)
if not node:
raise Exception("no node found")
service = None
for current_service in node.services:
if current_service.name == request.service:
service = current_service
break
response = core_pb2.GetNodeServiceFileResponse()
if not service:
return response
file_data = session.services.get_service_file(node, request.service, request.file)
response.data = file_data.data
return response
def ServiceAction(self, request, context):
session = self.coreemu.sessions.get(request.session)
if not session:
raise Exception("no session found")
node = session.get_object(request.id)
if not node:
raise Exception("no node found")
service = None
for current_service in node.services:
if current_service.name == request.service:
service = current_service
break
response = core_pb2.ServiceActionResponse()
response.result = False
if not service:
return response
status = -1
if request.action == core_pb2.START:
status = session.services.startup_service(node, service, wait=True)
elif request.action == core_pb2.STOP:
status = session.services.stop_service(node, service)
elif request.action == core_pb2.RESTART:
status = session.services.stop_service(node, service)
if not status:
status = session.services.startup_service(node, service, wait=True)
elif request.action == core_pb2.VALIDATE:
status = session.services.validate_service(node, service)
if not status:
response.result = True
return response
def GetWlanConfig(self, request, context):
session = self.coreemu.sessions.get(request.session)
if not session: