Merge pull request #394 from jonathan-chin/main

added script to allow healthcheck call in docker-compose

README.md

@@ -170,6 +170,15 @@ Run this version with:
 docker-compose -f docker-compose.cuda.yml up -d --build
 ```
 
+### Running Healthchecks
+
+You can enable healthchecks by including this in your docker-compose.yml file:
+
+```
+healthcheck:
+  test: ['CMD-SHELL', './venv/bin/python healthcheck.py']
+```
+
 ## Arguments
 
 | Argument                    | Description                                                                                                 | Default              | Env. name                    |

docker-compose.yml

@@ -9,6 +9,8 @@ services:
     restart: unless-stopped
     ports:
       - "5000:5000"
+    healthcheck:
+      test: ['CMD-SHELL', './venv/bin/python healthcheck.py']
     ## Uncomment above command and define your args if necessary
     # command: --ssl --ga-id MY-GA-ID --req-limit 100 --char-limit 500
     ## Uncomment this section and the `volumes` section if you want to backup your API keys

healthcheck.py

@@ -0,0 +1,11 @@
+import requests
+response = requests.post(
+    url='http://0.0.0.0:5000/translate',
+    headers={'Content-Type': 'application/json'},
+    json={
+         'q': 'Hello World!',
+         'source': 'en',
+         'target': 'en'
+    }
+)
+# if server unavailable then requests with raise exception and healthcheck will fail