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