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

util.ringbuffer: Add method for discarding buffered data without returning it to lua

Browse source
This commit is contained in:
Kim Alvefur 2018-02-24 14:45:06 +01:00
parent a92cdf75b1
commit 28ba10b6b4
1 changed files with 23 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

23
util-src/ringbuffer.c
View file

@ -78,6 +78,27 @@ int rb_find(lua_State *L) {
return 0; return 0;
} }
/*
* Move read position forward without returning the data
* (buffer, number) -> boolean
*/
int rb_discard(lua_State *L) {
ringbuffer *b = luaL_checkudata(L, 1, "ringbuffer_mt");
size_t r = luaL_checkinteger(L, 2);
if(r > b->blen) {
lua_pushboolean(L, 0);
return 1;
}
b->blen -= r;
b->rpos += r;
modpos(b);
lua_pushboolean(L, 1);
return 1;
}
/* /*
* Read bytes from buffer * Read bytes from buffer
* (buffer, number, boolean?) -> string * (buffer, number, boolean?) -> string
@ -210,6 +231,8 @@ int luaopen_util_ringbuffer(lua_State *L) {
{ {
lua_pushcfunction(L, rb_find); lua_pushcfunction(L, rb_find);
lua_setfield(L, -2, "find"); lua_setfield(L, -2, "find");
lua_pushcfunction(L, rb_discard);
lua_setfield(L, -2, "discard");
lua_pushcfunction(L, rb_read); lua_pushcfunction(L, rb_read);
lua_setfield(L, -2, "read"); lua_setfield(L, -2, "read");
lua_pushcfunction(L, rb_readuntil); lua_pushcfunction(L, rb_readuntil);