Merge pull request #323 from getsentry/txiao/chore/return-502-from-he…

…alth-endpoint-on-shutdown chore: Return 502 from health endpoint on shutdown
getsentry · Sep 27, 2023 · 3a6587a · 3a6587a
2 parents b248a9d + 2d78a9e
commit 3a6587a
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -15,6 +15,7 @@
 
 - Rename the frame drop issue title. ([#315](https://github.com/getsentry/vroom/pull/315))
 - Add new endpoint for regressed functions. ([#318](https://github.com/getsentry/vroom/pull/318))
+- Return 502 from health endpoint on shutdown. ([#323](https://github.com/getsentry/vroom/pull/323))
 
 ## 23.9.1
 

diff --git a/cmd/vroom/main.go b/cmd/vroom/main.go
@@ -225,7 +225,11 @@ func main() {
 }
 
 func (e *environment) getHealth(w http.ResponseWriter, _ *http.Request) {
-	w.WriteHeader(http.StatusNoContent)
+	if _, err := os.Stat("/tmp/vroom.down"); err == nil {
+		w.WriteHeader(http.StatusNoContent)
+	} else {
+		w.WriteHeader(http.StatusBadGateway)
+	}
 }
 
 type Filter struct {