Show progress when loading packages from reflist.

debian/list.go

@@ -2,6 +2,7 @@ package debian
 
 import (
 	"fmt"
+	"github.com/smira/aptly/aptly"
 	"github.com/smira/aptly/utils"
 	"sort"
 	"strings"
@@ -49,7 +50,7 @@ func NewPackageList() *PackageList {
 }
 
 // NewPackageListFromRefList loads packages list from PackageRefList
-func NewPackageListFromRefList(reflist *PackageRefList, collection *PackageCollection) (*PackageList, error) {
+func NewPackageListFromRefList(reflist *PackageRefList, collection *PackageCollection, progress aptly.Progress) (*PackageList, error) {
 	// empty reflist
 	if reflist == nil {
 		return NewPackageList(), nil
@@ -57,14 +58,25 @@ func NewPackageListFromRefList(reflist *PackageRefList, collection *PackageColle
 
 	result := &PackageList{packages: make(map[string]*Package, reflist.Len())}
 
+	if progress != nil {
+		progress.InitBar(int64(reflist.Len()), false)
+	}
+
 	err := reflist.ForEach(func(key []byte) error {
 		p, err := collection.ByKey(key)
 		if err != nil {
 			return fmt.Errorf("unable to load package with key %s: %s", key, err)
 		}
+		if progress != nil {
+			progress.AddBar(1)
+		}
 		return result.Add(p)
 	})
 
+	if progress != nil {
+		progress.ShutdownBar()
+	}
+
 	if err != nil {
 		return nil, err
 	}

debian/publish.go

@@ -105,7 +105,7 @@ func (p *PublishedRepo) Publish(packagePool aptly.PackagePool, publishedStorage
 	}
 
 	// Load all packages
-	list, err := NewPackageListFromRefList(p.snapshot.RefList(), packageCollection)
+	list, err := NewPackageListFromRefList(p.snapshot.RefList(), packageCollection, nil)
 	if err != nil {
 		return fmt.Errorf("unable to load packages: %s", err)
 	}

debian/reflist_test.go

@@ -46,18 +46,18 @@ func (s *PackageRefListSuite) TestNewPackageListFromRefList(c *C) {
 
 	reflist := NewPackageRefListFromPackageList(s.list)
 
-	_, err := NewPackageListFromRefList(reflist, coll)
+	_, err := NewPackageListFromRefList(reflist, coll, nil)
 	c.Assert(err, ErrorMatches, "unable to load package with key.*")
 
 	coll.Update(s.p5)
 	coll.Update(s.p6)
 
-	list, err := NewPackageListFromRefList(reflist, coll)
+	list, err := NewPackageListFromRefList(reflist, coll, nil)
 	c.Assert(err, IsNil)
 	c.Check(list.Len(), Equals, 4)
 	c.Check(list.Add(s.p4), ErrorMatches, "conflict in package.*")
 
-	list, err = NewPackageListFromRefList(nil, coll)
+	list, err = NewPackageListFromRefList(nil, coll, nil)
 	c.Assert(err, IsNil)
 	c.Check(list.Len(), Equals, 0)
 }