Support delete a fd event.

When you call the fd_add, it will return an object with `delete` method.
So you can delete that event if you want.

Signed-off-by: Xiongfei(Alex) Guo <xfguo@credosemi.com>
This commit is contained in:
Xiongfei Guo 2014-06-20 10:31:19 +00:00 committed by John Crispin
parent 79b56268b4
commit 02ca593347
2 changed files with 44 additions and 3 deletions

View file

@ -46,20 +46,31 @@ uloop.timer(
end, 2000 end, 2000
) )
uloop.fd_add(udp, function(ufd, events) udp_ev = uloop.fd_add(udp, function(ufd, events)
local words, msg_or_ip, port_or_nil = ufd:receivefrom() local words, msg_or_ip, port_or_nil = ufd:receivefrom()
print('Recv UDP packet from '..msg_or_ip..':'..port_or_nil..' : '..words) print('Recv UDP packet from '..msg_or_ip..':'..port_or_nil..' : '..words)
if words == "Stop!" then
udp_ev:delete()
end
end, uloop.ULOOP_READ) end, uloop.ULOOP_READ)
udp_count = 0
udp_send_timer = uloop.timer( udp_send_timer = uloop.timer(
function() function()
local s = socket.udp() local s = socket.udp()
local words = 'Hello!' local words
if udp_count > 3 then
words = "Stop!"
udp_send_timer:cancel()
else
words = 'Hello!'
udp_send_timer:set(1000)
end
print('Send UDP packet to 127.0.0.1:8080 :'..words) print('Send UDP packet to 127.0.0.1:8080 :'..words)
s:sendto(words, '127.0.0.1', 8080) s:sendto(words, '127.0.0.1', 8080)
s:close() s:close()
udp_send_timer:set(1000) udp_count = udp_count + 1
end, 3000 end, 3000
) )

View file

@ -172,6 +172,24 @@ static int get_sock_fd(lua_State* L, int idx) {
return fd; return fd;
} }
static int ul_ufd_delete(lua_State *L)
{
struct lua_uloop_fd *ufd = lua_touserdata(L, 1);
uloop_fd_delete(&ufd->fd);
lua_getglobal(state, "__uloop_cb");
luaL_unref(L, -1, ufd->r);
lua_getglobal(state, "__uloop_fds");
luaL_unref(L, -1, ufd->fd_r);
return 1;
}
static const luaL_Reg ufd_m[] = {
{ "delete", ul_ufd_delete },
{ NULL, NULL }
};
static int ul_ufd_add(lua_State *L) static int ul_ufd_add(lua_State *L)
{ {
struct lua_uloop_fd *ufd; struct lua_uloop_fd *ufd;
@ -205,6 +223,18 @@ static int ul_ufd_add(lua_State *L)
lua_pop(L, 1); lua_pop(L, 1);
ufd = lua_newuserdata(L, sizeof(*ufd)); ufd = lua_newuserdata(L, sizeof(*ufd));
lua_createtable(L, 0, 2);
lua_pushvalue(L, -1);
lua_setfield(L, -2, "__index");
lua_pushcfunction(L, ul_ufd_delete);
lua_setfield(L, -2, "__gc");
lua_pushvalue(L, -1);
lua_setmetatable(L, -3);
lua_pushvalue(L, -2);
luaI_openlib(L, NULL, ufd_m, 1);
lua_pushvalue(L, -2);
memset(ufd, 0, sizeof(*ufd)); memset(ufd, 0, sizeof(*ufd));
ufd->r = ref; ufd->r = ref;