Merge pull request #1557 from tonyhb/gc-logging

Utulize config log format within gc
This commit is contained in:
Aaron Lehmann 2016-03-24 15:53:26 -07:00
commit f4b6fc8d68
2 changed files with 12 additions and 7 deletions

View file

@ -17,9 +17,7 @@ import (
"github.com/spf13/cobra"
)
func markAndSweep(storageDriver driver.StorageDriver) error {
ctx := context.Background()
func markAndSweep(ctx context.Context, storageDriver driver.StorageDriver) error {
// Construct a registry
registry, err := storage.NewRegistry(ctx, storageDriver)
if err != nil {
@ -141,7 +139,14 @@ var GCCmd = &cobra.Command{
os.Exit(1)
}
err = markAndSweep(driver)
ctx := context.Background()
ctx, err = configureLogging(ctx, config)
if err != nil {
fmt.Fprintf(os.Stderr, "unable to configure logging with config: %s", err)
os.Exit(1)
}
err = markAndSweep(ctx, driver)
if err != nil {
fmt.Fprintf(os.Stderr, "failed to garbage collect: %v", err)
os.Exit(1)

View file

@ -161,7 +161,7 @@ func TestNoDeletionNoEffect(t *testing.T) {
}
// Run GC
err = markAndSweep(inmemoryDriver)
err = markAndSweep(context.Background(), inmemoryDriver)
if err != nil {
t.Fatalf("Failed mark and sweep: %v", err)
}
@ -193,7 +193,7 @@ func TestDeletionHasEffect(t *testing.T) {
manifests.Delete(ctx, image3.manifestDigest)
// Run GC
err = markAndSweep(inmemoryDriver)
err = markAndSweep(context.Background(), inmemoryDriver)
if err != nil {
t.Fatalf("Failed mark and sweep: %v", err)
}
@ -327,7 +327,7 @@ func TestOrphanBlobDeleted(t *testing.T) {
uploadRandomSchema2Image(t, repo)
// Run GC
err = markAndSweep(inmemoryDriver)
err = markAndSweep(context.Background(), inmemoryDriver)
if err != nil {
t.Fatalf("Failed mark and sweep: %v", err)
}