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

Logan

2025-05-26 ab2148ec182db2b5d18d38f2dd690922d2cf56b8

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

 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)

 runtime/golang/websocket/wavhandler.py

File was deleted