标签：插件 java String process 视频流 return add command new

兼容windows和linux

核心代码如下：

import org.springframework.stereotype.Component;

import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.OutputStream;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;


@Component
public class RtspToMP4 {

    public class In implements Runnable{
        private InputStream inputStream;

        public In(InputStream inputStream) {
            this.inputStream = inputStream;
        }
        @Override
        public void run() {
            try {
                //转成字符输入流
                InputStreamReader inputStreamReader = new InputStreamReader(inputStream, "gbk");
                int len = -1;
                char[] c = new char[1024];
                //读取进程输入流中的内容
                while ((len = inputStreamReader.read(c)) != -1) {
                    String s = new String(c, 0, len);
                    System.out.print(s);
                }
            }catch (Exception e) {
                log.error("run:"+e);
            }
        }
    }

    /**
     * 开始录制
     * @param ffmpegPath ffmpeg路径
     * @param streamUrl 视频流
     * @param FilePath 保存的视频文件路径
     * @return
     */
    public Process StartRecord(String ffmpegPath,String streamUrl, String FilePath){
        ProcessBuilder processBuilder = new ProcessBuilder();
        //定义命令内容
        List<String> command = new ArrayList<>();
        command.add(ffmpegPath);
        command.add("-rtsp_transport");
        command.add("tcp");
        command.add("-y");
        command.add("-i");
        command.add(streamUrl);
        command.add("-c");
        command.add("copy");
        command.add("-f");
        command.add("mp4");
        command.add(FilePath);
        processBuilder.command(command);
        System.out.println("脚本：" + command.toString());
        //将标准输入流和错误输入流合并，通过标准输入流读取信息
        processBuilder.redirectErrorStream(true);
        try {
            //启动进程
            Process process = processBuilder.start();
            System.out.println("开始时间：" + new SimpleDateFormat("yyyy-MM-dd HH:mm:ss").format(new Date(System.currentTimeMillis())));
            //获取输入流
            InputStream inputStream = process.getInputStream();
            Thread inThread = new Thread(new In(inputStream));
            inThread.start();
            return process;

        } catch (Exception e) {
            log.error("run:"+e);
        }
        return null;
    }

    /**
     * 停止录制
     * @param process
     * @return
     */
    public boolean stopRecord(Process process) {
        try {
            OutputStream os = process.getOutputStream();
            os.write("q".getBytes());
            // 一定要刷新
            os.flush();
            os.close();
        } catch (Exception err) {
            log.error("stopRecord:"+err);
            return false;
        }
        return true;
    }
}

方法调用如下：

@RestController
@RequestMapping("/RtspApi")
public class RtspToMP4Controller {
    @Autowired
    private RtspToMP4 rtspToMP4;

    private Map<Integer,Process> map=new HashMap<>();

    @ApiOperation(value = "开始录制")
    @GetMapping(value = "/videoRecordStart")
    public ResponseData Start(Integer id, String FileName) {
        /*String ffmpegPath="D:\\work\\Test\\ffmpeg.exe";*/
        String ffmpegPath="ffmpeg";
        String streamUrl="rtsp://admin:abcd1234@192.168.8.64:554/h264/ch1/main/av_stream";
        /*String FilePath="D:\\work\\Test\\2022-01-06\\"+FileName;*/
        String FilePath="/opt/videomp4/"+FileName;
        Process process = rtspToMP4.StartRecord(ffmpegPath, streamUrl, FilePath);
        if(null!=process){
            map.put(id,process);
            return new SuccessResponseData();
        }
        return new ErrorResponseData(1,"开始录制错误");
    }

    @ApiOperation(value = "结束录制")
    @GetMapping(value = "/videoRecordStop")
    public ResponseData stop(Integer id) {
        if(map.containsKey(id)){
            Process process = map.get(id);
            if(null!=process){
                rtspToMP4.stopRecord(process);
                return new SuccessResponseData();
            }
        }
        return new ErrorResponseData(1,"结束录制错误");
    }
}

如果在windows系统下，只需更换注释的内容即可

 其中streamUrl为海康的rtsp视频流地址，需改成自己的流地址

标签：插件,java,String,process,视频流,return,add,command,new
来源： https://www.cnblogs.com/ucaen/p/15985112.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。