Python基础15 - 协程、异步IO

 1 # 服务器端
 2 import sys
 3 import socket
 4 import time
 5 import gevent
 6 from gevent import socket,monkey
 7 monkey.patch_all()
 8 
 9 def server(port):
10     s = socket.socket()
11     s.bind((‘0.0.0.0‘,port))
12     s.listen(500)
13     while True:
14         conn ,addr = s.accept()
15         gevent.spawn(handle_request,conn)
16 
17 def handle_request(conn):
18     try:
19         while True:
20             data = conn.recv(1024)
21             print(‘recv:‘,data.decode())
22             conn.send(data)
23             if not data:
24                 conn.shutdown(socket.SHUT_WR)
25     except Exception as ex:
26         print(ex)
27     finally:
28         conn.close()
29 
30 if __name__ == ‘__main__‘:
31     server(8001)

 1 import socket
 2 
 3 HOST = ‘localhost‘
 4 PORT = 8001
 5 s = socket.socket()
 6 s.connect((HOST,PORT))
 7 while True:
 8     msg = input(‘>>:‘)
 9     s.send(msg.encode(‘utf-8‘))
10     data = s.recv(1024)
11     print(‘recv:‘,data.decode())
12 s.close()

 1 import socket
 2 import threading
 3 
 4 def sock_conn():
 5     client = socket.socket()
 6     client.connect((‘localhost‘,8001))
 7     count = 0
 8     while True:
 9         client.send((‘hello %s‘%count).encode(‘utf-8‘))
10         data = client.recv(1024)
11         print(‘[%s] recv from server:‘%threading.get_ident(),data.decode())
12         count += 1
13     client.close()
14 
15 for i in range(100):
16     t = threading.Thread(target=sock_conn)
17     t.start()

 1 import select
 2 import socket
 3 import queue
 4 
 5 server = socket.socket()
 6 server.bind((‘localhost‘,8001))
 7 server.listen(1000)
 8 
 9 server.setblocking(False) # 不阻塞
10 
11 msg_dic = {}
12 inputs = [server,]
13 outputs = []
14 while True:
15     readable, writeable, exceptional = select.select(inputs, outputs, inputs)
16     for r in readable:
17         if r is server: # 代表来了一个新连接
18             conn,addr = r.accept()
19             print(‘来了个新连接‘,addr)
20             inputs.append(conn)
21             msg_dic[conn] = queue.Queue() # 初始化一个队列，后面存要返回给这个客户端的数据
22         else:
23             try:
24                 data = r.recv(1024)
25                 print(‘收到数据‘,data)
26                 msg_dic[r].put(data)
27                 outputs.append(r) # 放入返回的连接队列里
28             except:break
29 
30     for w in writeable: # 要返回给客户端的连接列表
31         data_to_client = msg_dic[w].get()
32         w.send(data_to_client) # 返回给客户端数据
33         outputs.remove(w) # 确保下次循环的时候，不返回这个已经处理完的连接
34 
35     for e in exceptional:
36         if e in outputs:
37             outputs.remove(e)
38         inputs.remove(e)
39         del msg_dic[e]

 1 import socket
 2 import sys
 3 
 4 messages = [b‘This is the message. ‘,
 5             b‘It will be sent ‘,
 6             b‘in parts.‘,
 7             ]
 8 server_address = (‘127.0.0.1‘, 8001)
 9 
10 socks = [socket.socket() for i in range(100)]
11 
12 print (sys.stderr, ‘connecting to %s port %s‘ % server_address)
13 for s in socks:
14     s.connect(server_address)
15 
16 for message in messages:
17 
18     for s in socks:
19         # print >> sys.stderr, ‘%s: sending "%s"‘ % (s.getsockname(), message)
20         s.send(message)
21 
22     for s in socks:
23         data = s.recv(1024)
24         # print >> sys.stderr, ‘%s: received "%s"‘ % (s.getsockname(), data)
25         if not data:
26             print (sys.stderr, ‘closing socket‘, s.getsockname())
27             s.close()