lib/object-pool.c

   1 /*
   2  * Copyright (c) 2018 Philippe Proulx <pproulx@efficios.com>
   3  *
   4  * Permission is hereby granted, free of charge, to any person obtaining a copy
   5  * of this software and associated documentation files (the "Software"), to deal
   6  * in the Software without restriction, including without limitation the rights
   7  * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
   8  * copies of the Software, and to permit persons to whom the Software is
   9  * furnished to do so, subject to the following conditions:
  10  *
  11  * The above copyright notice and this permission notice shall be included in
  12  * all copies or substantial portions of the Software.
  13  *
  14  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  15  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  16  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  17  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  18  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  19  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
  20  * SOFTWARE.
  21  */
  22
  23 #define BT_LOG_TAG "OBJECT-POOL"
  24 #include <babeltrace/lib-logging-internal.h>
  25
  26 #include <stdint.h>
  27 #include <babeltrace/assert-internal.h>
  28 #include <babeltrace/object-pool-internal.h>
  29
  30 int bt_object_pool_initialize(struct bt_object_pool *pool,
  31                 bt_object_pool_new_object_func new_object_func,
  32                 bt_object_pool_destroy_object_func destroy_object_func,
  33                 void *data)
  34 {
  35         int ret = 0;
  36
  37         BT_ASSERT(new_object_func);
  38         BT_ASSERT(destroy_object_func);
  39         BT_LOGD("Initializing object pool: addr=%p, data-addr=%p",
  40                 pool, data);
  41         pool->objects = g_ptr_array_new();
  42         if (!pool->objects) {
  43                 BT_LOGE_STR("Failed to allocate a GPtrArray.");
  44                 goto error;
  45         }
  46
  47         pool->funcs.new_object = new_object_func;
  48         pool->funcs.destroy_object = destroy_object_func;
  49         pool->data = data;
  50         pool->size = 0;
  51         BT_LIB_LOGD("Initialized object pool: %!+o", pool);
  52         goto end;
  53
  54 error:
  55         if (pool) {
  56                 bt_object_pool_finalize(pool);
  57         }
  58
  59         ret = -1;
  60
  61 end:
  62         return ret;
  63 }
  64
  65 void bt_object_pool_finalize(struct bt_object_pool *pool)
  66 {
  67         uint64_t i;
  68
  69         BT_ASSERT(pool);
  70         BT_LIB_LOGD("Finalizing object pool: %!+o", pool);
  71
  72         if (pool->objects) {
  73                 for (i = 0; i < pool->size; i++) {
  74                         void *obj = pool->objects->pdata[i];
  75
  76                         if (obj) {
  77                                 pool->funcs.destroy_object(obj, pool->data);
  78                         }
  79                 }
  80
  81                 g_ptr_array_free(pool->objects, TRUE);
  82                 pool->objects = NULL;
  83         }
  84 }