引言

隨着直播行業迅猛發展，RTMP（Real-Time Messaging Protocol）作為廣泛使用的實時流媒體協議，已經成為推送直播流的標準選擇。然而，使用底層工具直接實現 RTMP 推流通常複雜且容易出現內存安全問題，給開發者帶來了不少挑戰。

本文將以 Rust 為背景，結合實際業務場景，探討一種更簡單、安全、高效地實現 RTMP 推流的方法，並給出具體的解決方案和代碼示例。

為什麼使用 Rust 實現 RTMP 推流？

RTMP 推流的常見痛點包括：

• 底層接口複雜，學習成本高。
• 直接調用 FFmpeg 底層 API 存在內存管理和安全隱患。
• 對實時性能和穩定性要求高。

而 Rust 語言天然具備以下優勢：

• 零成本抽象與內存安全。
• 性能接近 C/C++，但開發體驗更佳。
• 生態豐富，能夠無縫與 C/C++ 庫進行集成。

因此，Rust 是實現 RTMP 推流功能的絕佳選擇。

常見的 RTMP 推流場景

根據實際業務需求，RTMP 推流大致分為兩類場景：

1. 推送到公共直播平台

• 場景説明：例如抖音、B站、YouTube 等平台，面向大眾進行實時直播。
• 關鍵點：需要穩定性、低延遲、易用性。

2. 本地或局域網推流（測試、內網直播）

• 場景説明：適合內部直播、測試開發、局域網視頻監控等。
• 關鍵點：易部署、靈活、快速啓動、便於調試。

實踐方案：Rust 結合 ez-ffmpeg

針對以上痛點和需求，本文介紹使用 Rust 生態中對 FFmpeg 進行封裝的 ez-ffmpeg 庫，通過簡單的接口和自動化的內存管理，實現優雅高效的 RTMP 推流。

技術點解析：

• FFI 綁定：安全調用 FFmpeg 原生接口。
• 內存安全管理：自動化的資源回收，減少開發者的內存管理負擔。
• 鏈式調用：符合人體工學設計，開發體驗好。

快速入門示例

環境準備

安裝 FFmpeg

• macOS：

brew install ffmpeg

• Windows：

vcpkg install ffmpeg

Rust 依賴引入

[dependencies]
ez-ffmpeg = { version = "*", features = ["rtmp"] }

場景一：推送到公共 RTMP 服務器

以下代碼示例將本地視頻文件推送到公共 RTMP 直播平台（如 B站、抖音）：

use ez_ffmpeg::{FfmpegContext, Input, Output};
​
fn main() -> Result<(), Box<dyn std::error::Error>> {
    let input = Input::from("video.mp4");
    let output = Output::from("rtmp://your-platform-address/app/stream_key");
​
    FfmpegContext::builder()
        .input(input)
        .output(output)
        .build()?
        .start()?
        .wait()?;
​
    Ok(())
}

場景二：本地嵌入式 RTMP 服務器

適合本地測試開發，快速啓動一個 RTMP 服務器：

use ez_ffmpeg::rtmp::embed_rtmp_server::EmbedRtmpServer;
use ez_ffmpeg::{FfmpegContext, Input};
​
fn main() -> Result<(), Box<dyn std::error::Error>> {
    let mut server = EmbedRtmpServer::new("localhost:1935").start()?;
    let output = server.create_rtmp_input("test-app", "test-stream")?;
​
    let input = Input::from("video.mp4");
​
    FfmpegContext::builder()
        .input(input)
        .output(output)
        .build()?
        .start()?
        .wait()?;
​
    Ok(())
}

客户端可使用以下地址拉取流：

rtmp://localhost:1935/test-app/test-stream

總結

藉助 Rust 和 ez-ffmpeg 庫，開發者可以更高效、安全地實現 RTMP 推流需求。無論是公共直播平台還是本地環境，都能快速上手，讓直播應用開發變得簡單而穩定。

🔗 開源項目地址：https://github.com/YeautyYE/ez-ffmpeg