cthread_objcache.h

/* SPDX-License-Identifier: BSD-3-Clause
 * Copyright (c) 2019-2022 Intel Corporation
 */
 
#ifndef _CTHREAD_OBJCACHE_H_
#define _CTHREAD_OBJCACHE_H_
 
#include <string.h>
 
#include <cne_per_thread.h>
 
#include "cthread_int.h"
#include "cthread_queue.h"
 
#ifdef __cplusplus
extern "C" {
#endif
 
CNE_DECLARE_PER_THREAD(struct cthread_sched *, this_sched);
 
struct cthread_objcache {
    struct cthread_queue *q;      
    size_t obj_size;              
    int prealloc_size;            
    char name[CTHREAD_NAME_SIZE]; 
};
 
static inline struct cthread_objcache *
_cthread_objcache_create(const char *name, size_t obj_size, int prealloc_size)
{
    struct cthread_objcache *c = calloc(1, sizeof(struct cthread_objcache));
 
    if (c == NULL)
        return NULL;
 
    c->q = _cthread_queue_create("cache queue");
    if (c->q == NULL) {
        free(c);
        return NULL;
    }
    c->obj_size      = obj_size;
    c->prealloc_size = prealloc_size;
 
    c->name[0] = '\0';
    if (name != NULL)
        strlcpy(c->name, name, sizeof(c->name));
 
    return c;
}
 
static inline int
_cthread_objcache_destroy(struct cthread_objcache *c)
{
    if (c == NULL)
        return 0;
    if (_cthread_queue_destroy(c->q) == 0) {
        free(c);
        return 0;
    }
    return -1;
}
 
static inline void *
_cthread_objcache_alloc(struct cthread_objcache *c)
{
    int i;
    void *data;
    struct cthread_queue *q = c->q;
    size_t obj_size         = c->obj_size;
    int prealloc_size       = c->prealloc_size;
 
    data = _cthread_queue_remove(q);
 
    if (data == NULL) {
        for (i = 0; i < prealloc_size; i++) {
            data = calloc(1, obj_size);
            if (data == NULL)
                return NULL;
 
            _cthread_queue_insert_mp(q, data);
        }
        data = _cthread_queue_remove(q);
    }
    return data;
}
 
static inline void
_cthread_objcache_free(struct cthread_objcache *c, void *obj)
{
    _cthread_queue_insert_mp(c->q, obj);
}
 
#ifdef __cplusplus
}
#endif
 
#endif /* _CTHREAD_OBJCACHE_H_ */