Golang集成FFmpeg的音视频处理的实现

// GetMp4ImagesIntervalSeconds
// @Description: 指定时长截取图片
// @param videoBase64 base64视频
// @param intervalSeconds 截图间隔秒数
// @return imageSlice 截图切片
func GetMp4ImagesIntervalSeconds(videoBase64 string, intervalSeconds int) (imageSlice []ImageInfo, err error) {
    // 解码base64字符串
    videoData, err := base64.StdEncoding.DecodeString(videoBase64)
    if err != nil {
        Log.LogError("GetMp4ImagesIntervalSeconds DecodeString err：%v", err.Error())
        return nil, err
    }

    //判断目录是否存在
    if _, err = os.Stat("./temp"); os.IsNotExist(err) {
        err = os.Mkdir("./temp", 0755)
        if err != nil {
            Log.LogError("GetMp4ImagesIntervalSeconds os.Mkdir err：%v", err.Error())
            return nil, err
        }
    }

    // 创建临时视频文件
    tmpVideoFile, err := os.CreateTemp("./temp", "tempvideo_*.mp4")
    if err != nil {
        Log.LogError("GetMp4ImagesIntervalSeconds os.CreateTemp err：%v", err.Error())
        return nil, err
    }
    defer os.Remove(tmpVideoFile.Name())
    defer tmpVideoFile.Close()

    // 将视频数据写入临时文件
    _, err = tmpVideoFile.Write(videoData)
    if err != nil {
        Log.LogError("GetMp4ImagesIntervalSeconds tmpVideoFile.Write err：%v", err.Error())
        return nil, err
    }

    // 创建临时图片目录
    tempImageDir, err := os.MkdirTemp("./temp", "tempimage_")
    if err != nil {
        Log.LogError("GetMp4ImagesIntervalSeconds os.MkdirTemp err：%v", err.Error())
        return nil, err
    }
    defer os.RemoveAll(tempImageDir)

    // 使用ffmpeg截取关键帧并将图片放入指定目录
    /*
        -i tmpVideoFile.Name(): 指定输入文件为临时视频文件的名称。
        -vf select='eq(pict_type,I)',fps=1/intervalSeconds: 使用select滤镜选择关键帧并指定帧率为1/intervalSeconds，即每间隔intervalSeconds秒取一帧。
        filepath.Join(tempImageDir, "temp_%03d.jpg"): 指定输出的图片文件名的格式，temp_%03d.jpg表示输出文件名的格式为temp_001.jpg、temp_002.jpg等，%03d将会被替换成递增的数字序列。这里的 tempImageDir 是保存图片的临时目录。
    */
    cmd := exec.Command("ffmpeg", "-i", tmpVideoFile.Name(), "-vf", fmt.Sprintf("select='eq(pict_type,I)',fps=1/%d", intervalSeconds), filepath.Join(tempImageDir, "temp_%03d.jpg"))
    err = cmd.Run()
    if err != nil {
        Log.LogError("GetMp4ImagesIntervalSeconds exec.Command err：%v", err.Error())
        return nil, err
    }

    // 遍历临时图片目录，将图片文件名添加到切片中
    files, _ := filepath.Glob(filepath.Join(tempImageDir, "temp_*.jpg"))
    for _, fileName := range files {
        imageBase64, dhash, err := fileToBase64(fileName)
        if err != nil {
            Log.LogError("GetMp4ImagesIntervalSeconds fileToBase64(fileName) err：%v", err.Error())
            return nil, err
        }
        imageSlice = append(imageSlice, ImageInfo{
            ImageBase64: imageBase64,
            DHash:       dhash,
        })
        fmt.Println("dhash：", dhash)
    }
    Log.LogInfo("GetMp4ImagesIntervalSeconds success number:%v", len(imageSlice))
    return imageSlice, nil
}