eureka.py

from flask import Blueprint, Response

from .eurekaclient import EurekaClient
import json

eureka_bp = Blueprint('eureka', __name__)


@eureka_bp.route('/health')
def healthcheck():
    """
    Return 200 as default
    """
    payback={"status": "UP"}

    resp = Response(response=json.dumps(payback),
                    status=200,
                    mimetype="application/json;charset=UTF-8")


    return resp
    # return payback

@eureka_bp.route('/info')
def info():
    """
    Return 200 as default
    """
    payback={"id":"ks-nlp"}
    resp = Response(response=json.dumps(payback),
                    status=200,
                    mimetype="application/json;charset=UTF-8")
    return resp

class Eureka(object):
    def __init__(self, app=None, **kwargs):
        """
        Initialize the flask extension
        """

        self.kwargs = kwargs if kwargs else {}
        self.app = None

        if app is not None:
            self.init_app(app)

    def init_app(self, app):
        self.app = app

        if not hasattr(app, 'extensions'):
            app.extensions = {}
        if 'eureka' in app.extensions:
            raise RuntimeError('Flask application already initialized')
        app.extensions['eureka'] = self

    def register_service(self, name=None, **kwargs):
        """
        Register service with eureka service

        :param name: name of the eureka application
        """
        name = self.app.config.get('SERVICE_NAME', name)
        eureka_url = self.app.config.get(EurekaClient.EUREKA_SERVICE_URL, None)
        data_center = self.app.config.get(EurekaClient.EUREKA_INSTANCE_DATACENTER, None)
        host_name = self.app.config.get(EurekaClient.EUREKA_INSTANCE_HOSTNAME, None)
        heartbeat_interval = self.app.config.get(EurekaClient.EUREKA_HEARTBEAT_INTERVAL, None)
        service_path = self.app.config.get(EurekaClient.EUREKA_SERVICE_PATH, None)
        port = self.app.config.get(EurekaClient.EUREKA_INSTANCE_PORT, self._get_service_port())

        eureka_client = EurekaClient(name=name, host_name=host_name,
                                     eureka_url=eureka_url,
                                     data_center=data_center,
                                     heartbeat_interval=heartbeat_interval,
                                     service_path=service_path,
                                     port=port,
                                     **kwargs)
        eureka_client.star()

    def _get_service_port(self):
        """
        Retrieve the service port being used by the flask application
        """
        port = 5000
        server_name = self.app.config.get('SERVER_NAME', None)
        if server_name and ':' in server_name:
            port = int(server_name.rsplit(':', 1)[1])
        return port