Mercurial > prosody-modules

annotate mod_test_data/mod_test_data.lua @ 4362:116c88c28532

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
mod_http_admin_api: restructure group-related info in API - Return the members of the group right in the get_group_by_id call. This is an O(1) of extra work. - Remove the groups attribute from get_user_by_name as that is O(n) of work and rarely immediately needed. The replacement for the group membership information in the user is for now to use the group API and iterate; future work may fix that.
author Jonas Schäfer <jonas@wielicki.name>
date Wed, 20 Jan 2021 15:30:29 +0100
parents af824168729a
children
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
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