在python中编写socket服务端模块（一）：使用select

在linux上编写socket服务端程序一般可以用select、poll、epoll三种方式，本文主要介绍使用select编写socket服务端模块与客户端模块。

服务器端程序代码：

import socket
import select
import Queue

server=('10.0.2.15',21345)
#SOCK_STREAM（流套接字）、SOCK_DGRAM（数据报文套接字）、AF_INET（IPv4）	
sock=socket.socket(socket.AF_INET,socket.SOCK_STREAM)
#阻塞与端口复用前后顺序可换
sock.setblocking(False)
#SOL_SOCKET（套接字描述符）、SO_REUSEADDR（端口复用）
sock.setsockopt(socket.SOL_SOCKET,socket.SO_REUSEADDR,1)
sock.bind(server)
sock.listen(5)

rlists=[sock]
wlists=[]
msg_que={}

timeout=10

while True: 
    rs,ws,es=select.select(rlists,wlists,rlists,timeout)

    if not(rs or ws or es):
        print 'timeout...'
        continue
    
    #读部分
    for s in rs:
        #看s是否是本机上用于监听的socket，是则接受连接，不是则接收数据
        if s is sock:
            conn,addr=s.accept()
            #conn、addr分别是所接收到的socket对象和对方端口
            print 'connect by',addr
            conn.setblocking(False)
            rlists.append(conn)
            #使用字典将conn与一个队列相对应
            msg_que[conn]=Queue.Queue()            
        else:
            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:
            #get_nowait()跟get(0)一样
            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]    

客服端程序代码：

# -*- coding: utf-8 -*-  
#使用utf-8在命令行窗口里才能运行
import socket
import time

#getpeername()获得socket对方的地址

server=('10.0.2.15',21345)

msg=['hello','welcome','xiaoming','zhangsan','list','liuliu']

socks=[]
for i in range(10):
    sock=socket.socket(socket.AF_INET,socket.SOCK_STREAM)
    socks.append(sock)
    
for s in socks:
    s.connect(server)

counter =0
for m in msg:
    for s in socks:
        s.send('%d send %s' %(counter,m))
        counter=counter+1
    for s in socks:
        data =s.recv(1024)
        print '%s echo %s' %(s.getpeername(),data)
        if not data:
            s.close()
    time.sleep(2)

可以在编译器中运行服务端程序，然后使用命令行方式运行客户端程序进行测试。