✨ 优化Go版本API代理服务器的流式响应处理

- 根据Content-Type动态调整缓冲区大小，提升流式响应的性能
- 优化URL重写模式，减少字符串操作次数
- 增加服务器优雅关闭功能，确保在接收中断信号时安全关闭
- 更新日志信息，提供更清晰的服务器状态反馈

main.go

@@ -11,8 +11,10 @@ import (
 	"net/http"
 	"net/url"
 	"os"
+	"os/signal"
 	"strings"
 	"sync/atomic"
+	"syscall"
 	"time"
 
 	"github.com/gin-contrib/cors"
@@ -271,7 +273,26 @@ func apc_handleAsyncAPIRequest(asyncCtx *AsyncProxyContext, c *gin.Context, pref
 
 // apc_streamResponseBody 流式转发响应体
 func apc_streamResponseBody(asyncCtx *AsyncProxyContext, resp *http.Response) error {
-	buffer := make([]byte, 32*1024) // 32KB 缓冲区
+	// 检测是否为流式响应
+	contentType := resp.Header.Get("Content-Type")
+	isStreaming := strings.Contains(contentType, "text/event-stream") ||
+		strings.Contains(contentType, "application/stream+json") ||
+		strings.Contains(contentType, "text/plain") // OpenAI的流式响应通常是text/plain
+
+	var bufferSize int
+	if isStreaming {
+		bufferSize = 2 * 1024 // 2KB for streaming APIs (OpenAI, Claude等)
+	} else if strings.Contains(contentType, "text/html") {
+		bufferSize = 8 * 1024 // 8KB for HTML pages
+	} else if strings.Contains(contentType, "application/json") {
+		bufferSize = 16 * 1024 // 16KB for JSON APIs
+	} else if strings.Contains(contentType, "image/") || strings.Contains(contentType, "video/") {
+		bufferSize = 64 * 1024 // 64KB for media files
+	} else {
+		bufferSize = 32 * 1024 // 32KB for other content
+	}
+
+	buffer := make([]byte, bufferSize)
 
 	for {
 		select {
@@ -427,11 +448,26 @@ func apc_handleHTMLStreamRewrite(asyncCtx *AsyncProxyContext, resp *http.Respons
 		asyncCtx:  asyncCtx,
 		targetURL: targetURL,
 		proxyBase: proxyBase,
-		buffer:    make([]byte, 0, 4096),
+		buffer:    make([]byte, 0, 4096), // 初始4KB，动态增长
+	}
+
+	// 根据Content-Length动态调整缓冲区大小
+	contentLength := resp.ContentLength
+	var bufferSize int
+	if contentLength > 0 {
+		if contentLength < 1024*1024 { // 小于1MB
+			bufferSize = 4 * 1024 // 4KB
+		} else if contentLength < 10*1024*1024 { // 小于10MB
+			bufferSize = 16 * 1024 // 16KB
+		} else {
+			bufferSize = 64 * 1024 // 64KB for large pages
+		}
+	} else {
+		bufferSize = 8 * 1024 // 默认8KB
 	}
 
 	// 流式处理HTML内容
-	buffer := make([]byte, 8*1024) // 8KB 缓冲区
+	buffer := make([]byte, bufferSize)
 
 	for {
 		select {
@@ -486,7 +522,7 @@ func (h *AsyncHTMLRewriter) processBuffer() []byte {
 	baseURL := h.targetURL.Scheme + "://" + h.targetURL.Host
 	content := string(h.buffer)
 
-	// URL重写模式
+	// 优化的URL重写模式 - 使用更高效的字符串替换
 	patterns := []struct{ old, new string }{
 		{`href="` + baseURL, `href="` + h.proxyBase + baseURL},
 		{`src="` + baseURL, `src="` + h.proxyBase + baseURL},
@@ -494,14 +530,24 @@ func (h *AsyncHTMLRewriter) processBuffer() []byte {
 		{`href='` + baseURL, `href='` + h.proxyBase + baseURL},
 		{`src='` + baseURL, `src='` + h.proxyBase + baseURL},
 		{`action='` + baseURL, `action='` + h.proxyBase + baseURL},
+		// 添加更多常见的URL模式
+		{`url("` + baseURL, `url("` + h.proxyBase + baseURL},
+		{`url('` + baseURL, `url('` + h.proxyBase + baseURL},
 	}
 
+	// 批量替换，减少字符串操作次数
 	for _, pattern := range patterns {
 		content = strings.ReplaceAll(content, pattern.old, pattern.new)
 	}
 
-	// 保留缓冲区末尾防止URL跨块
-	const keepSize = 1024
+	// 动态调整保留大小，根据缓冲区大小优化
+	keepSize := 1024             // 默认保留1KB
+	if len(h.buffer) > 32*1024 { // 如果缓冲区大于32KB
+		keepSize = 2048 // 保留2KB，减少处理频率
+	} else if len(h.buffer) < 4*1024 { // 如果缓冲区小于4KB
+		keepSize = 512 // 只保留512B，更快处理
+	}
+
 	if len(h.buffer) > keepSize {
 		processed := []byte(content[:len(content)-keepSize])
 		h.buffer = []byte(content[len(content)-keepSize:])
@@ -629,8 +675,37 @@ func main() {
 	log.Printf("🚀 API代理服务器已启动 (Go优化版) 端口:%s", port)
 	log.Printf("🕒 统计数据每分钟自动刷新页面")
 	log.Printf("⚡ 性能优化：异步统计、内存优化、锁竞争减少")
+	log.Printf("📊 访问 http://localhost:%s 查看统计信息", port)
+
+	// 使用自定义HTTP服务器以更好地控制
+	srv := &http.Server{
+		Addr:    ":" + port,
+		Handler: r,
+	}
+
+	// 启动服务器
+	go func() {
+		if err := srv.ListenAndServe(); err != nil && err != http.ErrServerClosed {
+			log.Fatalf("服务器启动失败: %v", err)
+		}
+	}()
+
+	// 等待中断信号
+	quit := make(chan os.Signal, 1)
+	signal.Notify(quit, os.Interrupt, syscall.SIGTERM)
+	<-quit
+
+	log.Println("正在关闭服务器...")
+
+	// 优雅关闭
+	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
+	defer cancel()
+
+	if err := srv.Shutdown(ctx); err != nil {
+		log.Fatal("服务器强制关闭:", err)
+	}
 
-	r.Run(":" + port)
+	log.Println("服务器已关闭")
 }
 
 // handleIndex 处理首页