added startup_probe

BESTSELLER · Mar 12, 2024 · 5e858fd · 5e858fd
1 parent a787c56
commit 5e858fd
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/api/router.go b/api/router.go
@@ -2,6 +2,7 @@ package api
 
 import (
  "fmt"
+ "net/http"
  "os"
  "time"
 
@@ -47,6 +48,10 @@ func SetupRouter(webhookEnabled bool, workerEnabled bool, controllerEnabled bool
  server.Mux().Handle(pat.Post("/"), webhookHandler)
  }
 
+ server.Mux().HandleFunc(pat.Get("/healthz"), func(w http.ResponseWriter, r *http.Request) {
+ w.WriteHeader(http.StatusOK)
+ })
+
  if workerEnabled {
  server.Mux().HandleFunc(pat.Post("/start"), dependencyHandler)
  }

diff --git a/terraform/modules/cloud_run/main.tf b/terraform/modules/cloud_run/main.tf
@@ -53,6 +53,13 @@ resource "google_cloud_run_v2_service" "main" {
  name = "cloudsql"
  mount_path = "/cloudsql"
  }
+ startup_probe {
+ http_get {
+ path = "/healthz"
+ port = 3000
+ }
+ initial_delay_seconds = 2
+ }
  }
  containers {
  name = "secret-dumper"