golang 使用ffmpeg获取rtsp流截图,并生成缩略图

2023-05-16

golang 使用ffmpeg获取rtsp流截图,并生成缩略图

  1. golang 调用命令公共方法

    func CallCommandRun(cmdName string, args []string) (string, error) {
    	cmd := exec.Command(cmdName, args...)
    	fmt.Println("CallCommand Run 参数=> ", args)
    	fmt.Println("CallCommand Run 执行命令=> ", cmd)
    	bytes, err := cmd.Output()
    	if err != nil {
    		fmt.Println("CallCommand Run 出错了.....", err.Error())
    		fmt.Println(err)
    		return "", err
    	}
    	resp := string(bytes)
    	fmt.Println(resp)
    	fmt.Println("CallCommand Run 调用完成.....")
    	return resp, nil
    }
    
  2. golang直接使用命令调用ffmpeg 获取rtsp码流截图

    • ffmpeg命令

      ./ffmpeg -rtsp_transport tcp -timeout 10000000 -y -i rtsp://username:password@ip:554/chID=00000001-0000-0000-0000-000000000000 -ss 00:00:01 -vframes 1 -f image2 -vcodec png image.png
      
    • go 程序取流截图

      // 根据URL调用ffmpeg 获取截图
      func GetIpcScreenShot(ffmpegPath string, url string, screenShotPath string) string {
      	var params []string
      	params = append(params, "-rtsp_transport")
      	params = append(params, "tcp")
      	params = append(params, "-timeout")
      	// 10s
      	params = append(params, "10000000")
      	params = append(params, "-y")
      	params = append(params, "-i")
      	params = append(params, url)
      	params = append(params, "-ss")
      	params = append(params, "00:00:02")
      	params = append(params, "-vframes")
      	params = append(params, "1")
      	params = append(params, "-f")
      	params = append(params, "image2")
      	params = append(params, "-vcodec")
      	params = append(params, "png")
      	params = append(params, screenShotPath)
      
      	resp, err := CallCommandRun(ffmpegPath, params)
      	if err != nil {
      		fmt.Println(err.Error())
      		fmt.Println("获取截图出错,url为--->", url)
      	}
      	return resp
      }
      
  3. 使用imaging包将第二步获取的截图生成缩略图

    github.com/disintegration/imaging
    
    // 将截图生成缩略图
    func GetThumbnail(screenShotPath string) error {
    	src, err := imaging.Open(screenShotPath)
    	if err != nil {
    		fmt.Println("生成缩略图,打开截图文件失败--->", err)
    		return err
    	}
    
    	// 220 * 140 宽220px, 高140px
    	dsc := imaging.Resize(src, 220, 140, imaging.Lanczos)
    	err = imaging.Save(dsc, screenShotPath)
    	if err != nil {
    		fmt.Println("生成缩略图失败--->", err)
    		return err
    	}
    	return nil
    }
    
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

golang 使用ffmpeg获取rtsp流截图,并生成缩略图 的相关文章

随机推荐