JavaCV 本地视频推流实现依赖示例

<dependency>
    <groupId>org.bytedeco</groupId>
    <artifactId>javacv-platform</artifactId>
    <version>1.5.6</version>
</dependency>

import org.bytedeco.ffmpeg.global.avcodec;
import org.bytedeco.ffmpeg.global.avutil;
import org.bytedeco.javacv.*;
import java.nio.ByteBuffer;
public class LivePusher {
    public void pushLocalVideo2Rtmp(String localVideoPath, String rtmpAddress) throws Exception {
        FFmpegLogCallback.set();
        FFmpegFrameGrabber grabber = new FFmpegFrameGrabber(localVideoPath);
        grabber.setOption("nobuffer", "1");
        grabber.start();
        /*
         * 测试时推了一个剪映重新生成的高分辨率视频，其分辨率达到了 3840x2160；
         * 导致下面的推送速度跟不上实际消耗的速度；将会造成直播卡顿; 因此，可以重置其分辨率
         */
        int imageWidth = grabber.getImageWidth();
        int imageHeight = grabber.getImageHeight();
        if (imageWidth > 1920 || imageHeight > 1080) {
            double wScale = imageWidth * 1.0 / 1920;
            double hScale = imageHeight * 1.0 / 1080;
            double scale = Math.max(wScale, hScale);
            grabber.setImageWidth((int) (imageWidth / scale));
            grabber.setImageHeight((int) (imageHeight / scale));
        }
        if (grabber.getFormatContext() == null || grabber.getFormatContext().nb_streams() < 1) {
            throw new RuntimeException("本地视频中没有流数据");
        }
        FFmpegFrameRecorder recorder = new FFmpegFrameRecorder(
            rtmpAddress, grabber.getImageWidth(), grabber.getImageHeight()
        );
        recorder.setFrameRate(grabber.getFrameRate());  // 设置帧率
        recorder.setGopSize((int) (grabber.getFrameRate() * 2));  // 设置关键帧
        recorder.setVideoBitrate(grabber.getVideoBitrate());
        recorder.setVideoCodec(avcodec.AV_CODEC_ID_H264);   // 视频编码格式
        recorder.setPixelFormat(avutil.AV_PIX_FMT_YUV420P); // 视频源数据yuv
        recorder.setFormat("flv");
        recorder.setAudioCodec(avcodec.AV_CODEC_ID_AAC); // 音频编码格式
//        recorder.setAudioBitrate(grabber.getAudioBitrate());
        recorder.setAudioChannels(grabber.getAudioChannels());
        recorder.setSampleRate(grabber.getSampleRate());
        recorder.start();
        // 假设 1 秒中有 30 帧数据，那么两帧数据之间的时间间隔就是 1000 / 30；
        long interval = 1_000 / (int) grabber.getFrameRate();
        long startTime = System.currentTimeMillis();
        Frame frame;
        while ((frame = grabber.grab()) != null) {
            recorder.record(frame);
            long currentTime = 1_000 * (System.currentTimeMillis() - startTime);
            long frameTime = frame.timestamp;
            long sleepTime = (frameTime - currentTime) / 1_000;
            System.out.println("推流测试 >>>>>>>> " + getFrameTime(currentTime) + " >>>>>>>> " + getFrameTime(frameTime));
            try {
                if (interval > 0 && sleepTime > 500 + interval) {
                    Thread.sleep(interval);
                }
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
        recorder.close();
        grabber.close();
    }
    private String getFrameTime(long frameTime) {
        long mills = (frameTime / 1000) % 1000;
        int sec = (int) (frameTime / 1_000 / 1_000);
        int min = sec / 60;
        sec %= 60;
        if (min >= 60) {
            int hour = min / 60;
            min %= 60;
            return String.format("%02d:%02d:%02d.%03d", hour, min, sec, mills);
        } else {
            return String.format("%02d:%02d.%03d", min, sec, mills);
        }
    }
}