Merge pull request #16 from 10gen/demo

Add demo scripts

Author: aherlihy

packages/compass-serverstats/demo/rtss_demo.py

@@ -0,0 +1,58 @@
+import random
+import string
+import sys
+import time
+
+from pymongo import MongoClient
+
+port = 27017
+if len(sys.argv) == 2:
+    port = int(sys.argv[1])
+elif len(sys.argv) > 2:
+    print("usage: rtss_demo.py <port>")
+
+collections = ['coll1', 'coll2', 'coll3']
+dbs = ['db1', 'db2', 'db3']
+
+# Operations Chart + Network In/Out
+def many_ops(m):
+    coll = m[dbs[random.randint(0, len(dbs) - 1)]][collections[random.randint(0, len(collections) - 1)]]
+    [coll.update_one({"x": 1}, {"$inc": {"x": 1}}) for _ in range(
+        random.randint(0,100))]
+    [coll.find_one({"x": 1}) for _ in range(random.randint(0, 100))]
+    [coll.find_one({"x": 1}) for _ in range(random.randint(0, 100))]
+    [coll.find_one({"x": 1}) for _ in range(random.randint(0, 100))]
+    [coll.insert_one({"x": 1}) for _ in range(random.randint(0, 100))]
+    [coll.delete_one({"x": 1}) for _ in range(random.randint(0, 100))]
+
+# Network Connections
+def many_connections():
+    new_connections = random.randint(0, 5)
+    connections = [MongoClient(port=port) for _ in range(new_connections)]
+    time.sleep(random.randint(0, 2))
+    for m in connections:
+        m.close()
+
+# Memory
+def move_data(m):
+    randdata = ''.join(random.choice(
+        string.ascii_uppercase + string.digits) for _ in range(100))
+    randkey =''.join(random.choice(
+        string.ascii_uppercase + string.digits) for _ in range(100))
+    if random.randint(0, 1):
+        print('inserting')
+        m.test.coll.insert_many(
+            [{randkey: randdata, 'even': random.randint(0, 1)} for _ in range(1000)])
+    else:
+        print('deleting')
+        m.test.coll.delete_many({'even': random.randint(0, 1)})
+
+    if random.randint(0, 1000) == 5:
+        m.test.drop_collection('coll')
+
+while(True):
+    many_ops(client)
+    many_connections()
+    #move_data(client) keep out for now
+    time.sleep(random.randint(0, 3))
+

packages/compass-serverstats/demo/rtss_demo2.py

@@ -0,0 +1,40 @@
+from pymongo import MongoClient
+from multiprocessing import Process
+import random
+import time
+import sys
+
+
+port = 27017
+if len(sys.argv) == 2:
+    port = int(sys.argv[1])
+elif len(sys.argv) > 2:
+    print("usage: rtss_demo2.py <port>")
+
+client = MongoClient(port=port)
+def many_ops():
+    collections = ['coll1', 'coll2', 'coll3']
+    dbs = ['db1', 'db2', 'db3']
+    m = MongoClient(port=port)
+    coll = m[dbs[random.randint(0, len(dbs) - 1)]][collections[random.randint(0, len(collections) - 1)]]
+    [coll.find_one({"x": 1,  '$where': 'function() { sleep(' + str(random.randint(0, 2000)) + '); return true; }'}) for _ in range(random.randint(0, 100))]
+    sys.exit()
+
+try:
+    processes = []
+    while True:
+        p = Process(target=many_ops)
+        p.start()
+        processes.append(p)
+        time.sleep(random.randint(3, 5))
+
+# except KeyboardInterrupt:
+#     for p in processes:
+#         if p != None:
+#             p.kill()
+except:
+    sys.exit();
+#     for p in processes:
+#         if p != None:
+#             p.kill()
+#         raise

packages/compass-serverstats/demo/rtss_demo3.py

@@ -0,0 +1,35 @@
+
+
+from pymongo import MongoClient
+from multiprocessing import Process
+import random
+import time
+import sys
+
+def many_ops():
+    collections = ['coll1', 'coll2', 'coll3']
+    dbs = ['db1', 'db2', 'db3']
+    m = MongoClient()
+    coll = m[dbs[random.randint(0, len(dbs) - 1)]][collections[random.randint(0, len(collections) - 1)]]
+    [coll.find_one({"x": 1,  '$where': 'function() { sleep(2000); return true; }'}) for _ in range(random.randint(0, 100))]
+    sys.exit()
+
+try:
+    processes = []
+    for i in range(100):
+        p = Process(target=many_ops)
+        p.start()
+        processes.append(p)
+        #time.sleep(random.randint(0, 1))
+
+# except KeyboardInterrupt:
+#     for p in processes:
+#         if p != None:
+#             p.kill()
+except Exception as e:
+    print(e)
+    sys.exit();
+#     for p in processes:
+#         if p != None:
+#             p.kill()
+#         raise