2022-10-24 16:44:29 +00:00
|
|
|
/*
|
|
|
|
* Copyright (C) 2022 Jordan Bancino <@jordan:bancino.net>
|
|
|
|
*
|
|
|
|
* Permission is hereby granted, free of charge, to any person
|
|
|
|
* obtaining a copy of this software and associated documentation files
|
|
|
|
* (the "Software"), to deal in the Software without restriction,
|
|
|
|
* including without limitation the rights to use, copy, modify, merge,
|
|
|
|
* publish, distribute, sublicense, and/or sell copies of the Software,
|
|
|
|
* and to permit persons to whom the Software is furnished to do so,
|
|
|
|
* subject to the following conditions:
|
|
|
|
*
|
|
|
|
* The above copyright notice and this permission notice shall be
|
|
|
|
* included in all copies or portions of the Software.
|
|
|
|
*
|
|
|
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
|
|
|
|
* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
|
|
|
|
* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
|
|
|
|
* NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
|
|
|
|
* BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
|
|
|
|
* ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
|
|
|
|
* CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
|
|
|
* SOFTWARE.
|
|
|
|
*/
|
|
|
|
#include <Db.h>
|
|
|
|
|
|
|
|
#include <Memory.h>
|
2022-11-06 00:47:17 +00:00
|
|
|
#include <Json.h>
|
2022-10-24 16:44:29 +00:00
|
|
|
|
|
|
|
#include <pthread.h>
|
|
|
|
|
|
|
|
struct Db
|
|
|
|
{
|
|
|
|
const char *dir;
|
|
|
|
size_t cacheSize;
|
2022-10-31 12:12:42 +00:00
|
|
|
size_t maxCache;
|
2022-10-24 16:44:29 +00:00
|
|
|
pthread_mutex_t lock;
|
2022-10-31 12:12:42 +00:00
|
|
|
|
|
|
|
HashMap *cache;
|
2022-10-24 16:44:29 +00:00
|
|
|
};
|
|
|
|
|
|
|
|
struct DbRef
|
|
|
|
{
|
2022-10-31 12:12:42 +00:00
|
|
|
HashMap *json;
|
2022-10-24 16:44:29 +00:00
|
|
|
pthread_mutex_t lock;
|
2022-10-31 12:12:42 +00:00
|
|
|
|
|
|
|
unsigned long ts;
|
|
|
|
char *file;
|
2022-10-24 16:44:29 +00:00
|
|
|
};
|
|
|
|
|
2022-11-06 00:47:17 +00:00
|
|
|
static ssize_t
|
2022-11-08 01:05:28 +00:00
|
|
|
DbComputeSize(HashMap * json)
|
2022-11-06 00:47:17 +00:00
|
|
|
{
|
2022-11-08 01:05:28 +00:00
|
|
|
char *key;
|
|
|
|
JsonValue *val;
|
|
|
|
MemoryInfo *a;
|
|
|
|
size_t total;
|
|
|
|
|
|
|
|
if (!json)
|
|
|
|
{
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
total = 0;
|
|
|
|
|
|
|
|
a = MemoryInfoGet(json);
|
|
|
|
if (a)
|
|
|
|
{
|
|
|
|
total += MemoryInfoGetSize(a);
|
|
|
|
}
|
|
|
|
|
|
|
|
while (HashMapIterate(json, &key, (void **) &val))
|
|
|
|
{
|
|
|
|
a = MemoryInfoGet(key);
|
|
|
|
if (a)
|
|
|
|
{
|
|
|
|
total += MemoryInfoGetSize(a);
|
|
|
|
}
|
|
|
|
|
|
|
|
a = MemoryInfoGet(val);
|
|
|
|
if (a)
|
|
|
|
{
|
|
|
|
total += MemoryInfoGetSize(a);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return total;
|
2022-11-06 00:47:17 +00:00
|
|
|
}
|
|
|
|
|
2022-10-24 16:44:29 +00:00
|
|
|
Db *
|
|
|
|
DbOpen(const char *dir, size_t cache)
|
|
|
|
{
|
|
|
|
Db *db;
|
|
|
|
|
|
|
|
if (!dir)
|
|
|
|
{
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
db = Malloc(sizeof(Db));
|
|
|
|
if (!db)
|
|
|
|
{
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
db->dir = dir;
|
2022-10-31 12:12:42 +00:00
|
|
|
db->maxCache = cache;
|
2022-10-24 16:44:29 +00:00
|
|
|
|
|
|
|
pthread_mutex_init(&db->lock, NULL);
|
|
|
|
|
2022-10-31 12:12:42 +00:00
|
|
|
if (db->maxCache)
|
2022-10-24 16:44:29 +00:00
|
|
|
{
|
2022-10-31 12:12:42 +00:00
|
|
|
db->cache = HashMapCreate();
|
|
|
|
if (!db->cache)
|
|
|
|
{
|
|
|
|
return NULL;
|
|
|
|
}
|
2022-10-24 16:44:29 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
return db;
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
|
|
|
DbClose(Db * db)
|
|
|
|
{
|
|
|
|
if (!db)
|
|
|
|
{
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2022-10-31 12:12:42 +00:00
|
|
|
pthread_mutex_destroy(&db->lock);
|
|
|
|
HashMapFree(db->cache);
|
|
|
|
|
2022-10-24 16:44:29 +00:00
|
|
|
Free(db);
|
|
|
|
}
|
|
|
|
|
|
|
|
DbRef *
|
|
|
|
DbCreate(Db * db, char *prefix, char *key)
|
|
|
|
{
|
2022-10-31 12:12:42 +00:00
|
|
|
if (!db || !prefix || !key)
|
|
|
|
{
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
2022-10-24 16:44:29 +00:00
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
DbRef *
|
|
|
|
DbLock(Db * db, char *prefix, char *key)
|
|
|
|
{
|
2022-10-31 12:12:42 +00:00
|
|
|
if (!db || !prefix || !key)
|
|
|
|
{
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
pthread_mutex_lock(&db->lock);
|
|
|
|
|
|
|
|
pthread_mutex_unlock(&db->lock);
|
2022-10-24 16:44:29 +00:00
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
|
|
|
DbUnlock(Db * db, DbRef * ref)
|
|
|
|
{
|
2022-10-31 12:12:42 +00:00
|
|
|
if (!db || !ref)
|
|
|
|
{
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
pthread_mutex_lock(&db->lock);
|
|
|
|
|
|
|
|
pthread_mutex_unlock(&db->lock);
|
|
|
|
|
2022-10-24 16:44:29 +00:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
HashMap *
|
|
|
|
DbJson(DbRef * ref)
|
|
|
|
{
|
2022-10-31 12:12:42 +00:00
|
|
|
return ref ? ref->json : NULL;
|
2022-10-24 16:44:29 +00:00
|
|
|
}
|