Skip to content
Snippets Groups Projects
monitoring_server.py 2.46 KiB
Newer Older
Ricard Vilalta's avatar
Ricard Vilalta committed
#!/usr/bin/python

import os
from concurrent import futures
import grpc
Ricard Vilalta's avatar
Ricard Vilalta committed
import numpy
import time
Ricard Vilalta's avatar
Ricard Vilalta committed

import monitoring_pb2
import monitoring_pb2_grpc

from grpc_health.v1 import health
from grpc_health.v1 import health_pb2
from grpc_health.v1 import health_pb2_grpc

Ricard Vilalta's avatar
Ricard Vilalta committed

from logger import getJSONLogger
logger = getJSONLogger('monitoringservice-server')
logger.setLevel('DEBUG')

import threading

from prometheus_client import start_http_server, Summary
from prometheus_client import Counter, Gauge

MONITORING_GETINSTANTKPI_REQUEST_TIME = Summary('monitoring_getinstantkpi_processing_seconds', 'Time spent processing monitoring instant kpi request')
MONITORING_INCLUDEKPI_COUNTER = Counter('monitoring_includekpi_counter', 'Monitoring include kpi request counter')

Ricard Vilalta's avatar
Ricard Vilalta committed

class monitoringService(monitoring_pb2_grpc.MonitoringServiceServicer):
    def __init__(self):
        logger.info("init monitoringService")


    def IncludeKpi(self, request, context):
        #KPI, returns empty
        logger.info("IncludeKpi")
        MONITORING_INCLUDEKPI_COUNTER.inc()
Ricard Vilalta's avatar
Ricard Vilalta committed


    def MonitorKpi ( self, request, context):
        #KpiDevice returns (google.protobuf.Empty) {}
        logger.info("IncludeKpi")
Ricard Vilalta's avatar
Ricard Vilalta committed

    def GetStream_kpi ( self, request, context):
        #KpiId ) returns (stream Kpi) {}
        logger.info("IncludeKpi")
Ricard Vilalta's avatar
Ricard Vilalta committed

    @MONITORING_GETINSTANTKPI_REQUEST_TIME.time()
Ricard Vilalta's avatar
Ricard Vilalta committed
    def GetInstantKpi ( self, request, context):
        #KpiId ) returns ( Kpi) {}
        logger.info("IncludeKpi")
Ricard Vilalta's avatar
Ricard Vilalta committed


if __name__ == "__main__":
    logger.info("initializing monitoringService")

    port = os.environ.get('PORT', "7070")

    # create gRPC server
    serverGRPC = grpc.server(futures.ThreadPoolExecutor(max_workers=10)) # ,interceptors=(tracer_interceptor,))
Ricard Vilalta's avatar
Ricard Vilalta committed

    # add class to gRPC server
Lluis Gifre Renom's avatar
Lluis Gifre Renom committed
    service = monitoringService()
    monitoring_pb2_grpc.add_MonitoringServiceServicer_to_server(service, serverGRPC)
Ricard Vilalta's avatar
Ricard Vilalta committed

    health_servicer = health.HealthServicer(experimental_non_blocking=True,experimental_thread_pool=futures.ThreadPoolExecutor(max_workers=1))
    health_pb2_grpc.add_HealthServicer_to_server(health_servicer, serverGRPC)
Ricard Vilalta's avatar
Ricard Vilalta committed

    # start server
    logger.info("listening on port: " + port)
    serverGRPC.add_insecure_port('[::]:'+port)
    serverGRPC.start()
Ricard Vilalta's avatar
Ricard Vilalta committed
    health_servicer.set("", health_pb2.HealthCheckResponse.SERVING)

    # keep alive
    try:
         while True:
            time.sleep(0.1)
    except KeyboardInterrupt:
            serverGRPC.stop(0)
Ricard Vilalta's avatar
Ricard Vilalta committed