在音视频处理领域，将YUV格式的原始数据实时推送到RTMP服务器是一个常见需求。今天我们就来聊聊如何用Go语言结合FFmpeg高效完成这个任务。

背景与痛点

YUV是视频处理中最常用的原始数据格式之一，但直接处理YUV数据会遇到几个典型问题：

• 内存占用大：未经压缩的YUV数据体积庞大
• 性能瓶颈：单线程处理难以满足实时性要求
• 兼容性问题：不同设备的YUV格式可能存在差异

而推RTMP流时又需要考虑：

• 网络传输稳定性
• 编码效率
• 延迟控制

技术选型

我们有几种方案可以选择：

1. 纯Go实现
2. 优点：部署简单，无外部依赖

3. 缺点：编解码性能较差，开发成本高

4. FFmpeg命令行调用

5. 优点：实现简单

6. 缺点：进程间通信开销大

7. CGO调用FFmpeg库

8. 优点：性能好，控制粒度细
9. 缺点：需要CGO开发经验

综合考虑，我们选择第三种方案。

核心实现

下面是关键代码示例（已精简）：

package main

// #cgo pkg-config: libavformat libavcodec libavutil
// #include <libavformat/avformat.h>
// #include <libavcodec/avcodec.h>
import "C"
import "unsafe"

func pushStream(yuvData []byte, width, height int) {
    // 初始化FFmpeg相关上下文
    avFormatCtx := (*C.AVFormatContext)(unsafe.Pointer(nil))
    // ...其他初始化代码

    // 创建视频帧并填充YUV数据
    frame := C.av_frame_alloc()
    C.av_image_fill_arrays(
        // 填充YUV数据到frame
    )

    // 编码并发送数据包
    for {
        // 编码处理
        // 发送到RTMP服务器
    }
}

性能优化

经过实践，我们发现以下几个优化点特别重要：

1. 内存管理
2. 复用AVFrame和AVPacket对象

3. 使用内存池减少分配开销

4. 并发处理

5. 解码/编码/发送分离到不同goroutine

6. 合理设置缓冲区大小

7. 参数调优

8. 调整FFmpeg编码参数
9. 根据网络状况动态调整码率

避坑指南

在实际项目中，我们遇到过这些问题：

• 内存泄漏：CGO调用后忘记释放资源

• 解决方案：使用defer确保释放

• 线程安全问题：FFmpeg某些函数不是线程安全的

• 解决方案：加锁保护关键操作

• 时间戳处理：不正确的PTS/DTS导致播放异常

• 解决方案：严格按照帧率计算时间戳

总结

通过Go+FFmpeg的组合，我们实现了高性能的YUV数据处理和RTMP推流。未来可以考虑：

1. 支持更多视频格式
2. 实现自适应码率控制
3. 增加硬件加速支持

希望这篇分享对你有帮助！如果有任何问题，欢迎留言讨论。