73 lines
2.7 KiB
Python
73 lines
2.7 KiB
Python
# -*- coding: utf-8 -*-
|
||
import time
|
||
|
||
from alg_airport_ffmpeg.common.Constant import ConstantEnum
|
||
from alg_airport_ffmpeg.exception.CustomerException import ServiceException
|
||
from alg_airport_ffmpeg.util import YmlUtils, LogUtils
|
||
from loguru import logger
|
||
|
||
from alg_airport_ffmpeg.util.Cv2Utils import Cv2Util
|
||
|
||
'''
|
||
推流服务
|
||
'''
|
||
|
||
|
||
class PushStreamService:
|
||
|
||
# 初始化
|
||
def __init__(self):
|
||
# 获取airport环境所需要的配置
|
||
self.__content = YmlUtils.get_configs()
|
||
# 初始化日志
|
||
LogUtils.init_log(self.__content)
|
||
|
||
# 服务调用启动方法
|
||
def start_service(self):
|
||
logger.info(ConstantEnum.START_LOG.value[0])
|
||
# 循环消息处理
|
||
pull_url = self.__content["video"]["pullUrl"]
|
||
pushUrl = self.__content["video"]["pushUrl"]
|
||
cv2_tool = Cv2Util(pull_url, pushUrl)
|
||
cv2_init_num = 0
|
||
while True:
|
||
try:
|
||
time1 = time.time()
|
||
#################################拉流推流一体###################################
|
||
# 1，检查拉流通道是否有流，没有流7秒后重试
|
||
if not cv2_tool.is_video_stream(pull_url):
|
||
logger.info("开始重新获取视频信息: {}次", cv2_init_num)
|
||
time.sleep(1)
|
||
cv2_init_num += 1
|
||
cv2_tool.close_push_stream()
|
||
continue
|
||
cv2_init_num = 0
|
||
# 2. 推流存在跳过，不存在初始化推流
|
||
cv2_tool.start_push_stream()
|
||
# 3. 判断推流通道是否有流，进程是否存在
|
||
if cv2_tool.is_push_stream_ok():
|
||
cv2_tool.close_push_stream()
|
||
#################################先拉流再推流###################################
|
||
# if cv2_tool.check_config():
|
||
# time.sleep(1)
|
||
# logger.info("开始重新获取视频信息: {}次", cv2_init_num)
|
||
# cv2_init_num += 1
|
||
# # cv2_tool.build_cv2() # 方式1 cv2
|
||
# cv2_tool.get_video_info() # 方式2 ffmpeg
|
||
# continue
|
||
# cv2_init_num = 0
|
||
# # frame = cv2_tool.cv2_read() # 方式1 cv2
|
||
# frame = cv2_tool.read() # 方式2 ffmpeg
|
||
# if frame is None:
|
||
# continue
|
||
# cv2_tool.push_stream_write(frame)
|
||
# print(time.time()- time1)
|
||
except ServiceException as s:
|
||
logger.error("推流异常: {}", s.msg)
|
||
except Exception as e:
|
||
logger.error("推流异常: {}", e)
|
||
cv2_tool.close()
|
||
|
||
|
||
|