[Python]Socket高级 -- select I/O复用模型（二）

代码： 

客户端：

#!/usr/bin/env python
#coding:utf8
#python2.73 lession7_1_s.py
"""
这是一个多客户端 用来测试
有5个用户 7个消息 一共发送35次
"""

import socket
from time import sleep

msg = ['cs111','cs222','cs333','cs444','cs555','cs666','cs777']

ss = [] #客户端集合

saddr = ('localhost',30001)
for i in range(5):
s = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
ss.append(s)

for s in ss:
s.connect(saddr)

counts = 0
for m in msg:
for s in ss:
#向服务端发送信息
s.send("client num: %s meg: %s"%(counts,m))
counts+=1

for s in ss:
#从服务端获得返回信息
data = s.recv(1024)
print "%s echo %s"%(s.getpeername(),data)
if not data:
s.close()



服务端：

#!/usr/bin/env python
#coding:utf8
#python2.73 lession7_1_s.py
import socket
import select
import Queue

saddr = ('localhost',30001)
sock = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
sock.setblocking(False) #设置socket是非阻塞的
sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
sock.bind(saddr)#绑定地址
sock.listen(10)
#conn,addr = s.accept()
rlists = [sock] #select参数 可写状态集合
wlists = [] #select参数 可读状态集合

msg_que = {}
timeout = 20
#print 'connented with addr:',addr

while rlists:
rs,ws,es = select.select(rlists, wlists, rlists, timeout)
if not (rs or ws or es):
print 'timeout...'
continue
#大体思路就是对所有的返回组进行循环 查看状态作出动作
for s in rs:
if s is sock: #是否是服务端的sock
conn,addr = s.accept()
print 'connect by',addr
conn.setblocking(False)
rlists.append(conn)
msg_que[conn] = Queue.Queue()
else: #客户段的sock接受数据
data = s.recv(1024)
if data:
print data
msg_que[s].put(data)
if s not in wlists:
wlists.append(s)
else:
if s in wlists:
wlists.remove(s)
rlists.remove(s)
s.close()
del msg_que[s]

for s in ws:
try:
msg = msg_que[s].get_nowait()
except Queue.Empty:
print 'msg empty'
wlists.remove(s)
else:
s.send(msg)

for s in es:
print 'except ',s.getpeername()
if s in rlists:
rlists.remove(s)
if s in wlists:
wlists.remove(s)
s.close()
del msg_que[s]