fix incompleted reponse compute

faceair · May 3, 2023 · b769ae5 · b769ae5
1 parent e309db1
commit b769ae5
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/main.go b/main.go
@@ -153,7 +153,11 @@ func TestProxy(proxy C.Proxy, downloadSize int, timeout time.Duration) *Result {
 	}
 	ttfb := time.Since(start)
 
-	io.Copy(io.Discard, resp.Body)
+	written, _ := io.Copy(io.Discard, resp.Body)
+	if written == 0 {
+		return &Result{-1, -1}
+	}
+	downloadSize = int(written)
 	downloadTime := time.Since(start) - ttfb
 	bandwidth := float64(downloadSize) / downloadTime.Seconds()