engine: disable coroutines when invoking custom events.

Signed-off-by: Phillip Whelan <phillip.whelan@chronosphere.io>
fluent · Sep 3, 2024 · 18653cc · 18653cc
1 parent 0627a6f
commit 18653cc
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/flb_engine.c b/src/flb_engine.c
@@ -692,6 +692,7 @@ int flb_engine_start(struct flb_config *config)
     struct flb_bucket_queue *evl_bktq;
     struct flb_sched *sched;
     struct flb_net_dns dns_ctx;
+    struct flb_coro *prev_coro = NULL;
 
     /* Initialize the networking layer */
     flb_net_lib_init();
@@ -1038,7 +1039,14 @@ int flb_engine_start(struct flb_config *config)
                 flb_coro_resume(output_flush->coro);
             }
             else if (event->type == FLB_ENGINE_EV_CUSTOM) {
+                prev_coro = flb_coro_get();
+                if (prev_coro) {
+                    flb_coro_set(NULL);
+                }
                 event->handler(event);
+                if (prev_coro) {
+                    flb_coro_set(prev_coro);
+                }
             }
             else if (event->type == FLB_ENGINE_EV_THREAD) {
                 struct flb_connection *connection;