java是实现视频流Rtsp转Rtmp

简言：视频流格式大致分为：RTMP、RTSP、RTP、HLS、MPEG-DASH。
本文就只说一下Rtsp转Rtmp格式，转换成功之后Rtmp的视频流存储到srs上
Pom文件

1. <dependency>
2.     <groupId>org.bytedeco</groupId>
3.     <artifactId>javacv-platform</artifactId>
4.     <version>1.5.1</version>
5. </dependency>

复制代码

Constant类

1. public class Constant {
2.     public final static String rtmp="rtmp://";
3.     public final static String colon=":";
4.     public final static String slash="/";
5.     public final static String stop="stop";
6. }

复制代码

application.properties

1. rtmp.host:
2. rtmp.port:
3. rtmp.folder:

复制代码

rtmp.host: srs安装的ip所在
rtmp.port: srs的端口
rtmp.folder:视频流存储的位置
详细实现

1. import org.bytedeco.ffmpeg.global.avcodec;
2. import org.bytedeco.javacv.FFmpegFrameGrabber;
3. import org.bytedeco.javacv.FFmpegFrameRecorder;
4. import org.bytedeco.javacv.FFmpegLogCallback;
5. import org.bytedeco.javacv.Frame;
6. import org.springframework.beans.factory.annotation.Value;
7. import org.springframework.util.StringUtils;
8. import org.springframework.web.bind.annotation.CrossOrigin;
9. import org.springframework.web.bind.annotation.RequestMapping;
10. import org.springframework.web.bind.annotation.RestController;
12. /**
13. * @author: majinzhong
14. * @date: 2022/6/10 13:53
15. * @Version 1.0
16. */
17. @RestController
18. @CrossOrigin
19. @RequestMapping("/stream")
20. public class CameraPushController {
22.     @Value("${rtmp.host}")
23.     public String rtmpHost;
24.     @Value("${rtmp.port}")
25.     public String rtmpPort;
26.     @Value("${rtmp.folder}")
27.     public String rtmpFolder;
29.     @RequestMapping("/info")
30.     public String info(){
31.         return "项目启动成功";
32.     }
34.     /**
35.      *
36.      * @param rtsUrl rtsp的推流地址（在用户关闭时也要传值）
37.      * @param status 是否停止的状态（在用户关闭时传值，其他时候不传）
38.      * @return
39.      */
41.     @RequestMapping("/push")
42.     public String streamPush(String rtsUrl,String status)throws Exception{
43.         //取出rtsUrl入参中的最后一个/后的streamId
44.         String streamId = rtsUrl.substring(rtsUrl.lastIndexOf(Constant.slash) + 1);
45.         //对SRS的推流地址进行拼接
46.         String rtmpUrl = Constant.rtmp + rtmpHost + Constant.colon + rtmpPort + Constant.slash + rtmpFolder + Constant.slash + streamId;
47.         //实例化帧抓取器对象，将rtsp地址放入
48.         FFmpegFrameGrabber grabber = new FFmpegFrameGrabber(rtsUrl);
49.         grabber.setOption("rtsp_transport", "tcp");
50.         //初始化帧抓取器
51.         grabber.start();
52.         //实例化FFmpegFrameRecorder，将SRS地址传入
53.         FFmpegFrameRecorder recorder = new FFmpegFrameRecorder(rtmpUrl,
54.                 grabber.getImageWidth() * 2,
55.                 grabber.getImageHeight() * 2,
56.                 grabber.getAudioChannels());
58.         int v_rs = 25;
59.         //设置编码格式
60.         recorder.setVideoCodec(avcodec.AV_CODEC_ID_H264);
61.         //设置封装格式
62.         recorder.setFormat("flv");
63.         //一秒内的帧数
64.         recorder.setFrameRate(v_rs);
65.         //两个关键帧之间的帧数
66.         recorder.setGopSize(v_rs);
67.         //设置音频通道，与视频源的通道数相等
68.         recorder.setAudioChannels(grabber.getAudioChannels());
69.         FFmpegLogCallback.set();
70.         //初始化帧录制器
71.         recorder.start();
73.         try {
74.             //使用异步线程让视频流 从rtsp转为rtmp
75.             if(!StringUtils.isEmpty(status)) {
76.                 //获取当前所有在运行的线程
77.                 ThreadGroup currentGroup =
78.                         Thread.currentThread().getThreadGroup();
79.                 int noThreads = currentGroup.activeCount();
80.                 Thread[] lstThreads = new Thread[noThreads];
81.                 currentGroup.enumerate(lstThreads);
82.                 for (int i = 0; i < noThreads; i++){
83.                     //找到视频流的线程
84.                     if("push".equals(lstThreads[i].getName())){
85.                         //让视频流线程睡眠1秒，因为在执行的线程不能直接结束
86.                         lstThreads[i].sleep(1000);
87.                         //结束线程，阻止推流
88.                         lstThreads[i].suspend();
89.                     }
90.                 }
91.                 //如果是停止就关闭录制器和抓取器并退出程序，如果是替换，就继续创建新线程推流
92.                 if(Constant.stop.equals(status)) {
93.                     //关闭帧录制器
94.                     recorder.close();
95.                     //关闭帧抓取器
96.                     grabber.close();
97.                     return "停止推流成功！";
98.                 }
99.             }
100.             //创建新线程进行推流
101.             Thread thread = new Thread(() -> {
102.                 try {
103.                     //持续从视频源取帧
104.                     Frame frame;
105.                     while (null != (frame = grabber.grab())) {
106.                         //取出的每一帧都推送到SRS
107.                         recorder.record(frame);
108.                     }
109.                 } catch (Exception e) {
110.                     e.printStackTrace();
111.                 }
112.             });
113.             //给线程设置名称，方便后续查找
114.             thread.setName("push");
115.             //启动线程
116.             thread.start();
118.         }catch (Exception e){
119.             e.printStackTrace();
120.             System.out.println(e);
121.         }
122.         return rtmpUrl;
124.     }
125. }

复制代码

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。