* Copyright 2009, Ingo Weinhold, ingo_weinhold@gmx.de.
* Copyright 2011, Oliver Tappe <zooey@hirschkaefer.de>
* Distributed under the terms of the MIT License.
*/
#ifndef _PACKAGE__HPKG__PRIVATE__POOL_BUFFER_H_
#define _PACKAGE__HPKG__PRIVATE__POOL_BUFFER_H_
#include <stddef.h>
#include <util/DoublyLinkedList.h>
#include <package/hpkg/BufferPool.h>
namespace BPackageKit {
namespace BHPKG {
namespace BPrivate {
class PoolBuffer : public DoublyLinkedListLinkImpl<PoolBuffer> {
public:
PoolBuffer(size_t size);
~PoolBuffer();
void* Buffer() const { return fBuffer; }
size_t Size() const { return fSize; }
PoolBuffer** Owner() const { return fOwner; }
void SetOwner(PoolBuffer** owner)
{ fOwner = owner; }
void SetCached(bool cached) { fCached = cached; }
bool IsCached() const { return fCached; }
private:
PoolBuffer** fOwner;
void* fBuffer;
size_t fSize;
bool fCached;
};
class PoolBufferPutter {
public:
PoolBufferPutter(BBufferPool* pool, PoolBuffer** owner)
:
fPool(pool),
fOwner(owner),
fBuffer(NULL)
{
}
PoolBufferPutter(BBufferPool* pool, PoolBuffer* buffer)
:
fPool(pool),
fOwner(NULL),
fBuffer(buffer)
{
}
~PoolBufferPutter()
{
if (fPool != NULL) {
if (fOwner != NULL)
fPool->PutBufferAndCache(fOwner);
else if (fBuffer != NULL)
fPool->PutBuffer(&fBuffer);
}
}
private:
BBufferPool* fPool;
PoolBuffer** fOwner;
PoolBuffer* fBuffer;
};
}
}
}
#endif
