Merge pull request #111 from r3inbowari/fix/lowtest

fix: resolve slow test exception
showwin · Feb 27, 2023 · 1ed2df6 · 1ed2df6
2 parents 4d46617 + b5b434f
commit 1ed2df6
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/speedtest/data_manager.go b/speedtest/data_manager.go
@@ -362,6 +362,9 @@ func (dc *DataChunk) DownloadHandler(r io.Reader) error {
 	bufP := blackHolePool.Get().(*[]byte)
 	readSize := 0
 	for {
+		if !dc.manager.running {
+			return nil
+		}
 		readSize, dc.err = r.Read(*bufP)
 		rs := int64(readSize)
 
@@ -404,6 +407,9 @@ func (dc *DataChunk) GetParent() Manager {
 }
 
 func (dc *DataChunk) Read(b []byte) (n int, err error) {
+	if !dc.manager.running {
+		return n, io.EOF
+	}
 	if dc.remainOrDiscardSize < readChunkSize {
 		if dc.remainOrDiscardSize <= 0 {
 			dc.endTime = time.Now()