1 files changed, 56 insertions, 0 deletions
diff --git a/fluent-bit/src/flb_coro.c b/fluent-bit/src/flb_coro.c
new file mode 100644
index 000000000..0d4e67f09
--- /dev/null
+++ b/fluent-bit/src/flb_coro.c
@@ -0,0 +1,56 @@
+/* -*- Mode: C; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
+
+/*  Fluent Bit
+ *  ==========
+ *  Copyright (C) 2015-2022 The Fluent Bit Authors
+ *
+ *  Licensed under the Apache License, Version 2.0 (the "License");
+ *  you may not use this file except in compliance with the License.
+ *  You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ *  Unless required by applicable law or agreed to in writing, software
+ *  distributed under the License is distributed on an "AS IS" BASIS,
+ *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ *  See the License for the specific language governing permissions and
+ *  limitations under the License.
+ */
+
+#include <fluent-bit/flb_info.h>
+#include <fluent-bit/flb_thread_storage.h>
+#include <fluent-bit/flb_coro.h>
+
+FLB_TLS_DEFINE(struct flb_coro, flb_coro_key);
+
+static pthread_mutex_t coro_mutex_init;
+
+void flb_coro_init()
+{
+    FLB_TLS_INIT(flb_coro_key);
+    pthread_mutex_init(&coro_mutex_init, NULL);
+}
+
+void flb_coro_thread_init()
+{
+    size_t s;
+    cothread_t th;
+
+    pthread_mutex_lock(&coro_mutex_init);
+    th = co_create(256, NULL, &s);
+    co_delete(th);
+    pthread_mutex_unlock(&coro_mutex_init);
+}
+
+struct flb_coro *flb_coro_get()
+{
+    struct flb_coro *coro;
+
+    coro = FLB_TLS_GET(flb_coro_key);
+    return coro;
+}
+
+void flb_coro_set(struct flb_coro *coro)
+{
+    FLB_TLS_SET(flb_coro_key, coro);
+}