lang/c/msgpack: optimize zone::alloc()

git-svn-id: file:///Users/frsyuki/project/msgpack-git/svn/x@61 5a5092ae-2292-43ba-b2d5-dcab9c1a2731
This commit is contained in:
frsyuki 2009-02-15 09:09:57 +00:00
parent a721837ed0
commit 4d13f614b6
4 changed files with 86 additions and 73 deletions

View File

@ -69,28 +69,28 @@ class dynamic_packer;
struct object_class {
virtual ~object_class() {}
virtual bool isnil() const { return false; }
virtual bool xbool() const { throw cast_error(); }
virtual uint8_t xu8() const { throw cast_error(); }
virtual uint16_t xu16() const { throw cast_error(); }
virtual uint32_t xu32() const { throw cast_error(); }
virtual uint64_t xu64() const { throw cast_error(); }
virtual int8_t xi8() const { throw cast_error(); }
virtual int16_t xi16() const { throw cast_error(); }
virtual int32_t xi32() const { throw cast_error(); }
virtual int64_t xi64() const { throw cast_error(); }
virtual float xfloat() const { throw cast_error(); }
virtual double xdouble() const { throw cast_error(); }
virtual raw xraw() { throw cast_error(); }
virtual array& xarray() { throw cast_error(); }
virtual map& xmap() { throw cast_error(); }
virtual const_raw xraw() const { throw cast_error(); }
virtual bool isnil () const { return false; }
virtual bool xbool () const { throw cast_error(); }
virtual uint8_t xu8 () const { throw cast_error(); }
virtual uint16_t xu16 () const { throw cast_error(); }
virtual uint32_t xu32 () const { throw cast_error(); }
virtual uint64_t xu64 () const { throw cast_error(); }
virtual int8_t xi8 () const { throw cast_error(); }
virtual int16_t xi16 () const { throw cast_error(); }
virtual int32_t xi32 () const { throw cast_error(); }
virtual int64_t xi64 () const { throw cast_error(); }
virtual float xfloat () const { throw cast_error(); }
virtual double xdouble() const { throw cast_error(); }
virtual raw xraw () { throw cast_error(); }
virtual array& xarray () { throw cast_error(); }
virtual map& xmap () { throw cast_error(); }
virtual const_raw xraw () const { throw cast_error(); }
virtual const array& xarray() const { throw cast_error(); }
virtual const map& xmap() const { throw cast_error(); }
virtual const map& xmap () const { throw cast_error(); }
virtual bool operator== (const object_class* x) const { return false; }
bool operator!= (const object_class* x) const { return !(this->operator==(x)); }
virtual bool operator< (const object_class* x) const { throw cast_error(); }
virtual bool operator> (const object_class* x) const { throw cast_error(); }
bool operator!= (const object_class* x) const { return !(this->operator==(x)); }
virtual void pack(dynamic_packer& p) const = 0;
operator bool() const { return xbool(); } // FIXME !isnil();
operator uint8_t() const { return xu8(); }
@ -126,24 +126,24 @@ struct object {
object(object_class* v) : val(v) {}
//object(object_class& v) : val(&v) {}
~object() {}
bool isnil() const { return val->isnil(); }
bool xbool() const { return val->xbool(); }
uint8_t xu8() const { return val->xu8(); }
uint16_t xu16() const { return val->xu16(); }
uint32_t xu32() const { return val->xu32(); }
uint64_t xu64() const { return val->xu64(); }
int8_t xi8() const { return val->xi8(); }
int16_t xi16() const { return val->xi16(); }
int32_t xi32() const { return val->xi32(); }
int64_t xi64() const { return val->xi64(); }
float xfloat() const { return val->xfloat(); }
double xdouble() const { return val->xdouble(); }
raw xraw() { return val->xraw(); }
array& xarray() { return val->xarray(); }
map& xmap() { return val->xmap(); }
const_raw xraw() const { return const_cast<const object_class*>(val)->xraw(); }
bool isnil () const { return val->isnil(); }
bool xbool () const { return val->xbool(); }
uint8_t xu8 () const { return val->xu8(); }
uint16_t xu16 () const { return val->xu16(); }
uint32_t xu32 () const { return val->xu32(); }
uint64_t xu64 () const { return val->xu64(); }
int8_t xi8 () const { return val->xi8(); }
int16_t xi16 () const { return val->xi16(); }
int32_t xi32 () const { return val->xi32(); }
int64_t xi64 () const { return val->xi64(); }
float xfloat () const { return val->xfloat(); }
double xdouble() const { return val->xdouble(); }
raw xraw () { return val->xraw(); }
array& xarray () { return val->xarray(); }
map& xmap () { return val->xmap(); }
const_raw xraw () const { return const_cast<const object_class*>(val)->xraw(); }
const array& xarray() const { return const_cast<const object_class*>(val)->xarray(); }
const map& xmap() const { return const_cast<const object_class*>(val)->xmap(); }
const map& xmap () const { return const_cast<const object_class*>(val)->xmap(); }
bool operator== (object x) const { return val->operator== (x.val); }
bool operator!= (object x) const { return val->operator!= (x.val); }
bool operator< (object x) const { return val->operator< (x.val); }
@ -163,9 +163,9 @@ struct object {
operator raw() { return val->operator raw(); }
operator array&() { return val->operator array&(); }
operator map&() { return val->operator map&(); }
operator raw() const { return val->operator raw(); }
operator array&() const { return val->operator array&(); }
operator map&() const { return val->operator map&(); }
operator const_raw() const { return val->operator const_raw(); }
operator const array&() const { return val->operator const array&(); }
operator const map&() const { return val->operator const map&(); }
const object& inspect(std::ostream& s) const
{ val->inspect(s); return *this; }
private:

View File

@ -21,6 +21,7 @@
namespace msgpack {
struct unpacker::context {
context(zone* z)
{
@ -98,11 +99,13 @@ void unpacker::expand_buffer(size_t len)
else { next_size = UNPACKER_INITIAL_BUFFER_SIZE; }
while(next_size < len + m_used) { next_size *= 2; }
// FIXME realloc?
void* tmp = malloc(next_size);
void* tmp = realloc(m_buffer, next_size);
if(!tmp) { throw std::bad_alloc(); }
memcpy(tmp, m_buffer, m_used);
free(m_buffer);
m_buffer = tmp;
//void* tmp = malloc(next_size);
//if(!tmp) { throw std::bad_alloc(); }
//memcpy(tmp, m_buffer, m_used);
//free(m_buffer);
m_buffer = tmp;
m_free = next_size - m_used;

View File

@ -20,34 +20,33 @@
namespace msgpack {
void* zone::alloc()
// FIXME custom allocator?
void zone::expand_chunk()
{
if(m_pool.size()*ZONE_CHUNK_SIZE <= m_used) {
cell_t* chunk = (cell_t*)malloc(sizeof(cell_t)*ZONE_CHUNK_SIZE);
if(!chunk) { throw std::bad_alloc(); }
try {
m_pool.push_back(chunk);
} catch (...) {
free(chunk);
throw;
}
cell_t* chunk = (cell_t*)malloc(sizeof(cell_t)*ZONE_CHUNK_SIZE);
if(!chunk) { throw std::bad_alloc(); }
try {
m_pool.push_back(chunk);
} catch (...) {
free(chunk);
throw;
}
void* data = m_pool[m_used/ZONE_CHUNK_SIZE][m_used%ZONE_CHUNK_SIZE].data;
++m_used;
return data;
}
void zone::clear()
{
if(!m_pool.empty()) {
for(size_t b=0; b < m_used/ZONE_CHUNK_SIZE; ++b) {
size_t base_size = m_used / ZONE_CHUNK_SIZE;
size_t extend_size = m_used % ZONE_CHUNK_SIZE;
for(size_t b=0; b < base_size; ++b) {
cell_t* c(m_pool[b]);
for(size_t e=0; e < ZONE_CHUNK_SIZE; ++e) {
reinterpret_cast<object_class*>(c[e].data)->~object_class();
}
}
cell_t* c(m_pool.back());
for(size_t e=0; e < m_used%ZONE_CHUNK_SIZE; ++e) {
for(size_t e=0; e < extend_size; ++e) {
reinterpret_cast<object_class*>(c[e].data)->~object_class();
}
@ -60,7 +59,8 @@ void zone::clear()
}
m_used = 0;
for(user_finalizer_t::reverse_iterator it(m_user_finalizer.rbegin()), it_end(m_user_finalizer.rend());
for(user_finalizer_t::reverse_iterator it(m_user_finalizer.rbegin()),
it_end(m_user_finalizer.rend());
it != it_end;
++it) {
it->call();

View File

@ -25,7 +25,7 @@
#include <stdexcept>
#ifndef MSGPACK_ZONE_CHUNK_SIZE
#define MSGPACK_ZONE_CHUNK_SIZE 8*1024
#define MSGPACK_ZONE_CHUNK_SIZE 1024
#endif
namespace msgpack {
@ -44,19 +44,19 @@ public:
void push_finalizer(void (*func)(void* obj, void* user), T* obj, void* user);
public:
object_nil* nnil() { return new (alloc()) object_nil(); }
object_true* ntrue() { return new (alloc()) object_true(); }
object_false* nfalse() { return new (alloc()) object_false(); }
object_u8* nu8( uint8_t v) { return new (alloc()) object_u8(v); }
object_u16* nu16(uint16_t v) { return new (alloc()) object_u16(v); }
object_u32* nu32(uint32_t v) { return new (alloc()) object_u32(v); }
object_u64* nu64(uint64_t v) { return new (alloc()) object_u64(v); }
object_i8* ni8( int8_t v) { return new (alloc()) object_i8(v); }
object_i16* ni16( int16_t v) { return new (alloc()) object_i16(v); }
object_i32* ni32( int32_t v) { return new (alloc()) object_i32(v); }
object_i64* ni64( int64_t v) { return new (alloc()) object_i64(v); }
object_float* nfloat( float v) { return new (alloc()) object_float(v); }
object_double* ndouble( double v) { return new (alloc()) object_double(v); }
object_nil* nnil () { return new (alloc()) object_nil(); }
object_true* ntrue () { return new (alloc()) object_true(); }
object_false* nfalse () { return new (alloc()) object_false(); }
object_u8* nu8 (uint8_t v) { return new (alloc()) object_u8(v); }
object_u16* nu16 (uint16_t v) { return new (alloc()) object_u16(v); }
object_u32* nu32 (uint32_t v) { return new (alloc()) object_u32(v); }
object_u64* nu64 (uint64_t v) { return new (alloc()) object_u64(v); }
object_i8* ni8 (int8_t v) { return new (alloc()) object_i8(v); }
object_i16* ni16 (int16_t v) { return new (alloc()) object_i16(v); }
object_i32* ni32 (int32_t v) { return new (alloc()) object_i32(v); }
object_i64* ni64 (int64_t v) { return new (alloc()) object_i64(v); }
object_float* nfloat (float v) { return new (alloc()) object_float(v); }
object_double* ndouble(double v) { return new (alloc()) object_double(v); }
object_raw_ref* nraw_ref(void* ptr, uint32_t len)
{ return new (alloc()) object_raw_ref(ptr, len); }
@ -149,7 +149,7 @@ private:
user_finalizer_t m_user_finalizer;
private:
void resize_pool(size_t n);
void expand_chunk();
public:
static void finalize_free(void* obj, void* user)
@ -173,6 +173,16 @@ inline bool zone::empty() const
return m_used == 0 && m_user_finalizer.empty();
}
inline void* zone::alloc()
{
if(m_pool.size() <= m_used/ZONE_CHUNK_SIZE) {
expand_chunk();
}
void* data = m_pool[m_used/ZONE_CHUNK_SIZE][m_used%ZONE_CHUNK_SIZE].data;
++m_used;
return data;
}
} // namespace msgpack