eUtils/c_src/enlfq/enlfq_nif.cc

#include <string.h>
#include "enlfq_nif.h"
#include "enlfq.h"
#include "nif_utils.h"

const char kAtomOk[] = "ok";
const char kAtomError[] = "error";
const char kAtomTrue[] = "true";
//const char kAtomFalse[] = "false";
//const char kAtomUndefined[] = "undefined";
const char kAtomEmpty[] = "empty";

atoms ATOMS;

void open_resources(ErlNifEnv *env, shared_data *data) {
    ErlNifResourceFlags flags = static_cast<ErlNifResourceFlags>(ERL_NIF_RT_CREATE | ERL_NIF_RT_TAKEOVER);
    data->resQueueInstance = enif_open_resource_type(env, NULL, "enlfq_instance", nif_enlfq_free, flags, NULL);
}

int on_nif_load(ErlNifEnv *env, void **priv_data, ERL_NIF_TERM) {

    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.atomEmpty = make_atom(env, kAtomEmpty);

    shared_data *data = static_cast<shared_data *>(enif_alloc(sizeof(shared_data)));
    open_resources(env, data);

    *priv_data = data;
    return 0;
}

void on_nif_unload(ErlNifEnv *, void *priv_data) {
    shared_data *data = static_cast<shared_data *>(priv_data);
    enif_free(data);
}

int on_nif_upgrade(ErlNifEnv *env, void **priv, void **, ERL_NIF_TERM) {
    shared_data *data = static_cast<shared_data *>(enif_alloc(sizeof(shared_data)));
    open_resources(env, data);

    *priv = data;
    return 0;
}

static ErlNifFunc nif_funcs[] =
        {
                {"new",  0, nif_enlfq_new},
                {"push", 2, nif_enlfq_push},
                {"pop",  1, nif_enlfq_pop}
        };

ERL_NIF_INIT(enlfq, nif_funcs, on_nif_load, NULL, on_nif_upgrade, on_nif_unload)