kopia/cli/command_storage_sweep.go

package cli

import (
	"fmt"
	"time"

	"github.com/kopia/kopia/repo"

	kingpin "gopkg.in/alecthomas/kingpin.v2"
)

var (
	storageSweepCommand      = storageCommands.Command("sweep", "Remove unused storage blocks")
	storageSweepAgeThreshold = storageSweepCommand.Flag("min-age", "Minimum age of storage block to delete").Default("24h").Duration()
	storageSweepPrintUnused  = storageSweepCommand.Flag("print", "Do not delete blocks, but print which ones would be deleted").Short('n').Bool()
)

func runStorageSweepAction(context *kingpin.ParseContext) error {
	rep := mustOpenRepository(nil)
	defer rep.Close()

	indexBlocks, err := rep.Blocks.ActiveIndexBlocks()
	if err != nil {
		return err
	}

	inUseIndexBlocks := map[string]bool{}
	for _, ib := range indexBlocks {
		inUseIndexBlocks[ib.BlockID] = true
	}

	ch, cancel := rep.Storage.ListBlocks("")
	defer cancel()

	for bm := range ch {
		if bm.Error != nil {
			return bm.Error
		}

		age := time.Since(bm.TimeStamp)
		inUse := rep.Blocks.IsStorageBlockInUse(bm.BlockID) || inUseIndexBlocks[bm.BlockID] || bm.BlockID == repo.FormatBlockID
		keep := inUse || age < *storageSweepAgeThreshold
		if !keep {
			if *storageSweepPrintUnused {
				fmt.Printf("unused block %v age %v\n", bm.BlockID, age)
			} else {
				fmt.Printf("deleting unused block %v age %v\n", bm.BlockID, age)
				if err := rep.Storage.DeleteBlock(bm.BlockID); err != nil {
					return err
				}
			}
		}
	}

	return nil
}

func init() {
	storageSweepCommand.Action(runStorageSweepAction)
}