File

+
−-- 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, "prosody.util.table"); -- For pre-allocation of table
+
−
+
−local function new(size, allow_wrapping)
+
−	-- 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
+
−	--luacheck: ignore 212/self
+
−	return {
+
−		_items = t;
+
−		size = size;
+
−		count = function (self) return items; end;
+
−		push = function (self, item)
+
−			if items >= size then
+
−				if allow_wrapping then
+
−					tail = (tail%size)+1; -- Advance to next oldest item
+
−					items = items - 1;
+
−				else
+
−					return nil, "queue full";
+
−				end
+
−			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;
+
−		replace = function (self, data)
+
−			if items == 0 then
+
−				return self:push(data);
+
−			end
+
−			t[tail] = data;
+
−			return true;
+
−		end;
+
−		items = function (self)
+
−			return function (_, pos)
+
−				if pos >= items then
+
−					return nil;
+
−				end
+
−				local read_pos = tail + pos;
+
−				if read_pos > self.size then
+
−					read_pos = (read_pos%size);
+
−				end
+
−				return pos+1, t[read_pos];
+
−			end, self, 0;
+
−		end;
+
−		consume = function (self)
+
−			return self.pop, self;
+
−		end;
+
−	};
+
−end
+
−
+
−return {
+
−	new = new;
+
−};
+
−