selectors and non-blocking socket

Author: CoderHuo

studyLib/socket/my_tcp_client.py

@@ -2,21 +2,22 @@
 # -*- coding: utf-8 -*-
 
 import socket
+import time
 
 __author__ = 'Mr.Huo'
 
 
-
 def main():
-    server_addr = ('146.11.22.166', 20000)
+    server_addr = ('146.11.22.150', 20000)
     client = socket.socket()
     client.connect(server_addr)
     send_data1 = [b'1aheuo\r\n', b'2ashxao\r\n', b'3bhaua\r\n']
     print(client.getpeername())
-    for data in send_data1:
-        client.send(data)
-        print(client.recv(1024))
-    client.send(b'exit\r\n')
+    while True:
+        for data in send_data1:
+            client.send(data)
+            print(client.recv(1024))
+            time.sleep(2)
     client.close()
     pass
 

studyLib/socket/network_setting.py

@@ -10,3 +10,8 @@
 UDP_PORT_C = 10002
 TCP_PORT_S = 20000
 TCP_PORT_C = 20002
+
+TcpSerAddr = (LOCALIP, TCP_PORT_S)
+TcpCliAddr = (LOCALIP, TCP_PORT_C)
+UdpSerAddr = (LOCALIP, UDP_PORT_S)
+UdpCliAddr = (LOCALIP, UDP_PORT_C)

studyLib/socket/sockServer.py

@@ -29,8 +29,11 @@ def main():
     # TCPServer，IO多路复用，但是StreamRequestHandler还是同步阻塞IO
     # 同步阻塞IO、同步非阻塞IO、IO多路复用、异步IO
     tcpServ = TCP(TcpSerAddr, MyRequestHandler)
+    print(tcpServ.fileno())
     with selectors.SelectSelector() as selector:
         key = selector.register(tcpServ,selectors.EVENT_READ)
+        ready = selector.select()
+        print(ready)
         fd = selector._fileobj_lookup(tcpServ)
         fd1 = selectors._fileobj_to_fd(tcpServ)
         print(tcpServ.fileno())

studyLib/socket/socket_selectors.py

@@ -0,0 +1,52 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+
+import selectors, socket
+from time import ctime
+from network_setting import *
+
+__author__ = 'Mr.Huo'
+
+sel = selectors.DefaultSelector()
+
+
+def accept(sock, mask):
+    conn, addr = sock.accept()
+    print('Accept %s from %s' % (conn, addr))
+    #conn.setblocking(False)
+    sel.register(conn, selectors.EVENT_READ, read)
+
+
+def read(conn, mask):
+    data = conn.recv(1000)
+    if data:
+        print('Recv data from [%s]. Data is [%s]' % (conn.getpeername(), data))
+        #send_data = ('[' + ctime() + '] ').encode() + data
+        send_data = data
+        try:
+            conn.sendall(send_data)
+        except BlockingIOError:
+            print('10035')
+    else:
+        print('Closing ', conn)
+        sel.unregister(conn)
+        conn.close()
+
+
+def main():
+    sock = socket.socket()
+    sock.bind(TcpSerAddr)
+    sock.listen(100)
+    sock.setblocking(False)
+    sel.register(sock, selectors.EVENT_READ, accept)
+
+    while True:
+        events = sel.select()
+        print(events)
+        for key, mask in events:
+            callback = key.data
+            callback(key.fileobj,mask)
+
+
+if __name__ == '__main__':
+    main()