diff --git a/cmd/manager/Dockerfile b/cmd/manager/Dockerfile index a25a0310d07..9a03b257645 100644 --- 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 index 1a50b8997b1..f42c4b84ed2 100644 --- 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