Add / handler for health checks

viamrobotics · Nov 5, 2024 · 596b486 · 596b486
1 parent 8aa32a5
commit 596b486
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/robot/web/web.go b/robot/web/web.go
@@ -688,7 +688,11 @@ func (svc *webService) initHTTPServer(listenerTCPAddr *net.TCPAddr, options webo
 // Initialize multiplexer between http handlers.
 func (svc *webService) initMux(options weboptions.Options) *goji.Mux {
 	mux := goji.NewMux()
-	mux.Handle(pat.New("/"), http.NotFoundHandler())
+  // Note: used by viam-agent for health checks
+	mux.HandleFunc(pat.New("/"), func(w http.ResponseWriter, _ *http.Request) {
+    w.WriteHeader(200)
+    w.Write([]byte("healthy"))
+	})
 
 	if options.Pprof {
 		mux.HandleFunc(pat.New("/debug/pprof/"), pprof.Index)