Diff

mod_storage_gdbm/mod_storage_gdbm.lua @ 1596:b362e6c00fd1

mod_storage_gdbm: Add archive support
author Kim Alvefur <zash@zash.se>
date Sun, 25 Jan 2015 22:18:05 +0100
parent 1595:6288591d5edf
child 1602:3912a53fd4db
line wrap: on
line diff
--- a/mod_storage_gdbm/mod_storage_gdbm.lua	Sun Jan 25 22:17:20 2015 +0100
+++ b/mod_storage_gdbm/mod_storage_gdbm.lua	Sun Jan 25 22:18:05 2015 +0100
@@ -9,10 +9,22 @@
 local gdbm = require"gdbm";
 local path = require"util.paths";
 local lfs = require"lfs";
+local uuid = require"util.uuid".generate;
 local serialization = require"util.serialization";
+local st = require"util.stanza";
 local serialize = serialization.serialize;
 local deserialize = serialization.deserialize;
 
+local function id(v) return v; end
+
+local function is_stanza(s)
+	return getmetatable(s) == st.stanza_mt;
+end
+
+local function ifelse(cond, iftrue, iffalse)
+	if cond then return iftrue; end return iffalse;
+end
+
 local base_path = path.resolve_relative_path(prosody.paths.data, module.host);
 lfs.mkdir(base_path);
 
@@ -33,8 +45,65 @@
 	return deserialize(data);
 end
 
+local archive = {};
+local archive_mt = { __index = archive, suffix = ".adb" };
+
+archive.get = keyval.get;
+archive.set = keyval.set;
+
+function archive:append(username, key, when, with, value)
+	key = key or uuid();
+	local meta = self:get(username);
+	if not meta then
+		meta = {};
+	end
+	local i = meta[key] or #meta+1;
+	local type;
+	if is_stanza(value) then
+		type, value = "stanza", st.preserialize(value);
+	end
+	meta[i] = { key = key, when = when, with = with, type = type };
+	meta[key] = i;
+	local ok, err = self:set(username, meta);
+	if not ok then return nil, err; end
+	ok, err = self:set(key, value);
+	if not ok then return nil, err; end
+	return key;
+end
+
+local deserialize = {
+	stanza = st.deserialize;
+};
+
+function archive:find(username, query)
+	local meta = self:get(username);
+	local r = query.reverse;
+	local d = r and -1 or 1;
+	local s = meta[ifelse(r, query.before, meta.after)];
+	if s then
+		s = s + d;
+	else
+		s = ifelse(r, #meta, 1)
+	end
+	local e = ifelse(r, 1, #meta);
+	return function ()
+		local item, value;
+		for i = s, e, d do
+			item = meta[i];
+			if (not query.with or item.with == query.with)
+			and (not query.start or item.when >= query.start)
+			and (not query["end"] or item.when >= query["end"]) then
+				s = i + d;
+				value = self:get(item.key);
+				return item.key, (deserialize[item.type] or id)(value), item.when, item.with;
+			end
+		end
+	end
+end
+
 local drivers = {
 	keyval = keyval_mt;
+	archive = archive_mt;
 }
 
 function open(_, store, typ)