Viewing file:  swoole_coroutine_context.h (2.43 KB) -rw-r--r--
Select action/file-type:
 (+) |  (+) |  (+) | Code (+) | Session (+) |  (+) | SDB (+) |  (+) |  (+) |  (+) |  (+) |  (+) |
/*
+----------------------------------------------------------------------+
| Swoole |
+----------------------------------------------------------------------+
| This source file is subject to version 2.0 of the Apache license, |
| that is bundled with this package in the file LICENSE, and is |
| available through the world-wide-web at the following url: |
| http://www.apache.org/licenses/LICENSE-2.0.html |
| If you did not receive a copy of the Apache2.0 license and are unable|
| to obtain it through the world-wide-web, please send a note to |
| license@swoole.com so we can mail you a copy immediately. |
+----------------------------------------------------------------------+
| Author: Tianfeng Han <mikan.tenny@gmail.com> |
| Twosee <twose@qq.com> |
+----------------------------------------------------------------------+
*/
#pragma once
#include "swoole.h"
#ifdef SW_USE_THREAD_CONTEXT
#include <thread>
#include <mutex>
#elif !defined(SW_USE_ASM_CONTEXT)
#define USE_UCONTEXT 1
#ifndef _XOPEN_SOURCE
#define _XOPEN_SOURCE
#endif
#include <ucontext.h>
#else
#define USE_ASM_CONTEXT 1
#include "swoole_asm_context.h"
#endif
#if defined(HAVE_VALGRIND) && !defined(HAVE_KQUEUE)
#define USE_VALGRIND 1
#include <valgrind/valgrind.h>
#endif
#ifdef USE_UCONTEXT
typedef ucontext_t coroutine_context_t;
#elif defined(USE_ASM_CONTEXT)
typedef fcontext_t coroutine_context_t;
#endif
typedef std::function<void(void *)> CoroutineFunc;
namespace swoole {
namespace coroutine {
class Context {
public:
Context(size_t stack_size, const CoroutineFunc &fn, void *private_data);
~Context();
bool swap_in();
bool swap_out();
#if !defined(SW_USE_THREAD_CONTEXT) && defined(SW_CONTEXT_DETECT_STACK_USAGE)
ssize_t get_stack_usage();
#endif
inline bool is_end() {
return end_;
}
protected:
CoroutineFunc fn_;
#ifdef SW_USE_THREAD_CONTEXT
std::thread thread_;
std::mutex lock_;
std::mutex *swap_lock_;
#else
coroutine_context_t ctx_;
coroutine_context_t swap_ctx_;
char *stack_;
uint32_t stack_size_;
#endif
#ifdef USE_VALGRIND
uint32_t valgrind_stack_id;
#endif
void *private_data_;
bool end_;
static void context_func(void *arg);
};
} // namespace coroutine
} // namespace swoole
|