系统相关
首页 > 系统相关> > java封装FFmpeg命令,支持原生ffmpeg全部命令,实现FFmpeg多进程处理与多线程输出控制(开启、关闭、查询),rtsp/rtmp推流、拉流

java封装FFmpeg命令,支持原生ffmpeg全部命令,实现FFmpeg多进程处理与多线程输出控制(开启、关闭、查询),rtsp/rtmp推流、拉流

作者:互联网

前言:

之前已经对FFmpeg命令进行了封装http://blog.csdn.net/eguid_1/article/details/51787646,但是当时没有考虑到扩展性,所以总体设计不是太好,需要改动的地方也比较多,也不支持原生ffmpeg命令,所以本次版本推翻了前面的版本重新设计接口和实现,全面支持各个流程注入自己的实现,并且在原有命令组装基础上增加一个接口用来支持全部原生FFmpeg命令。

概述:

提供一个管理器用于方便管理FFmpeg命令的执行、停止和执行信息持久化。

可以方便的使用ffmpeg来进行推流,拉流,转流等任务

实现的功能:

①开启一个进程+一个输出线程来执行原生ffmpeg命令②开启一个进程+一个输出线程来执行组装命令③查询执行任务信息④查询全部正在执行的任务⑤停止进程和输出线程⑥停止全部正在执行的任务

源码包下载:http://download.csdn.net/detail/eguid_1/9668143

github项目地址:https://github.com/eguid/FFmpegCommandHandler4java

1、接口设计

1.1、发布任务接口
通过原生ffmpeg命令发布处理任务
通过map组装成ffmpeg命令来处理任务
1.2、终止任务接口
结束任务
结束全部任务
1.3、任务查询接口
查询
查询全部

1.4、接口实现注入

执行处理器注入

命令组装器注入

持久化实现注入

2、内部实现

2.1、任务处理器
开启一个进程用于执行ffmpeg命令
开启一个子线程用于输出ffmpeg执行过程
停止进程
停止输出线程(需要在进程关闭前停止输出线程)
按照正确顺序停止进程和线程2.2、输出线程处理器
用于输出ffmpeg执行过程

package cc.eguid.FFmpegCommandManager.service;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;

import cc.eguid.FFmpegCommandManager.entity.TaskEntity;
/**
 * 任务处理实现
 * @author eguid
 * @since jdk1.7
 * @version 2016年10月29日
 */
public class TaskHandlerImpl implements TaskHandler {
	private Runtime runtime = null;

	@Override
	public TaskEntity process(String id, String command) {
		Process process = null;
		OutHandler outHandler = null;
		TaskEntity tasker = null;
		try {
			if (runtime == null) {
				runtime = Runtime.getRuntime();
			}
			process = runtime.exec(command);// 执行本地命令获取任务主进程
			outHandler = new OutHandler(process.getErrorStream(), id);
			outHandler.start();
			tasker = new TaskEntity(id, process, outHandler);
		} catch (IOException e) {
			stop(outHandler);
			stop(process);
			// 出现异常说明开启失败,返回null
			return null;
		}
		return tasker;
	}

	@Override
	public boolean stop(Process process) {
		if (process != null && process.isAlive()) {
			process.destroy();
			return true;
		}
		return false;
	}

	@Override
	public boolean stop(Thread outHandler) {
		if (outHandler != null && outHandler.isAlive()) {
			outHandler.stop();
			outHandler.destroy();
			return true;
		}
		return false;
	}

	@Override
	public boolean stop(Process process, Thread thread) {
		boolean ret;
		ret=stop(thread);
		ret=stop(process);
		return ret;
	}
}

/**
 * 
 * @author eguid
 *
 */
class OutHandler extends Thread {
	/**
	 * 控制状态
	 */
	private volatile boolean desstatus = true;

	/**
	 * 读取输出流
	 */
	private BufferedReader br = null;

	/**
	 * 输出类型
	 */
	private String type = null;

	public OutHandler(InputStream is, String type) {
		br = new BufferedReader(new InputStreamReader(is));
		this.type = type;
	}

