123 lines
2.3 KiB
C
123 lines
2.3 KiB
C
#include "first.h"
|
|
|
|
#include "mod_magnet_cache.h"
|
|
#include "log.h"
|
|
#include "stat_cache.h"
|
|
|
|
#include <stdlib.h>
|
|
#include <time.h>
|
|
|
|
#include <lualib.h>
|
|
#include <lauxlib.h>
|
|
|
|
static script *script_init() {
|
|
script *sc;
|
|
|
|
sc = calloc(1, sizeof(*sc));
|
|
sc->name = buffer_init();
|
|
sc->etag = buffer_init();
|
|
|
|
return sc;
|
|
}
|
|
|
|
static void script_free(script *sc) {
|
|
if (!sc) return;
|
|
|
|
lua_pop(sc->L, 1); /* the function copy */
|
|
|
|
buffer_free(sc->name);
|
|
buffer_free(sc->etag);
|
|
|
|
lua_close(sc->L);
|
|
|
|
free(sc);
|
|
}
|
|
|
|
#if 0
|
|
script_cache *script_cache_init() {
|
|
script_cache *p = calloc(1, sizeof(script_cache));
|
|
force_assert(p);
|
|
return p;
|
|
}
|
|
#endif
|
|
|
|
void script_cache_free_data(script_cache *p) {
|
|
size_t i;
|
|
|
|
if (!p) return;
|
|
|
|
for (i = 0; i < p->used; i++) {
|
|
script_free(p->ptr[i]);
|
|
}
|
|
|
|
free(p->ptr);
|
|
}
|
|
|
|
lua_State *script_cache_get_script(script_cache *cache, buffer *name, int etag_flags) {
|
|
script *sc = NULL;
|
|
stat_cache_entry *sce;
|
|
|
|
for (uint32_t i = 0; i < cache->used; ++i, sc = NULL) {
|
|
sc = cache->ptr[i];
|
|
if (!buffer_is_equal(name, sc->name)) continue;
|
|
|
|
sc->last_used = log_epoch_secs;
|
|
|
|
/* oops, the script failed last time */
|
|
|
|
if (lua_gettop(sc->L) == 0) break;
|
|
force_assert(lua_gettop(sc->L) == 1);
|
|
|
|
sce = stat_cache_get_entry(sc->name);
|
|
if (NULL == sce) {
|
|
lua_pop(sc->L, 1); /* pop the old function */
|
|
break;
|
|
}
|
|
|
|
const buffer *etag = stat_cache_etag_get(sce, etag_flags);
|
|
if (NULL == etag || !buffer_is_equal(sc->etag, etag)) {
|
|
/* the etag is outdated, reload the function */
|
|
lua_pop(sc->L, 1);
|
|
break;
|
|
}
|
|
|
|
force_assert(lua_isfunction(sc->L, -1));
|
|
|
|
return sc->L;
|
|
}
|
|
|
|
/* if the script was script already loaded but either got changed or
|
|
* failed to load last time */
|
|
if (sc == NULL) {
|
|
sc = script_init();
|
|
|
|
if (cache->used == cache->size) {
|
|
cache->size += 16;
|
|
cache->ptr = realloc(cache->ptr, cache->size * sizeof(*(cache->ptr)));
|
|
}
|
|
|
|
cache->ptr[cache->used++] = sc;
|
|
|
|
buffer_copy_buffer(sc->name, name);
|
|
|
|
sc->L = luaL_newstate();
|
|
luaL_openlibs(sc->L);
|
|
}
|
|
|
|
sc->last_used = log_epoch_secs;
|
|
|
|
if (0 != luaL_loadfile(sc->L, name->ptr)) {
|
|
/* oops, an error, return it */
|
|
return sc->L;
|
|
}
|
|
|
|
sce = stat_cache_get_entry(sc->name);
|
|
if (sce) {
|
|
buffer_copy_buffer(sc->etag, stat_cache_etag_get(sce, etag_flags));
|
|
}
|
|
|
|
force_assert(lua_isfunction(sc->L, -1));
|
|
|
|
return sc->L;
|
|
}
|