plugins/lua/ts_lua_coroutine.h

/*
  Licensed to the Apache Software Foundation (ASF) under one
  or more contributor license agreements.  See the NOTICE file
  distributed with this work for additional information
  regarding copyright ownership.  The ASF licenses this file
  to you 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.
*/

#pragma once

#include <stdio.h>

extern "C" {
#include "lua.h"
#include "lualib.h"
#include "lauxlib.h"
} // extern "C"

#include <ts/ts.h>

struct ts_lua_async_item;
using async_clean = int (*)(ts_lua_async_item *item);

/* context stats */
struct ts_lua_ctx_stats {
  TSMutex mutexp;      // mutex for the following stats
  int     gc_kb;       // last recorded gc kbytes
  int     gc_kb_max;   // maximum recorded gc kbytes
  int     threads;     // associated coroutines
  int     threads_max; // max coroutines
};

/* main context*/
struct ts_lua_main_ctx {
  lua_State        *lua;    // basic lua vm, injected
  TSMutex           mutexp; // mutex for lua vm
  int               gref;   // reference for lua vm self, in reg table
  ts_lua_ctx_stats *stats;  // per vm stats
};

/* coroutine */
struct ts_lua_coroutine {
  ts_lua_main_ctx *mctx;
  lua_State       *lua; // derived lua_thread
  int              ref; // reference for lua_thread, in REG Table
};

/* continuation info */
struct ts_lua_cont_info {
  ts_lua_coroutine   routine;
  TSCont             contp;       // continuation for the routine
  TSMutex            mutex;       // mutex for continuation
  ts_lua_async_item *async_chain; // ts_lua_async_item list
};

/* asynchronous item */
struct ts_lua_async_item {
  ts_lua_async_item *next;
  ts_lua_cont_info  *cinfo;

  TSCont contp; // continuation for the async operation
  void  *data;  // private data

  async_clean  cleanup; // cleanup function
  unsigned int deleted : 1;
};

ts_lua_async_item *ts_lua_async_create_item(TSCont cont, async_clean func, void *d, ts_lua_cont_info *ci);
void               ts_lua_release_cont_info(ts_lua_cont_info *ci);