fix: postgres failures

Signed-off-by: Tanisha goyal <[email protected]>

Author: tanishagoyal2

data-models/pkg/protos/health_event.pb.go

@@ -33,8 +33,9 @@ message HealthEvents {
 // EXECUTE_REMEDIATION: normal behavior; downstream modules may update cluster state.
 // STORE_ONLY: observability-only behavior; event should be persisted/exported but should not modify cluster resources.
 enum ProcessingStrategy {
-  EXECUTE_REMEDIATION = 0;
-  STORE_ONLY = 1;
+  UNSPECIFIED = 0;
+  EXECUTE_REMEDIATION = 1;
+  STORE_ONLY = 2;
 }
 
 enum RecommendedAction {

data-models/protobufs/health_event.proto

@@ -359,6 +359,9 @@ func (e *Engine) mapMaintenanceEventToHealthEvent(
 		Metadata:           event.Metadata, // Pass along metadata
 		NodeName:           event.NodeName, // K8s node name
 		GeneratedTimestamp: timestamppb.New(time.Now()),
+		// TODO: Remove hardcoded processing strategy and make it configurable via the config file.
+		// PR: https://github.com/NVIDIA/NVSentinel/pull/641
+		ProcessingStrategy: pb.ProcessingStrategy_EXECUTE_REMEDIATION,
 	}
 
 	return healthEvent, nil

health-monitors/csp-health-monitor/pkg/triggerengine/trigger.go

@@ -240,8 +240,9 @@ func TestMapMaintenanceEventToHealthEvent(t *testing.T) {
 				EntitiesImpacted: []*pb.Entity{
 					{EntityType: "gce_instance", EntityValue: "instance-123"},
 				},
-				Metadata: map[string]string{"key": "value"},
-				NodeName: "node-a",
+				Metadata:           map[string]string{"key": "value"},
+				NodeName:           "node-a",
+				ProcessingStrategy: pb.ProcessingStrategy_EXECUTE_REMEDIATION,
 			},
 		},
 		{
@@ -268,7 +269,8 @@ func TestMapMaintenanceEventToHealthEvent(t *testing.T) {
 				EntitiesImpacted: []*pb.Entity{
 					{EntityType: "EC2", EntityValue: "i-abcdef"},
 				},
-				NodeName: "node-b",
+				NodeName:           "node-b",
+				ProcessingStrategy: pb.ProcessingStrategy_EXECUTE_REMEDIATION,
 			},
 		},
 		{
@@ -331,7 +333,8 @@ func TestMapMaintenanceEventToHealthEvent(t *testing.T) {
 				EntitiesImpacted: []*pb.Entity{
 					{EntityType: "gce_instance", EntityValue: "instance-789"},
 				},
-				NodeName: "node-e",
+				NodeName:           "node-e",
+				ProcessingStrategy: pb.ProcessingStrategy_EXECUTE_REMEDIATION,
 			},
 		},
 	}

health-monitors/csp-health-monitor/pkg/triggerengine/trigger_test.go

@@ -13,6 +13,7 @@ DESCRIPTOR: _descriptor.FileDescriptor
 
 class ProcessingStrategy(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
     __slots__ = ()
+    UNSPECIFIED: _ClassVar[ProcessingStrategy]
     EXECUTE_REMEDIATION: _ClassVar[ProcessingStrategy]
     STORE_ONLY: _ClassVar[ProcessingStrategy]
 
@@ -28,6 +29,7 @@ class RecommendedAction(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
     RUN_DCGMEUD: _ClassVar[RecommendedAction]
     UNKNOWN: _ClassVar[RecommendedAction]
 
+UNSPECIFIED: ProcessingStrategy
 EXECUTE_REMEDIATION: ProcessingStrategy
 STORE_ONLY: ProcessingStrategy
 NONE: RecommendedAction

health-monitors/gpu-health-monitor/gpu_health_monitor/protos/health_event_pb2.py

@@ -401,6 +401,7 @@ var fieldNameMapping = map[string]string{
 	"faultremediated":        "faultremediated",
 	"status":                 "status",
 	"message":                "message",
+	"processingstrategy":     "processingStrategy",
 }
 
 // isOperationTypeField checks if the field is the operationType field.

health-monitors/gpu-health-monitor/gpu_health_monitor/protos/health_event_pb2.pyi

@@ -16,5 +16,6 @@
             "entityValue": "0"
         }
     ],
-    "nodeName": "NODE_NAME"
+    "nodeName": "NODE_NAME",
+    "processingStrategy": 1
 }

store-client/pkg/datastore/providers/postgresql/sql_filter_builder.go

@@ -16,5 +16,6 @@
             "entityValue": "0"
         }
     ],
-    "nodeName": "NODE_NAME"
+    "nodeName": "NODE_NAME",
+    "processingStrategy": 1
 }

tests/data/fatal-health-event.json

@@ -16,5 +16,6 @@
             "entityValue": "0"
         }
     ],
-    "nodeName": "NODE_NAME"
+    "nodeName": "NODE_NAME",
+    "processingStrategy": 1
 }