From 836919876a381fae859991ad423e04656cea24b4 Mon Sep 17 00:00:00 2001
From: Antonio Murdaca <runcom@redhat.com>
Date: Thu, 7 Sep 2017 17:09:04 +0200
Subject: [PATCH] pkg: storage: image: get size from image ID

Signed-off-by: Antonio Murdaca <runcom@redhat.com>
---
 pkg/storage/image.go | 24 ++++++++++++------------
 1 file changed, 12 insertions(+), 12 deletions(-)

diff --git a/pkg/storage/image.go b/pkg/storage/image.go
index 29e3c746..32ca1513 100644
--- a/pkg/storage/image.go
+++ b/pkg/storage/image.go
@@ -90,10 +90,12 @@ func (svc *imageService) ListImages(systemContext *types.SystemContext, filter s
 			if err != nil {
 				return nil, err
 			}
+			size := imageSize(img)
+			img.Close()
 			results = append(results, ImageResult{
 				ID:    image.ID,
 				Names: image.Names,
-				Size:  imageSize(img),
+				Size:  size,
 			})
 		}
 	} else {
@@ -102,18 +104,16 @@ func (svc *imageService) ListImages(systemContext *types.SystemContext, filter s
 			return nil, err
 		}
 		for _, image := range images {
-			var size *uint64
-			if len(image.Names) != 0 {
-				ref, err := svc.getRef(image.Names[0])
-				if err != nil {
-					return nil, err
-				}
-				img, err := ref.NewImage(systemContext)
-				if err != nil {
-					return nil, err
-				}
-				size = imageSize(img)
+			ref, err := istorage.Transport.ParseStoreReference(svc.store, "@"+image.ID)
+			if err != nil {
+				return nil, err
 			}
+			img, err := ref.NewImage(systemContext)
+			if err != nil {
+				return nil, err
+			}
+			size := imageSize(img)
+			img.Close()
 			results = append(results, ImageResult{
 				ID:    image.ID,
 				Names: image.Names,