From 15c519eb2fa4ab7a704f2f3817c41233e78c7624 Mon Sep 17 00:00:00 2001 From: Kim Alvefur Date: Fri, 7 Oct 2016 17:18:29 +0200 Subject: net.server_epoll: Separate ok from err branch more --- net/server_epoll.lua | 33 +++++++++++++++++++-------------- 1 file changed, 19 insertions(+), 14 deletions(-) (limited to 'net/server_epoll.lua') diff --git a/net/server_epoll.lua b/net/server_epoll.lua index 9ef14a78..a9896923 100644 --- a/net/server_epoll.lua +++ b/net/server_epoll.lua @@ -290,24 +290,28 @@ end -- Called when socket is readable function interface:onreadable() local data, err, partial = self.conn:receive(self._pattern); + if data then + self:on("incoming", data); + else + if partial then + self:on("incoming", partial, err); + end + if err == "wantread" then + self:setflags(true, nil); + elseif err == "wantwrite" then + self:setflags(nil, true); + elseif err ~= "timeout" then + self:on("disconnect", err); + self:destroy() + return; + end + end if self.conn:dirty() then self:setreadtimeout(false); self:pausefor(cfg.read_retry_delay); else self:setreadtimeout(); end - if data or partial then - self:on("incoming", data or partial, err); - end - if err == "wantread" then - self:setflags(true, nil); - elseif err == "wantwrite" then - self:setflags(nil, true); - elseif not data and err ~= "timeout" then - self:on("disconnect", err); - self:destroy() - return; - end end -- Called when socket is writable @@ -323,7 +327,8 @@ function interface:onwriteable() self:setwritetimeout(false); self:ondrain(); -- Be aware of writes in ondrain return; - elseif partial then + end + if partial then buffer[1] = data:sub(partial+1); for i = #buffer, 2, -1 do buffer[i] = nil; @@ -334,7 +339,7 @@ function interface:onwriteable() self:setflags(nil, true); elseif err == "wantread" then self:setflags(true, nil); - elseif err and err ~= "timeout" then + elseif err ~= "timeout" then self:on("disconnect", err); self:destroy(); end -- cgit v1.2.3