使用javaCV录制直播流

直播流如何保存到本地

1. 使用javaCV的FFmpegFrameGrabber帧捕捉器捕捉流的音频帧和视频帧
2. 使用javacpp的FFmpeg模块调用C/C++类库将帧录制到本地文件

相关资料

1. 在windows下搭建、配置nginx流媒体服务器，并进行rtmp流的推流、拉流测试

2. 关于使用FFmpeg推流时，live目录的理解

3. 在windows下配置nginx流媒体服务器，使其支持hls协议的直播和rtmp点播

4. 使用FFmpeg命令进行hls切片，得到的ts文件时长不准确

5. java使用ffmpeg实现视频切割

6. 流媒体协议介绍（rtp/rtcp/rtsp/rtmp/mms/hls）

所需依赖

<dependency>
    <groupId>org.bytedeco</groupId>
    <artifactId>javacv</artifactId>
    <version>1.4.3</version>
</dependency>
<dependency>
    <groupId>org.bytedeco.javacpp-presets</groupId>
    <artifactId>ffmpeg-platform</artifactId>
    <version>4.0.2-1.4.3</version>
</dependency>
<!--这个依赖区分Linux系统和windows系统-->
<dependency>
    <groupId>com.github.hoary.ffmpeg</groupId>
    <artifactId>FFmpeg-windows-x86_64</artifactId>
    <version>2.0.1.1</version>
</dependency>

代码实现，这里启用线程录制

import org.bytedeco.javacpp.avcodec;
import org.bytedeco.javacv.*;

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

/**
 * Description: 直播流保存本地线程
 *
 * @author ZhuZiKai
 * @date 2020/8/5 0005
 */
public class RecordVideoThread implements Runnable {

    private String streamURL;// 流地址（测试可以用obs推流）
    private String filePath;// 文件路径
    private Long timesSec = 100L;// 停止录制时长 0为不限制时长
    private String fileFormat = "flv";//录制的文件格式
    private boolean isAudio = false;//是否录制声音

    public void setStreamURL(String streamURL) {
        this.streamURL = streamURL;
    }

    public void setFilePath(String filePath) {
        this.filePath = filePath;
    }

    public static void main(String[] args) {
//        "E:/video/one.flv"
//        "rtmp://192.168.1.224:1935/live/home"
        RecordVideoThread recordVideoThread = new RecordVideoThread();
        recordVideoThread.filePath = "E:/video/one.flv";
        recordVideoThread.timesSec = 0L;
        recordVideoThread.streamURL = "rtmp://192.168.1.224:1935/live";// 最好设置结束时长 如直接停止程序会造成输出文件的损坏无法正常播放
        recordVideoThread.isAudio = true;
        new Thread(recordVideoThread).start();
//        RecordVideoThread thread = new RecordVideoThread();
//        thread.pushVideoAsRTSP(1, null);
    }

    @Override
    public void run() {
        System.out.println(streamURL);
        // 获取视频源
        FFmpegFrameGrabber grabber = new FFmpegFrameGrabber(streamURL);
        FFmpegFrameRecorder recorder = null;
        try {
            grabber.start();
            Frame frame = grabber.grabFrame();
            createFile();
            //"rtmp://192.168.1.225:1935/live/home"
            // 流媒体输出地址，分辨率（长，高），是否录制音频（0:不录制/1:录制）
            recorder = new FFmpegFrameRecorder(filePath, 1280, 720, isAudio ? 1 : 0);
            recorder.setVideoCodec(avcodec.AV_CODEC_ID_H264);// 直播流格式
            recorder.setFormat(fileFormat);// 录制的视频格式
            recorder.setFrameRate(25);// 帧数
            //百度翻译的比特率，默认400000，但是我400000贼模糊，调成800000比较合适
            recorder.setVideoBitrate(800000);
            recorder.start();
            System.out.println("推流开始");
//            new Thread(new Runnable() {
//                @Override
//                public void run() {
//                    RecordVideoThread thread = new RecordVideoThread();
//                    thread.pushVideoAsRTSP(1, null);
//                }
//            }).start();
            // 计算结束时间
            long endTime = System.currentTimeMillis() + timesSec * 1000;
            // 如果没有到录制结束时间并且获取到了下一帧则继续录制
            while ((System.currentTimeMillis() < endTime) && (frame != null)) {
                System.out.println("开始");
                recorder.record(frame);//录制
                frame = grabber.grabFrame();//获取下一帧
                System.out.println("结束");
            }
            recorder.record(frame);

        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            //停止录制
            if (null != grabber) {
                try {
                    grabber.stop();
                } catch (FrameGrabber.Exception e) {
                    e.printStackTrace();
                }
            }
            if (recorder != null) {
                try {
                    recorder.stop();
                } catch (FrameRecorder.Exception e) {
                    e.printStackTrace();
                }
            }
            System.out.println("录制完成，录制时长：" + timesSec + "秒");
        }
    }
     private void createFile() {
        File outFile = new File(filePath);
        if (filePath.isEmpty() || !outFile.exists() || outFile.isFile()) {
            try {
                outFile.createNewFile();
            } catch (IOException e) {
                // TODO Auto-generated catch block
                e.printStackTrace();
            }
        }
    }