lihao/skynet
1
0
Fork 0
mirror of https://github.com/cloudwu/skynet.git synced 2026-07-23 19:43:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

use stdatomic (#1317)

Browse Source
This commit is contained in:
云风
2021-01-11 10:54:34 +08:00
committed by GitHub
parent 284df5430b
commit b164e3a8a9
12 changed files with 201 additions and 138 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
lualib-src/lua-multicast.c
View File

@@ -10,7 +10,7 @@
#include "atomic.h"
struct mc_package {
int reference;
ATOM_INT reference;
uint32_t size;
void *data;
};
@@ -18,7 +18,7 @@ struct mc_package {
static int
pack(lua_State *L, void *data, size_t size) {
struct mc_package * pack = skynet_malloc(sizeof(struct mc_package));
pack->reference = 0;
ATOM_INIT(&pack->reference, 0);
pack->size = (uint32_t)size;
pack->data = data;
struct mc_package ** ret = skynet_malloc(sizeof(*ret));
@@ -91,10 +91,10 @@ static int
mc_bindrefer(lua_State *L) {
struct mc_package ** pack = lua_touserdata(L,1);
int ref = luaL_checkinteger(L,2);
if ((*pack)->reference != 0) {
if (ATOM_LOAD(&(*pack)->reference) != 0) {
return luaL_error(L, "Can't bind a multicast package more than once");
}
(*pack)->reference = ref;
ATOM_STORE(&(*pack)->reference , ref);
lua_pushlightuserdata(L, *pack);
@@ -110,7 +110,7 @@ static int
mc_closelocal(lua_State *L) {
struct mc_package *pack = lua_touserdata(L,1);
int ref = ATOM_DEC(&pack->reference);
int ref = ATOM_FDEC(&pack->reference)-1;
if (ref <= 0) {
skynet_free(pack->data);
skynet_free(pack);