Go语言学习之旅 | 青训营笔记

在项目实践中遇到了需要处理视频编码和获取图片第一帧的需求，通过查阅资料了解到通过使用ffmpeg-go包来实现

• ffmpeg-go 是一个 Go 语言的 FFmpeg 封装库，提供了简单易用的 API，可以方便地调用 FFmpeg 进行音视频处理。

下面是一个简单的示例代码，使用 ffmpeg-go 将视频文件转换为 FLV 格式，并将转换后的视频流推送到指定的 RTMP 服务器上：

package main

import (
	"fmt"
	"log"

	"github.com/goodboynet/ffmpeg-go/ffmpeg"
)

func main() {
	// 创建 FFmpeg 实例
	ffmpegBin, err := ffmpeg.LookPath()
	if err != nil {
		log.Fatal(err)
	}
	ffmpeg := ffmpeg.New(ffmpegBin)

	// 执行转换命令
	cmd := ffmpeg.Input("input.mp4").Output("rtmp://localhost/live/stream", ffmpeg.KwArgs{"c:v": "libx264", "c:a": "aac", "f": "flv"}).GlobalArgs("-loglevel", "error")

	// 开始执行命令
	if err := Welcome to GitHub Pages(); err != nil {
		fmt.Println(err)
		return
	}
}

上面的代码中，我们使用 ffmpeg-go 包创建了一个 FFmpeg 实例，然后调用 Input()和 Output() 方法设置输入和输出参数，最后使用 Run()方法执行转换命令。在设置输出参数时，我们使用了 ffmpeg.KwArgs类型的参数，可以方便地设置 FFmpeg 命令中的关键字参数。

运行上面的代码，可以看到 FFmpeg 的输出信息以及转换后的视频流被推送到了指定的 RTMP 服务器上。

需要注意的是，使用 ffmpeg-go 进行视频转换需要先安装 FFmpeg 工具，并将其加入系统环境变量中。同时，如果需要将视频流推送到 RTMP 服务器上，还需要安装相应的推流工具（如 nginx-rtmp、srs 等）。

• 调用 github.com/u2takey/ffmpeg-go 实现视频流转换

​ 首先，您需要安装 FFmpeg 库。在 Ubuntu 系统上，可以使用以下命令安装 FFmpeg：

sudo apt-get update
sudo apt-get install ffmpeg

1. 在其他操作系统上，可以从 FFmpeg 官网下载并安装相应的软件包。
2. 接下来，在 Go 中引入 github.com/u2takey/ffmpeg-go 库：

import (
    "github.com/u2takey/ffmpeg-go"
)

然后，您可以使用 ffmpeg.InputFile() 函数指定输入文件或流，并使用 ffmpeg.Output() 函数指定输出文件或流。例如，以下代码将从名为 input.mp4 的文件创建一个输入流，并将其转换为名为 output.avi 的 AVI 文件：

input := ffmpeg.InputFile("input.mp4")
output := ffmpeg.Output("output.avi", ffmpeg.KwArgs{})

接下来，您可以使用 output.WithVideoCodec() 函数指定视频编解码器，并使用 output.WithBitrate() 函数指定比特率。例如，以下代码将使用 H.264 编码器和 500k 比特率编码视频：

output = output.WithVideoCodec("libx264").WithBitrate("500k")

最后，您可以使用 output.Run() 函数运行转换，如下所示：

err := ffmpeg.
    ConcatDemuxer([]string{input.Path()}).
    WithOutput(output).
    Run(context.Background())
if err != nil {
    // 处理错误
}

1. 在此示例中，我们使用了 ConcatDemuxer() 函数将输入文件作为分段传入，确保每个输入段都被完全读取。

这样就完成了视频流转换的基本过程。您还可以根据需要添加其他参数和选项，如音频编解码器、帧率、分辨率等。