基於python+ffmpeg的視訊併發直播壓力測試

通過python與ffmpeg結合使用，可生成進行視訊點播、直播的壓力測試指令碼。可支援不同型別的視訊流，比如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,time
import threading
import subprocess
import os, base64
import sys
import Queue

queue = 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 ip


class 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 "程式退出"