From bd93b2d2aa47908c894d4bb986f1cb11258bf9a9 Mon Sep 17 00:00:00 2001
From: Sergey Zhuk <sergey.zhuk@heartex.com>
Date: Mon, 19 Jun 2023 10:50:50 +0100
Subject: [PATCH] ci: fb-DEV-2304: add healthcheck for rqworker

---
 heartex/label-studio/Chart.yaml                         | 2 +-
 heartex/label-studio/templates/rqworker-deployment.yaml | 8 +++++---
 heartex/label-studio/values.yaml                        | 3 +--
 3 files changed, 7 insertions(+), 6 deletions(-)

diff --git a/heartex/label-studio/Chart.yaml b/heartex/label-studio/Chart.yaml
index 0fe1be0..9ee552c 100644
--- a/heartex/label-studio/Chart.yaml
+++ b/heartex/label-studio/Chart.yaml
@@ -5,7 +5,7 @@ home: https://labelstud.io/
 type: application
 icon: https://raw.githubusercontent.com/heartexlabs/label-studio/master/images/logo.png
 # Chart version
-version: 1.1.4
+version: 1.1.5
 # Label Studio release version
 appVersion: "1.8.0"
 kubeVersion: ">= 1.14.0-0"
diff --git a/heartex/label-studio/templates/rqworker-deployment.yaml b/heartex/label-studio/templates/rqworker-deployment.yaml
index 7d85a13..95f5113 100644
--- a/heartex/label-studio/templates/rqworker-deployment.yaml
+++ b/heartex/label-studio/templates/rqworker-deployment.yaml
@@ -90,9 +90,11 @@ spec:
           {{- end }}
           {{- if $.Values.rqworker.livenessProbe.enabled }}
           livenessProbe:
-            httpGet:
-              path: {{ $.Values.rqworker.livenessProbe.path | quote }}
-              port: 8000
+            exec:
+              command:
+                - python3
+                - /label-studio-enterprise/label_studio_enterprise/check_rqworker.py
+                - rqworker_selfcheck
             failureThreshold: {{ $.Values.rqworker.livenessProbe.failureThreshold }}
             initialDelaySeconds: {{ $.Values.rqworker.livenessProbe.initialDelaySeconds }}
             periodSeconds: {{ $.Values.rqworker.livenessProbe.periodSeconds }}
diff --git a/heartex/label-studio/values.yaml b/heartex/label-studio/values.yaml
index 1b4dcf5..99919e9 100644
--- a/heartex/label-studio/values.yaml
+++ b/heartex/label-studio/values.yaml
@@ -478,8 +478,7 @@ rqworker:
     timeoutSeconds: 3
   # Used to enable a livenessProbe for the pods
   livenessProbe:
-    enabled: false
-    path: "/health"
+    enabled: true
     # When a probe fails, Kubernetes will try failureThreshold times before giving up
     failureThreshold: 2
     # Number of seconds after the container has started before probe initiates