Make BatchDownload honour maxSize when uploading to redis. (#306)

thought-machine · Apr 29, 2024 · 947353e · 947353e
1 parent f152c8d
commit 947353e
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 2 deletions.
diff --git a/ChangeLog b/ChangeLog
@@ -1,3 +1,7 @@
+Version 11.9.12
+--------------
+    * Make BatchDownload honour maxSize when uploading to redis.
+
 Version 11.9.11
 --------------
     * Allow setting pool timeout on redis client.

diff --git a/VERSION b/VERSION
@@ -1 +1 @@
-11.9.11
+11.9.12
diff --git a/mettle/worker/redis.go b/mettle/worker/redis.go
@@ -196,7 +196,9 @@ func (r *elanRedisWrapper) BatchDownload(dgs []digest.Digest) (map[digest.Digest
 	uploads := make([]interface{}, 0, 2*len(m))
 	for k, v := range m {
 		ret[k] = v
-		uploads = append(uploads, k.Hash, v)
+		if k.Size < r.maxSize {
+			uploads = append(uploads, k.Hash, v)
+		}
 	}
 	go r.writeBlobs(uploads)
 	return ret, nil