annotate bin/load_test_rdfdb @ 1917:713e56e8e2b9

rdfdb load tester Ignore-this: ca714a9f84b3c0f0aa05b3805cd7344b
author Drew Perttula <drewp@bigasterisk.com>
date Sat, 01 Jun 2019 20:07:27 +0000
parents
children 4718ca6f812e
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
713e56e8e2b9 rdfdb load tester
Drew Perttula <drewp@bigasterisk.com>
parents:
diff changeset
12 class BusyClient:
713e56e8e2b9 rdfdb load tester
Drew Perttula <drewp@bigasterisk.com>
parents:
diff changeset
13 def __init__(self, subj, rate):
713e56e8e2b9 rdfdb load tester
Drew Perttula <drewp@bigasterisk.com>
parents:
diff changeset
14 self.subj = subj
713e56e8e2b9 rdfdb load tester
Drew Perttula <drewp@bigasterisk.com>
parents:
diff changeset
15 self.rate = rate
713e56e8e2b9 rdfdb load tester
Drew Perttula <drewp@bigasterisk.com>
parents:
diff changeset
16
713e56e8e2b9 rdfdb load tester
Drew Perttula <drewp@bigasterisk.com>
parents:
diff changeset
17 self.graph = SyncedGraph(networking.rdfdb.url, "collector")
713e56e8e2b9 rdfdb load tester
Drew Perttula <drewp@bigasterisk.com>
parents:
diff changeset
18 self.graph.initiallySynced.addCallback(self.go)
713e56e8e2b9 rdfdb load tester
Drew Perttula <drewp@bigasterisk.com>
parents:
diff changeset
19 def go(self, _):
713e56e8e2b9 rdfdb load tester
Drew Perttula <drewp@bigasterisk.com>
parents:
diff changeset
20 task.LoopingCall(self.loop).start(1 / self.rate)
713e56e8e2b9 rdfdb load tester
Drew Perttula <drewp@bigasterisk.com>
parents:
diff changeset
21
713e56e8e2b9 rdfdb load tester
Drew Perttula <drewp@bigasterisk.com>
parents:
diff changeset
22
713e56e8e2b9 rdfdb load tester
Drew Perttula <drewp@bigasterisk.com>
parents:
diff changeset
23 def loop(self):
713e56e8e2b9 rdfdb load tester
Drew Perttula <drewp@bigasterisk.com>
parents:
diff changeset
24 self.graph.patchObject(showconfig.showUri() + '/loadTestContext',
713e56e8e2b9 rdfdb load tester
Drew Perttula <drewp@bigasterisk.com>
parents:
diff changeset
25 subject=self.subj,
713e56e8e2b9 rdfdb load tester
Drew Perttula <drewp@bigasterisk.com>
parents:
diff changeset
26 predicate=L9['time'],
713e56e8e2b9 rdfdb load tester
Drew Perttula <drewp@bigasterisk.com>
parents:
diff changeset
27 newObject=Literal(str(time.time())))
713e56e8e2b9 rdfdb load tester
Drew Perttula <drewp@bigasterisk.com>
parents:
diff changeset
28
713e56e8e2b9 rdfdb load tester
Drew Perttula <drewp@bigasterisk.com>
parents:
diff changeset
29 def main():
713e56e8e2b9 rdfdb load tester
Drew Perttula <drewp@bigasterisk.com>
parents:
diff changeset
30 log.setLevel(logging.INFO)
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 clients = [BusyClient(L9['loadTest_%d' % i], 20) for i in range(10)]
713e56e8e2b9 rdfdb load tester
Drew Perttula <drewp@bigasterisk.com>
parents:
diff changeset
33 reactor.run()
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 if __name__ == "__main__":
713e56e8e2b9 rdfdb load tester
Drew Perttula <drewp@bigasterisk.com>
parents:
diff changeset
36 main()
713e56e8e2b9 rdfdb load tester
Drew Perttula <drewp@bigasterisk.com>
parents:
diff changeset
37