Mercurial > prosody-modules
annotate mod_test_data/mod_test_data.lua @ 4876:0f5f2d4475b9
mod_http_xep227: Add support for import via APIs rather than direct store manipulation
In particular this transitions PEP nodes and data to be imported via mod_pep's
APIs, fixing issues with importing at runtime while PEP data may already be
live in RAM.
Next obvious candidate for this approach is rosters, so clients get immediate
roster pushes and other special handling (such as emitting subscribes to reach
the desired subscription state).
author | Matthew Wild <mwild1@gmail.com> |
---|---|
date | Tue, 18 Jan 2022 17:01:18 +0000 |
parents | af824168729a |
children |
rev | line source |
---|---|
3357
af824168729a
mod_test_data: New module to generate test data in Prosody's data store
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
1 local users = { "fezziwig", "badger", "nupkins", "pumblechook", "rouncewell" }; |
af824168729a
mod_test_data: New module to generate test data in Prosody's data store
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
2 local host = "localhost"; |
af824168729a
mod_test_data: New module to generate test data in Prosody's data store
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
3 |
af824168729a
mod_test_data: New module to generate test data in Prosody's data store
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
4 local id = require "util.id"; |
af824168729a
mod_test_data: New module to generate test data in Prosody's data store
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
5 local st = require "util.stanza"; |
af824168729a
mod_test_data: New module to generate test data in Prosody's data store
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
6 local sm = require "core.storagemanager"; |
af824168729a
mod_test_data: New module to generate test data in Prosody's data store
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
7 |
af824168729a
mod_test_data: New module to generate test data in Prosody's data store
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
8 -- Return a random number from 1..max excluding n |
af824168729a
mod_test_data: New module to generate test data in Prosody's data store
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
9 function random_other(n, max) return ((math.random(1, max-1)+(n-1))%max)+1; end |
af824168729a
mod_test_data: New module to generate test data in Prosody's data store
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
10 |
af824168729a
mod_test_data: New module to generate test data in Prosody's data store
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
11 local new_time; |
af824168729a
mod_test_data: New module to generate test data in Prosody's data store
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
12 do |
af824168729a
mod_test_data: New module to generate test data in Prosody's data store
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
13 local _current_time = os.time(); |
af824168729a
mod_test_data: New module to generate test data in Prosody's data store
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
14 function new_time() |
af824168729a
mod_test_data: New module to generate test data in Prosody's data store
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
15 _current_time = _current_time + math.random(1, 3600); |
af824168729a
mod_test_data: New module to generate test data in Prosody's data store
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
16 return _current_time; |
af824168729a
mod_test_data: New module to generate test data in Prosody's data store
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
17 end |
af824168729a
mod_test_data: New module to generate test data in Prosody's data store
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
18 end |
af824168729a
mod_test_data: New module to generate test data in Prosody's data store
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
19 |
af824168729a
mod_test_data: New module to generate test data in Prosody's data store
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
20 function module.command(arg) --luacheck: ignore arg |
af824168729a
mod_test_data: New module to generate test data in Prosody's data store
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
21 sm.initialize_host(host); |
af824168729a
mod_test_data: New module to generate test data in Prosody's data store
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
22 local archive = sm.open(host, "archive", "archive"); |
af824168729a
mod_test_data: New module to generate test data in Prosody's data store
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
23 |
af824168729a
mod_test_data: New module to generate test data in Prosody's data store
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
24 for _ = 1, 100000 do |
af824168729a
mod_test_data: New module to generate test data in Prosody's data store
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
25 local random = math.random(1, #users); |
af824168729a
mod_test_data: New module to generate test data in Prosody's data store
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
26 local user, contact = users[random], users[random_other(random, #users)]; |
af824168729a
mod_test_data: New module to generate test data in Prosody's data store
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
27 local user_jid, contact_jid = user.."@"..host, contact.."@"..host; |
af824168729a
mod_test_data: New module to generate test data in Prosody's data store
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
28 |
af824168729a
mod_test_data: New module to generate test data in Prosody's data store
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
29 local stanza = st.message({ to = contact_jid, from = user_jid, type="chat" }) |
af824168729a
mod_test_data: New module to generate test data in Prosody's data store
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
30 :tag("body"):text(id.long()); |
af824168729a
mod_test_data: New module to generate test data in Prosody's data store
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
31 |
af824168729a
mod_test_data: New module to generate test data in Prosody's data store
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
32 archive:append(user, nil, stanza, new_time(), contact_jid) |
af824168729a
mod_test_data: New module to generate test data in Prosody's data store
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
33 |
af824168729a
mod_test_data: New module to generate test data in Prosody's data store
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
34 local stanza2 = st.clone(stanza); |
af824168729a
mod_test_data: New module to generate test data in Prosody's data store
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
35 stanza2.attr.from, stanza2.attr.to = stanza.attr.to, stanza.attr.from; |
af824168729a
mod_test_data: New module to generate test data in Prosody's data store
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
36 archive:append(contact, nil, stanza2, new_time(), user_jid) |
af824168729a
mod_test_data: New module to generate test data in Prosody's data store
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
37 end |
af824168729a
mod_test_data: New module to generate test data in Prosody's data store
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
38 end |