Add /healthz and /info endpoints

bundlegen/rabbitmq/app.py

@@ -0,0 +1,13 @@
+from flask import Flask
+from info import Info
+
+app = Flask(__name__)
+app_info = Info()
+
+@app.route("/healthz")
+def healthz() -> str:
+    return 'OK'
+
+@app.route('/info')
+def info():
+    return app_info.get()

bundlegen/rabbitmq/info.py

@@ -0,0 +1,29 @@
+from datetime import datetime
+import os
+import socket
+
+class Info:
+    """Get information about service"""
+
+    def __init__(self) -> None:
+        """Read info"""
+        self.app_start_time = datetime.utcnow()
+        self.host_name = socket.gethostname()
+        self.app_branch = os.getenv("APP_BRANCH", "undefined")
+        self.app_name = os.getenv("APP_NAME", "undefined")
+        self.app_build_time = os.getenv("APP_BUILD_TIME", "undefined")
+        self.app_version = os.getenv("APP_VERSION", "undefined")
+        self.stack_name = os.getenv("STACK_NAME", "undefined")
+        self.app_revision = os.getenv("APP_REVISION", "undefined")
+
+    def get(self) -> dict:
+        info = {}
+        info['APP_START_TIME'] = self.app_start_time.strftime('%Y-%m-%dT%H:%M:%SZ')
+        info['HOST_NAME'] = self.host_name
+        info['APP_BRANCH'] = self.app_branch
+        info['APP_NAME'] = self.app_name
+        info['APP_BUILD_TIME'] = self.app_build_time
+        info['APP_VERSION'] = self.app_version
+        info['STACK_NAME'] = self.stack_name
+        info['APP_REVISION'] = self.app_revision
+        return info

docker/rabbitmq/Dockerfile

@@ -53,4 +53,4 @@ RUN cd ./bundlegen && \
 WORKDIR /home/bundlegenuser/bundlegen
 
 # Start BundleGen consumer
-ENTRYPOINT [ "bundlegen-rabbitmq", "-vv", "start" ]
+ENTRYPOINT [ "./entrypoint.sh" ]

entrypoint.sh

@@ -0,0 +1,7 @@
+#!/usr/bin/env bash
+set -e
+
+echo "Starting Gunicorn with Flask application"
+gunicorn --bind 0.0.0.0:8081 --chdir /home/bundlegenuser/bundlegen/bundlegen/rabbitmq app:app --daemon
+
+bundlegen-rabbitmq -vv start

requirements.txt

@@ -1,8 +1,10 @@
 autopep8==1.5.3
-click==7.1.2
+click>=8.1.3
 humanfriendly==8.2
 loguru==0.5.1
 msgpack==1.0.2
 pika==1.2.0
 python-dotenv==0.18.0
-jsonschema
+jsonschema
+gunicorn>=3.5
+flask~=3.0.0