Add BaseContext to http servers

This commit is contained in:
H1JK 2023-04-08 18:46:35 +08:00 committed by 世界
parent 5c1de2bb06
commit a548e45ad7
No known key found for this signature in database
GPG key ID: CD109927C34A63C4
4 changed files with 12 additions and 0 deletions

View file

@ -90,6 +90,9 @@ func (n *Naive) Start() error {
n.httpServer = &http.Server{ n.httpServer = &http.Server{
Handler: n, Handler: n,
TLSConfig: tlsConfig, TLSConfig: tlsConfig,
BaseContext: func(listener net.Listener) context.Context {
return n.ctx
},
} }
go func() { go func() {
var sErr error var sErr error

View file

@ -51,6 +51,9 @@ func NewServer(ctx context.Context, options option.V2RayGRPCOptions, tlsConfig t
} }
server.httpServer = &http.Server{ server.httpServer = &http.Server{
Handler: server, Handler: server,
BaseContext: func(net.Listener) context.Context {
return ctx
},
} }
server.h2cHandler = h2c.NewHandler(server, server.h2Server) server.h2cHandler = h2c.NewHandler(server, server.h2Server)
return server, nil return server, nil

View file

@ -70,6 +70,9 @@ func NewServer(ctx context.Context, options option.V2RayHTTPOptions, tlsConfig t
Handler: server, Handler: server,
ReadHeaderTimeout: C.TCPTimeout, ReadHeaderTimeout: C.TCPTimeout,
MaxHeaderBytes: http.DefaultMaxHeaderBytes, MaxHeaderBytes: http.DefaultMaxHeaderBytes,
BaseContext: func(net.Listener) context.Context {
return ctx
},
} }
server.h2cHandler = h2c.NewHandler(server, server.h2Server) server.h2cHandler = h2c.NewHandler(server, server.h2Server)
return server, nil return server, nil

View file

@ -53,6 +53,9 @@ func NewServer(ctx context.Context, options option.V2RayWebsocketOptions, tlsCon
Handler: server, Handler: server,
ReadHeaderTimeout: C.TCPTimeout, ReadHeaderTimeout: C.TCPTimeout,
MaxHeaderBytes: http.DefaultMaxHeaderBytes, MaxHeaderBytes: http.DefaultMaxHeaderBytes,
BaseContext: func(net.Listener) context.Context {
return ctx
},
} }
return server, nil return server, nil
} }