Mercurial > prosody-modules

annotate mod_test_data/mod_test_data.lua @ 4980:da151f9af861

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
replaced 'session' with 'origin' in push_disable session is not defined in this function, trying to access it leads to an error. The correct reference seems to be 'origin'. (This may have come about by copying from the similar code in process_stanza_queue.)
author arcseconds
date Sat, 30 Jul 2022 21:07:47 +1200
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