quick patch for negative workers

rule110-io · Jan 4, 2021 · 5dddd26 · 5dddd26
1 parent f161907
commit 5dddd26
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/surge/SurgeCore.go b/surge/SurgeCore.go
@@ -264,6 +264,9 @@ func WriteChunk(FileID string, ChunkID int32, Chunk []byte) {
 	defer RecoverAndLog()
 
 	workerCount--
+	if workerCount < 0 {
+		workerCount = 0
+	}
 
 	fileInfo, err := dbGetFile(FileID)
 	if err != nil {

diff --git a/surge/download.go b/surge/download.go
@@ -198,6 +198,10 @@ func downloadChunks(file *File, randomChunks []int) {
 					appendChunkLock.Unlock()
 
 					workerCount--
+					//TODO: Remove this clamp, dont double count timeouted arrivals
+					if workerCount < 0 {
+						workerCount = 0
+					}
 
 					//This file was not available at this time from this seeder, drop seeder for file.
 					mutateSeederLock.Lock()
@@ -271,6 +275,10 @@ func downloadChunks(file *File, randomChunks []int) {
 					appendChunkLock.Unlock()
 
 					workerCount--
+					//TODO: Remove this clamp, dont double count timeouted arrivals
+					if workerCount < 0 {
+						workerCount = 0
+					}
 
 					//This file was not available at this time from this seeder, drop seeder for file.
 					mutateSeederLock.Lock()