1 files changed, 93 insertions, 0 deletions
diff --git a/examples/streams-stress-test.py b/examples/streams-stress-test.py
new file mode 100755
index 0000000..33cc20c
--- /dev/null
+++ b/examples/streams-stress-test.py
@@ -0,0 +1,93 @@
+#!/usr/bin/env python3
+# vim: set sts=2 sw=2 et tw=0 :
+#
+
+import json
+import time
+import random
+import atexit
+import datetime
+import subprocess
+import http.client
+from itertools import zip_longest
+
+PROXY_HOST = "localhost:8000"
+GST_LAUNCH_BASE = "gst-launch-1.0 -e videotestsrc is-live=TRUE ! videoscale ! \
+  video/x-raw, height=120, width=160 ! timeoverlay font-desc=80px ! tee name=t \
+  t. ! queue ! videoconvert ! vp8enc ! webmmux ! tee name=s \
+    "
+GST_LAUNCH_STREAMER = ""
+
+rtp_caps = "application/x-rtp, media=(string)video"
+start_port = 5004
+end_port = 5104
+token_table = {}
+processes = []
+stream_count = 0
+
+for port in range(start_port, end_port+1):
+  someid = "{}-{}".format(datetime.datetime.now().strftime("%H%M%S.%f"),
+                          random.randint(0, 100))
+  clients = "localhost:{}".format(port)
+  token_table.update({someid: clients})
+
+def cleanup():
+  for p in processes:
+    p.term()
+  for sid, clients in token_table.items():
+    revoke_token(conn, sid).read()
+
+def grouper(n, iterable, fillvalue=None):
+  "grouper(3, 'ABCDEFG', 'x') --> ABC DEF Gxx"
+  args = [iter(iterable)] * n
+  return zip_longest(fillvalue=fillvalue, *args)
+
+def add_token(conn, someid, clients):
+  conn.request("POST", "/add-token?sessionid={}&type=http,rtp-udp&udp-clients={}".format(someid, clients))
+  return conn.getresponse()
+
+def list_tokens(conn):
+  conn.request("GET", "/list-tokens")
+  return conn.getresponse()
+
+def revoke_token(conn, someid):
+  conn.request("DELETE", "/revoke-token?sessionid={}".format(someid))
+  return conn.getresponse()
+
+def list_sessionids(conn):
+  r = list_tokens(conn)
+  l = json.loads(r)
+  return [e['sessionid'] for e in l]
+
+def list_udp_clients(conn):
+  r = list_tokens(conn)
+  l = json.loads(r)
+  return [e['udp-clients'] for e in l]
+
+def generate_souphttpclientsink(sid, clients):
+  base = "s. ! queue ! souphttpclientsink location=\"http://" + PROXY_HOST
+  return "{}/{}?type=http,rtp-udp&udp-clients={}\" ".format(base, sid, clients)
+
+conn = http.client.HTTPConnection(PROXY_HOST)
+atexit.register(cleanup)
+
+# Add tokens to valid list
+for sid, clients in token_table.items():
+  add_token(conn, sid, clients).read()
+list_tokens(conn).read()
+
+for group in grouper(10, token_table.items(), fillvalue=(None, None)):
+  sinks = []
+  for sid, clients in group:
+    if not sid or not clients:
+      continue
+    sink = generate_souphttpclientsink(sid, clients)
+    sinks.append(sink)
+    stream_count += 1
+  streamer = GST_LAUNCH_BASE + " ".join(sinks)
+  print(streamer)
+  processes.append(subprocess.Popen(streamer, shell=True))
+  print("Currently streaming {} HTTP streams".format(stream_count))
+  time.sleep(2)
+
+[p.wait() for p in processes]