odboy
c266da2858
- scanner: filepath.Ext 替代 path.Ext，修复 Windows 点号目录名扩展名解析错误 - scanner: Walk 遇不可访问路径仅告警跳过，不再整体中止扫描 - probe: ffprobe 探测增加 30s 超时，避免损坏文件卡死程序 - mover: copyFile 失败时清理半截目标文件，消除双重 Close - 新增代码审查报告 bug20260814.md
88 lines
2.2 KiB
Go
88 lines
2.2 KiB
Go
// Package probe 负责媒体信息探测：图片尺寸读取、ffprobe 视频/音频解析与内嵌解码器释放。
|
|
package probe
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"os/exec"
|
|
"strconv"
|
|
"time"
|
|
)
|
|
|
|
// ProbeResult 媒体探测结果（图片/视频/音频通用）
|
|
type ProbeResult struct {
|
|
HasVideo bool
|
|
HasAudio bool
|
|
Width int
|
|
Height int
|
|
Duration float64
|
|
VideoCodec string
|
|
AudioCodec string
|
|
Format string
|
|
RawJSON string
|
|
}
|
|
|
|
type streamInfo struct {
|
|
Index int `json:"index"`
|
|
CodecType string `json:"codec_type"`
|
|
CodecName string `json:"codec_name"`
|
|
Width int `json:"width"`
|
|
Height int `json:"height"`
|
|
}
|
|
|
|
type formatInfo struct {
|
|
Duration string `json:"duration"`
|
|
Format string `json:"format_name"`
|
|
}
|
|
|
|
type probeOutput struct {
|
|
Streams []streamInfo `json:"streams"`
|
|
Format formatInfo `json:"format"`
|
|
}
|
|
|
|
// ffprobeTimeout ffprobe 探测超时时间，超时视为探测失败（避免损坏文件卡死程序）
|
|
const ffprobeTimeout = 30 * time.Second
|
|
|
|
// ProbeMediaFile 用 ffprobe 探测媒体文件，一次调用同时获取视频/音频流与时长
|
|
func ProbeMediaFile(ffprobePath, filePath string) (*ProbeResult, error) {
|
|
ctx, cancel := context.WithTimeout(context.Background(), ffprobeTimeout)
|
|
defer cancel()
|
|
cmd := exec.CommandContext(ctx, ffprobePath, "-v", "error",
|
|
"-show_entries", "stream=index,codec_type,codec_name,width,height:format=duration,format_name",
|
|
"-of", "json", filePath)
|
|
output, err := cmd.Output()
|
|
if err != nil {
|
|
return nil, fmt.Errorf("ffprobe failed: %w", err)
|
|
}
|
|
var out probeOutput
|
|
if err := json.Unmarshal(output, &out); err != nil {
|
|
return nil, fmt.Errorf("parse ffprobe output: %w", err)
|
|
}
|
|
res := &ProbeResult{
|
|
Format: out.Format.Format,
|
|
RawJSON: string(output),
|
|
}
|
|
if d, err := strconv.ParseFloat(out.Format.Duration, 64); err == nil {
|
|
res.Duration = d
|
|
}
|
|
for _, s := range out.Streams {
|
|
switch s.CodecType {
|
|
case "video":
|
|
res.HasVideo = true
|
|
if res.VideoCodec == "" {
|
|
res.VideoCodec = s.CodecName
|
|
}
|
|
if res.Width == 0 && s.Width > 0 && s.Height > 0 {
|
|
res.Width, res.Height = s.Width, s.Height
|
|
}
|
|
case "audio":
|
|
res.HasAudio = true
|
|
if res.AudioCodec == "" {
|
|
res.AudioCodec = s.CodecName
|
|
}
|
|
}
|
|
}
|
|
return res, nil
|
|
}
|