From 28ba10b6b49b37a6786fde91a9f2fd5b7cd3fa52 Mon Sep 17 00:00:00 2001 From: Kim Alvefur Date: Sat, 24 Feb 2018 14:45:06 +0100 Subject: util.ringbuffer: Add method for discarding buffered data without returning it to lua --- util-src/ringbuffer.c | 23 +++++++++++++++++++++++ 1 file changed, 23 insertions(+) (limited to 'util-src') 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 @@ -78,6 +78,27 @@ int rb_find(lua_State *L) { 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 * (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); -- cgit v1.2.3