25개 이상의 토픽을 선택하실 수 없습니다.
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
122 lines
3.4 KiB
122 lines
3.4 KiB
#include "erl_nif.h"
|
|
#include <atomic>
|
|
|
|
using namespace std;
|
|
|
|
const int LockSize = 2097152;
|
|
atomic<uint64_t> LockSlot[LockSize];
|
|
|
|
ERL_NIF_TERM atomTrue;
|
|
ERL_NIF_TERM atomFalse;
|
|
ERL_NIF_TERM atomUndefined;
|
|
|
|
int nifLoad(ErlNifEnv *env, void **priv_data, ERL_NIF_TERM load_info){
|
|
atomTrue = enif_make_atom(env, "true");
|
|
atomFalse = enif_make_atom(env, "false");
|
|
atomUndefined = enif_make_atom(env, "undefined");
|
|
return 0;
|
|
}
|
|
|
|
inline bool lockOne(ErlNifEnv *env, ErlNifPid *ThePid, int KeyIx, uint64_t Val){
|
|
uint64_t Expected = 0;
|
|
if (LockSlot[KeyIx].compare_exchange_strong(Expected, Val)){
|
|
return true;
|
|
}else{
|
|
ThePid->pid = (ERL_NIF_TERM)Expected;
|
|
if (enif_is_process_alive(env, ThePid)){
|
|
return false;
|
|
}else{
|
|
return LockSlot[KeyIx].compare_exchange_strong(Expected, Val);
|
|
}
|
|
}
|
|
}
|
|
|
|
ERL_NIF_TERM tryLock(ErlNifEnv *env, int argc, const ERL_NIF_TERM argv[]){
|
|
int KeyIx;
|
|
enif_get_int(env, argv[0], &KeyIx);
|
|
ErlNifPid ThePid;
|
|
enif_self(env, &ThePid);
|
|
uint64_t Val = (uint64_t)(ThePid.pid);
|
|
|
|
return lockOne(env, &ThePid, KeyIx, Val) ? atomTrue : atomFalse;
|
|
}
|
|
|
|
ERL_NIF_TERM tryLocks(ErlNifEnv *env, int argc, const ERL_NIF_TERM argv[]){
|
|
ERL_NIF_TERM allList = argv[0];
|
|
ERL_NIF_TERM head;
|
|
ErlNifPid ThePid;
|
|
enif_self(env, &ThePid);
|
|
uint64_t Val = (uint64_t)(ThePid.pid);
|
|
int KeyIx;
|
|
int cnt = -1;
|
|
|
|
while (enif_get_list_cell(env, allList, &head, &allList)){
|
|
enif_get_int(env, head, &KeyIx);
|
|
if(lockOne(env, &ThePid, KeyIx, Val)){
|
|
cnt++;
|
|
}else{
|
|
allList = argv[0];
|
|
for(int i = 0; i <= cnt; i++){
|
|
enif_get_list_cell(env, allList, &head, &allList);
|
|
enif_get_int(env, head, &KeyIx);
|
|
LockSlot[KeyIx].store(0);
|
|
}
|
|
return atomFalse;
|
|
}
|
|
}
|
|
return atomTrue;
|
|
}
|
|
|
|
ERL_NIF_TERM releaseLock(ErlNifEnv *env, int argc, const ERL_NIF_TERM argv[]){
|
|
int KeyIx;
|
|
enif_get_int(env, argv[0], &KeyIx);
|
|
ErlNifPid ThePid;
|
|
enif_self(env, &ThePid);
|
|
uint64_t Expected = (uint64_t)(ThePid.pid);
|
|
|
|
return LockSlot[KeyIx].compare_exchange_strong(Expected, 0) ? atomTrue : atomFalse;
|
|
}
|
|
|
|
ERL_NIF_TERM releaseLocks(ErlNifEnv *env, int argc, const ERL_NIF_TERM argv[]){
|
|
ERL_NIF_TERM allList = argv[0];
|
|
ERL_NIF_TERM head;
|
|
ErlNifPid ThePid;
|
|
enif_self(env, &ThePid);
|
|
uint64_t Expected = (uint64_t)(ThePid.pid);
|
|
uint64_t RExpected;
|
|
int KeyIx;
|
|
int isAllOk = 1;
|
|
|
|
while (enif_get_list_cell(env, allList, &head, &allList)){
|
|
enif_get_int(env, head, &KeyIx);
|
|
RExpected = Expected;
|
|
if (!LockSlot[KeyIx].compare_exchange_strong(RExpected, 0)){
|
|
isAllOk = 0;
|
|
}
|
|
}
|
|
return isAllOk > 0 ? atomTrue : atomFalse;
|
|
}
|
|
|
|
ERL_NIF_TERM getLockPid(ErlNifEnv *env, int argc, const ERL_NIF_TERM argv[]){
|
|
int KeyIx;
|
|
enif_get_int(env, argv[0], &KeyIx);
|
|
ErlNifPid ThePid;
|
|
|
|
uint64_t Var = LockSlot[KeyIx].load();
|
|
if (Var > 0){
|
|
ThePid.pid = (ERL_NIF_TERM)Var;
|
|
return enif_make_pid(env, &ThePid);
|
|
}else{
|
|
return atomUndefined;
|
|
}
|
|
}
|
|
|
|
static ErlNifFunc nifFuncs[] = {
|
|
{"tryLock", 1, tryLock},
|
|
{"tryLocks", 1, tryLocks},
|
|
{"releaseLock", 1, releaseLock},
|
|
{"releaseLocks", 1, releaseLocks},
|
|
{"getLockPid", 1, getLockPid}
|
|
};
|
|
|
|
ERL_NIF_INIT(eNifLock, nifFuncs, &nifLoad, NULL, NULL, NULL)
|