如何用select或epoll实现LuaJIT非阻塞？解决epoll_wait阻塞问题

核心思路：避免epoll_wait无限阻塞 #

LuaJIT本身不支持安全的多线程并行执行，因此不能照搬C语言开独立线程跑epoll的方案。核心解决方向是让epoll_wait不要无限等待，通过设置超时时间，让程序在无事件触发时能回到Lua层执行其他任务，再循环进入epoll监听。

方法1：给epoll_wait设置超时时间 #

将epoll_wait的最后一个参数（超时时间，单位毫秒）从-1（无限阻塞）改为一个较小的数值（比如10或50），即使没有IO事件触发，epoll_wait也会在超时后返回，此时即可执行其他任务，之后再次进入监听循环。

修改后的示例代码：

local ctx = ffi.new_tcp("0.0.0.0", 5555)
local server_socket = ffi.tcp_listen(ctx)
local epoll_fd = ffi.C.epoll_create1(0)
-- 完成server_socket注册等前置操作...

-- 定义超时时间（10毫秒）
local EPOLL_TIMEOUT = 10

while true do
    -- 带超时的epoll_wait，无事件时超时返回0
    local nfds = ffi.C.epoll_wait(epoll_fd, events, NB_CONNECTION, EPOLL_TIMEOUT)
    
    if nfds > 0 then
        -- 处理触发的IO事件
        for i = 0, nfds - 1 do
            -- 处理events[i]对应的socket事件（如连接、读写）
        end
    elseif nfds == 0 then
        -- 超时触发，执行其他任务
        do_other_tasks() -- 替换为你的自定义任务函数
    else
        -- 处理epoll_wait错误
        local err = ffi.errno()
        -- 若为信号中断（EINTR），继续循环即可
        if err ~= ffi.C.EINTR then
            break
        end
    end
end

方法2：结合协程实现伪并发 #

如果其他任务也是IO密集型，可以用LuaJIT的协程拆分任务，将epoll循环作为调度核心，每次超时或处理完事件后切换协程执行其他任务，实现伪并发效果。

示例框架：

-- 存储需要执行的任务协程
local task_coroutines = {}

-- 添加自定义任务协程
table.insert(task_coroutines, coroutine.create(function()
    while true do
        -- 自定义任务逻辑（如定时检查、非阻塞IO）
        run_scheduled_task()
        coroutine.yield() -- 让出控制权给epoll循环
    end
end))

local EPOLL_TIMEOUT = 10

while true do
    local nfds = ffi.C.epoll_wait(epoll_fd, events, NB_CONNECTION, EPOLL_TIMEOUT)
    
    if nfds > 0 then
        -- 处理IO事件
        for i = 0, nfds - 1 do
            -- 处理对应socket事件
        end
    end
    
    -- 切换协程执行其他任务
    for _, co in ipairs(task_coroutines) do
        if coroutine.status(co) ~= "dead" then
            coroutine.resume(co)
        end
    end
end

用select实现非阻塞的思路 #

select的处理逻辑和epoll完全一致：给select设置超时时间，避免无限阻塞。示例伪代码：

local fd_set = ffi.new("fd_set")
local tv = ffi.new("struct timeval")
tv.tv_sec = 0
tv.tv_usec = 10000 -- 10毫秒超时

while true do
    ffi.C.FD_ZERO(fd_set)
    ffi.C.FD_SET(server_socket, fd_set)
    -- 将其他需要监听的socket添加到fd_set中
    
    local ret = ffi.C.select(server_socket + 1, fd_set, nil, nil, tv)
    
    if ret > 0 then
        if ffi.C.FD_ISSET(server_socket, fd_set) then
            -- 处理server_socket的连接事件
        end
        -- 检查其他socket是否有事件触发
    elseif ret == 0 then
        -- 超时，执行其他任务
        do_other_tasks()
    end
end

注意select有默认的fd数量限制（通常为1024），高并发场景下性能不如epoll。

关键注意事项 #

• 绝对不要用-1作为epoll_wait或select的超时参数，否则会无限阻塞，无法处理其他任务。
• 处理epoll_wait返回的错误时，若为EINTR（被信号中断），无需退出循环，继续执行即可。
• 如果其他任务是CPU密集型，要避免单次执行时间过长，可拆分任务到每次循环中执行一小部分，保证IO事件的响应速度。

内容的提问来源于stack exchange，提问作者tvon