使用javaCV录制直播流

r囧r小猫 2023-03-06 05:13 45阅读 0赞

# 直播流如何保存到本地 #

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

## 相关资料 ##

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

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

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

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

###### 5. [java使用ffmpeg实现视频切割][java_ffmpeg] ######

###### 6. [流媒体协议介绍（rtp/rtcp/rtsp/rtmp/mms/hls）][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>
    
    <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();
                }
            }
        }

[windows_nginx_rtmp]: https://blog.csdn.net/u014552102/article/details/100906058
[FFmpeg_live]: https://blog.csdn.net/u014552102/article/details/103093607
[windows_nginx_hls_rtmp]: https://blog.csdn.net/u014552102/article/details/103191245
[FFmpeg_hls_ts]: https://blog.csdn.net/u014552102/article/details/103302731
[java_ffmpeg]: https://blog.csdn.net/qq_34445142/article/details/106570745
[rtp_rtcp_rtsp_rtmp_mms_hls]: https://blog.csdn.net/tttyd/article/details/12032357/