Go语言中用FFmpeg处理HTTP请求体[]byte视频提取单帧转PNG

你已经从HTTP请求体里拿到了[]byte格式的视频数据，完全不用多此一举先把它存成文件再处理——直接用FFmpeg的管道功能，让它从标准输入读视频内容，同时把生成的PNG帧输出到标准输出，既省时间又减少IO开销。

结合你现有的代码，我给你写了完整的实现：

package main

import (
	"bytes"
	"io/ioutil"
	"net/http"
	"os/exec"
)

const UPLOAD_ERROR = http.StatusInternalServerError

func uploadHandler(w http.ResponseWriter, r *http.Request) {
	// 读取HTTP请求体中的视频数据
	videoData, err := ioutil.ReadAll(r.Body)
	if err != nil {
		w.WriteHeader(UPLOAD_ERROR)
		w.Write([]byte("Error uploading the file"))
		return
	}
	defer r.Body.Close()

	// 配置FFmpeg命令：从标准输入读视频，输出1帧PNG到标准输出
	// 如果需要提取特定时间的帧，把"-ss", "00:00:01"放在"-i"前面（更快定位）
	cmd := exec.Command("ffmpeg",
		"-i", "pipe:0",  // 从管道0（标准输入）读取视频数据
		"-vframes", "1", // 只提取1帧
		"-f", "png",     // 指定输出格式为PNG
		"-",             // 输出到标准输出
	)

	// 用buffer捕获FFmpeg输出的PNG数据
	var pngBuffer bytes.Buffer
	cmd.Stdout = &pngBuffer

	// 获取FFmpeg的标准输入管道，用于写入视频数据
	stdin, err := cmd.StdinPipe()
	if err != nil {
		w.WriteHeader(http.StatusInternalServerError)
		w.Write([]byte("Failed to create input pipe for FFmpeg"))
		return
	}

	// 启动FFmpeg进程
	if err := cmd.Start(); err != nil {
		w.WriteHeader(http.StatusInternalServerError)
		w.Write([]byte("Failed to start FFmpeg process"))
		stdin.Close()
		return
	}

	// 将内存中的视频数据写入FFmpeg的标准输入
	_, err = stdin.Write(videoData)
	if err != nil {
		w.WriteHeader(http.StatusInternalServerError)
		w.Write([]byte("Failed to send video data to FFmpeg"))
		stdin.Close()
		cmd.Process.Kill() // 终止进程避免资源泄漏
		return
	}

	// 关闭标准输入，告诉FFmpeg数据已全部发送
	stdin.Close()

	// 等待FFmpeg处理完成
	if err := cmd.Wait(); err != nil {
		w.WriteHeader(http.StatusInternalServerError)
		w.Write([]byte("FFmpeg processing failed: " + err.Error()))
		return
	}

	// 将生成的PNG数据返回给客户端
	w.Header().Set("Content-Type", "image/png")
	w.Write(pngBuffer.Bytes())
}

一些实用的参数调整建议： #

• 提取特定时间的帧：如果不需要默认的第一帧，可以在-i前加上-ss参数指定时间，比如"-ss", "00:00:02"（提取第2秒的帧），这样FFmpeg会先快速定位到目标时间点再解码，效率更高。
• 调整输出分辨率：加上"-s", "640x360"可以把帧调整到指定的宽高。
• 指定输入格式：如果视频格式比较特殊，FFmpeg自动识别有问题，可以在-i前加上"-f", "mp4"（替换成你的视频格式）来强制指定输入格式。

注意事项： #

• 确保你的服务器已经安装了FFmpeg，并且ffmpeg命令在系统PATH中（如果不在，要写完整路径，比如"/usr/local/bin/ffmpeg"）。
• 如果处理超大视频，ioutil.ReadAll会把整个视频加载到内存，可能导致内存占用过高——这种情况可以直接把r.Body作为输入流传给FFmpeg的标准输入，不用一次性读入内存。

内容的提问来源于stack exchange，提问作者thelearner