forked from TrueCloudLab/distribution
9c88801a12
Back in the before time, the best practices surrounding usage of Context weren't quite worked out. We defined our own type to make usage easier. As this packaged was used elsewhere, it make it more and more challenging to integrate with the forked `Context` type. Now that it is available in the standard library, we can just use that one directly. To make usage more consistent, we now use `dcontext` when referring to the distribution context package. Signed-off-by: Stephen J Day <stephen.day@docker.com>
68 lines
1.5 KiB
Go
68 lines
1.5 KiB
Go
package storage
|
|
|
|
import (
|
|
"context"
|
|
"path"
|
|
|
|
dcontext "github.com/docker/distribution/context"
|
|
"github.com/docker/distribution/registry/storage/driver"
|
|
"github.com/opencontainers/go-digest"
|
|
)
|
|
|
|
// vacuum contains functions for cleaning up repositories and blobs
|
|
// These functions will only reliably work on strongly consistent
|
|
// storage systems.
|
|
// https://en.wikipedia.org/wiki/Consistency_model
|
|
|
|
// NewVacuum creates a new Vacuum
|
|
func NewVacuum(ctx context.Context, driver driver.StorageDriver) Vacuum {
|
|
return Vacuum{
|
|
ctx: ctx,
|
|
driver: driver,
|
|
}
|
|
}
|
|
|
|
// Vacuum removes content from the filesystem
|
|
type Vacuum struct {
|
|
driver driver.StorageDriver
|
|
ctx context.Context
|
|
}
|
|
|
|
// RemoveBlob removes a blob from the filesystem
|
|
func (v Vacuum) RemoveBlob(dgst string) error {
|
|
d, err := digest.Parse(dgst)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
blobPath, err := pathFor(blobPathSpec{digest: d})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
dcontext.GetLogger(v.ctx).Infof("Deleting blob: %s", blobPath)
|
|
|
|
err = v.driver.Delete(v.ctx, blobPath)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// RemoveRepository removes a repository directory from the
|
|
// filesystem
|
|
func (v Vacuum) RemoveRepository(repoName string) error {
|
|
rootForRepository, err := pathFor(repositoriesRootPathSpec{})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
repoDir := path.Join(rootForRepository, repoName)
|
|
dcontext.GetLogger(v.ctx).Infof("Deleting repo: %s", repoDir)
|
|
err = v.driver.Delete(v.ctx, repoDir)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|