libimage: add LayersDiskUsage

Add an API to query the sum of the layer sizes. This data is needed to fix containers/podman/issues/16135. Signed-off-by: Valentin Rothberg <[email protected]>
vrothberg · Oct 17, 2022 · cbe8b45 · cbe8b45
1 parent bbbc188
commit cbe8b45
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/libimage/corrupted_test.go b/libimage/corrupted_test.go
@@ -43,6 +43,8 @@ func TestCorruptedLayers(t *testing.T) {
 	// Disk usage works.
 	_, err = runtime.DiskUsage(ctx)
 	require.NoError(t, err, "disk usage works on healthy image")
+	_, err = runtime.LayersDiskUsage(ctx)
+	require.NoError(t, err, "layers disk usage works on healthy image")
 
 	// Now remove one layer from the layers.json index in the storage.  The
 	// image will still be listed in the container storage but attempting

diff --git a/libimage/disk_usage.go b/libimage/disk_usage.go
@@ -5,6 +5,20 @@ import (
 	"time"
 )
 
+// LayersDiskUsage returns the sum of the size of all layers in the current store.
+func (r *Runtime) LayersDiskUsage(ctx context.Context) (int64, error) {
+	layers, err := r.store.Layers()
+	if err != nil {
+		return -1, err
+	}
+
+	var size int64
+	for _, l := range layers {
+		size += l.UncompressedSize
+	}
+
+	return size, nil
+}
 // ImageDiskUsage reports the total size of an image.  That is the size
 type ImageDiskUsage struct {
 	// Number of containers using the image.