refactor: 重构项目架构，拆分代码到模块化目录

将单文件tts_server.go重构为模块化项目结构，拆分出common、dto、middleware、router、controller、service、adapter、setting等目录，优化代码组织提升可维护性
2026-05-23 20:32:12 +08:00
parent 1b84a6c9ee
commit 4e1820d45b
21 changed files with 1536 additions and 895 deletions
@@ -0,0 +1,167 @@
+package volcano
+
+import (
+	"bufio"
+	"bytes"
+	"context"
+	"encoding/base64"
+	"encoding/json"
+	"fmt"
+	"io"
+	"log"
+	"net/http"
+	"time"
+
+	"github.com/google/uuid"
+	"github.com/volcano-tts/tts-api/common"
+	"github.com/volcano-tts/tts-api/dto"
+)
+
+type HTTPClient struct {
+	client *http.Client
+}
+
+func NewHTTPClient() *HTTPClient {
+	return &HTTPClient{
+		client: &http.Client{
+			Timeout: common.DefaultTimeout,
+			Transport: &http.Transport{
+				MaxIdleConns:        100,
+				MaxIdleConnsPerHost: 20,
+				IdleConnTimeout:     90 * time.Second,
+				TLSHandshakeTimeout: 10 * time.Second,
+			},
+		},
+	}
+}
+
+func (h *HTTPClient) PostStream(url string, headers map[string]string, body []byte, timeout time.Duration) (*http.Response, error) {
+	req, err := http.NewRequest(http.MethodPost, url, bytes.NewBuffer(body))
+	if err != nil {
+		return nil, err
+	}
+	for key, value := range headers {
+		req.Header.Set(key, value)
+	}
+
+	ctx, cancel := context.WithTimeout(context.Background(), timeout)
+	defer cancel()
+
+	req = req.WithContext(ctx)
+	return h.client.Do(req)
+}
+
+func convertSpeedToSpeechRate(speed float64) int {
+	if speed <= 0.5 {
+		return -50
+	}
+	if speed >= 2.0 {
+		return 100
+	}
+	return int((speed - 1.0) * 100)
+}
+
+func Synthesis(config *dto.ByteDanceTTSConfig, httpClient *HTTPClient, text string, speed float64) (*dto.SynthesisResult, error) {
+	reqID := uuid.NewString()
+	speechRate := convertSpeedToSpeechRate(speed)
+
+	params := map[string]interface{}{
+		"user": map[string]interface{}{
+			"uid": "uid",
+		},
+		"namespace": "BidirectionalTTS",
+		"req_params": map[string]interface{}{
+			"text":    text,
+			"speaker": config.Speaker,
+			"audio_params": map[string]interface{}{
+				"format":      "wav",
+				"sample_rate": 24000,
+				"speech_rate": speechRate,
+			},
+		},
+	}
+
+	headers := map[string]string{
+		"Content-Type":      "application/json",
+		"Connection":        "keep-alive",
+		"X-Api-Resource-Id": config.ResourceId,
+		"X-Api-Request-Id":  reqID,
+		"X-Api-Key":         config.ApiKey,
+	}
+
+	bodyStr, err := json.Marshal(params)
+	if err != nil {
+		log.Printf("JSON marshal fail: %v", err)
+		return nil, err
+	}
+
+	resp, err := httpClient.PostStream(config.URL, headers, bodyStr, config.Timeout)
+	if err != nil {
+		log.Printf("http post fail: %v", err)
+		return nil, err
+	}
+	defer resp.Body.Close()
+
+	if resp.StatusCode != http.StatusOK {
+		body, err := io.ReadAll(resp.Body)
+		if err != nil {
+			log.Printf("Failed to read error response body: %v", err)
+		} else {
+			log.Printf("TTS service error: status=%d, body=%s", resp.StatusCode, string(body))
+		}
+		return nil, fmt.Errorf("TTS service error: status %d", resp.StatusCode)
+	}
+
+	var audioData []byte
+	scanner := bufio.NewScanner(resp.Body)
+	scanner.Buffer(make([]byte, 1024*1024), 8*1024*1024)
+
+	for scanner.Scan() {
+		line := scanner.Bytes()
+		if len(line) == 0 {
+			continue
+		}
+
+		var v3Resp dto.V3TTSResponse
+		if err := json.Unmarshal(line, &v3Resp); err != nil {
+			log.Printf("unmarshal chunk fail: %v, line: %s", err, string(line))
+			continue
+		}
+
+		if v3Resp.Code == 20000000 {
+			if v3Resp.Usage != nil {
+				log.Printf("TTS synthesis completed, usage: %+v", v3Resp.Usage)
+			}
+			for scanner.Scan() {
+			}
+			break
+		}
+
+		if v3Resp.Code != 0 {
+			log.Printf("TTS service error: code=%d, message=%s", v3Resp.Code, v3Resp.Message)
+			return nil, fmt.Errorf("TTS service error: %s", v3Resp.Message)
+		}
+
+		if v3Resp.Data != "" {
+			chunk, err := base64.StdEncoding.DecodeString(v3Resp.Data)
+			if err != nil {
+				log.Printf("base64 decode fail: %v", err)
+				return nil, err
+			}
+			audioData = append(audioData, chunk...)
+		} else if v3Resp.Sentence != "" {
+			log.Printf("Received sentence info (sequence %d): %s", v3Resp.Sequence, v3Resp.Sentence)
+		}
+	}
+
+	if err := scanner.Err(); err != nil {
+		log.Printf("read stream fail: %v", err)
+		return nil, err
+	}
+
+	if len(audioData) == 0 {
+		return nil, fmt.Errorf("no audio data received")
+	}
+
+	return &dto.SynthesisResult{AudioData: audioData, ReqID: reqID}, nil
+}