blob: 20b5f6d12cd1dde9e11f9bf535251e13f4fca778 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
|
// Copyright (C) by Josh Blum. See LICENSE.txt for licensing information.
#include <gras/thread_pool.hpp>
#include <gras_impl/block_actor.hpp>
#include <Theron/Framework.h>
#include <iostream>
#include <set>
#include <map>
using namespace gras;
/***********************************************************************
* Active framework implementation
**********************************************************************/
static boost::weak_ptr<Theron::Framework> weak_framework;
void ThreadPool::set_active(void)
{
weak_framework = *this;
}
/***********************************************************************
* Map of thread pools to actors - not used externally yet
**********************************************************************/
typedef std::pair<ThreadPool, std::set<BlockActor *> > ThreadPoolMapPair;
typedef std::map<ThreadPool, std::set<BlockActor *> > ThreadPoolMap;
static ThreadPoolMap &get_tpm(void)
{
static ThreadPoolMap tpm;
return tpm;
}
static boost::mutex tpm_mutex;
/***********************************************************************
* Block actor factory - gets active framework
**********************************************************************/
BlockActor *BlockActor::make(const ThreadPool &tp)
{
//thread pool provided, use it
if (tp) return new BlockActor(tp);
//was the thread per block env specified
if (getenv("GRAS_TPP"))
{
ThreadPoolConfig config;
config.thread_count = 1;
return new BlockActor(ThreadPool(config));
}
//otherwise, create/use active pool
else
{
ThreadPool active(weak_framework);
if (not active)
{
active = ThreadPool(ThreadPoolConfig());
active.set_active();
std::cout << "Created default thread pool with " << active->GetNumThreads() << " threads." << std::endl;
}
return new BlockActor(active);
}
}
BlockActor::BlockActor(const ThreadPool &tp):
Theron::Actor(*tp)
{
this->thread_pool = tp;
this->register_handlers();
this->prio_token = Token::make();
//enter the actor into the thread pool map
boost::mutex::scoped_lock lock(tpm_mutex);
get_tpm()[tp].insert(this);
}
BlockActor::~BlockActor(void)
{
const ThreadPool &tp = this->thread_pool;
//clear the actor from the thread pool map
boost::mutex::scoped_lock lock(tpm_mutex);
get_tpm()[tp].erase(this);
if (get_tpm()[tp].empty()) get_tpm().erase(tp);
}
|
