Mercurial > prosody-modules

annotate mod_test_data/mod_test_data.lua @ 4537:53ee391ca689

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
mod_smacks: Fix traceback due to session being destroyed in send() Sending something can cause the OS to notice that the connection is dead and then the connection can be dead at this point. More likely if opportunistic_writes is enabled.
author Kim Alvefur <zash@zash.se>
date Thu, 01 Apr 2021 11:35:26 +0200
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