sync: fixed bandwidth limit bug when streaming (#5046)

juicedata · Aug 5, 2024 · cba7747 · cba7747
1 parent ff4f502
commit cba7747
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/pkg/sync/sync.go b/pkg/sync/sync.go
@@ -393,9 +393,6 @@ func doCopySingle(src, dst object.ObjectStorage, key string, size int64) error {
 }
 
 func doCopySingle0(src, dst object.ObjectStorage, key string, size int64) error {
-	if limiter != nil {
-		limiter.Wait(size)
-	}
 	concurrent <- 1
 	defer func() {
 		<-concurrent
@@ -432,6 +429,9 @@ type withProgress struct {
 }
 
 func (w *withProgress) Read(b []byte) (int, error) {
+	if limiter != nil {
+		limiter.Wait(int64(len(b)))
+	}
 	n, err := w.r.Read(b)
 	copiedBytes.IncrInt64(int64(n))
 	return n, err