在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)
可以在编译器中运行服务端程序,然后使用命令行方式运行客户端程序进行测试。