grpc: added config service actions and update the gui to leverage them as the default for node context menus of running nodes

This commit is contained in:
Blake Harnden 2022-01-11 16:29:55 -08:00
parent 58b8d1cd24
commit 8f89488fd5
5 changed files with 88 additions and 4 deletions

View file

@ -810,6 +810,30 @@ class CoreGrpcClient:
response = self.stub.ServiceAction(request)
return response.result
def config_service_action(
self,
session_id: int,
node_id: int,
service: str,
action: wrappers.ServiceAction,
) -> bool:
"""
Send an action to a config service for a node.
:param session_id: session id
:param node_id: node id
:param service: config service name
:param action: action for service (start, stop, restart,
validate)
:return: True for success, False otherwise
:raises grpc.RpcError: when session or node doesn't exist
"""
request = ServiceActionRequest(
session_id=session_id, node_id=node_id, service=service, action=action.value
)
response = self.stub.ConfigServiceAction(request)
return response.result
def get_wlan_config(
self, session_id: int, node_id: int
) -> Dict[str, wrappers.ConfigOption]:

View file

@ -72,6 +72,7 @@ from core.api.grpc.wlan_pb2 import (
WlanLinkRequest,
WlanLinkResponse,
)
from core.configservice.base import ConfigServiceBootError
from core.emane.modelmanager import EmaneModelManager
from core.emulator.coreemu import CoreEmu
from core.emulator.data import InterfaceData, LinkData, LinkOptions
@ -986,6 +987,48 @@ class CoreGrpcServer(core_pb2_grpc.CoreApiServicer):
return ServiceActionResponse(result=result)
def ConfigServiceAction(
self, request: ServiceActionRequest, context: ServicerContext
) -> ServiceActionResponse:
"""
Take action whether to start, stop, restart, validate the config service or
none of the above.
:param request: service action request
:param context: context object
:return: service action response about status of action
"""
logger.debug("service action: %s", request)
session = self.get_session(request.session_id, context)
node = self.get_node(session, request.node_id, context, CoreNode)
service = node.config_services.get(request.service)
if not service:
context.abort(grpc.StatusCode.NOT_FOUND, "config service not found")
result = False
if request.action == ServiceAction.START:
try:
service.start()
result = True
except ConfigServiceBootError:
pass
elif request.action == ServiceAction.STOP:
service.stop()
result = True
elif request.action == ServiceAction.RESTART:
service.stop()
try:
service.start()
result = True
except ConfigServiceBootError:
pass
elif request.action == ServiceAction.VALIDATE:
try:
service.run_validation()
result = True
except ConfigServiceBootError:
pass
return ServiceActionResponse(result=result)
def GetWlanConfig(
self, request: GetWlanConfigRequest, context: ServicerContext
) -> GetWlanConfigResponse: