【发布时间】:2018-03-20 22:07:46
【问题描述】:
使用 Go,我正在获取 RTMP 流,将其转码为 FLAC(使用 ffmpeg)并尝试流式传输到 Google 的语音 API 以转录音频。但是,我在发送数据时不断收到EOF 错误。我在文档中找不到有关此错误的任何信息,因此我不确定是什么原因造成的。
我将接收到的数据分块为 3 秒剪辑(长度不相关,只要它小于流式识别请求的最大长度)。
这是我的代码的核心:
func main() {
done := make(chan os.Signal)
received := make(chan []byte)
go receive(received)
go transcribe(received)
signal.Notify(done, os.Interrupt, syscall.SIGTERM)
select {
case <-done:
os.Exit(0)
}
}
func receive(received chan<- []byte) {
var b bytes.Buffer
stdout := bufio.NewWriter(&b)
cmd := exec.Command("ffmpeg", "-i", "rtmp://127.0.0.1:1935/live/key", "-f", "flac", "-ar", "16000", "-")
cmd.Stdout = stdout
if err := cmd.Start(); err != nil {
log.Fatal(err)
}
duration, _ := time.ParseDuration("3s")
ticker := time.NewTicker(duration)
for {
select {
case <-ticker.C:
stdout.Flush()
log.Printf("Received %d bytes", b.Len())
received <- b.Bytes()
b.Reset()
}
}
}
func transcribe(received <-chan []byte) {
ctx := context.TODO()
client, err := speech.NewClient(ctx)
if err != nil {
log.Fatal(err)
}
stream, err := client.StreamingRecognize(ctx)
if err != nil {
log.Fatal(err)
}
// Send the initial configuration message.
if err = stream.Send(&speechpb.StreamingRecognizeRequest{
StreamingRequest: &speechpb.StreamingRecognizeRequest_StreamingConfig{
StreamingConfig: &speechpb.StreamingRecognitionConfig{
Config: &speechpb.RecognitionConfig{
Encoding: speechpb.RecognitionConfig_FLAC,
LanguageCode: "en-GB",
SampleRateHertz: 16000,
},
},
},
}); err != nil {
log.Fatal(err)
}
for {
select {
case data := <-received:
if len(data) > 0 {
log.Printf("Sending %d bytes", len(data))
if err := stream.Send(&speechpb.StreamingRecognizeRequest{
StreamingRequest: &speechpb.StreamingRecognizeRequest_AudioContent{
AudioContent: data,
},
}); err != nil {
log.Printf("Could not send audio: %v", err)
}
}
}
}
}
运行此代码会给出以下输出:
2017/10/09 16:05:00 Received 191704 bytes
2017/10/09 16:05:00 Saving 191704 bytes
2017/10/09 16:05:00 Sending 191704 bytes
2017/10/09 16:05:00 Could not send audio: EOF
2017/10/09 16:05:03 Received 193192 bytes
2017/10/09 16:05:03 Saving 193192 bytes
2017/10/09 16:05:03 Sending 193192 bytes
2017/10/09 16:05:03 Could not send audio: EOF
2017/10/09 16:05:06 Received 193188 bytes
2017/10/09 16:05:06 Saving 193188 bytes
2017/10/09 16:05:06 Sending 193188 bytes // Notice that this doesn't error
2017/10/09 16:05:09 Received 191704 bytes
2017/10/09 16:05:09 Saving 191704 bytes
2017/10/09 16:05:09 Sending 191704 bytes
2017/10/09 16:05:09 Could not send audio: EOF
请注意,并非所有Sends 都失败了。
谁能在这里指出我正确的方向?它与 FLAC 标头有关吗?我还想知道是否可能重置缓冲区会导致一些数据被丢弃(即这是一个不平凡的操作,实际上需要一些时间才能完成)并且它不喜欢这种丢失的信息?
任何帮助将不胜感激。
【问题讨论】:
-
虽然这完全是随机的,但是您是否尝试过使用比赛检测器运行它?它可能不会显示任何内容,但你永远不知道......
-
@MiloChristiansen 好电话!有一个竞争条件。我尝试创建
bufio.Writer的线程安全版本,但该库使用ReadFrom,它不断地从读者那里读取。它仅在读取所有内容后才释放锁:-/
标签: go ffmpeg google-cloud-platform google-speech-api