python+ffmpeg視頻並發直播壓力測試，pythonffmpeg

python+ffmpeg視頻並發直播壓力測試，pythonffmpeg

通過python與ffmpeg結合使用，可產生進行ApsaraVideo for VOD、直播的壓力測試指令碼。可支援不同類型的視頻流，比如rtmp或者hls形式。
通過如下方式執行指令碼：python multiRealPlay.py [rtmp|http] [thread counts] [interval Time]
[rtmp | http]：視頻播放的不同形式
[thread counts]：並發線程數
[interval Time]：啟動每個線程的間隔時間

代碼：

#!/usr/bin/python# -*- coding: utf-8 -*-  '''Created on 2015年7月22日@author: LiBiao'''import datetime,timeimport threadingimport subprocessimport os, base64import sysimport Queuequeue = Queue.Queue()#需要手動設定的參數#啟動序號SLEEP_TIME = 0#直播地址FULL_ADDR = {}#需要手動設定的參數RTMP_ADDR = 'rtmp://192.168.1.208:1935/live/'HTTP_ADDR = 'http://192.168.1.208:80/live'liveID = '100002750'  #來自於萬視無憂中建立的直播urlKey = 'a1e5c680f7bfc85851de8ab2e63b0a33'  #來自於萬視無憂安全設定模組liveResCode = '71ac6c06d3'  #直播源碼#產生MD5值def getMD5_Value(inputdata):  try:    import hashlib    hash = hashlib.md5(inputdata.encode('utf-8'))  except ImportError:    #for python << 2.5    import md5    hash = md5.new()  return hash.hexdigest()#直播地址組裝def build_live_addr():  t = time.strftime('%Y%m%d%H%M%S',time.localtime())[2:]  data = '%s#%s#%s' %(liveID, t, urlKey)  secret = getMD5_Value(data)  rtmp_addr = '%s%s?liveID=%s&time=%s&secret=%s' %(RTMP_ADDR, liveResCode, liveID, t, secret)  http_addr = '%s/%s/playlist.m3u8?liveID=%s&time=%s&secret=%s' %(HTTP_ADDR, liveResCode, liveID, t, secret)  FULL_ADDR['rtmp'] = rtmp_addr  FULL_ADDR['http'] = http_addr  return FULL_ADDR#擷取本機ip地址，用來產生區別於其他機器的資料def get_local_ip():  try:    ip = os.popen("ifconfig | grep 'inet addr' | awk '{print $2}'").read()    ip = ip[ip.find(':') + 1:ip.find('\n')]  except Exception,e:    print e  return ipclass Video_To_Live(threading.Thread):  def __init__(self,queue):    threading.Thread.__init__(self)    self.queue = queue  def run(self):    liveAddr = self.queue.get()    #print liveAddr    try:      print liveAddr      subprocess.call('./ffmpeg -i \"%s\" -c:v copy -c:a copy -bsf:a aac_adtstoasc -y -f flv -timeout 4000 /dev/null 2>/dev/null' %liveAddr,stdout=subprocess.PIPE,shell=True)    except Exception as e:      wiriteLog('ERROR',str(e))    self.queue.task_done()if __name__ == "__main__":  time.sleep(SLEEP_TIME)  parser = argparse.ArgumentParser(description = "Live Play")  parser.add_argument('--liveType',action = "store",dest = "liveType",required = False)  parser.add_argument('--pnum',action = "store",dest = "pnum",type = int,required = False)  parser.add_argument('--itime',action = "store",dest = "itime",required = False)  given_args = parser.parse_args()  liveType = given_args.liveType   threadNum = given_args.pnum  intervalTime = given_args.itime  print "%d 個 %s 進程開始運行........" %(threadNum, Video_To_Live)  for i in xrange(threadNum):    videotolive = Video_To_Live(queue)    videotolive.setDaemon(True)    videotolive.start()  for i in xrange(threadNum):    if liveType in ["http","rtmp"]:        addr = build_live_addr()      liveaddr = addr[liveType]    queue.put(liveaddr)    time.sleep(intervalTime)  queue.join()  print "進程退出"

以上就是本文的全部內容，希望對大家的學習有所協助，也希望大家多多支援幫客之家。