Compare commits
2 Commits
v1.0.1
...
v1.0.2-rc2
| Author | SHA1 | Date | |
|---|---|---|---|
| 2331d0c73f | |||
| 5ee8e15965 |
@@ -10,16 +10,24 @@ import (
|
||||
func SoundStart() leaf.HandlerFunc {
|
||||
return func(c *leaf.Context) {
|
||||
pm := leaf.Value[*schema.PlayModal](c, PayloadJSONKey)
|
||||
tts.DefaultTTS.Sound(pm.TTS.Start)
|
||||
|
||||
// 使用请求的 context,支持取消和超时
|
||||
if pm.TTS.Start != "" {
|
||||
tts.DefaultTTS.SoundWithContext(c, pm.TTS.Start)
|
||||
}
|
||||
|
||||
defer func() {
|
||||
var text string
|
||||
switch leaf.Value[leaf.EndType](c, leaf.EndKey) {
|
||||
case leaf.End:
|
||||
tts.DefaultTTS.Sound(pm.TTS.End)
|
||||
text = pm.TTS.End
|
||||
case leaf.EndTimeout:
|
||||
tts.DefaultTTS.Sound(pm.TTS.Timeout)
|
||||
text = pm.TTS.Timeout
|
||||
case leaf.EndStop:
|
||||
tts.DefaultTTS.Sound(pm.TTS.Stop)
|
||||
text = pm.TTS.Stop
|
||||
}
|
||||
if text != "" {
|
||||
tts.DefaultTTS.SoundWithContext(c, text)
|
||||
}
|
||||
}()
|
||||
|
||||
|
||||
@@ -22,29 +22,37 @@ func init() {
|
||||
}
|
||||
|
||||
func PlayWav(c context.Context, r io.Reader) {
|
||||
zap.S().Debugln("开始 WAV 解码")
|
||||
streamer, format, err := wav.Decode(r)
|
||||
if err != nil {
|
||||
zap.S().Errorln("WAV解码失败: ", err)
|
||||
return
|
||||
}
|
||||
defer streamer.Close()
|
||||
|
||||
zap.S().Debugln("WAV解码成功,采样率:", format.SampleRate)
|
||||
s := beep.Resample(4, format.SampleRate, DefaultSampleRate, streamer)
|
||||
|
||||
ctrl := &beep.Ctrl{Streamer: s}
|
||||
done := make(chan struct{})
|
||||
speaker.Play(beep.Seq(ctrl, beep.Callback(func() {
|
||||
zap.S().Debugln("音频播放完成")
|
||||
close(done)
|
||||
})))
|
||||
|
||||
zap.S().Debugln("等待音频播放完成...")
|
||||
for {
|
||||
select {
|
||||
case <-done:
|
||||
zap.S().Infoln("音频播放正常结束")
|
||||
return
|
||||
case <-c.Done():
|
||||
{
|
||||
zap.S().Infoln("音频播放被 context 取消")
|
||||
speaker.Lock()
|
||||
ctrl.Streamer = nil
|
||||
speaker.Unlock()
|
||||
return
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -53,6 +61,7 @@ func PlayWav(c context.Context, r io.Reader) {
|
||||
func PlayMP3(c context.Context, r io.ReadCloser) {
|
||||
streamer, format, err := mp3.Decode(r)
|
||||
if err != nil {
|
||||
zap.S().Errorln("MP3解码失败: ", err)
|
||||
return
|
||||
}
|
||||
defer streamer.Close()
|
||||
@@ -74,6 +83,7 @@ func PlayMP3(c context.Context, r io.ReadCloser) {
|
||||
speaker.Lock()
|
||||
ctrl.Streamer = nil
|
||||
speaker.Unlock()
|
||||
return
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@@ -45,9 +45,28 @@ func (tts *AliTTS) Sound(text string) {
|
||||
if text == "" {
|
||||
return
|
||||
}
|
||||
zap.S().Infof("开始播放TTS: %s", text)
|
||||
buf, err := tts.Get(text)
|
||||
if err == nil && buf != nil {
|
||||
zap.S().Debugln("TTS合成成功,开始播放")
|
||||
audio.PlayWav(tts.ctx, buf)
|
||||
zap.S().Infof("TTS播放完成: %s", text)
|
||||
} else {
|
||||
zap.S().Errorln("AliTTS 请求异常: ", err)
|
||||
}
|
||||
}
|
||||
|
||||
// SoundWithContext 使用指定的 context 播放 TTS,支持取消和超时
|
||||
func (tts *AliTTS) SoundWithContext(ctx context.Context, text string) {
|
||||
if text == "" {
|
||||
return
|
||||
}
|
||||
zap.S().Infof("开始播放TTS: %s", text)
|
||||
buf, err := tts.Get(text)
|
||||
if err == nil && buf != nil {
|
||||
zap.S().Debugln("TTS合成成功,开始播放")
|
||||
audio.PlayWav(ctx, buf)
|
||||
zap.S().Infof("TTS播放完成: %s", text)
|
||||
} else {
|
||||
zap.S().Errorln("AliTTS 请求异常: ", err)
|
||||
}
|
||||
@@ -110,13 +129,22 @@ func (tts *AliTTS) Get(text string) (io.Reader, error) {
|
||||
case done := <-ch:
|
||||
{
|
||||
if !done {
|
||||
zap.S().Errorln("TTS合成失败: done=false")
|
||||
return ttsData.Data, errorsx.ThirdPartyErr
|
||||
}
|
||||
size := ttsData.Data.(*bytes.Buffer).Len()
|
||||
zap.S().Debugf("TTS合成成功,数据大小: %d bytes", size)
|
||||
if size == 0 {
|
||||
zap.S().Errorln("TTS合成数据为空")
|
||||
return ttsData.Data, errorsx.ThirdPartyErr
|
||||
}
|
||||
return ttsData.Data, nil
|
||||
}
|
||||
case <-time.After(time.Duration(tts.Timeout) * time.Second):
|
||||
zap.S().Errorln("TTS合成超时")
|
||||
return ttsData.Data, errorsx.DriverTimeoutErr
|
||||
case <-tts.ctx.Done():
|
||||
zap.S().Errorln("TTS合成被取消")
|
||||
return ttsData.Data, errorsx.DriverCancelErr
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user