更新go client 的原生实现 (#2532)

Logan

2025-05-26 ab2148ec182db2b5d18d38f2dd690922d2cf56b8

 runtime/golang/websocket/go_ws_client.go

@@ -2,19 +2,18 @@

import (
   "bufio"
   "bytes"
   "encoding/base64"
   "encoding/json"
   "fmt"
   "log"
   "net/url"
   "os"
   "os/exec"
   "strconv"
   "strings"
   "sync"
   "time"

   "github.com/go-audio/wav"
   "github.com/gorilla/websocket"
)

@@ -47,6 +46,63 @@
   Stride     int    `json:"stride"`
   ChunkNum   int    `json:"chunk_num"`
   AudioBytes string `json:"audio_bytes"`
}

// 完全模拟Python wave库的行为读取WAV文件
func readWAVFile(filePath string, chunkSize []int, chunkInterval int) (*AudioData, error) {
   file, err := os.Open(filePath)
   if err != nil {
      return nil, fmt.Errorf("failed to open WAV file: %v", err)
   }
   defer file.Close()

   // 使用wav库解码WAV文件，但只获取基本信息
   decoder := wav.NewDecoder(file)
   if !decoder.IsValidFile() {
      return nil, fmt.Errorf("invalid WAV file format")
   }

   // 获取WAV文件格式信息
   format := decoder.Format()
   sampleRate := int(format.SampleRate)

   // 读取所有音频数据
   buf, err := decoder.FullPCMBuffer()
   if err != nil {
      return nil, fmt.Errorf("failed to read full PCM data: %v", err)
   }

   // 获取原始音频字节数据（等同于Python的bytes(frames)）
   audioBytes := make([]byte, len(buf.Data)*2)
   for i, sample := range buf.Data {
      // 确保与Python wave库的字节序一致（小端序16位）
      sample16 := int16(sample)
      audioBytes[i*2] = byte(sample16 & 0xFF)
      audioBytes[i*2+1] = byte((sample16 >> 8) & 0xFF)
   }

   // 完全按照Python的计算方式计算参数
   // Python: stride = int(60 * chunk_size[1] / chunk_interval / 1000 * sample_rate * 2)
   // 注意：Python中的除法是浮点除法，Go中需要显式转换避免整数除法截断
   stride := int(float64(60*chunkSize[1]) / float64(chunkInterval) / 1000.0 * float64(sampleRate) * 2.0)

   // 添加安全检查防止除零
   if stride <= 0 {
      return nil, fmt.Errorf("calculated stride is zero or negative: %d", stride)
   }

   // Python: chunk_num = (len(audio_bytes) - 1) // stride + 1
   chunkNum := (len(audioBytes)-1)/stride + 1

   // 编码为Base64（与Python的base64.b64encode().decode('utf-8')一致）
   audioBase64 := base64.StdEncoding.EncodeToString(audioBytes)

   return &AudioData{
      SampleRate: sampleRate,
      Stride:     stride,
      ChunkNum:   chunkNum,
      AudioBytes: audioBase64,
   }, nil
}

func IntSlicetoString(nums []int) string {
@@ -109,20 +165,10 @@
   }

   if strings.HasSuffix(wav_path, ".wav") {
      cmd := exec.Command("python", "wavhandler.py", wav_path, IntSlicetoString(args.chunk_size), strconv.Itoa(args.chunk_interval))

      var out bytes.Buffer
      cmd.Stdout = &out
      err := cmd.Run()
      // 使用Go原生实现替换Python调用
      audioData, err := readWAVFile(wav_path, args.chunk_size, args.chunk_interval)
      if err != nil {
         fmt.Println("Error running Python script:", err)
         return
      }

      var audioData AudioData
      err = json.Unmarshal(out.Bytes(), &audioData)
      if err != nil {
         fmt.Println("Error parsing JSON:", err)
         fmt.Println("Error reading WAV file:", err)
         return
      }

@@ -222,7 +268,7 @@
   var ibestWriter *os.File
   var err error
   if args.output_dir != "" {
      filePath := fmt.Sprintf("%s/text.%d", args.output_dir, id)
      filePath := fmt.Sprintf("%s/text.%s", args.output_dir, id)
      ibestWriter, err = os.OpenFile(filePath, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
      if err != nil {
         log.Fatalf("failed to open file: %v", err)