diff options
| author | Kim Alvefur <zash@zash.se> | 2021-08-16 12:34:52 +0200 |
|---|---|---|
| committer | Kim Alvefur <zash@zash.se> | 2021-08-16 12:34:52 +0200 |
| commit | d06ae1629513e0a8ab8c1a24566226dddc93e8e1 (patch) | |
| tree | 1734b65d5c33a62a267fdf8394066b89fa732570 /net/server_event.lua | |
| parent | b686dda4afa8f519830997dc88386a37e371d47e (diff) | |
| download | prosody-d06ae1629513e0a8ab8c1a24566226dddc93e8e1.tar.gz prosody-d06ae1629513e0a8ab8c1a24566226dddc93e8e1.zip | |
net.server: Add a predrain callaback just before writes
Allows sneaking in things in the write buffer just before it's sent to
the network stack. For example ack requests, compression flushes or
other things that make sense to send after stanzas or other things.
This ensures any additional trailing data sent is included in the same
write, and possibly the same TCP packet. Other methods used such as
timers or nextTick might not have the same effect as it depends on
scheduling.
Diffstat (limited to 'net/server_event.lua')
| -rw-r--r-- | net/server_event.lua | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/net/server_event.lua b/net/server_event.lua index f7e1f448..139c7e5f 100644 --- a/net/server_event.lua +++ b/net/server_event.lua @@ -449,6 +449,7 @@ function interface_mt:setlistener(listener, data) self.onstatus = listener.onstatus; self.ondetach = listener.ondetach; self.onattach = listener.onattach; + self.onpredrain = listener.onpredrain; self.ondrain = listener.ondrain; self:onattach(data); end @@ -464,6 +465,8 @@ function interface_mt:ontimeout() end function interface_mt:onreadtimeout() end +function interface_mt:onpredrain() +end function interface_mt:ondrain() end function interface_mt:ondetach() @@ -490,6 +493,7 @@ local function handleclient( client, ip, port, server, pattern, listener, sslctx onincoming = listener.onincoming; -- will be called when client sends data ontimeout = listener.ontimeout; -- called when fatal socket timeout occurs onreadtimeout = listener.onreadtimeout; -- called when socket inactivity timeout occurs + onpredrain = listener.onpredrain; -- called before writes ondrain = listener.ondrain; -- called when writebuffer is empty ondetach = listener.ondetach; -- called when disassociating this listener from this connection onstatus = listener.onstatus; -- called for status changes (e.g. of SSL/TLS) @@ -540,6 +544,7 @@ local function handleclient( client, ip, port, server, pattern, listener, sslctx interface.eventwritetimeout = false end end + interface:onpredrain(); interface.writebuffer = { t_concat(interface.writebuffer) } local succ, err, byte = interface.conn:send( interface.writebuffer[1], 1, interface.writebufferlen ) --vdebug( "write data:", interface.writebuffer, "error:", err, "part:", byte ) |