You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
74 lines
2.1 KiB
74 lines
2.1 KiB
#include <string.h> /* memcmp,strlen */
|
|
#include "epqueue_nif.h"
|
|
#include "epqueue.h"
|
|
#include "epqueue_item.h"
|
|
#include "nif_utils.h"
|
|
#include "macros.h"
|
|
|
|
const char kAtomOk[] = "ok";
|
|
const char kAtomError[] = "error";
|
|
const char kAtomTrue[] = "true";
|
|
const char kAtomFalse[] = "false";
|
|
const char kAtomUndefined[] = "undefined";
|
|
const char kAtomGlobalLock[] = "global_lock";
|
|
|
|
atoms ATOMS;
|
|
|
|
void open_resources(ErlNifEnv* env, epqueue_data* data)
|
|
{
|
|
ErlNifResourceFlags flags = static_cast<ErlNifResourceFlags>(ERL_NIF_RT_CREATE | ERL_NIF_RT_TAKEOVER);
|
|
data->resPQueueInstance = enif_open_resource_type(env, NULL, "pqueue_instance", nif_epqueue_free, flags, NULL);
|
|
data->resPQueueItem = enif_open_resource_type(env, NULL, "pqueue_item", epqueue_item_free, flags, NULL);
|
|
}
|
|
|
|
int on_nif_load(ErlNifEnv* env, void** priv_data, ERL_NIF_TERM load_info)
|
|
{
|
|
UNUSED(load_info);
|
|
|
|
ATOMS.atomOk = make_atom(env, kAtomOk);
|
|
ATOMS.atomError = make_atom(env, kAtomError);
|
|
ATOMS.atomTrue = make_atom(env, kAtomTrue);
|
|
ATOMS.atomFalse = make_atom(env, kAtomFalse);
|
|
ATOMS.atomUndefined = make_atom(env, kAtomUndefined);
|
|
|
|
ATOMS.atomGlobalLock = make_atom(env, kAtomGlobalLock);
|
|
|
|
epqueue_data* data = static_cast<epqueue_data*>(enif_alloc(sizeof(epqueue_data)));
|
|
open_resources(env, data);
|
|
|
|
*priv_data = data;
|
|
return 0;
|
|
}
|
|
|
|
void on_nif_unload(ErlNifEnv* env, void* priv_data)
|
|
{
|
|
UNUSED(env);
|
|
|
|
epqueue_data* data = static_cast<epqueue_data*>(priv_data);
|
|
enif_free(data);
|
|
}
|
|
|
|
int on_nif_upgrade(ErlNifEnv* env, void** priv, void** old_priv, ERL_NIF_TERM info)
|
|
{
|
|
UNUSED(old_priv);
|
|
UNUSED(info);
|
|
|
|
epqueue_data* data = static_cast<epqueue_data*>(enif_alloc(sizeof(epqueue_data)));
|
|
open_resources(env, data);
|
|
|
|
*priv = data;
|
|
return 0;
|
|
}
|
|
|
|
static ErlNifFunc nif_funcs[] =
|
|
{
|
|
{"new", 1, nif_epqueue_new},
|
|
{"insert", 3, nif_epqueue_insert},
|
|
{"remove", 2, nif_epqueue_remove},
|
|
{"pop", 1, nif_epqueue_pop},
|
|
{"peek", 1, nif_epqueue_peek},
|
|
{"size", 1, nif_epqueue_size}
|
|
};
|
|
|
|
ERL_NIF_INIT(epqueue_nif, nif_funcs, on_nif_load, NULL, on_nif_upgrade, on_nif_unload)
|
|
|