	/**
	 * 重写线程销毁方法,安全的关闭线程
	 */
	@Override
	public void destroy() {
		setDesStatus(false);
	}

	public void setDesStatus(boolean desStatus) {
		this.desstatus = desStatus;
	}

	/**
	 * 执行输出线程
	 */
	@Override
	public void run() {
		String msg = null;
		int index = 0;
		int errorIndex = 0;
		int status = 10;
		try {
			System.out.println(type + "开始推流!");
			while (desstatus && (msg = br.readLine()) != null) {
				if (msg.indexOf("[rtsp") != -1) {
					System.out.println("接收" + status + "个数据包" + msg);
					System.out.println(type + ",网络异常丢包,丢包次数:" + errorIndex++ + ",消息体:" + msg);
					status = 10;
					index = 0;
				}

				if (index % status == 0) {
					System.out.println("接收" + status + "个数据包" + msg);
					status *= 2;
				}
				index++;
			}
		} catch (IOException e) {
			System.out.println("发生内部异常错误,自动关闭[" + this.getId() + "]线程");
			destroy();
		} finally {
			if (this.isAlive()) {
				destroy();
			}
		}
	}
}




2.3、持久化服务
增加任务信息
删除任务信息
删除全部任务信息
查询任务信息
查询全部任务信息


        任务是否存在

package cc.eguid.FFmpegCommandManager.dao;

import java.util.Collection;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentMap;

import cc.eguid.FFmpegCommandManager.entity.TaskEntity;

/**
 * 任务信息持久层实现
 * 
 * @author eguid
 * @since jdk1.7
 * @version 2016年10月29日
 */
public class TaskDaoImpl implements TaskDao {
	// 存放任务信息
	private ConcurrentMap<String, TaskEntity> map = null;

	public TaskDaoImpl(int size) {
		map = new ConcurrentHashMap<>(size);
	}

	@Override
	public TaskEntity get(String id) {
		return map.get(id);
	}

	@Override
	public Collection<TaskEntity> getAll() {
		return map.values();
	}

	@Override
	public int add(TaskEntity taskEntity) {
		String id = taskEntity.getId();
		if (id != null && !map.containsKey(id)) {
			if (map.put(taskEntity.getId(), taskEntity) != null) {
				return 1;
			}
		}
		return 0;
	}

	@Override
	public int remove(String id) {
		if(map.remove(id) != null){
			return 1;
		};
		return 0;
	}

	@Override
	public int removeAll() {
		int size = map.size();
		try {
			map.clear();
		} catch (Exception e) {
			return 0;
		}
		return size;
	}

	@Override
	public boolean isHave(String id) {
		return map.containsKey(id);
	}

}



2.3命令组装器
用于将参数组装成对应的ffmpeg命令


package cc.eguid.FFmpegCommandManager.util;

import java.util.Map;

