add health check support for ArgoCD Application

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

Author: ksdpmx

pkg/health/health.go

@@ -120,6 +120,8 @@ func GetHealthCheckFunc(gvk schema.GroupVersionKind) func(obj *unstructured.Unst
 		switch gvk.Kind {
 		case "Workflow":
 			return getArgoWorkflowHealth
+		case "Application":
+			return getArgoApplicationHealth
 		}
 	case "apiregistration.k8s.io":
 		switch gvk.Kind {

pkg/health/health_argo.go

@@ -26,6 +26,15 @@ type argoWorkflow struct {
 	}
 }
 
+type argoApplication struct {
+	Status struct {
+		Health struct {
+			Status  HealthStatusCode
+			Message string
+		}
+	}
+}
+
 func getArgoWorkflowHealth(obj *unstructured.Unstructured) (*HealthStatus, error) {
 	var wf argoWorkflow
 	err := runtime.DefaultUnstructuredConverter.FromUnstructured(obj.Object, &wf)
@@ -42,3 +51,12 @@ func getArgoWorkflowHealth(obj *unstructured.Unstructured) (*HealthStatus, error
 	}
 	return &HealthStatus{Status: HealthStatusUnknown, Message: wf.Status.Message}, nil
 }
+
+func getArgoApplicationHealth(obj *unstructured.Unstructured) (*HealthStatus, error) {
+	var app argoApplication
+	err := runtime.DefaultUnstructuredConverter.FromUnstructured(obj.Object, &app)
+	if err != nil {
+		return nil, err
+	}
+	return &HealthStatus{Status: app.Status.Health.Status, Message: app.Status.Health.Message}, nil
+}

pkg/health/health_test.go

@@ -167,3 +167,8 @@ func TestGetArgoWorkflowHealth(t *testing.T) {
 	assert.Equal(t, "", health.Message)
 
 }
+
+func TestGetArgoApplicationHealth(t *testing.T) {
+	assertAppHealth(t, "./testdata/application-healthy.yaml", HealthStatusHealthy)
+	assertAppHealth(t, "./testdata/application-degraded.yaml", HealthStatusDegraded)
+}