mirror of
https://github.com/coder/coder.git
synced 2026-06-04 13:38:21 +00:00
Spike Curtis
61ae5b81ab
Fixes an issue introduce in #21288 The default sdkclient created by the CLI root includes several additional http.RoundTripper wrappers to check versions and attach telemetry, so `DupClientCopyingHeaders` would break and scale tests would fail. Instead of explicitly adding support for these additional wrappers to `DupClientCopyingHeaders` I think we should just stop unwrapping and move on. Scale tests don't need these wrapped functions. This is a bit fragile, since it depends on the fact that the headers wrapper needs to be outermost, but that needs to be true for other uses, since things like dialing DERP do a similar thing where they unwrap and extract the auth headers. More long term this needs a refactor to make HTTP headers in the SDK a more first-class resource instead of this hacky RoundTripper wrapping, but that's for a different day.
52 lines
1.7 KiB
Go
52 lines
1.7 KiB
Go
package loadtestutil
|
|
|
|
import (
|
|
"maps"
|
|
"net/http"
|
|
|
|
"golang.org/x/xerrors"
|
|
|
|
"github.com/coder/coder/v2/codersdk"
|
|
)
|
|
|
|
// DupClientCopyingHeaders duplicates the Client, but with an independent underlying HTTP transport, so that it will not
|
|
// share connections with the client being duplicated. It copies any headers already on the existing transport as
|
|
// [codersdk.HeaderTransport] and add the headers in the argument.
|
|
func DupClientCopyingHeaders(client *codersdk.Client, header http.Header) (*codersdk.Client, error) {
|
|
nc := codersdk.New(client.URL, codersdk.WithLogger(client.Logger()))
|
|
nc.SessionTokenProvider = client.SessionTokenProvider
|
|
newHeader, t, err := extractHeaderAndInnerTransport(client.HTTPClient.Transport)
|
|
if err != nil {
|
|
return nil, xerrors.Errorf("extract headers: %w", err)
|
|
}
|
|
maps.Copy(newHeader, header)
|
|
|
|
nc.HTTPClient.Transport = &codersdk.HeaderTransport{
|
|
Transport: t.Clone(),
|
|
Header: newHeader,
|
|
}
|
|
return nc, nil
|
|
}
|
|
|
|
func extractHeaderAndInnerTransport(rt http.RoundTripper) (http.Header, *http.Transport, error) {
|
|
if t, ok := rt.(*http.Transport); ok {
|
|
// base case
|
|
return make(http.Header), t, nil
|
|
}
|
|
if ht, ok := rt.(*codersdk.HeaderTransport); ok {
|
|
headers, t, err := extractHeaderAndInnerTransport(ht.Transport)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
maps.Copy(headers, ht.Header)
|
|
return headers, t, nil
|
|
}
|
|
// unrecognized RoundTripper. Just return a default transport, since we only care about preserving headers.
|
|
t, ok := http.DefaultTransport.(*http.Transport)
|
|
if !ok {
|
|
// unhittable, unless the Go stdlib changes.
|
|
return nil, nil, xerrors.New("DefaultTransport is not *http.Transport")
|
|
}
|
|
return make(http.Header), t, nil
|
|
}
|