/******************************************************************************
|
* Copyright (C) 2015 Shenzhen Penguin Network Technology Co., Ltd
|
* All Rights Reserved.
|
* 本软件为深圳市企鹅网络科技有限公司开发研制。未经本公司正式书面同意，其他任何个人、团体
|
* 不得使用、复制、修改或发布本软件.
|
*****************************************************************************/
|
|
package com.qxueyou.scc.teach.live.service.impl;
|
|
import java.io.File;
|
import java.io.FilenameFilter;
|
import java.util.Arrays;
|
import java.util.Calendar;
|
import java.util.Comparator;
|
import java.util.GregorianCalendar;
|
import java.util.List;
|
import java.util.concurrent.TimeUnit;
|
|
import org.apache.commons.io.FileUtils;
|
import org.apache.commons.lang3.StringUtils;
|
import org.apache.logging.log4j.LogManager;
|
import org.apache.logging.log4j.Logger;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.data.redis.core.StringRedisTemplate;
|
import org.springframework.scheduling.annotation.Scheduled;
|
import org.springframework.stereotype.Service;
|
|
import com.qxueyou.scc.base.service.ICacheService;
|
import com.qxueyou.scc.base.service.impl.CommonAppService;
|
import com.qxueyou.scc.base.util.CollectionUtils;
|
import com.qxueyou.scc.base.util.TraceUtils;
|
import com.qxueyou.scc.base.util.UUIDUtils;
|
import com.qxueyou.scc.config.SccConfig;
|
import com.qxueyou.scc.teach.live.model.MediaVideoLive;
|
import com.qxueyou.scc.teach.live.model.MediaVideoLiveReplay;
|
import com.qxueyou.scc.teach.live.service.IMediaVideoLivePlayBackService;
|
import com.qxueyou.scc.teach.live.utils.FfmpegFileVO;
|
import com.qxueyou.scc.teach.live.utils.FfmpegMediaHelper;
|
import com.qxueyou.scc.teach.live.utils.FfmpegVideoInfo;
|
|
@Service
|
public class MediaVideoLivePlayBackService extends CommonAppService implements IMediaVideoLivePlayBackService {
|
|
private static final Logger log = LogManager.getLogger(MediaVideoLivePlayBackService.class);
|
|
//直播回放队列
|
public static final String LIVE_PLAYBACK_LST = "LIVE_PLAYBACK_LST";
|
|
//直播回放全局锁
|
public static final String LIVE_PLAYBACK_LOCK = "LIVE_PLAYBACK_LOCK";
|
|
@Autowired
|
SccConfig cfg;
|
|
@Autowired
|
ICacheService cacheService;
|
|
@Autowired
|
private StringRedisTemplate redisTemplate;
|
|
@Override
|
public List<MediaVideoLiveReplay> queryVideoLiveReplay(String videoLiveId){
|
StringBuffer hql = new StringBuffer(500);
|
hql.append("from MediaVideoLiveReplay where videoLiveId=? and deleteFlag is false ");
|
List<Object> params = CollectionUtils.newList(videoLiveId);
|
|
hql.append(" order by orderNo asc");
|
return this.find(hql.toString(),params, MediaVideoLiveReplay.class);
|
}
|
|
//生成回放(需修改为晚上2:00 开始执行)
|
@Scheduled(cron = " 0 0/2 * * * ?")
|
protected void doTimer(){
|
if(this.lock()){
|
String liveId = null;
|
try {
|
liveId = cacheService.lstLeftPop(LIVE_PLAYBACK_LST);
|
|
if(StringUtils.isEmpty(liveId)){
|
return ;
|
}
|
|
//判断是否是回放状态
|
MediaVideoLive live = this.read(MediaVideoLive.class, liveId);
|
|
if(live.getStatus()==MediaVideoLive.STATUS_LIVE_DOWNLOAD){
|
|
log.info("doTimer live: "+live.getVideoLiveId());
|
|
this.doVideoLivePlayBack(live);
|
|
//更新状态
|
live.setStatus(MediaVideoLive.STATUS_LIVE_REVIEW);
|
this.save(live);
|
}
|
} catch (Exception e) {
|
//重新处理
|
if(StringUtils.isNotEmpty(liveId)){
|
cacheService.lstRightPush(LIVE_PLAYBACK_LST,liveId);
|
}
|
}finally{
|
this.releaseLock();
|
}
|
}
|
}
|
|
@Override
|
public void testPlayBack(String liveId){
|
//判断是否是回放状态
|
MediaVideoLive live = this.read(MediaVideoLive.class, liveId);
|
|
if(live.getStatus()==MediaVideoLive.STATUS_LIVE_DOWNLOAD){
|
try {
|
this.doVideoLivePlayBack(live);
|
|
//更新状态
|
live.setStatus(MediaVideoLive.STATUS_LIVE_REVIEW);
|
this.save(live);
|
|
} catch (Exception e) {
|
//重新处理
|
cacheService.lstRightPush(LIVE_PLAYBACK_LST,liveId);
|
}
|
}
|
}
|
|
|
public void doVideoLivePlayBack(MediaVideoLive videoLive) throws Exception{
|
//读取原文件
|
File srcRootFile = new File(cfg.getSrcLivePath());
|
|
String[] fileNames= srcRootFile.list(new FilenameFilter(){
|
@Override
|
public boolean accept(File dir, String name) {
|
return name.startsWith(videoLive.getWyLiveNumber());
|
}
|
});
|
|
log.info("doVideoLivePlayBack fileNames: "+fileNames==null?"xxx":Arrays.toString(fileNames));
|
|
//排序
|
Arrays.sort(fileNames, new Comparator<String>(){
|
@Override
|
public int compare(String o1, String o2) {
|
if(StringUtils.isNotEmpty(o1) && StringUtils.isNotEmpty(o2)){
|
return getStartTime(o1).compareTo(getStartTime(o2));
|
}
|
return 0;
|
}
|
});
|
|
//读取
|
if(fileNames!=null && fileNames.length>0){
|
short order = 1 ;
|
for(String srcFileName:fileNames){
|
log.info("doCreateVideoLiveReplay srcFileName: "+srcFileName);
|
doCreateVideoLiveReplay(videoLive,srcFileName,order++); //生成回放
|
}
|
}
|
|
//清理源文件
|
if(fileNames!=null && fileNames.length>0){
|
for(String srcFileName:fileNames){
|
log.info("删除原文件：filePath"+srcRootFile.getAbsolutePath().concat(File.separator).concat(srcFileName));
|
FileUtils.deleteQuietly(new File(srcRootFile.getAbsolutePath().concat(File.separator).concat(srcFileName)));
|
}
|
}
|
}
|
|
|
//生成VideoLiveReplay信息
|
private String doCreateVideoLiveReplay(MediaVideoLive videoLive,String srcFileName,short order) throws Exception{
|
//视频文件后缀
|
String mediaSuffix = srcFileName.substring(srcFileName.lastIndexOf("."));
|
|
//回放地址
|
String dstPath =this.generateDstPath(videoLive.getVideoLiveId(), mediaSuffix);
|
|
File srcFile = new File(cfg.getSrcLivePath() + srcFileName);
|
|
File dstFileDir = new File(cfg.getResRootPath().concat(dstPath)).getParentFile();
|
|
//创建目录
|
dstFileDir.mkdirs();
|
|
log.info("开始生成 doCreateVideoLiveReplay: srcFile"+srcFile.getAbsolutePath());
|
|
String dstFile = this.convertFlvToMp4(srcFile, dstFileDir);
|
|
log.info("结束生成 doCreateVideoLiveReplay: dstFile"+ dstFile);
|
|
//获取视频时长
|
long playTime = this.getMediaPlayTime(srcFile.getAbsolutePath());
|
|
MediaVideoLiveReplay liveReplay = new MediaVideoLiveReplay();
|
liveReplay.setName((videoLive.getName() + "-" + order + mediaSuffix).replace(".flv", ".mp4"));
|
liveReplay.setVideoLiveId(videoLive.getVideoLiveId());
|
liveReplay.setOrderNo(order);
|
liveReplay.setBeginTime(this.getStartTime(srcFileName));
|
liveReplay.setEndTime(this.getStartTime(srcFileName)+playTime);
|
liveReplay.setDuration(playTime);
|
liveReplay.setInitialSize(srcFile.length());
|
liveReplay.setUrl(dstFile.substring(dstFile.indexOf("live")).replace("\\", "/"));
|
TraceUtils.setCreateTrace(liveReplay);
|
this.save(liveReplay);
|
|
return liveReplay.getLiveReplayId();
|
}
|
|
|
|
/**
|
* 获取文件目标路径
|
*
|
* @param name
|
* @return
|
*/
|
private String generateDstPath(String liveId,String suffix) {
|
StringBuffer path = new StringBuffer(256);
|
path.append("live");
|
Calendar now = new GregorianCalendar();
|
|
path.append('/');
|
path.append(now.get(Calendar.YEAR));
|
path.append(StringUtils.leftPad(String.valueOf(now.get(Calendar.MONTH)), 2, '0'));
|
path.append('/');
|
path.append(now.get(Calendar.DAY_OF_MONTH));
|
path.append('/');
|
path.append(liveId);
|
path.append('/');
|
path.append(UUIDUtils.UUID());
|
path.append(suffix);
|
|
return path.toString();
|
}
|
|
|
private long getMediaPlayTime(String filePath) throws Exception{
|
FfmpegFileVO fileVO = new FfmpegFileVO();
|
fileVO.setInputPath(filePath);
|
FfmpegVideoInfo videoInfo = new FfmpegVideoInfo();
|
FfmpegMediaHelper.mediaInfo(fileVO, videoInfo);
|
return videoInfo.getPlayTime();
|
}
|
|
private String convertFlvToMp4(File srcFile,File dstFile) throws Exception{
|
// 获取直播视频片段信息
|
FfmpegFileVO videoFileVO = new FfmpegFileVO();
|
videoFileVO.setInputPath(srcFile.getPath());
|
|
// 转码视频
|
videoFileVO.setOutputPath(dstFile.getPath() + File.separator);
|
|
return FfmpegMediaHelper.converUploadVideoToLiveMp4(videoFileVO, new FfmpegVideoInfo());
|
|
//copy备份原片段(暂不备份)
|
// FileUtils.copyFile(srcFile, dstFile);
|
}
|
|
|
//格式： d72de336c7ed45beb14641a79074f9bb-1541497652-rec ， 时间戳单位: 秒
|
private Long getStartTime(String filePath){
|
return Long.valueOf(filePath.split("-")[1]);
|
}
|
|
private boolean lock(){
|
boolean lockResult = redisTemplate.opsForValue().setIfAbsent(LIVE_PLAYBACK_LOCK, "lock");
|
//可能发生死锁，需要手动清除锁
|
if(lockResult==true){
|
redisTemplate.expire(LIVE_PLAYBACK_LOCK, 6, TimeUnit.HOURS);
|
}
|
return lockResult;
|
}
|
|
private void releaseLock(){
|
redisTemplate.delete(LIVE_PLAYBACK_LOCK);
|
}
|
|
}
|
