package ttslaunch import ( "context" "fmt" "io" "log/slog" "net/http" "time" ) // probeBodyLimit 是探活响应的最大读取量:探活只关心"有没有人应答",回应体一律丢弃, // 但也要有界(别被一个话痨服务的无限流拖住)。 const probeBodyLimit = 4 << 10 // httpProbe 用一次 HTTP GET 回答「这个端口上已经有人在应答吗」。 // // 判定口径(刻意宽松):**任何** HTTP 回应都算"在跑",只有传输层错误(连不上 / // 超时 / 连接被拒)才算"没起"。理由是这个服务的根路径回什么状态码并不稳定 // (404/405/200 都可能),而"端口有人说话"已经足以证明不该再拉第二个实例 —— // 误判成"没起"的代价是去起一个抢同一个端口的进程,代价明显更大。 // // `Proxy: nil` 是必须的:探的是回环地址,走环境里的 HTTP_PROXY 只会问到代理上, // 得到"有人在跑"的假结论。 func httpProbe(url string, timeout time.Duration) probeFunc { client := &http.Client{ Timeout: timeout, Transport: &http.Transport{ Proxy: nil, DisableKeepAlives: true, }, } return func(ctx context.Context) error { reqCtx, cancel := context.WithTimeout(ctx, timeout) defer cancel() req, err := http.NewRequestWithContext(reqCtx, http.MethodGet, url, nil) if err != nil { return fmt.Errorf("ttslaunch: build probe request: %w", err) } resp, err := client.Do(req) if err != nil { return fmt.Errorf("ttslaunch: probe %s: %w", url, err) } defer func() { _ = resp.Body.Close() }() _, _ = io.Copy(io.Discard, io.LimitReader(resp.Body, probeBodyLimit)) return nil } } // discardLogger 是没注入日志器时的兜底(绝不 nil 解引用)。 func discardLogger() *slog.Logger { return slog.New(slog.NewTextHandler(io.Discard, nil)) }