Python实现socket信息发送和监

发布时间:2019-09-21 10:47:08编辑:auto阅读(1522)

    最近在研究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


关键字