298 lines
8.9 KiB
C
298 lines
8.9 KiB
C
/*
|
|
**********************************************************************************************
|
|
* File: maat_ex_data.c
|
|
* Description: ex data
|
|
* Authors: Liu WenTan <liuwentan@geedgenetworks.com>
|
|
* Date: 2022-10-31
|
|
* Copyright: (c) Since 2022 Geedge Networks, Ltd. All rights reserved.
|
|
***********************************************************************************************
|
|
*/
|
|
|
|
#include <stddef.h>
|
|
#include <stdio.h>
|
|
#include <assert.h>
|
|
|
|
#include "uthash/uthash.h"
|
|
#include "uthash/utarray.h"
|
|
#include "log/log.h"
|
|
#include "maat_utils.h"
|
|
#include "maat_ex_data.h"
|
|
|
|
#define MODULE_EX_DATA module_name_str("maat.ex_data")
|
|
|
|
struct ex_data_runtime {
|
|
UT_array *cache_rows;
|
|
size_t cache_row_num;
|
|
size_t cache_size;
|
|
|
|
struct rcu_hash_table *htable; // store ex_container
|
|
struct ex_container_schema *ref_container_schema;
|
|
int gc_timeout_s;
|
|
int table_id;
|
|
|
|
struct log_handle *logger;
|
|
};
|
|
|
|
void cache_row_free(void *p)
|
|
{
|
|
free(*(char **)p);
|
|
}
|
|
|
|
UT_icd ut_cache_row_icd = {sizeof(char*), NULL, NULL, cache_row_free};
|
|
|
|
struct ex_data_runtime *
|
|
ex_data_runtime_new(int table_id, int gc_timeout_s, struct log_handle *logger)
|
|
{
|
|
if (table_id < 0 || NULL == logger) {
|
|
return NULL;
|
|
}
|
|
|
|
struct ex_data_runtime *ex_data_rt = ALLOC(struct ex_data_runtime, 1);
|
|
|
|
utarray_new(ex_data_rt->cache_rows, &ut_cache_row_icd);
|
|
ex_data_rt->htable = NULL;
|
|
ex_data_rt->gc_timeout_s = gc_timeout_s;
|
|
ex_data_rt->table_id = table_id;
|
|
ex_data_rt->logger = logger;
|
|
|
|
return ex_data_rt;
|
|
}
|
|
|
|
void ex_data_runtime_free(struct ex_data_runtime *ex_data_rt)
|
|
{
|
|
if (NULL == ex_data_rt) {
|
|
return;
|
|
}
|
|
|
|
if (ex_data_rt->cache_rows != NULL) {
|
|
utarray_free(ex_data_rt->cache_rows);
|
|
ex_data_rt->cache_rows = NULL;
|
|
ex_data_rt->cache_row_num = 0;
|
|
}
|
|
|
|
if (ex_data_rt->htable != NULL) {
|
|
rcu_hash_free(ex_data_rt->htable);
|
|
ex_data_rt->htable = NULL;
|
|
}
|
|
|
|
FREE(ex_data_rt);
|
|
}
|
|
|
|
void ex_data_runtime_commit(struct ex_data_runtime *ex_data_rt)
|
|
{
|
|
if (NULL == ex_data_rt) {
|
|
return;
|
|
}
|
|
|
|
rcu_hash_commit(ex_data_rt->htable);
|
|
}
|
|
|
|
void ex_data_runtime_cache_row_put(struct ex_data_runtime *ex_data_rt, const char *row)
|
|
{
|
|
if (NULL == ex_data_rt || NULL == row) {
|
|
return;
|
|
}
|
|
|
|
size_t row_len = strlen(row);
|
|
char *row_copy = ALLOC(char, row_len + 1);
|
|
|
|
memcpy(row_copy, row, row_len);
|
|
ex_data_rt->cache_size += row_len;
|
|
utarray_push_back(ex_data_rt->cache_rows, &row_copy);
|
|
ex_data_rt->cache_row_num++;
|
|
}
|
|
|
|
const char *ex_data_runtime_cached_row_get(struct ex_data_runtime *ex_data_rt, size_t index)
|
|
{
|
|
if (NULL == ex_data_rt) {
|
|
return NULL;
|
|
}
|
|
|
|
const char **row = NULL;
|
|
row = (const char **)utarray_eltptr(ex_data_rt->cache_rows, index);
|
|
return *row;
|
|
}
|
|
|
|
size_t ex_data_runtime_cached_row_count(struct ex_data_runtime *ex_data_rt)
|
|
{
|
|
if (NULL == ex_data_rt) {
|
|
return 0;
|
|
}
|
|
|
|
return ex_data_rt->cache_row_num;
|
|
}
|
|
|
|
void ex_data_runtime_clear_row_cache(struct ex_data_runtime *ex_data_rt)
|
|
{
|
|
if (NULL == ex_data_rt) {
|
|
return;
|
|
}
|
|
|
|
if (ex_data_rt->cache_rows != NULL) {
|
|
utarray_free(ex_data_rt->cache_rows);
|
|
ex_data_rt->cache_rows = NULL;
|
|
}
|
|
|
|
ex_data_rt->cache_row_num = 0;
|
|
ex_data_rt->cache_size = 0;
|
|
}
|
|
|
|
void ex_data_runtime_set_ex_container_schema(struct ex_data_runtime *ex_data_rt,
|
|
struct ex_container_schema *container_schema)
|
|
{
|
|
ex_data_rt->ref_container_schema = container_schema;
|
|
}
|
|
|
|
void *ex_data_runtime_row2ex_data(struct ex_data_runtime *ex_data_rt,
|
|
const char *table_name, const char *row,
|
|
const char *key, size_t key_len)
|
|
{
|
|
void *ex_data = NULL;
|
|
struct ex_container_schema *container_schema = ex_data_rt->ref_container_schema;
|
|
container_schema->ex_schema.new_func(table_name, ex_data_rt->table_id, key, row,
|
|
&ex_data, container_schema->ex_schema.argl,
|
|
container_schema->ex_schema.argp);
|
|
return ex_data;
|
|
}
|
|
|
|
struct ex_container *ex_container_new(void *ex_data, void *custom_data)
|
|
{
|
|
struct ex_container *ex_container = ALLOC(struct ex_container, 1);
|
|
|
|
ex_container->ex_data = ex_data;
|
|
ex_container->custom_data = custom_data;
|
|
|
|
return ex_container;
|
|
}
|
|
|
|
void ex_container_free(void *user_ctx, void *data)
|
|
{
|
|
/* schema is NULL if not call ex_data_runtime_set_ex_container_schema */
|
|
if (NULL == user_ctx || NULL == data) {
|
|
return;
|
|
}
|
|
|
|
struct ex_container *container = (struct ex_container *)data;
|
|
struct ex_container_schema *container_schema = (struct ex_container_schema *)user_ctx;
|
|
|
|
/* free ex_container->custom_data */
|
|
if (container->custom_data != NULL && container_schema->custom_data_free != NULL) {
|
|
container_schema->custom_data_free(container->custom_data);
|
|
container->custom_data = NULL;
|
|
}
|
|
|
|
/* free ex_container->ex_data */
|
|
if (container->ex_data != NULL && container_schema->ex_schema.free_func != NULL) {
|
|
container_schema->ex_schema.free_func(container_schema->table_id,
|
|
&(container->ex_data),
|
|
container_schema->ex_schema.argl,
|
|
container_schema->ex_schema.argp);
|
|
container->ex_data = NULL;
|
|
}
|
|
|
|
FREE(container);
|
|
}
|
|
|
|
int ex_data_runtime_add_ex_container(struct ex_data_runtime *ex_data_rt,
|
|
const char *key, size_t key_len,
|
|
struct ex_container *ex_container)
|
|
{
|
|
if (NULL == ex_data_rt || NULL == key || 0 == key_len ||
|
|
NULL == ex_container) {
|
|
return -1;
|
|
}
|
|
|
|
if (NULL == ex_data_rt->htable) {
|
|
/* ex_data_rt->ref_container_schema has been set */
|
|
assert(ex_data_rt->ref_container_schema != NULL);
|
|
ex_data_rt->htable = rcu_hash_new(ex_container_free,
|
|
ex_data_rt->ref_container_schema,
|
|
ex_data_rt->gc_timeout_s);
|
|
}
|
|
|
|
return rcu_hash_add(ex_data_rt->htable, key, key_len, ex_container);
|
|
}
|
|
|
|
int ex_data_runtime_del_ex_container(struct ex_data_runtime *ex_data_rt,
|
|
const char *key, size_t key_len)
|
|
{
|
|
if (NULL == ex_data_rt || NULL == key || 0 == key_len ||
|
|
NULL == ex_data_rt->htable) {
|
|
return -1;
|
|
}
|
|
|
|
return rcu_hash_del(ex_data_rt->htable, key, key_len);
|
|
}
|
|
|
|
void *ex_data_runtime_get_ex_data_by_key(struct ex_data_runtime *ex_data_rt,
|
|
const char *key, size_t key_len)
|
|
{
|
|
if (NULL == ex_data_rt || NULL == key || 0 == key_len) {
|
|
return NULL;
|
|
}
|
|
|
|
struct ex_container_schema *container_schema = ex_data_rt->ref_container_schema;
|
|
if (NULL == container_schema || (0 == container_schema->set_flag)) {
|
|
return NULL;
|
|
}
|
|
|
|
struct ex_container *ex_container = (struct ex_container *)rcu_hash_find(ex_data_rt->htable,
|
|
key, key_len);
|
|
if (NULL == ex_container) {
|
|
return NULL;
|
|
}
|
|
|
|
void *dup_ex_data = NULL;
|
|
container_schema->ex_schema.dup_func(ex_data_rt->table_id, &dup_ex_data,
|
|
&(ex_container->ex_data),
|
|
container_schema->ex_schema.argl,
|
|
container_schema->ex_schema.argp);
|
|
return dup_ex_data;
|
|
}
|
|
|
|
void *ex_data_runtime_get_ex_data_by_container(struct ex_data_runtime *ex_data_rt,
|
|
struct ex_container *ex_container)
|
|
{
|
|
if (NULL == ex_data_rt || NULL == ex_container) {
|
|
return NULL;
|
|
}
|
|
|
|
struct ex_container_schema *container_schema = ex_data_rt->ref_container_schema;
|
|
if (NULL == container_schema || (0 == container_schema->set_flag)) {
|
|
return NULL;
|
|
}
|
|
|
|
void *dup_ex_data = NULL;
|
|
container_schema->ex_schema.dup_func(ex_data_rt->table_id, &dup_ex_data,
|
|
&(ex_container->ex_data),
|
|
container_schema->ex_schema.argl,
|
|
container_schema->ex_schema.argp);
|
|
return dup_ex_data;
|
|
}
|
|
|
|
size_t ex_data_runtime_ex_container_count(struct ex_data_runtime *ex_data_rt)
|
|
{
|
|
return rcu_hash_count(ex_data_rt->htable);
|
|
}
|
|
|
|
int ex_data_runtime_is_updating(struct ex_data_runtime *ex_data_rt)
|
|
{
|
|
return rcu_hash_is_updating(ex_data_rt->htable);
|
|
}
|
|
|
|
size_t ex_data_runtime_list_ex_container(struct ex_data_runtime *ex_data_rt,
|
|
struct ex_container ***ex_container)
|
|
{
|
|
return rcu_hash_list(ex_data_rt->htable, (void ***)ex_container);
|
|
}
|
|
|
|
void ex_data_runtime_garbage_collect_routine(struct ex_data_runtime *ex_data_rt)
|
|
{
|
|
if (NULL == ex_data_rt) {
|
|
return;
|
|
}
|
|
|
|
if (ex_data_rt->htable != NULL) {
|
|
rcu_hash_garbage_collect_routine(ex_data_rt->htable);
|
|
}
|
|
} |