Fix HTTP response body not being closed properly

nezhahq · Dec 5, 2023 · abf3a71 · abf3a71
1 parent 793dafc
commit abf3a71
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/main.go b/main.go
@@ -405,6 +405,10 @@ func handleHttpGetTask(task *pb.Task, result *pb.TaskResult) {
 }
 
 func checkHttpResp(taskUrl string, start time.Time, resp *http.Response, err error, result *pb.TaskResult) {
+	if err == nil {
+		defer resp.Body.Close()
+		_, err = io.Copy(io.Discard, resp.Body)
+	}
 	if err == nil {
 		// 检查 HTTP Response 状态
 		result.Delay = float32(time.Since(start).Microseconds()) / 1000.0