Newer
Older
Lluis Gifre Renom
committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
import grpc, logging
from prometheus_client import Counter, Histogram
from google.protobuf.json_format import MessageToDict
from context.proto.context_pb2 import DeviceId, Empty
from device.proto.device_pb2_grpc import DeviceServiceServicer
LOGGER = logging.getLogger(__name__)
ADDDEVICE_COUNTER_STARTED = Counter ('device_adddevice_counter_started',
'Device:AddDevice counter of requests started' )
ADDDEVICE_COUNTER_COMPLETED = Counter ('device_adddevice_counter_completed',
'Device:AddDevice counter of requests completed')
ADDDEVICE_COUNTER_FAILED = Counter ('device_adddevice_counter_failed',
'Device:AddDevice counter of requests failed' )
ADDDEVICE_HISTOGRAM_DURATION = Histogram('device_adddevice_histogram_duration',
'Device:AddDevice histogram of request duration')
CONFIGUREDEVICE_COUNTER_STARTED = Counter ('device_configuredevice_counter_started',
'Device:ConfigureDevice counter of requests started' )
CONFIGUREDEVICE_COUNTER_COMPLETED = Counter ('device_configuredevice_counter_completed',
'Device:ConfigureDevice counter of requests completed')
CONFIGUREDEVICE_COUNTER_FAILED = Counter ('device_configuredevice_counter_failed',
'Device:ConfigureDevice counter of requests failed' )
CONFIGUREDEVICE_HISTOGRAM_DURATION = Histogram('device_configuredevice_histogram_duration',
'Device:ConfigureDevice histogram of request duration')
DELETEDEVICE_COUNTER_STARTED = Counter ('device_deletedevice_counter_started',
'Device:DeleteDevice counter of requests started' )
DELETEDEVICE_COUNTER_COMPLETED = Counter ('device_deletedevice_counter_completed',
'Device:DeleteDevice counter of requests completed')
DELETEDEVICE_COUNTER_FAILED = Counter ('device_deletedevice_counter_failed',
'Device:DeleteDevice counter of requests failed' )
DELETEDEVICE_HISTOGRAM_DURATION = Histogram('device_deletedevice_histogram_duration',
'Device:DeleteDevice histogram of request duration')
class DeviceServiceServicerImpl(DeviceServiceServicer):
def __init__(self, database):
LOGGER.debug('Creating Servicer...')
self.database = database
LOGGER.debug('Servicer Created')
@ADDDEVICE_HISTOGRAM_DURATION.time()
def AddDevice(self, request, context):
# request=context.Device(), returns=context.DeviceId()
ADDDEVICE_COUNTER_STARTED.inc()
try:
LOGGER.info('AddDevice request: {}'.format(str(request)))
reply = DeviceId(**self.database.add_device(MessageToDict(request)))
LOGGER.info('AddDevice reply: {}'.format(str(reply)))
ADDDEVICE_COUNTER_COMPLETED.inc()
return reply
except:
LOGGER.exception('AddDevice exception')
ADDDEVICE_COUNTER_FAILED.inc()
context.set_code(grpc.StatusCode.INTERNAL)
return DeviceId()
@CONFIGUREDEVICE_HISTOGRAM_DURATION.time()
def ConfigureDevice(self, request, context):
# request=context.DeviceConfig(), returns=context.DeviceId()
CONFIGUREDEVICE_COUNTER_STARTED.inc()
try:
LOGGER.info('ConfigureDevice request: {}'.format(str(request)))
reply = DeviceId(**self.database.configure_device(MessageToDict(request)))
LOGGER.info('ConfigureDevice reply: {}'.format(str(reply)))
CONFIGUREDEVICE_COUNTER_COMPLETED.inc()
return reply
except:
LOGGER.exception('ConfigureDevice exception')
CONFIGUREDEVICE_COUNTER_FAILED.inc()
context.set_code(grpc.StatusCode.INTERNAL)
return DeviceId()
@DELETEDEVICE_HISTOGRAM_DURATION.time()
def DeleteDevice(self, request, context):
# request=context.DeviceId(), returns=context.Empty()
DELETEDEVICE_COUNTER_STARTED.inc()
try:
LOGGER.info('DeleteDevice request: {}'.format(str(request)))
reply = Empty(**self.database.delete_device(MessageToDict(request)))
LOGGER.info('DeleteDevice reply: {}'.format(str(reply)))
DELETEDEVICE_COUNTER_COMPLETED.inc()
return reply
except:
LOGGER.exception('DeleteDevice exception')
DELETEDEVICE_COUNTER_FAILED.inc()
context.set_code(grpc.StatusCode.INTERNAL)
return Empty()