acl/lib_fiber/cpp/include/fiber/go_fiber.hpp

135 lines
2.2 KiB
C++
Raw Normal View History

2020-06-13 14:33:08 +08:00
#pragma once
#include "fiber_cpp_define.hpp"
#include <thread>
2020-06-13 14:49:31 +08:00
#include <functional>
2020-06-13 14:33:08 +08:00
#include "fiber.hpp"
#include "fiber_tbox.hpp"
struct ACL_FIBER;
namespace acl {
class FIBER_CPP_API fiber_ctx
{
public:
fiber_ctx(std::function<void()> fn) : fn_(fn) {}
~fiber_ctx() = default;
std::function<void()> fn_;
};
#define go acl::go_fiber()>
#define go_stack(size) acl::go_fiber(size)>
#define go_wait_fiber acl::go_fiber()<
#define go_wait_thread acl::go_fiber()<<
#define go_wait go_wait_thread
class FIBER_CPP_API go_fiber
{
public:
go_fiber(void) {}
go_fiber(size_t stack_size) : stack_size_(stack_size) {}
void operator > (std::function<void()> fn)
{
fiber_ctx* ctx = new fiber_ctx(fn);
fiber::fiber_create(fiber_main, (void*) ctx, stack_size_);
}
void operator < (std::function<void()> fn)
{
fiber_tbox<int> box;
go[&] {
fn();
box.push(NULL);
};
(void) box.pop();
}
void operator << (std::function<void()> fn)
{
fiber_tbox<int> box;
std::thread thread([&]() {
fn();
box.push(NULL);
});
thread.detach();
(void) box.pop();
}
private:
size_t stack_size_ = 320000;
static void fiber_main(ACL_FIBER*, void* ctx)
{
fiber_ctx* fc = (fiber_ctx *) ctx;
std::function<void()> fn = fc->fn_;
delete fc;
fn();
}
};
} // namespace acl
/**
* static void fiber1(void)
* {
* printf("fiber: %d\r\n", acl::fiber::self());
* }
*
* static void fiber2(acl::string& buf)
* {
* printf("in fiber: %d, buf: %s\r\n", acl::fiber::self(), buf.c_str());
* buf = "world";
* }
*
* static void fiber3(const acl::string& buf)
* {
* printf("in fiber: %d, buf: %s\r\n", acl::fiber::self(), buf.c_str());
* }
*
* static void incr(int& n)
* {
* n++;
* }
*
* static void waiter(void)
* {
* int n = 100;
*
* // run in thread and wait for result
* go_wait_thread[&] { incr(n); };
* // here: n should be 101
*
* n = 200;
*
* // run in fiber and wait for result
* go_wait_fiber[&] { incr(n); };
* // here: n should be 201
* }
*
* static test(void)
* {
* go fiber1;
*
* acl::string buf("hello");
*
* go[&] {
* fiber2(buf);
* };
*
* go[=] {
* fiber3(buf);
* };
*
* go[&] {
* fiber3(buf);
* };
* }
*/