mapleafgo
baa32fedc3
问题: - 7秒的音频不到1秒就播完 - player.IsPlaying() 只检查播放状态，不检查数据是否播放完 修复: - 添加等待播放器启动的逻辑 - 等待 IsPlaying() 返回 false - 额外等待 200ms 确保缓冲区数据完全播放 - 添加调试日志输出音频格式信息 测试: - 所有单元测试通过（6/6）
116 lines
2.6 KiB
Go
116 lines
2.6 KiB
Go
package audio
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"fmt"
|
|
"io"
|
|
"time"
|
|
|
|
"github.com/youpy/go-wav"
|
|
"github.com/hajimehoshi/go-mp3"
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
// PlayWav 播放 WAV 文件(阻塞),直到完成或 context 取消
|
|
func PlayWav(ctx context.Context, r io.ReadCloser) error {
|
|
otoCtx, err := initContext()
|
|
if err != nil {
|
|
return fmt.Errorf("音频上下文初始化失败: %w", err)
|
|
}
|
|
|
|
// Read the entire file into memory since wav.NewReader needs ReadAt
|
|
data, err := io.ReadAll(r)
|
|
if err != nil {
|
|
r.Close()
|
|
return fmt.Errorf("读取 WAV 文件失败: %w", err)
|
|
}
|
|
r.Close()
|
|
|
|
// Create a reader from the buffered data
|
|
dec := wav.NewReader(bytes.NewReader(data))
|
|
|
|
// 获取音频格式信息
|
|
format, err := dec.Format()
|
|
if err == nil {
|
|
duration, _ := dec.Duration()
|
|
zap.S().Debugf("WAV 格式: %d ch, %d Hz, %d bits, 时长: %v",
|
|
format.NumChannels, format.SampleRate, format.BitsPerSample, duration)
|
|
}
|
|
|
|
player := otoCtx.NewPlayer(dec)
|
|
defer player.Close()
|
|
|
|
player.Play()
|
|
|
|
// 等待播放完成 - 确保 Play() 调用后数据都被播放
|
|
done := make(chan struct{})
|
|
go func() {
|
|
// 先确保播放器开始播放
|
|
for !player.IsPlaying() {
|
|
time.Sleep(10 * time.Millisecond)
|
|
}
|
|
// 等待播放结束（播放完所有数据）
|
|
for player.IsPlaying() {
|
|
time.Sleep(10 * time.Millisecond)
|
|
}
|
|
// 额外等待 200ms 确保缓冲区数据完全播放
|
|
time.Sleep(200 * time.Millisecond)
|
|
close(done)
|
|
}()
|
|
|
|
select {
|
|
case <-done:
|
|
return nil
|
|
case <-ctx.Done():
|
|
return ctx.Err()
|
|
}
|
|
}
|
|
|
|
// PlayMP3 播放 MP3 文件(阻塞),直到完成或 context 取消
|
|
func PlayMP3(ctx context.Context, r io.ReadCloser) error {
|
|
otoCtx, err := initContext()
|
|
if err != nil {
|
|
return fmt.Errorf("音频上下文初始化失败: %w", err)
|
|
}
|
|
|
|
dec, err := mp3.NewDecoder(r)
|
|
if err != nil {
|
|
r.Close()
|
|
return fmt.Errorf("MP3 解码失败: %w", err)
|
|
}
|
|
defer r.Close()
|
|
|
|
// MP3 解码器信息
|
|
zap.S().Debugf("MP3 采样率: %d Hz, 时长: %d samples",
|
|
dec.SampleRate(), dec.Length())
|
|
|
|
player := otoCtx.NewPlayer(dec)
|
|
defer player.Close()
|
|
|
|
player.Play()
|
|
|
|
// 等待播放完成 - 确保 Play() 调用后数据都被播放
|
|
done := make(chan struct{})
|
|
go func() {
|
|
// 先确保播放器开始播放
|
|
for !player.IsPlaying() {
|
|
time.Sleep(10 * time.Millisecond)
|
|
}
|
|
// 等待播放结束（播放完所有数据）
|
|
for player.IsPlaying() {
|
|
time.Sleep(10 * time.Millisecond)
|
|
}
|
|
// 额外等待 200ms 确保缓冲区数据完全播放
|
|
time.Sleep(200 * time.Millisecond)
|
|
close(done)
|
|
}()
|
|
|
|
select {
|
|
case <-done:
|
|
return nil
|
|
case <-ctx.Done():
|
|
return ctx.Err()
|
|
}
|
|
}
|