wjqserver/touka
1
0
Fork 0
mirror of https://github.com/infinite-iroha/touka.git synced 2026-06-13 15:47:38 +08:00
Code Issues Projects Releases Packages Wiki Activity Actions 13

refactor: cache ResponseController in H2ReadWriteCloser for better performance

Browse source
This commit is contained in:
wjqserver 2026-04-02 19:44:02 +08:00
parent 7abedc1ace
commit 20dc6e4047
2 changed files with 7 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

9
reverseproxy.go
View file

@ -96,20 +96,21 @@ type reverseProxyExtendedConnectBridge struct {
type reverseProxyH2ReadWriteCloser struct {
io.ReadCloser
ResponseWriter
controller *http.ResponseController
}
func (rwc reverseProxyH2ReadWriteCloser) Write(p []byte) (int, error) {
func (rwc *reverseProxyH2ReadWriteCloser) Write(p []byte) (int, error) {
n, err := rwc.ResponseWriter.Write(p)
if err != nil {
return n, err
}
if err := http.NewResponseController(reverseProxyBaseResponseWriter(rwc.ResponseWriter)).Flush(); err != nil && !errors.Is(err, http.ErrNotSupported) {
if err := rwc.controller.Flush(); err != nil && !errors.Is(err, http.ErrNotSupported) {
return n, err
}
return n, nil
}
func (rwc reverseProxyH2ReadWriteCloser) Close() error {
func (rwc *reverseProxyH2ReadWriteCloser) Close() error {
if rwc.ReadCloser == nil {
return nil
}
@ -1012,7 +1013,7 @@ func (p *reverseProxyHandler) handleBridgedExtendedConnectResponse(c *Context, r
return &reverseProxyStatusError{status: http.StatusBadGateway, err: err}
}
conn := reverseProxyH2ReadWriteCloser{ReadCloser: bridge.body, ResponseWriter: c.Writer}
conn := &reverseProxyH2ReadWriteCloser{ReadCloser: bridge.body, ResponseWriter: c.Writer, controller: controller}
backConnClosed := make(chan struct{})
go func() {