caffee bitsflow ctrl

CortexFoundation · Dec 29, 2023 · 2d0d016 · 2d0d016
1 parent 1914237
commit 2d0d016
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/backend/caffe/t.go b/backend/caffe/t.go
@@ -109,6 +109,17 @@ func (t *Torrent) SetBytesRequested(bytesRequested int64) {
 	//t.Lock()
 	//defer t.Unlock()
 	//t.bytesRequested = bytesRequested
+	if bytesRequested <= t.bytesRequested.Load() {
+		return
+	}
+
+	if t.Info() != nil {
+		if bytesRequested > t.Length() {
+			t.bytesRequested.Store(t.Length())
+			return
+		}
+	}
+
 	t.bytesRequested.Store(bytesRequested)
 }