Mercurial > prosody-modules
diff mod_mam/mod_mam.lua @ 1682:6b2122630b92
mod_mam: Support the mandatory 'complete' attribute by requesting one extra item from storage
| author | Kim Alvefur <zash@zash.se> |
|---|---|
| date | Sun, 03 May 2015 13:21:15 +0200 |
| parents | d20cfc5ba827 |
| children | cc3fad4198bc |
line wrap: on
line diff
--- a/mod_mam/mod_mam.lua Sun May 03 13:20:04 2015 +0200 +++ b/mod_mam/mod_mam.lua Sun May 03 13:21:15 2015 +0200 @@ -124,7 +124,7 @@ local data, err = archive:find(origin.username, { start = qstart; ["end"] = qend; -- Time range with = qwith; - limit = qmax; + limit = qmax + 1; before = before; after = after; reverse = reverse; total = true; @@ -141,7 +141,14 @@ -- Wrap it in stuff and deliver local fwd_st, first, last; + local count = 0; + local complete = "true"; for id, item, when in data do + count = count + 1; + if count > qmax then + complete = nil; + break; + end fwd_st = st.message(msg_reply_attr) :tag("result", { xmlns = xmlns_mam, queryid = qid, id = id }) :tag("forwarded", { xmlns = xmlns_forward }) @@ -163,7 +170,7 @@ if reverse then first, last = last, first; end origin.send(st.message(msg_reply_attr) - :tag("fin", { xmlns = xmlns_mam, queryid = qid }) + :tag("fin", { xmlns = xmlns_mam, queryid = qid, complete = complete }) :add_child(rsm.generate { first = first, last = last, count = count })); return true;