python通信测试_python http(通讯一个测试的例子Teraservertest)

#!-*- coding: cp936 -*-"

#Teraservertest

import threading, time, socket, random,sys

import base64,zipfile

thread_count =500

TEST_COUNT = 1

SPTIME=0.02

ISLOG=0

MAX_PAGE = 10000

HOST = "192.168.6.12"

PORT=5534

h={}

b={}

h[1] =[ "POST /httpEntry HTTP/1.1/r/n" ,/

"X-Online-Host: 192.168.6.12:5534/r/n" ,/

"Content-Length: 32/r/n"  ,/

"Accept: */*/r/n" ,/

"Host: 192.168.1.26:5533/r/n" ,/

"x-up-calling-line-id: 13977777777/r/n",

"x-extern-ver: 1.00/r/n",

"x-extern-msgid: 002/r/n",

"x-extern-devid: 01/r/n",

"x-extern-control-flag: 00000011/r/n",

"x-extern-branch-id: 1.02.01/r/n",

"x-extern-platform-id: 03/r/n" ,

"Connection: close/r/n/r/n"

]

#"x-extern-enterprise-id: 0000/r/n" ,/

b[1]='WZM                 1234                '

devid=['1.02.03',/

'1.02.02',/

'1.02.01']

FMT1=  '%Y%m%d'

def ctime():

FMT = '%Y-%m-%d %H:%M:%S'

return time.strftime(FMT, time.localtime())

def log(text):

print ctime(), 'log:', text

sys.stdout.flush()

def message(text):

print ctime(), 'msg:', text.encode('utf-8')

sys.stdout.flush()

def log_message(path,hit,text=''):

file = open(path, 'a')

file.write('%s%s/t%s' % (ctime(),hit,text))

file.close()

def makebody(data):

return base64.b64encode(zipfile.zlib.compress(data))

def initlog(logfile):  #日志

import logging

logger = logging.getLogger()

hdlr = logging.FileHandler(logfile)

formatter = logging.Formatter('%(asctime)s %(levelname)s %(message)s')

hdlr.setFormatter(formatter)

logger.addHandler(hdlr)

logger.setLevel(logging.ERROR)

return logger

err_log=initlog("err_log.log")

class RequestThread(threading.Thread):

def __init__(self, thread_name,islog=0):

threading.Thread.__init__(self)

self.test_count = 0

self.not_recv=0

self.terminalid=thread_name

self.islog=islog

self.path=time.strftime(FMT1, time.localtime())+thread_name+'.log' #日志文件名

def run(self):

i = 0

while i < TEST_COUNT:

j=1#random.randrange(2,5)

HEADS=h[j]

tb=b[j]

if j==1:  #订单上传的流水号要更新

seq=time.strftime('%y%m%d%H%M%S', time.localtime())

tb=seq+seq+tb

BODY=makebody(tb)

HEADS[2]="Content-Length: "+str(len(BODY))+"/r/n" #包体长度

HEADS[11]="x-extern-devid: "+devid[random.randrange(0,2)]+"/r/n"  #设备号random.randrange(0,12)

self.connectSocket(HOST,PORT,HEADS,BODY)

i += 1

time.sleep(2)

def test_performace(self):

conn = httplib.HTTPConnection(SERVER_NAME)

try:

conn.request("POST", url,BODY,HEADS)

rsps = conn.getresponse()

if rsps.status == 200:

data = rsps.read()

print data

except:

pass

conn.close()

def connectSocket(self, host, textport, heads,body):

try:

s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

except socket.error, e:

log( "%s Strange error creating socket: %s" % (self.terminalid,e))

return

try:

port = int(textport)

except ValueError:

# That didn't work.  Look it up instread.

try:

port = socket.getservbyname(textport, 'tcp')

except socket.error, e:

log( "Couldn't find your port: %s" % e)

return

try:

s.connect((host, port))

#print host, port,self.islog, "*"*30

if self.islog:

log_message(self.path,'终端连接成功/r/n')

except socket.gaierror, e:

log( "Address-related error connecting to server: %s" % e)

return

except socket.error, e:

