107 lines
3.2 KiB
C++
107 lines
3.2 KiB
C++
// a thread manager for iter-thread communicatios, etc.
|
||
|
||
#ifndef BASE_THREAD_THREAD_MANAGER_H_
|
||
#define BASE_THREAD_THREAD_MANAGER_H_
|
||
|
||
#include <map>
|
||
#include <memory>
|
||
#include "base/base_export.h"
|
||
#include "base/thread/framework_thread.h"
|
||
#include "base/memory/singleton.h"
|
||
|
||
/**
|
||
* 一些预定义的线程 ID 值，提供开发者使用
|
||
*/
|
||
enum ThreadIds
|
||
{
|
||
kThreadMain,
|
||
kThreadDatabase,
|
||
kThreadGlobalMisc,
|
||
};
|
||
|
||
namespace nbase
|
||
{
|
||
class MessageLoop;
|
||
class MessageLoopProxy;
|
||
|
||
class ThreadMap
|
||
{
|
||
public:
|
||
SINGLETON_DEFINE(ThreadMap);
|
||
static bool AquireAccess();
|
||
|
||
bool RegisterThread(int self_identifier);
|
||
bool UnregisterThread();
|
||
int QueryThreadId(const FrameworkThread *thread);
|
||
std::shared_ptr<MessageLoopProxy> GetMessageLoop(int identifier) const;
|
||
FrameworkThread* QueryThreadInternal(int identifier) const;
|
||
|
||
private:
|
||
ThreadMap() { }
|
||
NLock lock_;
|
||
std::map<int, FrameworkThread*> threads_;
|
||
};
|
||
|
||
// 使用ThreadManager可以极大地方便线程间通信
|
||
// 注意：只有受ThreadManager托管的线程（通过Register托管）才允许调用除Register和Post族外的成员函数
|
||
class BASE_EXPORT ThreadManager
|
||
{
|
||
public:
|
||
|
||
// 托管当前FrameworkThread线程
|
||
// identifier >= 0
|
||
// 必须在self的线程过程内被调用
|
||
static bool RegisterThread(int self_identifier);
|
||
// 取消当前线程托管
|
||
// 线程运行结束之前必须调用UnregisterThread取消托管
|
||
static bool UnregisterThread();
|
||
|
||
// 查找
|
||
static FrameworkThread* CurrentThread();
|
||
template<typename T> static T* CurrentThreadT();
|
||
static int QueryThreadId(const FrameworkThread *thread);
|
||
|
||
static bool PostTask(const StdClosure &task);
|
||
static bool PostTask(int identifier, const StdClosure &task);
|
||
|
||
static bool PostDelayedTask(const StdClosure &task, TimeDelta delay);
|
||
static bool PostDelayedTask(int identifier, const StdClosure &task, TimeDelta delay);
|
||
|
||
static const int TIMES_FOREVER = -1;
|
||
static void PostRepeatedTask(const WeakCallback<StdClosure>& task, const TimeDelta& delay, int times = TIMES_FOREVER);
|
||
static void PostRepeatedTask(int thread_id, const WeakCallback<StdClosure>& task, const TimeDelta& delay, int times = TIMES_FOREVER);
|
||
|
||
static bool PostNonNestableTask(const StdClosure &task);
|
||
static bool PostNonNestableTask(int identifier, const StdClosure &task);
|
||
|
||
static bool PostNonNestableDelayedTask(const StdClosure &task, TimeDelta delay);
|
||
static bool PostNonNestableDelayedTask(int identifier, const StdClosure &task, TimeDelta delay);
|
||
|
||
template<typename T1, typename T2>
|
||
static bool Await(int identifier, const std::function<T1> &task, const std::function<T2> &reply)
|
||
{
|
||
std::shared_ptr<MessageLoopProxy> message_loop =
|
||
ThreadMap::GetInstance()->GetMessageLoop(identifier);
|
||
if (message_loop == NULL)
|
||
return false;
|
||
message_loop->PostTaskAndReply(task, reply);
|
||
return true;
|
||
}
|
||
|
||
private:
|
||
static void RunRepeatedly(const WeakCallback<StdClosure>& task, const TimeDelta& delay, int times);
|
||
static void RunRepeatedly2(int thread_id, const WeakCallback<StdClosure>& task, const TimeDelta& delay, int times);
|
||
|
||
DISALLOW_COPY_AND_ASSIGN(ThreadManager);
|
||
};
|
||
|
||
template<typename T>
|
||
T* ThreadManager::CurrentThreadT()
|
||
{
|
||
return static_cast<T *>(CurrentThread());
|
||
}
|
||
|
||
}
|
||
|
||
#endif // BASE_THREAD_THREAD_MANAGER_H_
|