annotate bin/load_test_rdfdb @ 2035:f9faa2ec824f

WIP collector using starlette and vite
author drewp@bigasterisk.com
date Sat, 09 Apr 2022 02:49:14 -0700
parents 4718ca6f812e
children
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
1917
713e56e8e2b9 rdfdb load tester
Drew Perttula <drewp@bigasterisk.com>
parents:
diff changeset
1 #!bin/python
713e56e8e2b9 rdfdb load tester
Drew Perttula <drewp@bigasterisk.com>
parents:
diff changeset
2 from run_local import log
713e56e8e2b9 rdfdb load tester
Drew Perttula <drewp@bigasterisk.com>
parents:
diff changeset
3 from twisted.internet import reactor, task, defer
713e56e8e2b9 rdfdb load tester
Drew Perttula <drewp@bigasterisk.com>
parents:
diff changeset
4 from rdflib import URIRef, Literal
713e56e8e2b9 rdfdb load tester
Drew Perttula <drewp@bigasterisk.com>
parents:
diff changeset
5 from twisted.internet.defer import ensureDeferred
713e56e8e2b9 rdfdb load tester
Drew Perttula <drewp@bigasterisk.com>
parents:
diff changeset
6 from rdfdb.syncedgraph import SyncedGraph
713e56e8e2b9 rdfdb load tester
Drew Perttula <drewp@bigasterisk.com>
parents:
diff changeset
7 import time, logging
713e56e8e2b9 rdfdb load tester
Drew Perttula <drewp@bigasterisk.com>
parents:
diff changeset
8
713e56e8e2b9 rdfdb load tester
Drew Perttula <drewp@bigasterisk.com>
parents:
diff changeset
9 from light9 import networking, showconfig
713e56e8e2b9 rdfdb load tester
Drew Perttula <drewp@bigasterisk.com>
parents:
diff changeset
10 from light9.namespaces import L9
713e56e8e2b9 rdfdb load tester
Drew Perttula <drewp@bigasterisk.com>
parents:
diff changeset
11
1927
4718ca6f812e autoformat
Drew Perttula <drewp@bigasterisk.com>
parents: 1917
diff changeset
12
1917
713e56e8e2b9 rdfdb load tester
Drew Perttula <drewp@bigasterisk.com>
parents:
diff changeset
13 class BusyClient:
1927
4718ca6f812e autoformat
Drew Perttula <drewp@bigasterisk.com>
parents: 1917
diff changeset
14
1917
713e56e8e2b9 rdfdb load tester
Drew Perttula <drewp@bigasterisk.com>
parents:
diff changeset
15 def __init__(self, subj, rate):
713e56e8e2b9 rdfdb load tester
Drew Perttula <drewp@bigasterisk.com>
parents:
diff changeset
16 self.subj = subj
713e56e8e2b9 rdfdb load tester
Drew Perttula <drewp@bigasterisk.com>
parents:
diff changeset
17 self.rate = rate
1927
4718ca6f812e autoformat
Drew Perttula <drewp@bigasterisk.com>
parents: 1917
diff changeset
18
1917
713e56e8e2b9 rdfdb load tester
Drew Perttula <drewp@bigasterisk.com>
parents:
diff changeset
19 self.graph = SyncedGraph(networking.rdfdb.url, "collector")
713e56e8e2b9 rdfdb load tester
Drew Perttula <drewp@bigasterisk.com>
parents:
diff changeset
20 self.graph.initiallySynced.addCallback(self.go)
1927
4718ca6f812e autoformat
Drew Perttula <drewp@bigasterisk.com>
parents: 1917
diff changeset
21
1917
713e56e8e2b9 rdfdb load tester
Drew Perttula <drewp@bigasterisk.com>
parents:
diff changeset
22 def go(self, _):
713e56e8e2b9 rdfdb load tester
Drew Perttula <drewp@bigasterisk.com>
parents:
diff changeset
23 task.LoopingCall(self.loop).start(1 / self.rate)
713e56e8e2b9 rdfdb load tester
Drew Perttula <drewp@bigasterisk.com>
parents:
diff changeset
24
713e56e8e2b9 rdfdb load tester
Drew Perttula <drewp@bigasterisk.com>
parents:
diff changeset
25 def loop(self):
713e56e8e2b9 rdfdb load tester
Drew Perttula <drewp@bigasterisk.com>
parents:
diff changeset
26 self.graph.patchObject(showconfig.showUri() + '/loadTestContext',
713e56e8e2b9 rdfdb load tester
Drew Perttula <drewp@bigasterisk.com>
parents:
diff changeset
27 subject=self.subj,
1927
4718ca6f812e autoformat
Drew Perttula <drewp@bigasterisk.com>
parents: 1917
diff changeset
28 predicate=L9['time'],
4718ca6f812e autoformat
Drew Perttula <drewp@bigasterisk.com>
parents: 1917
diff changeset
29 newObject=Literal(str(time.time())))
4718ca6f812e autoformat
Drew Perttula <drewp@bigasterisk.com>
parents: 1917
diff changeset
30
1917
713e56e8e2b9 rdfdb load tester
Drew Perttula <drewp@bigasterisk.com>
parents:
diff changeset
31
713e56e8e2b9 rdfdb load tester
Drew Perttula <drewp@bigasterisk.com>
parents:
diff changeset
32 def main():
713e56e8e2b9 rdfdb load tester
Drew Perttula <drewp@bigasterisk.com>
parents:
diff changeset
33 log.setLevel(logging.INFO)
713e56e8e2b9 rdfdb load tester
Drew Perttula <drewp@bigasterisk.com>
parents:
diff changeset
34
713e56e8e2b9 rdfdb load tester
Drew Perttula <drewp@bigasterisk.com>
parents:
diff changeset
35 clients = [BusyClient(L9['loadTest_%d' % i], 20) for i in range(10)]
713e56e8e2b9 rdfdb load tester
Drew Perttula <drewp@bigasterisk.com>
parents:
diff changeset
36 reactor.run()
1927
4718ca6f812e autoformat
Drew Perttula <drewp@bigasterisk.com>
parents: 1917
diff changeset
37
4718ca6f812e autoformat
Drew Perttula <drewp@bigasterisk.com>
parents: 1917
diff changeset
38
1917
713e56e8e2b9 rdfdb load tester
Drew Perttula <drewp@bigasterisk.com>
parents:
diff changeset
39 if __name__ == "__main__":
713e56e8e2b9 rdfdb load tester
Drew Perttula <drewp@bigasterisk.com>
parents:
diff changeset
40 main()