Add readiness/liveness probes to vizier-core

Signed-off-by: Koichiro Den <[email protected]>
kubeflow · Dec 4, 2018 · 2fea722 · 2fea722
1 parent ee1a1a2
commit 2fea722
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/cmd/manager/Dockerfile b/cmd/manager/Dockerfile
@@ -10,3 +10,6 @@ COPY --from=build-env /go/src/github.com/kubeflow/katib/cmd/manager/vizier-manag
 COPY --from=build-env /go/src/github.com/kubeflow/katib/pkg/manager/visualise /
 ENTRYPOINT ["./vizier-manager"]
 CMD ["-w", "kubernetes"]
+RUN GRPC_HEALTH_PROBE_VERSION=v0.2.0 && \
+    wget -qO/bin/grpc_health_probe https://github.com/grpc-ecosystem/grpc-health-probe/releases/download/${GRPC_HEALTH_PROBE_VERSION}/grpc_health_probe-linux-amd64 && \
+    chmod +x /bin/grpc_health_probe
diff --git a/manifests/vizier/core/deployment.yaml b/manifests/vizier/core/deployment.yaml
@@ -30,6 +30,14 @@ spec:
         ports:
         - name: api
           containerPort: 6789
+        readinessProbe:
+          exec:
+            command: ["/bin/grpc_health_probe", "-addr=:6789"]
+          initialDelaySeconds: 5
+        livenessProbe:
+          exec:
+            command: ["/bin/grpc_health_probe", "-addr=:6789"]
+          initialDelaySeconds: 10
 #        resources:
 #          requests:
 #            cpu: 500m