Golang調用FFmpeg轉換視頻流的實現

更新時間：2023年02月24日 10:52:12 作者：Edward.W

本文主要介紹了Golang調用FFmpeg轉換視頻流，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧

問題背景

問題背景是在，由于視頻采集端使用的是H264編碼采集的裸流，而網絡流媒體大多是以FLV為主的直播方式進行的，為了實現實時直播，當前是打算直接使用FFmpeg將H264裸流實時轉成FLV視頻流。

為什么是使用FLV視頻流呢，因為相對簡單，加上FLV Header后將每個NALU打包成Tag并進去大致就行了。但是這塊怕有疏忽，最終還是想使用成熟的工具FFmpeg。

方法實現

1. 使用FFmpeg-go封裝好的方法

FFmpeg-go

上面有具體的Demo，但是只有流轉文件，也只有文件轉流的方法，對于流轉流還是需要自己動手處理一下。

import ffmpeg "github.com/u2takey/ffmpeg-go"
?
err := ffmpeg.Input("pipe:", ffmpeg.KwArgs{
?? ??? ?"format": ? ? "rawvideo",
?? ??? ?"video_size": fmt.Sprintf("%dx%d", 480, 1064)}).WithInput(filein).
?? ??? ?Output("pipe:",
?? ??? ??? ?ffmpeg.KwArgs{"c:v": "libx264", "f": "flv", "crf": "24"}).
?? ??? ?WithOutput(buf, errorbuf).
?? ??? ?Run()
if err != nil {
?? ??? ?panic(err)
}
bufs := make([]byte, 1024)
out, _ := os.OpenFile("res2.flv", os.O_CREATE|os.O_RDWR|os.O_APPEND, 0644)
for {
?? ?n, err := buf.Read(bufs)
?? ?if n == 0 || err == io.EOF {
?? ??? ?out.Close()
?? ??? ?break
?? ?} else {
?? ??? ?//此處處理輸出流，這邊簡單地寫到文件里
?? ??? ?out.Write(bufs)
?? ?}
}

其中

filein 是輸入的H264視頻裸流 buf是輸出的視頻流
雖說代碼是這么寫，但是實際上是run不起來的，不清楚具體是什么原因，因此后續(xù)還是打算直接調用ffmpeg可執(zhí)行程序。

2. 直接調用FFmpeg

首先需要在電腦上下載好ffmpeg并且添加到環(huán)境變量，保證執(zhí)行Powshell或者cmd指令的ffmpeg時能夠正常運行，接下來還是直接上代碼。

cmd := exec.Command("ffmpeg", "-re", "-r", "30", "-i", "pipe:0", "-vcodec", "libx264", "-f", "flv", "pipe:1", "-y", "another.flv")
 
//獲取輸入流
stdin, err := cmd.StdinPipe()
if err != nil {
    fmt.Println("Error getting stdout pipe:", err)
    return
}
 
//要寫東西進去的時候只需要
stdin.Write(bytes) 
 
//獲取輸出流
stdout, err := cmd.StdoutPipe()
if err != nil {
    fmt.Println("Error getting stdout pipe:", err)
    return
}
 
// Start the command
err = cmd.Start()
if err != nil {
    fmt.Println("Error starting command:", err)
    return
}

這邊需要注意到的點是，如果不加上"-re"的話，需要等stdin.close()之后，整個ffmpeg才會運行，才能見到有輸出流。

其次，這邊加了 “-y” "another.flv"，是用來進行對比的，這里ffmpeg將會把輸出同時放在兩個地方，一個寫入到文件another.flv里（-y 是覆蓋原有文件），另一個通過stdout傳出來，我將stdout的數據流保存成out.flv后發(fā)現同another.flv對比發(fā)現并不相同。或許視頻流和文件不應該這么處理。