log( "Connection error: %s" % e)

return

try:

for data in heads:  #往SOCKET写HTTP数据头

s.sendall(data)

if self.islog:

log_message(self.path,"发送报头数据:",data)

sn=self.gaussrandom(SPTIME,0.1)

time.sleep(sn)

data=body

while True:   #往SOCKET写HTTP数据体

i=random.randrange(1,200)

sd=data[0:i]

if not sd:

break

s.sendall(sd)

if self.islog:

log_message(self.path,"发送包体数据:",data)

data=data[i:]

sn=self.gaussrandom(SPTIME,0.1)

time.sleep(sn)

except socket.error, e:

log( "Error sending data: %s" % e)

s.close()

return

#        try:

#            s.shutdown(1)

#        except socket.error, e:

#            log( "Error sending data (detected by shutdown): %s" % e)

#            s.close()

#            return

data=''

while 1:

try:

s.settimeout(60)

st=time.time()   #开始接收的时间点

buf = s.recv(1024)

if not buf :

break

data=data+buf

if self.isrecv(data):#如果数据接收完成退出接收

break

sn=self.gaussrandom(SPTIME,0.1)

time.sleep(sn)

except socket.timeout,e:

log( "%s Error receiving data: %s,sptime: %f" % (self.terminalid,e,time.time()-st))

break

except socket.error, e:

log( self.terminalid+"Error receiving data(not time out): %s,sptime: %f" % (e,time.time()-st))

break

if not data:

self.not_recv+=1

log(self.terminalid+"未收到返回业务号 %s:,已等待时间: %f,可能被服务器断开" % (heads[9],time.time()-st))

if self.islog:

log_message(self.path,"接收到数据:",data)

s.close()

self.test_count += 1

def gaussrandom(self,mu,sigma):#生成正态分布的随机数

return abs(random.gauss(mu,sigma))

def isrecv(self,data):#判断是否接收所有数据,暂时不用

ls=data.split('/r/n/r/n')

if len(ls)==2:

heads=ls[0].split('/r/n')

tmplist=[element for element in heads if element.find('-Length:')>0]

lenlist=tmplist[0].split(': ')

if int(lenlist[1])==len(ls[1]):

return True

else:

return False

else:

return False

#    def __del__(self):

#        self.log_file.close()

if __name__=="__main__":

if len(sys.argv)<> 7:

print '参数不正确,请输入线程数,执行数,间隔时间,是否写日志'

print "usage: performancetest 500 100 0.01 0 127.0.0.1 9877"

sys.exit(1)

thread_count = int(sys.argv[1])

TEST_COUNT = int(sys.argv[2])

SPTIME=float(sys.argv[3])

ISLOG=int(sys.argv[4])

HOST=sys.argv[5]

PORT=sys.argv[6]

start_time = time.time()

threads = []

# 并发的线程数

i = 0

while i < thread_count:

t = RequestThread("thread" + str(i),ISLOG) #线程,是否写日志文件

threads.append(t)

t.setDaemon(True)  #线程结束自动关闭

t.start()

i += 1

# 接受统计的命令

word = ""

while True:

word = raw_input("请输入s实时统计结果,e退出测试:")

try:

if word.rstrip() == "s":

time_span = time.time() - start_time

all_count = 0.00

not_recv=0.00

for t in threads:

all_count += t.test_count

not_recv +=t.not_recv

print '需执行:',thread_count*TEST_COUNT,',已连接:',all_count,'用时:',time_span,'秒'

print '成功返回:',all_count-not_recv,'个,未返回:',not_recv,'个'

print "%s 请求数/秒" % str(all_count / time_span)

print "%s 已连接/需执行;" % str(all_count/(thread_count*TEST_COUNT)),/

"%s 已返回/需执行;" %str((all_count-not_recv)/(thread_count*TEST_COUNT)),/

"%s 已返回/连接数." %str((all_count-not_recv)/all_count),'/r/n'

elif word.rstrip() == "e":

# 准备退出 其实 X 掉 窗口更加容易，没什么浪费的资源

TEST_COUNT = 0

for t in threads:

t.join(0)

break

except:

pass