【python】python的socket编程

之前Bavon有问过我能否用python去实现一个测试客户端实现对于服务端的测试，当时时间比较紧，所以没有给出具体的方案，最近打算总结下之前的工作，把这块的内容简单的总结一下，慢慢的详细，当然，如果有时间的话。

这里先给出一个python实现的socket的例子：

server端的例子：

import sys
import socket

def main():
    sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    sock.bind(('127.0.0.1', 62001))
    sock.listen(5)
    while True:
        con, addr = sock.accept()
        print "connect from:" + str(addr)
        try:
            con.settimeout(5)
            buf = con.recv(1024)
            strReturn = ""
            try:
                intTmp = int(buf)
                strReturn = "welcome to this simple server:" + str(intTmp + 1)
            except:
                strReturn = "welcome to this simple server:" + buf
            con.send(strReturn)
        except socket.timeout:
            print "transfer time out!"
        con.close()

if __name__ == "__main__":
    main()

client端的例子：

import sys
import socket
import time

def main():
    if len(sys.argv) != 3:
        print "usage:python client.py sleeptime sendbuf"
        sys.exit(1)
    intSleepTime = int(sys.argv[1])
    strSendBuf = sys.argv[2]
    sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    sock.connect(('127.0.0.1', 62001))
    time.sleep(intSleepTime)
    sock.send(strSendBuf)
    strGet = sock.recv(1024)
    print strGet
    sock.close()

if __name__ == "__main__":
    main()

例子上当然都很简单，关于具体的过程解释，网上一搜一大把，这里就不详细解释了。

熟悉了过后，按照自己的通信协议去做丰富即可，当然效率这块按照这个例子去做肯定是非常的底下的，不过测试桩的编写大部分是用来做功能测试的，对于效率的要求也不是那么高，真的需要高性能的做性能测试或者压力测试的话，还是老老实实的用c去实现一个测试端或者继续深入的研究python。