diff options
| author | Kim Alvefur <zash@zash.se> | 2016-10-07 17:18:29 +0200 |
|---|---|---|
| committer | Kim Alvefur <zash@zash.se> | 2016-10-07 17:18:29 +0200 |
| commit | e3232fbc9185dd2ccb0bba74d3f5f1fcbd2013ed (patch) | |
| tree | 4fce544e19e3ea082219dad361a1769abf40e900 /net | |
| parent | 58686a0f39cea79956b747bd5bb4f091a67b9db9 (diff) | |
| download | prosody-e3232fbc9185dd2ccb0bba74d3f5f1fcbd2013ed.tar.gz prosody-e3232fbc9185dd2ccb0bba74d3f5f1fcbd2013ed.zip | |
net.server_epoll: Separate ok from err branch more
Diffstat (limited to 'net')
| -rw-r--r-- | net/server_epoll.lua | 33 |
1 files changed, 19 insertions, 14 deletions
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 |