mirrors/prosody
1
0
Fork 0
mirror of https://github.com/bjc/prosody.git synced 2025-04-04 05:37:39 +03:00
Code Issues Projects Releases Packages Wiki Activity Actions

util.dbuffer: Add read_until() method

Browse source
This commit is contained in:
Matthew Wild 2021-06-29 13:48:14 +01:00
parent 08253e2671
commit 4e56658eb3
2 changed files with 58 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

14
util/dbuffer.lua
View file

@ -76,6 +76,20 @@ function dbuffer_methods:read(requested_bytes)
return table.concat(chunks);
end
-- Read to, and including, the specified character sequence (return nil if not found)
function dbuffer_methods:read_until(char)
local buffer_pos = 0;
for i, chunk in self.items:items() do
local start = 1 + self.front_consumed;
local char_pos = chunk:find(char, start, true);
if char_pos then
return self:read(buffer_pos + (char_pos - start) + #char);
end
buffer_pos = buffer_pos + #chunk;
end
return nil;
end
function dbuffer_methods:discard(requested_bytes)
if requested_bytes > self._length then
return nil;