author | Kim Alvefur <zash@zash.se> |
Sat, 25 Jan 2020 01:31:49 +0100 | |
changeset 3861 | 8752e5b5dd08 |
parent 3361 | af824168729a |
permissions | -rw-r--r-- |
3361
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 |