TRKJ/EdgeNode
2
0
Fork 0
mirror of https://github.com/TeaOSLab/EdgeNode.git synced 2025-12-10 20:10:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

修复内容为空时无法缓存的Bug

Browse Source
This commit is contained in:
刘祥超
2022-06-09 20:26:36 +08:00
parent b2fc785543
commit d82d16e28d
3 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
internal/nodes/http_request_reverse_proxy.go
View File

@@ -328,7 +328,12 @@ func (this *HTTPRequest) doReverseProxy() {
// 是否有内容 // 是否有内容
if resp.ContentLength == 0 && len(resp.TransferEncoding) == 0 { if resp.ContentLength == 0 && len(resp.TransferEncoding) == 0 {
// 即使内容为0，也需要读取一次，以便于触发相关事件
var buf = utils.BytePool4k.Get()
_, _ = io.CopyBuffer(this.writer, resp.Body, buf)
utils.BytePool4k.Put(buf)
_ = resp.Body.Close() _ = resp.Body.Close()
this.writer.SetOk() this.writer.SetOk()
return return
} }

2
internal/nodes/http_writer.go
View File

@@ -836,7 +836,7 @@ func (this *HTTPWriter) HeaderData() []byte {
return nil return nil
} }
resp := &http.Response{} var resp = &http.Response{}
resp.Header = this.Header() resp.Header = this.Header()
if this.statusCode == 0 { if this.statusCode == 0 {
this.statusCode = http.StatusOK this.statusCode = http.StatusOK

4
internal/utils/readers/reader_closer_tee.go
View File

@@ -2,7 +2,9 @@
package readers package readers
import "io" import (
"io"
)
type TeeReaderCloser struct { type TeeReaderCloser struct {
r io.Reader r io.Reader