diff options
Diffstat (limited to 'util-src')
-rw-r--r-- | util-src/ringbuffer.c | 23 |
1 files changed, 23 insertions, 0 deletions
diff --git a/util-src/ringbuffer.c b/util-src/ringbuffer.c index 233b9058..10fe0209 100644 --- a/util-src/ringbuffer.c +++ b/util-src/ringbuffer.c @@ -79,6 +79,27 @@ int rb_find(lua_State *L) { } /* + * 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 * (buffer, number, boolean?) -> string */ @@ -210,6 +231,8 @@ int luaopen_util_ringbuffer(lua_State *L) { { lua_pushcfunction(L, rb_find); lua_setfield(L, -2, "find"); + lua_pushcfunction(L, rb_discard); + lua_setfield(L, -2, "discard"); lua_pushcfunction(L, rb_read); lua_setfield(L, -2, "read"); lua_pushcfunction(L, rb_readuntil); |