ObjectPool.c

 
#include <winpr/config.h>
 
#include <winpr/crt.h>
#include <winpr/assert.h>
 
#include <winpr/collections.h>
 
struct s_wObjectPool
{
  size_t size;
  size_t capacity;
  void** array;
  CRITICAL_SECTION lock;
  wObject object;
  BOOL synchronized;
};
 
static void ObjectPool_Lock(wObjectPool* pool)
{
  WINPR_ASSERT(pool);
  if (pool->synchronized)
    EnterCriticalSection(&pool->lock);
}
 
static void ObjectPool_Unlock(wObjectPool* pool)
{
  WINPR_ASSERT(pool);
  if (pool->synchronized)
    LeaveCriticalSection(&pool->lock);
}
 
void* ObjectPool_Take(wObjectPool* pool)
{
  void* obj = NULL;
 
  ObjectPool_Lock(pool);
 
  if (pool->size > 0)
    obj = pool->array[--(pool->size)];
 
  if (!obj)
  {
    if (pool->object.fnObjectNew)
      obj = pool->object.fnObjectNew(NULL);
  }
 
  if (pool->object.fnObjectInit)
    pool->object.fnObjectInit(obj);
 
  ObjectPool_Unlock(pool);
 
  return obj;
}
 
void ObjectPool_Return(wObjectPool* pool, void* obj)
{
  ObjectPool_Lock(pool);
 
  if ((pool->size + 1) >= pool->capacity)
  {
    size_t new_cap = 0;
    void** new_arr = NULL;
 
    new_cap = pool->capacity * 2;
    new_arr = (void**)realloc(pool->array, sizeof(void*) * new_cap);
    if (!new_arr)
      goto out;
 
    pool->array = new_arr;
    pool->capacity = new_cap;
  }
 
  pool->array[(pool->size)++] = obj;
 
  if (pool->object.fnObjectUninit)
    pool->object.fnObjectUninit(obj);
 
out:
  ObjectPool_Unlock(pool);
}
 
wObject* ObjectPool_Object(wObjectPool* pool)
{
  WINPR_ASSERT(pool);
  return &pool->object;
}
 
void ObjectPool_Clear(wObjectPool* pool)
{
  ObjectPool_Lock(pool);
 
  while (pool->size > 0)
  {
    (pool->size)--;
 
    if (pool->object.fnObjectFree)
      pool->object.fnObjectFree(pool->array[pool->size]);
  }
 
  ObjectPool_Unlock(pool);
}
 
wObjectPool* ObjectPool_New(BOOL synchronized)
{
  wObjectPool* pool = NULL;
 
  pool = (wObjectPool*)calloc(1, sizeof(wObjectPool));
 
  if (pool)
  {
    pool->capacity = 32;
    pool->size = 0;
    pool->array = (void**)calloc(pool->capacity, sizeof(void*));
    if (!pool->array)
    {
      free(pool);
      return NULL;
    }
    pool->synchronized = synchronized;
 
    if (pool->synchronized)
      InitializeCriticalSectionAndSpinCount(&pool->lock, 4000);
  }
 
  return pool;
}
 
void ObjectPool_Free(wObjectPool* pool)
{
  if (pool)
  {
    ObjectPool_Clear(pool);
 
    if (pool->synchronized)
      DeleteCriticalSection(&pool->lock);
 
    free(pool->array);
 
    free(pool);
  }
}