python+ffmpeg视频并发直播压力测试
作者:LZT-LHZ 时间:2023-09-02 10:30:33
通过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 "进程退出"
来源:http://blog.csdn.net/u011881908/article/details/50944555
标签:python,ffmpeg,视频,直播
0
投稿
猜你喜欢
Jsp+Servlet实现文件上传下载 删除上传文件(三)
2023-06-27 16:29:29
pycharm快捷键汇总
2022-12-30 22:43:40
如何在浏览器地址栏显示自己的Favicons.ico图标
2007-10-22 11:45:00
SQL Server触发器及触发器中的事务学习
2024-01-27 08:48:02
简单介绍Python中的round()方法
2023-05-01 11:57:10
dedecms 批量提取第一张图片最为缩略图的代码(文章+软件)
2024-04-28 09:45:40
深入浅出的SQL server 查询优化
2010-07-02 20:58:00
Python中的单下划线和双下划线使用场景详解
2021-10-28 05:08:48
英文罚抄引发的艺术创意
2008-05-13 12:02:00
一文轻松掌握python语言命名规范规则
2023-03-23 02:11:43
python实现微信每日一句自动发送给喜欢的人
2022-10-13 02:09:25
2009年情人节网站logo欣赏
2009-02-15 12:13:00
python playwright 自动等待和断言详解
2021-04-04 19:39:01
Python数据挖掘Pandas详解
2021-08-04 13:11:44
分享11个Python自动化操作Excel的方法
2022-11-18 08:10:39
Python 多进程并发操作中进程池Pool的实例
2022-06-28 16:31:37
如何通过python计算圆周率PI
2021-01-24 10:29:25
node.js实现BigPipe详解
2024-05-05 09:22:17
全国省市区县最全最新数据表(数据来源谷歌)
2024-01-12 17:45:55
vue下拉菜单组件(含搜索)的实现代码
2024-05-09 15:18:57