85 lines
2.2 KiB
Python
85 lines
2.2 KiB
Python
|
#!/usr/bin/env python
|
||
|
|
|
||
|
|
# Copyright 2016 NXP
|
||
|
|
# Copyright 2020 ACRIOS Systems s.r.o.
|
||
|
|
# All rights reserved.
|
||
|
|
#
|
||
|
|
# SPDX-License-Identifier: BSD-3-Clause
|
||
|
|
|
||
|
|
from .codec import MessageType
|
||
|
|
from .client import RequestError
|
||
|
|
|
||
|
|
|
||
|
|
class Service(object):
|
||
|
|
def __init__(self, serviceId):
|
||
|
|
self._id = serviceId
|
||
|
|
self._methods = {}
|
||
|
|
|
||
|
|
@property
|
||
|
|
def service_id(self):
|
||
|
|
return self._id
|
||
|
|
|
||
|
|
def handle_invocation(self, methodId, sequence, codec):
|
||
|
|
try:
|
||
|
|
self._methods[methodId](sequence, codec)
|
||
|
|
except Exception as e:
|
||
|
|
raise RequestError(
|
||
|
|
"invalid method ID (%d) or method implementation: %s" % (methodId, str(e)))
|
||
|
|
|
||
|
|
|
||
|
|
class Server(object):
|
||
|
|
def __init__(self, transport=None, codecClass=None):
|
||
|
|
self._transport = transport
|
||
|
|
self._codecClass = codecClass
|
||
|
|
self._services = []
|
||
|
|
|
||
|
|
@property
|
||
|
|
def transport(self):
|
||
|
|
return self._transport
|
||
|
|
|
||
|
|
@transport.setter
|
||
|
|
def transport(self, value):
|
||
|
|
self._transport = value
|
||
|
|
|
||
|
|
@property
|
||
|
|
def codec_class(self):
|
||
|
|
return self._codecClass
|
||
|
|
|
||
|
|
@codec_class.setter
|
||
|
|
def codec_class(self, value):
|
||
|
|
self._codecClass = value
|
||
|
|
|
||
|
|
@property
|
||
|
|
def services(self):
|
||
|
|
return self._services
|
||
|
|
|
||
|
|
def get_service_with_id(self, theId):
|
||
|
|
matches = [x for x in self._services if x.service_id == theId]
|
||
|
|
if len(matches):
|
||
|
|
return matches[0]
|
||
|
|
else:
|
||
|
|
return None
|
||
|
|
|
||
|
|
def add_service(self, service):
|
||
|
|
assert self.get_service_with_id(service.service_id) is None
|
||
|
|
self._services.append(service)
|
||
|
|
|
||
|
|
def run(self):
|
||
|
|
raise NotImplementedError()
|
||
|
|
|
||
|
|
def stop(self):
|
||
|
|
pass
|
||
|
|
|
||
|
|
def _process_request(self, codec):
|
||
|
|
info = codec.start_read_message()
|
||
|
|
if info.type not in [MessageType.kInvocationMessage, MessageType.kOnewayMessage]:
|
||
|
|
raise RequestError("invalid type of incoming request")
|
||
|
|
|
||
|
|
service = self.get_service_with_id(info.service)
|
||
|
|
if service is None:
|
||
|
|
raise RequestError("invalid service ID (%d)" % info.service)
|
||
|
|
service.handle_invocation(info.request, info.sequence, codec)
|
||
|
|
|
||
|
|
if info.type is MessageType.kOnewayMessage:
|
||
|
|
codec.reset()
|