# HG changeset patch # User Matthew Wild # Date 1431532647 -3600 # Node ID 343ca80ceb36e76d25367cf86a4491ba47ca9291 # Parent 1089f45c5e67e5188e1a795b9ba4e3d32066c5e3 util.queue: Small fast FIFO/ringbuffer/queue library diff -r 1089f45c5e67 -r 343ca80ceb36 util/queue.lua --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/util/queue.lua Wed May 13 16:57:27 2015 +0100 @@ -0,0 +1,54 @@ +-- Prosody IM +-- Copyright (C) 2008-2015 Matthew Wild +-- Copyright (C) 2008-2015 Waqas Hussain +-- +-- This project is MIT/X11 licensed. Please see the +-- COPYING file in the source package for more information. +-- + +-- Small ringbuffer library (i.e. an efficient FIFO queue with a size limit) +-- (because unbounded dynamically-growing queues are a bad thing...) + +local have_utable, utable = pcall(require, "util.table"); -- For pre-allocation of table + +local function new(size) + -- Head is next insert, tail is next read + local head, tail = 1, 1; + local items = 0; -- Number of stored items + local t = have_utable and utable.create(size, 0) or {}; -- Table to hold items + + return { + size = size; + count = function (self) return items; end; + push = function (self, item) + if items >= size then + return nil, "queue full"; + end + t[head] = item; + items = items + 1; + head = (head%size)+1; + return true; + end; + pop = function (self) + if items == 0 then + return nil; + end + local item; + item, t[tail] = t[tail], 0; + tail = (tail%size)+1; + items = items - 1; + return item; + end; + peek = function (self) + if items == 0 then + return nil; + end + return t[tail]; + end; + }; +end + +return { + new = new; +}; +