public class CommandAssemblyUtil {
	/**
	 * 
	 * @param map
	 *            -要组装的map
	 * @param id
	 *            -返回参数:id
	 * @param id
	 *            -返回参数:组装好的命令
	 * @return
	 */
	public static String assembly(Map<String, String> paramMap) {
		try {
			if (paramMap.containsKey("ffmpegPath")) {
				String ffmpegPath = (String) paramMap.get("ffmpegPath");
				// -i:输入流地址或者文件绝对地址
				StringBuilder comm = new StringBuilder(ffmpegPath + " -i ");
				// 是否有必输项:输入地址,输出地址,应用名,twoPart:0-推一个元码流;1-推一个自定义推流;2-推两个流(一个是自定义,一个是元码)
				if (paramMap.containsKey("input") && paramMap.containsKey("output") && paramMap.containsKey("appName")
						&& paramMap.containsKey("twoPart")) {
					String input = (String) paramMap.get("input");
					String output = (String) paramMap.get("output");
					String appName = (String) paramMap.get("appName");
					String twoPart = (String) paramMap.get("twoPart");
					String codec = (String) paramMap.get("codec");
					// 默认h264解码
					codec = (codec == null ? "h264" : (String) paramMap.get("codec"));
					// 输入地址
					comm.append(input);
					// 当twoPart为0时,只推一个元码流
					if ("0".equals(twoPart)) {
						comm.append(" -vcodec " + codec + " -f flv -an " + output + appName);
					} else {
						// -f :转换格式,默认flv
						if (paramMap.containsKey("fmt")) {
							String fmt = (String) paramMap.get("fmt");
							comm.append(" -f " + fmt);
						}
						// -r :帧率,默认25;-g :帧间隔
						if (paramMap.containsKey("fps")) {
							String fps = (String) paramMap.get("fps");
							comm.append(" -r " + fps);
							comm.append(" -g " + fps);
						}
						// -s 分辨率 默认是原分辨率
						if (paramMap.containsKey("rs")) {
							String rs = (String) paramMap.get("rs");
							comm.append(" -s " + rs);
						}
						// 输出地址+发布的应用名
						comm.append(" -an " + output + appName);
						// 当twoPart为2时推两个流,一个自定义流,一个元码流
						if ("2".equals(twoPart)) {
							// 一个视频源,可以有多个输出,第二个输出为拷贝源视频输出,不改变视频的各项参数并且命名为应用名+HD
							comm.append(" -vcodec copy  -f flv -an ").append(output + appName + "HD");
						}
					}
					return comm.toString();
				}
			}
		} catch (Exception e) {
			return null;
		}
		return null;
	}
}




2.4、配置文件读取器
读取配置文件中的ffmpeg路径配置
读取默认位置的ffmpeg执行文件

package cc.eguid.FFmpegCommandManager.util;

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.util.Properties;

/**
 * 读取配置文件并加载FFmpeg
 * 
 * @author eguid
 * @since jdk1.7
 * @version 2016年10月29日
 */
public class LoadConfig {
	private volatile static boolean isHave = false;
	private volatile static String ffmpegPath = null;

	public LoadConfig() {
		super();
		if (readConfig()) {
			System.out.println("读取FFmpeg执行路径成功!");
			isHave = true;
		} else if (initConfInfo()) {
			// 配置文件读取失败,自动从项目路径加载ffmpeg
			isHave = true;
		} else {
			isHave = false;
		}

	}

	protected boolean readConfig() {
		String path = null;
		File confFile = new File(getClass().getResource("/").getPath() + "loadFFmpeg.properties");
		System.out.println("读取FFMPEG配置文件:" + confFile.getPath());
		if (confFile != null && confFile.exists() && confFile.isFile() && confFile.canRead()) {
			Properties prop = new Properties();
			try {
				prop.load(new FileInputStream(confFile));
				path = prop.getProperty("path");
				if (path != null) {
					System.out.println("读取配置文件中的ffmpeg路径:" + path);
					ffmpegPath = path;
					return true;
				}
			} catch (IOException e) {
				System.err.println("读取配置文件失败!");
				return false;
			}
		}
		System.err.println("读取配置文件失败!");
		return false;
	}

	/**
	 * 从配置文件中初始化参数
	 */
	protected boolean initConfInfo() {
		String path = getClass().getResource("../").getPath() + "ffmpeg/ffmpeg.exe";
		System.out.print("预加载默认项目路径FFMPEG配置:" + path);
		File ffmpeg = new File(path);
		ffmpegPath = ffmpeg.getPath();
		if (isHave = ffmpeg.isFile()) {
			return true;
		}
		System.out.println("加载ffmpeg失败!");
		return false;
	}

	/**
	 * 判断ffmpeg环境配置
	 * 
	 * @return true:配置成功;false:配置失败
	 */
	public boolean isHave() {
		return isHave;
	}

	/**
	 * 获取ffmpeg环境调用地址 添加方法功能描述
	 * 
	 * @return
	 */
	public String getPath() {
		return ffmpegPath;
	}

	public static void main(String[] args) {
		LoadConfig conf = new LoadConfig();
	}
}








标签:null,return,FFmpeg,paramMap,命令,ffmpeg,多线程,public,String
来源: https://blog.51cto.com/u_6329998/2718441