[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]