Go语言接入ChatGPT实战指南：从API调用到生产环境部署

1次阅读

共计 2035 个字符，预计需要花费 6 分钟才能阅读完成。

ChatGPT API 正在重塑多个技术领域：在客服场景中实现 7×24 小时智能响应，为开发者提供代码自动补全和文档生成能力，还能快速构建个性化内容推荐系统。本文将手把手带你完成从零接入到生产部署的全流程。

标准库 net/http
优点：零依赖、完全控制请求生命周期
缺点：需要手动处理连接复用、JSON 序列化等基础功能
Resty
优点：链式调用、自动重试、支持中间件
缺点：抽象层级较高，调试复杂请求时不够透明
推荐方案
对于需要精细控制的中大型项目，建议基于标准库封装；快速原型开发可使用 Resty。本示例选择标准库实现以便展示底层细节。

安全认证客户端

type ChatGPTClient struct {
    baseURL    string
    apiKey     string
    httpClient *http.Client
    // 连接池参数
    maxIdleConns: 100,
    idleTimeout: 90 * time.Second,
}

// 带 JWT 的请求构造
func (c *ChatGPTClient) newRequest(ctx context.Context, method, path string) (*http.Request, error) {req, err := http.NewRequestWithContext(ctx, method, c.baseURL+path, nil)
    req.Header.Set("Authorization", "Bearer"+c.apiKey)
    req.Header.Set("Content-Type", "application/json")
    return req, err
}

流式响应处理

// 处理 Server-Sent Events(SSE)
func processStream(resp *http.Response) chan string {ch := make(chan string)
    scanner := bufio.NewScanner(resp.Body)

    go func() {defer close(ch)
        for scanner.Scan() {line := scanner.Text()
            if strings.HasPrefix(line, "data:") {ch <- strings.TrimPrefix(line, "data:")
            }
        }
    }()

    return ch
}

智能重试机制

// 指数退避重试
func withRetry(fn func() error, maxAttempts int) error {
    baseDelay := 100 * time.Millisecond
    for attempt := 0; ; attempt++ {err := fn()
        if err == nil || attempt >= maxAttempts {return err}

        delay := time.Duration(math.Pow(2, float64(attempt))) * baseDelay
        time.Sleep(delay)
    }
}

连接池调优
设置 MaxIdleConnsPerHost 与 CPU 核心数相同
KeepAlive 时间建议 90-120 秒
禁用 HTTP/ 2 可降低延迟（当使用长连接时）
压测对比
| 配置项 | QPS(未优化) | QPS(优化后) |
|—————-|————|————|
| 默认连接池 | 832 | – |
| 调优连接池 | – | 2147 |
| 启用压缩 | 1856 | 2412 |

敏感词过滤

// GDPR 关键数据过滤
var sensitivePattern = regexp.MustCompile(`(?i)(password|credit_card|ssn)`)

func sanitizeInput(input string) string {return sensitivePattern.ReplaceAllString(input, "[REDACTED]")
}

监控指标

使用 Prometheus 统计:

apiDuration := prometheus.NewHistogramVec(prometheus.HistogramOpts{
    Name: "chatgpt_api_duration_seconds",
    Help: "API response time distribution",
}, []string{"endpoint"})