go net/http/httptest: Server.Close 变异 http.DefaultTransport

9ceoxa92  于 3个月前  发布在  Go
关注(0)|答案(1)|浏览(82)

Go版本:go1.21.6 linux/amd64

go env 在你的模块/工作区中的输出:

$ go env
GO111MODULE=''
GOARCH='amd64'
GOBIN=''
GOCACHE='/home/akashem/.cache/go-build'
GOENV='/home/akashem/.config/go/env'
GOEXE=''
GOEXPERIMENT=''
GOFLAGS=''
GOHOSTARCH='amd64'
GOHOSTOS='linux'
GOINSECURE=''
GOMODCACHE='/home/akashem/go/pkg/mod'
GONOPROXY=''
GONOSUMDB=''
GOOS='linux'
GOPATH='/home/akashem/go'
GOPRIVATE=''
GOPROXY='https://proxy.golang.org,direct'
GOROOT='/usr/lib/go'
GOSUMDB='sum.golang.org'
GOTMPDIR=''
GOTOOLCHAIN='auto'
GOTOOLDIR='/usr/lib/go/pkg/tool/linux_amd64'
GOVCS=''
GOVERSION='go1.21.6'
GCCGO='gccgo'
GOAMD64='v1'
AR='ar'
CC='gcc'
CXX='g++'
CGO_ENABLED='1'
GOMOD='/home/akashem/go/src/k8s.io/kubernetes/go.mod'
GOWORK=''
CGO_CFLAGS='-O2 -g'
CGO_CPPFLAGS=''
CGO_CXXFLAGS='-O2 -g'
CGO_FFLAGS='-O2 -g'
CGO_LDFLAGS='-O2 -g'
PKG_CONFIG='pkg-config'
GOGCCFLAGS='-fPIC -m64 -pthread -Wl,--no-gc-sections -fmessage-length=0 -ffile-prefix-map=/tmp/go-build2922644753=/tmp/go-build -gno-record-gcc-switches'

你做了什么?
似乎默认的传输对象 http.DefaultTransportClose 方法修改了。下面是一个演示它的测试:

func TestHTTPTestServerCloseMutatesDefaultTransport(t *testing.T) {
	server := httptest.NewUnstartedServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		w.Write([]byte("pong"))
	}))
	server.EnableHTTP2 = true
	server.StartTLS()

	client := server.Client()
	client.Get(server.URL + "/ping")

	want := http.DefaultTransport.(*http.Transport).TLSClientConfig
	if want != nil {
		t.Errorf("expected TLSClientConfig of http.DefaultTransport to be nil")
	}
	server.Close()
	if got := http.DefaultTransport.(*http.Transport).TLSClientConfig; want != got {
		t.Errorf("TLSClientConfig has been mutated")
	}
}

DefaultTransport 对象的 TLSClientConfig 字段最初是 nil,但在 Close 方法返回后,它被一个非空示例替换。以下是变异的调用链:
httptest/Server.Close 调用默认 DefaultTransport 对象的 CloseIdleConnections 方法:
go/src/net/http/httptest/server.go
第237行到第242行:| | // 不是 httptest.Server 正确性的一部分,但假设大多数 | | // httptest.Server 的用户将使用标准传输,所以帮助他们并为他们关闭任何空闲连接。 | | ift, ok:=http.DefaultTransport.(closeIdleTransport); ok { | | t.CloseIdleConnections() | | } |
go/src/net/http/transport.go
第783行到第784行:| | func (t*Transport) CloseIdleConnections() { | | t.nextProtoOnce.Do(t.onceSetNextProtoDefaults) | TLSClientConfig.NextProtos 被设置:
go/src/net/http/h2_bundle.go
第7269行到第7274行:| | ift1.TLSClientConfig==nil { | | t1.TLSClientConfig=new(tls.Config) | | } | | if!http2strSliceContains(t1.TLSClientConfig.NextProtos, "h2") { | | t1.TLSClientConfig.NextProtos=append([]string{"h2"}, t1.TLSClientConfig.NextProtos...) | | } |

我不确定生产代码是否经常会执行:

if t, ok := http.DefaultTransport.(closeIdleTransport); ok {
		t.CloseIdleConnections()
	}

我在验证给定传输的 TLSClientConfig 字段不会意外地被修改时遇到了这个问题。我可以修改测试来解决它。但我想把它提给你注意,这样你可以评估它是否会影响生产代码。

相关问题