add version api endpoint to gateway (#1279)

* add version api endpoint to gateway

Signed-off-by: Akihiko Kuroda <[email protected]>

.github/workflows/update-component-versions.yaml

@@ -43,6 +43,7 @@ jobs:
           sed -i "s/tag: \"${OLDNUM}\"/tag: \"${NEWNUM}\"/" charts/quantum-serverless/values.yaml
           sed -i "s/tag: \"${OLDNUM}-py39\"/tag: \"${NEWNUM}-py39\"/" charts/quantum-serverless/values.yaml
           sed -i "s/ray-node:${OLDNUM}/ray-node:${NEWNUM}/" charts/quantum-serverless/values.yaml
+          sed -i "s/version: ${OLDNUM}/version: ${NEWNUM}/" charts/quantum-serverless/values.yaml
           cd charts/quantum-serverless
           helm dependency update
           cd -

charts/quantum-serverless/charts/gateway/templates/deployment.yaml

@@ -84,6 +84,8 @@ spec:
           resources:
             {{- toYaml .Values.resources | nindent 12 }}
           env:
+            - name: VERSION
+              value: {{ .Values.global.version | quote }}
             - name: DEBUG
               value: {{ .Values.application.debug | quote }}
             - name: DJANGO_SECRET_KEY

charts/quantum-serverless/values.yaml

@@ -1,3 +1,9 @@
+# ===================
+# Quantum Serverless Info
+# ===================
+global:
+  version: 0.9.0
+
 # ===================
 # Quantum Serverless configs
 # ===================

gateway/main/settings.py

@@ -17,6 +17,7 @@
 from pathlib import Path
 from utils import sanitize_file_path
 
+RELEASE_VERSION = os.environ.get("VERSION", "UNKNOWN")
 
 # Build paths inside the project like this: BASE_DIR / 'subdir'.
 BASE_DIR = Path(__file__).resolve().parent.parent

gateway/main/urls.py

@@ -22,6 +22,7 @@
 from drf_yasg.views import get_schema_view
 from drf_yasg import openapi
 import probes.views
+import version.views
 
 schema = get_schema_view(  # pylint: disable=invalid-name
     openapi.Info(
@@ -45,6 +46,7 @@
     path("accounts/", include("allauth.urls")),
     path("readiness/", probes.views.readiness, name="readiness"),
     path("liveness/", probes.views.liveness, name="liveness"),
+    path("version/", version.views.version, name="version"),
     path("", include("django_prometheus.urls")),
     re_path(r"^api/v1/", include(("api.v1.urls", "api"), namespace="v1")),
     # docs

gateway/tests/version/test_version.py

@@ -0,0 +1,14 @@
+"""Test version."""
+
+from django.urls import reverse
+from rest_framework import status
+from rest_framework.test import APITestCase
+
+
+class TestProbes(APITestCase):
+    """TestVersion."""
+
+    def test_version(self):
+        """Tests version."""
+        jobs_response = self.client.get(reverse("version"))
+        self.assertEqual(jobs_response.status_code, status.HTTP_200_OK)

gateway/version/views.py

@@ -0,0 +1,10 @@
+from django.http import HttpResponse
+import json
+import os
+import requests
+import main.settings
+
+
+def version(request):
+    info = {"version": main.settings.RELEASE_VERSION}
+    return HttpResponse(json.dumps(info))