conditionals: add context support for routing conditions

This patch extends the conditionals interface to support context
variables for conditional routing.

Signed-off-by: Eduardo Silva <[email protected]>

Author: edsiper

include/fluent-bit/flb_conditionals.h

@@ -28,10 +28,20 @@
 #include <monkey/mk_core.h>
 #include <fluent-bit/flb_mp_chunk.h>
 
+struct flb_condition_rule;
+
+typedef struct cfl_variant *(*flb_condition_get_variant_fn)(struct flb_condition_rule *rule,
+                                                            void *ctx);
+
 /* Context types enum */
 enum record_context_type {
-    RECORD_CONTEXT_BODY    = 0,
-    RECORD_CONTEXT_METADATA    = 1
+    RECORD_CONTEXT_BODY = 0,
+    RECORD_CONTEXT_METADATA = 1,
+    RECORD_CONTEXT_GROUP_METADATA,
+    RECORD_CONTEXT_GROUP_ATTRIBUTES,
+    RECORD_CONTEXT_OTEL_RESOURCE_ATTRIBUTES,
+    RECORD_CONTEXT_OTEL_SCOPE_ATTRIBUTES,
+    RECORD_CONTEXT_OTEL_SCOPE_METADATA
 };
 
 struct flb_condition;
@@ -88,6 +98,9 @@ int flb_condition_add_rule(struct flb_condition *cond,
 void flb_condition_destroy(struct flb_condition *cond);
 
 /* Evaluation function */
+int flb_condition_evaluate_ex(struct flb_condition *cond,
+                             void *ctx,
+                             flb_condition_get_variant_fn get_variant);
 int flb_condition_evaluate(struct flb_condition *cond,
                           struct flb_mp_chunk_record *record);
 

src/flb_conditionals.c

@@ -25,15 +25,16 @@
 #include <fluent-bit/flb_mp.h>
 #include <fluent-bit/flb_conditionals.h>
 
-/* Function to get the record variant based on context */
-static inline struct cfl_variant *get_record_variant(struct flb_mp_chunk_record *record,
-                                                    enum record_context_type context_type)
+static struct cfl_variant *default_get_record_variant(struct flb_condition_rule *rule,
+                                                      void *ctx)
 {
-    if (!record) {
+    struct flb_mp_chunk_record *record = (struct flb_mp_chunk_record *) ctx;
+
+    if (!record || !rule) {
         return NULL;
     }
 
-    switch (context_type) {
+    switch (rule->context) {
     case RECORD_CONTEXT_METADATA:
         if (record->cobj_metadata) {
             return record->cobj_metadata->variant;
@@ -45,6 +46,9 @@ static inline struct cfl_variant *get_record_variant(struct flb_mp_chunk_record
             return record->cobj_record->variant;
         }
         break;
+
+    default:
+        break;
     }
 
     return NULL;
@@ -356,8 +360,9 @@ static int evaluate_rule(struct flb_condition_rule *rule,
     return result;
 }
 
-int flb_condition_evaluate(struct flb_condition *cond,
-                           struct flb_mp_chunk_record *record)
+int flb_condition_evaluate_ex(struct flb_condition *cond,
+                              void *ctx,
+                              flb_condition_get_variant_fn get_variant)
 {
     struct mk_list *head;
     struct flb_condition_rule *rule;
@@ -366,11 +371,16 @@ int flb_condition_evaluate(struct flb_condition *cond,
     int any_rule_evaluated = FLB_FALSE;
     int any_rule_matched = FLB_FALSE;
 
-    if (!cond || !record) {
-        flb_trace("[condition] NULL condition or record, returning TRUE");
+    if (!cond) {
+        flb_trace("[condition] NULL condition, returning TRUE");
         return FLB_TRUE;
     }
 
+    if (!get_variant) {
+        flb_trace("[condition] missing variant provider, returning FALSE");
+        return FLB_FALSE;
+    }
+
     flb_trace("[condition] evaluating condition with %d rules", mk_list_size(&cond->rules));
 
     if (mk_list_size(&cond->rules) == 0) {
@@ -382,8 +392,7 @@ int flb_condition_evaluate(struct flb_condition *cond,
         rule = mk_list_entry(head, struct flb_condition_rule, _head);
         flb_trace("[condition] processing rule with op=%d", rule->op);
 
-        /* Get the variant for this rule's context */
-        record_variant = get_record_variant(record, rule->context);
+        record_variant = get_variant(rule, ctx);
         any_rule_evaluated = FLB_TRUE;
         if (!record_variant) {
             flb_trace("[condition] no record variant found for context %d", rule->context);
@@ -427,3 +436,13 @@ int flb_condition_evaluate(struct flb_condition *cond,
     flb_trace("[condition] final evaluation result: TRUE");
     return FLB_TRUE;
 }
+
+int flb_condition_evaluate(struct flb_condition *cond,
+                           struct flb_mp_chunk_record *record)
+{
+    if (!cond || !record) {
+        return FLB_TRUE;
+    }
+
+    return flb_condition_evaluate_ex(cond, record, default_get_record_variant);
+}