gather

Author: HFrost0

README.md

@@ -45,7 +45,9 @@ official python [selectors](https://docs.python.org/3/library/selectors.html) mo
 - [x] get_event_loop
 - [x] selectors block
 - [x] async sock
-- [ ] gather
+- [x] gather
+- [ ] creat server
+- [ ] cancel
 
 ## quote
 

io_test.py

@@ -3,23 +3,26 @@
 import taio
 
 
-async def main():
+async def client(message):
     sock = socket.socket(family=socket.AF_INET, type=socket.SOCK_STREAM)
     sock.setblocking(False)
     loop = taio.get_event_loop()
     await loop.sock_connect(sock, ('127.0.0.1', 6666))
-    sock.send(b'hey server, Im taio')
+    sock.send(message)
     data = await loop.sock_recv(sock, 1024)
     print(data)
     sock.close()
+    return data
+
+
+async def main():
+    cors = []
+    for i in range(5):
+        cors.append(client(f'{i} taio client'.encode('utf8')))
+    datas = await taio.gather(*cors)
+    print('All: ', datas)
 
 
 if __name__ == '__main__':
     loop = taio.get_event_loop()
-    loop.create_task(main())
-    loop.create_task(main())
-    loop.create_task(main())
-    loop.create_task(main())
-    loop.create_task(main())
-    loop.call_later(1.5, loop.stop)
-    loop.run_forever()
+    loop.run_until_complete(main())

taio/tasks.py

@@ -32,6 +32,36 @@ def __wakeup(self, future):
             self.__step()
 
 
+def gather(*coros_or_futures):
+    loop = events.get_event_loop()
+    if not coros_or_futures:
+        outer = loop.create_future()
+        outer.set_result([])
+        return outer
+
+    def _done_callback(fut):
+        nonlocal nfinished
+        nfinished += 1
+        if nfinished == nfuts:
+            results = []
+            for c in children:
+                results.append(c.result())
+            outer.set_result(results)
+
+    arg_to_fut = {}
+    nfuts = 0
+    nfinished = 0
+    children = []
+    for arg in coros_or_futures:
+        fut = loop.create_task(arg)  # ensure future
+        nfuts += 1
+        arg_to_fut[arg] = fut
+        fut.add_done_callback(_done_callback)
+        children.append(fut)
+    outer = loop.create_future()
+    return outer
+
+
 async def sleep(delay):
     future = futures.Future()
     events.get_event_loop().call_later(delay, future.set_result, None)