From bcdf9bb009ef32dac156ea518c1b113c0773877c Mon Sep 17 00:00:00 2001
From: Miguel Grinberg <miguel.grinberg@gmail.com>
Date: Sun, 13 Dec 2020 23:38:15 +0000
Subject: [PATCH] Performace tuning

---
 socketio/base_manager.py              |  8 +++++---
 tests/performance/README.md           |  5 +++++
 tests/performance/binary_packet.py    | 22 ++++++++++++++++++++++
 tests/performance/json_packet.py      | 19 +++++++++++++++++++
 tests/performance/namespace_packet.py | 19 +++++++++++++++++++
 tests/performance/run.sh              |  7 +++++++
 tests/performance/server_receive.py   | 21 +++++++++++++++++++++
 tests/performance/server_send.py      | 26 ++++++++++++++++++++++++++
 tests/performance/text_packet.py      | 19 +++++++++++++++++++
 9 files changed, 143 insertions(+), 3 deletions(-)
 create mode 100644 tests/performance/README.md
 create mode 100644 tests/performance/binary_packet.py
 create mode 100644 tests/performance/json_packet.py
 create mode 100644 tests/performance/namespace_packet.py
 create mode 100755 tests/performance/run.sh
 create mode 100644 tests/performance/server_receive.py
 create mode 100644 tests/performance/server_send.py
 create mode 100644 tests/performance/text_packet.py

diff --git a/socketio/base_manager.py b/socketio/base_manager.py
index 24cd288..795bb93 100644
--- a/socketio/base_manager.py
+++ b/socketio/base_manager.py
@@ -38,7 +38,7 @@ class BaseManager(object):
 
     def get_participants(self, namespace, room):
         """Return an iterable with the active participants in a room."""
-        for sid, eio_sid in self.rooms[namespace][room].copy().items():
+        for sid, eio_sid in self.rooms[namespace][room]._fwdm.copy().items():
             yield sid, eio_sid
 
     def connect(self, eio_sid, namespace):
@@ -59,8 +59,10 @@ class BaseManager(object):
             pass
 
     def sid_from_eio_sid(self, eio_sid, namespace):
-        if namespace in self.rooms:
-            return self.rooms[namespace][None].inverse.get(eio_sid)
+        try:
+            return self.rooms[namespace][None]._invm[eio_sid]
+        except KeyError:
+            pass
 
     def eio_sid_from_sid(self, sid, namespace):
         if namespace in self.rooms:
diff --git a/tests/performance/README.md b/tests/performance/README.md
new file mode 100644
index 0000000..afce173
--- /dev/null
+++ b/tests/performance/README.md
@@ -0,0 +1,5 @@
+Performance
+===========
+
+This directory contains several scripts and tools to test the performance of
+the project.
diff --git a/tests/performance/binary_packet.py b/tests/performance/binary_packet.py
new file mode 100644
index 0000000..b9084ee
--- /dev/null
+++ b/tests/performance/binary_packet.py
@@ -0,0 +1,22 @@
+import time
+from socketio import packet
+
+
+def test():
+    p = packet.Packet(packet.EVENT, {'foo': b'bar'})
+    start = time.time()
+    count = 0
+    while True:
+        eps = p.encode()
+        p = packet.Packet(encoded_packet=eps[0])
+        for ep in eps[1:]:
+            p.add_attachment(ep)
+        count += 1
+        if time.time() - start >= 5:
+            break
+    return count
+
+
+if __name__ == '__main__':
+    count = test()
+    print('binary_packet:', count, 'packets processed.')
diff --git a/tests/performance/json_packet.py b/tests/performance/json_packet.py
new file mode 100644
index 0000000..6861a5c
--- /dev/null
+++ b/tests/performance/json_packet.py
@@ -0,0 +1,19 @@
+import time
+from socketio import packet
+
+
+def test():
+    p = packet.Packet(packet.EVENT, {'foo': 'bar'})
+    start = time.time()
+    count = 0
+    while True:
+        p = packet.Packet(encoded_packet=p.encode())
+        count += 1
+        if time.time() - start >= 5:
+            break
+    return count
+
+
+if __name__ == '__main__':
+    count = test()
+    print('json_packet:', count, 'packets processed.')
diff --git a/tests/performance/namespace_packet.py b/tests/performance/namespace_packet.py
new file mode 100644
index 0000000..4d560a7
--- /dev/null
+++ b/tests/performance/namespace_packet.py
@@ -0,0 +1,19 @@
+import time
+from socketio import packet
+
+
+def test():
+    p = packet.Packet(packet.EVENT, 'hello', namespace='/foo')
+    start = time.time()
+    count = 0
+    while True:
+        p = packet.Packet(encoded_packet=p.encode())
+        count += 1
+        if time.time() - start >= 5:
+            break
+    return count
+
+
+if __name__ == '__main__':
+    count = test()
+    print('namespace_packet:', count, 'packets processed.')
diff --git a/tests/performance/run.sh b/tests/performance/run.sh
new file mode 100755
index 0000000..328e6ae
--- /dev/null
+++ b/tests/performance/run.sh
@@ -0,0 +1,7 @@
+#!/bin/bash
+python text_packet.py
+python binary_packet.py
+python json_packet.py
+python namespace_packet.py
+python server_receive.py
+python server_send.py
diff --git a/tests/performance/server_receive.py b/tests/performance/server_receive.py
new file mode 100644
index 0000000..77f6161
--- /dev/null
+++ b/tests/performance/server_receive.py
@@ -0,0 +1,21 @@
+import time
+import socketio
+
+
+def test():
+    s = socketio.Server(async_handlers=False)
+    start = time.time()
+    count = 0
+    s._handle_eio_connect('123', 'environ')
+    s._handle_eio_message('123', '0')
+    while True:
+        s._handle_eio_message('123', '2["test","hello"]')
+        count += 1
+        if time.time() - start >= 5:
+            break
+    return count
+
+
+if __name__ == '__main__':
+    count = test()
+    print('server_receive:', count, 'packets received.')
diff --git a/tests/performance/server_send.py b/tests/performance/server_send.py
new file mode 100644
index 0000000..6c89911
--- /dev/null
+++ b/tests/performance/server_send.py
@@ -0,0 +1,26 @@
+import time
+import socketio
+
+
+class Server(socketio.Server):
+    def _send_packet(self, eio_sid, pkt):
+        pass
+
+
+def test():
+    s = Server()
+    start = time.time()
+    count = 0
+    s._handle_eio_connect('123', 'environ')
+    s._handle_eio_message('123', '0')
+    while True:
+        s.emit('test', 'hello')
+        count += 1
+        if time.time() - start >= 5:
+            break
+    return count
+
+
+if __name__ == '__main__':
+    count = test()
+    print('server_send:', count, 'packets received.')
diff --git a/tests/performance/text_packet.py b/tests/performance/text_packet.py
new file mode 100644
index 0000000..a28dfd1
--- /dev/null
+++ b/tests/performance/text_packet.py
@@ -0,0 +1,19 @@
+import time
+from socketio import packet
+
+
+def test():
+    p = packet.Packet(packet.EVENT, 'hello')
+    start = time.time()
+    count = 0
+    while True:
+        p = packet.Packet(encoded_packet=p.encode())
+        count += 1
+        if time.time() - start >= 5:
+            break
+    return count
+
+
+if __name__ == '__main__':
+    count = test()
+    print('text_packet:', count, 'packets processed.')