瀏覽代碼

Close s3 request body on error if any

DarthSim 2 年之前
父節點
當前提交
d2cca005d3
共有 1 個文件被更改,包括 4 次插入0 次删除
  1. 4 0
      transport/s3/s3.go

+ 4 - 0
transport/s3/s3.go

@@ -64,6 +64,10 @@ func (t transport) RoundTrip(req *http.Request) (resp *http.Response, err error)
 	s3req, _ := t.svc.GetObjectRequest(input)
 
 	if err := s3req.Send(); err != nil {
+		if s3req.HTTPResponse != nil && s3req.HTTPResponse.Body != nil {
+			s3req.HTTPResponse.Body.Close()
+		}
+
 		if s3err, ok := err.(awserr.RequestFailure); !ok || s3err.StatusCode() < 100 || s3err.StatusCode() == 301 {
 			return nil, err
 		} else {