Java opencv拉取网络摄像头视频流推送到流媒体服务器

Java opencv拉取网络摄像头视频流推送到流媒体服务器

1. 新建maven项目,添加如下maven依赖，注意添加lombok的版本号。

	<dependency>
	       <groupId>org.bytedeco</groupId>
	       <artifactId>javacv-platform</artifactId>
	       <version>1.5.1</version>
    	</dependency>
	<dependency>
              <groupId>org.projectlombok</groupId>
              <artifactId>lombok</artifactId>
	</dependency>

2. 代码实现如下。封装好的可以直接使用。下面有测试的main方法。我这里是使用的是nginx作为流媒体服务器，关于流媒体服务器搭建网上有很多的资料。具体的实现方案可以使用源码编译安装nginx ,添加nginx-http-flv-module模块。

@Slf4j
public class ConvertVideoPakcet {


	private static final Map<String,ConvertVideoPakcet> convertVideoPakcets = new HashMap<>();

	private FFmpegFrameGrabber grabber = null;
	private FFmpegFrameRecorder record = null;
	private int width = -1, height = -1;

	// 视频参数
	private int audiocodecid;
	private int codecid;
	private double framerate;// 帧率
	private int bitrate;// 比特率

	// 音频参数
	private int audioChannels;
	private int audioBitrate;
	private int sampleRate;

	//控制程序循环
	private Boolean flag = true;


	private static ConvertVideoPakcet get(String deviceId){
		return convertVideoPakcets.get(deviceId);
	}

	public static Boolean start(String deviceId,String formUrl,String toUrl){

		if(null != get(deviceId)) return true;

		final ConvertVideoPakcet convertVideoPakcet = new ConvertVideoPakcet();
		convertVideoPakcets.put(deviceId,convertVideoPakcet);

		new Thread(new Runnable() {
			@Override
			public void run() {
				log.info("start device");
				try {
					convertVideoPakcet.rtsp(formUrl).rtmp(toUrl).start();
				} catch (IOException e) {
					log.error("start dvice error,{}",e);
				}
			}
		}).start();

		log.info("start device finish!");
		return true;
	}

	/**
	 * 停止当前直播
	 * @param id
	 * @return
	 */
	public static Boolean stop(String id){
		log.info("stop device ,{}",id);
		ConvertVideoPakcet convertVideoPakcet = get(id);

		if(null != convertVideoPakcet){
			convertVideoPakcets.remove(id);
			return convertVideoPakcet.stop();
		}

		return false;
	}

	/**
	 * 拉取摄像头视频源
	 * 
	 * @param src rtsp数据源地址
	 * @author JW
	 * @throws Exception
	 */
    private ConvertVideoPakcet rtsp(String src) throws Exception {
		// 采集/抓取器
		grabber = new FFmpegFrameGrabber(src);
		if (src.indexOf("rtsp") >= 0) {
			grabber.setOption("rtsp_transport", "tcp");
		}
		grabber.start();// 开始之后ffmpeg会采集视频信息，之后就可以获取音视频信息
		if (width < 0 || height < 0) {
			width = grabber.getImageWidth();
			height = grabber.getImageHeight();
		}
		// 视频参数
		audiocodecid = grabber.getAudioCodec();
		log.warn("音频编码：{}",audiocodecid);
		codecid = grabber.getVideoCodec();
		framerate = grabber.getVideoFrameRate();// 帧率
		bitrate = grabber.getVideoBitrate();// 比特率
		// 音频参数
		// 想要录制音频，这三个参数必须有：audioChannels > 0 && audioBitrate > 0 && sampleRate > 0
		audioChannels = grabber.getAudioChannels();
		audioBitrate = grabber.getAudioBitrate();
		if (audioBitrate < 1) {
			audioBitrate = 128 * 1000;// 默认音频比特率
		}
		return this;
	}

	/**
	 * rtmp输出推流到nginx媒体流服务器
	 * 
	 * @param out t\ rtmp媒体流服务器地址
	 * @author JW
	 * @throws IOException
	 */
    private ConvertVideoPakcet rtmp(String out) throws IOException {
		// 录制/推流器
		record = new FFmpegFrameRecorder(out, width, height);
		record.setVideoOption("crf", "30");
		record.setGopSize(2);
		record.setFrameRate(framerate);
		record.setVideoBitrate(bitrate);

		record.setAudioChannels(audioChannels);
		record.setAudioBitrate(audioBitrate);
		record.setSampleRate(sampleRate);

		AVFormatContext fc = null;
		
		if (out.indexOf("rtmp") >= 0 || out.indexOf("flv") > 0) {
			// 封装格式flv
			record.setFormat("flv");
			record.setAudioCodecName("aac");
			record.setVideoCodec(codecid);
			fc = grabber.getFormatContext();
		}
		
		record.start(fc);
		
		return this;
	}

	/**
	 * 转封装
	 * 
	 * @author eguid
	 * @throws IOException
	 */
	private void start() throws IOException {

	    //刷新开始的测试数据
		if(null != grabber)
			grabber.flush();

		while (flag) {
			AVPacket pkt = null;
			try {
				// 没有解码的音视频帧
				pkt = grabber.grabPacket();
				if (pkt == null || pkt.size() <= 0 || pkt.data() == null) {
					continue;
				}

				// 不需要编码直接把音视频帧推出去
				record.recordPacket(pkt);
				avcodec.av_packet_unref(pkt);

				try {
					Thread.sleep(0,1000);
				} catch (InterruptedException e) {
					log.error("推流发生等待异常,{}",e);
				}

			} catch (Exception e) {
				log.error("推流发生异常,{}",e);
			} catch (IOException e) {
				log.error("推流发生IO异常,{}",e);
			}
		}
	}

	private Boolean stop() {

		//控制退出循环
		flag = false;

		if(null != record){
			try {
				record.release();
			} catch (FrameRecorder.Exception e) {
				log.error("stop record error ,{}",e);
				return false;
			}
		}

		if(null != grabber){
			try {
				grabber.release();
			} catch (Exception e) {
				log.error("stop grabber error ,{}",e);
				return false;
			}
		}


		return true;
	}

	public static void main(String[] args) throws Exception, IOException {

		// 运行，设置视频源和推流地址
		new ConvertVideoPakcet().rtsp("rtsp://admin:a1234567@192.168.0.121:554/h264/ch1/main/av_stream")
				.rtmp("rtmp://www.2edge.cn:1985/factory/test").start();
	}
}