Python实现socket信息发送和监听的简单代码示例

最近在研究boost C++库，用于工作中处理大规模高并发TCP连接数据响应，想测试，也可以用boost::asio库来写，但不利于测试代码的灵活修改。

于是就研究了一下python怎么做。学习过程中看到网上给出的代码示例虽然很基础易懂，对于python零基础的人来说，完全可以稍微复杂些，配上一些注释，就能更多掌握一些socket和python的基本语法知识。

于是根据我的使用需要，修改python的socket server和client的范例代码如下：

server端（单线程）

# -*- coding: cp936 -*-
'''
建立一个python server，监听指定端口，
如果该端口被远程连接访问，则获取远程连接，然后接收数据，
并且做出相应反馈。
'''
if __name__=="__main__":
import socket
print "Server is starting"
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sock.bind(('localhost', 8001))  #配置soket，绑定IP地址和端口号
sock.listen(5) #设置最大允许连接数，各连接和server的通信遵循FIFO原则
print "Server is listenting port 8001, with max connection 5"
while True:  #循环轮询socket状态，等待访问

connection,address = sock.accept()
try:
connection.settimeout(50)
#获得一个连接，然后开始循环处理这个连接发送的信息
'''
如果server要同时处理多个连接，则下面的语句块应该用多线程来处理，
否则server就始终在下面这个while语句块里被第一个连接所占用，
无法去扫描其他新连接了，但多线程会影响代码结构，所以记得在连接数大于1时
下面的语句要改为多线程即可。
'''
while True:
buf = connection.recv(1024)
print "Get value " +buf
if buf == '1':
print "send welcome"
connection.send('welcome to server!')
elif buf!='0':
connection.send('please go out!')
print "send refuse"
else:
print "close"
break  #退出连接监听循环
except socket.timeout:  #如果建立连接后，该连接在设定的时间内无数据发来，则time out
print 'time out'

                print "closing one connection" #当一个连接监听循环退出后，连接可以关掉
                connection.close() 

           

client端，每隔3秒轮番发送不同的数据给server：

if __name__=="__main__":
import socket
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sock.connect(('localhost', 8001))
import time
flag = '1'
while True:
time.sleep(3)
print 'send to server with value: '+ flag
sock.send(flag)
print sock.recv(1024)
flag = (flag=='1') and '2' or '1' #change to another type of value each time

sock.close()

 执行上面两段代码后，可以得到如下结果：
先执行server.py：

C:\Users\YanBo>D:\server.py

Server is starting

Server is listenting port 8001, with max connection 5

执行client后：

C:\Users\YanBo>D:\client.py

send to server with value: 1

welcome to server!

send to server with value: 2

please go out!

send to server with value: 1

welcome to server!

send to server with value: 2

please go out!

.....

server则相应输出：

Get value 1

send welcome

Get value 2

send refuse

.....

如果在linux下，用 

$ lsof -i:8001

则可以列出正在执行的server和client进程，因为他们占用了端口8001 

yanbo@yanbo-linux64-01:~$ lsof -i:8001

COMMAND  PID  USER   FD   TYPE DEVICE SIZE/OFF NODE NAME

python  2413 yanbo    3u  IPv4  12907      0t0  TCP localhost:8001 (LISTEN)

python  2413 yanbo    4u  IPv4  12910      0t0  TCP localhost:8001->localhost:38609 (ESTABLISHED)

python  2469 yanbo    3u  IPv4  12222      0t0  TCP localhost:38609->localhost:8001 (ESTABLISHED)

可以看出，进程2413是server.py 2469是client.py