blob: 03f87b1f1083a601a684f2690154405fcb85db0b (
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
|
// Copyright (C) by Josh Blum. See LICENSE.txt for licensing information.
#include <gras/sbuffer.hpp>
#include "alloc_on_node.hpp"
#include <boost/bind.hpp>
using namespace gras;
void gras::sbuffer_handle_deref(SBufferImpl *impl)
{
//call the deleter if possible
boost::shared_ptr<SBufferDeleter> token_deleter = impl->config.token.lock();
if (token_deleter)
{
SBuffer buff;
buff.reset(impl);
(*token_deleter)(buff);
}
else if (impl->config.deleter)
{
SBuffer buff;
buff.reset(impl);
impl->config.deleter(buff);
impl->config.deleter = SBufferDeleter(); //reset deleter, so we dont double delete
}
else
{
delete impl; //its really dead now
}
}
SBufferConfig::SBufferConfig(void)
{
memory = NULL;
length = 0;
affinity = -1;
user_index = ~0;
}
static void numa_mem_deleter(SBuffer &buff)
{
FreeOnNode(buff.get_actual_memory(), buff.get_actual_length());
}
static void default_allocator_deleter(SBuffer &, char *m)
{
delete [] m;
}
static void default_allocator(SBufferConfig &config)
{
if (config.affinity == -1)
{
char *m = new char[config.length + GRAS_MAX_ALIGNMENT - 1];
size_t x = size_t(m) + GRAS_MAX_ALIGNMENT - 1;
x -= x % GRAS_MAX_ALIGNMENT;
config.memory = (void *)x;
config.deleter = boost::bind(&default_allocator_deleter, _1, m);
}
else
{
config.memory = AllocOnNode(config.affinity, config.length);
config.deleter = boost::bind(&numa_mem_deleter, _1);
//deal with numa failue case //TODO print warning message
if (config.memory == NULL)
{
config.affinity = -1;
default_allocator(config);
}
}
}
SBuffer::SBuffer(const SBufferConfig &config):
offset(0),
length(0)
{
this->reset(new SBufferImpl(config));
if (config.memory == NULL)
{
default_allocator((*this)->config);
}
this->length = this->get_actual_length();
}
|
