feat: Healthz api check appId support

related: dapr/components-contrib#2489

Signed-off-by: AlbertHuang <[email protected]>

pkg/http/api.go

@@ -2198,7 +2198,13 @@ func (a *api) onGetHealthz(reqCtx *fasthttp.RequestCtx) {
 		respond(reqCtx, withError(fasthttp.StatusInternalServerError, msg))
 		log.Debug(msg)
 	} else {
-		respond(reqCtx, withEmpty())
+		matchAppID := string(reqCtx.QueryArgs().Peek("app-id"))
+		if matchAppID != "" && matchAppID != a.id {
+			msg := NewErrorResponse("ERR_HEALTH_APPID_NOT_MATCH", messages.ErrHealthAppIDNotMatch)
+			respond(reqCtx, withError(fasthttp.StatusInternalServerError, msg))
+		} else {
+			respond(reqCtx, withEmpty())
+		}
 	}
 }
 

pkg/http/api_test.go

@@ -4998,8 +4998,10 @@ func (l *fakeWorkflowComponent) GetComponentMetadata() map[string]string {
 func TestV1HealthzEndpoint(t *testing.T) {
 	fakeServer := newFakeHTTPServer()
 
+	appId := "test"
 	testAPI := &api{
 		actor: nil,
+		id:    appId,
 	}
 
 	fakeServer.StartServer(testAPI.constructHealthzEndpoints())
@@ -5019,6 +5021,20 @@ func TestV1HealthzEndpoint(t *testing.T) {
 		assert.Equal(t, 204, resp.StatusCode)
 	})
 
+	t.Run("Healthz - 500 No AppId Match", func(t *testing.T) {
+		apiPath := "v1.0/healthz"
+		testAPI.MarkStatusAsReady()
+		resp := fakeServer.DoRequest("GET", apiPath, nil, map[string]string{"app-id": "not-test"})
+		assert.Equal(t, 500, resp.StatusCode)
+	})
+
+	t.Run("Healthz - 204 AppId Match", func(t *testing.T) {
+		apiPath := "v1.0/healthz"
+		testAPI.MarkStatusAsReady()
+		resp := fakeServer.DoRequest("GET", apiPath, nil, map[string]string{"app-id": appId})
+		assert.Equal(t, 204, resp.StatusCode)
+	})
+
 	fakeServer.Shutdown()
 }
 

pkg/messages/predefined.go

@@ -84,6 +84,7 @@ const (
 	// Healthz.
 	ErrHealthNotReady         = "dapr is not ready"
 	ErrOutboundHealthNotReady = "dapr outbound is not ready"
+	ErrHealthAppIDNotMatch    = "dapr appId is not match"
 
 	// Configuration.
 	ErrConfigurationStoresNotConfigured = "configuration stores not configured"