128 lines
3.7 KiB
Go
128 lines
3.7 KiB
Go
package daemon // import "github.com/docker/docker/daemon"
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/docker/docker/api/server/router/system"
|
|
"github.com/docker/docker/api/types"
|
|
"github.com/docker/docker/api/types/container"
|
|
"github.com/docker/docker/api/types/filters"
|
|
"github.com/docker/docker/api/types/image"
|
|
"github.com/docker/docker/api/types/volume"
|
|
"github.com/pkg/errors"
|
|
"golang.org/x/sync/errgroup"
|
|
)
|
|
|
|
// containerDiskUsage obtains information about container data disk usage
|
|
// and makes sure that only one calculation is performed at the same time.
|
|
func (daemon *Daemon) containerDiskUsage(ctx context.Context) ([]*types.Container, error) {
|
|
res, _, err := daemon.usageContainers.Do(ctx, struct{}{}, func(ctx context.Context) ([]*types.Container, error) {
|
|
// Retrieve container list
|
|
containers, err := daemon.Containers(ctx, &container.ListOptions{
|
|
Size: true,
|
|
All: true,
|
|
})
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to retrieve container list: %v", err)
|
|
}
|
|
return containers, nil
|
|
})
|
|
return res, err
|
|
}
|
|
|
|
// imageDiskUsage obtains information about image data disk usage from image service
|
|
// and makes sure that only one calculation is performed at the same time.
|
|
func (daemon *Daemon) imageDiskUsage(ctx context.Context) ([]*image.Summary, error) {
|
|
imgs, _, err := daemon.usageImages.Do(ctx, struct{}{}, func(ctx context.Context) ([]*image.Summary, error) {
|
|
// Get all top images with extra attributes
|
|
imgs, err := daemon.imageService.Images(ctx, types.ImageListOptions{
|
|
Filters: filters.NewArgs(),
|
|
SharedSize: true,
|
|
ContainerCount: true,
|
|
})
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "failed to retrieve image list")
|
|
}
|
|
return imgs, nil
|
|
})
|
|
|
|
return imgs, err
|
|
}
|
|
|
|
// localVolumesSize obtains information about volume disk usage from volumes service
|
|
// and makes sure that only one size calculation is performed at the same time.
|
|
func (daemon *Daemon) localVolumesSize(ctx context.Context) ([]*volume.Volume, error) {
|
|
volumes, _, err := daemon.usageVolumes.Do(ctx, struct{}{}, func(ctx context.Context) ([]*volume.Volume, error) {
|
|
volumes, err := daemon.volumes.LocalVolumesSize(ctx)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return volumes, nil
|
|
})
|
|
return volumes, err
|
|
}
|
|
|
|
// layerDiskUsage obtains information about layer disk usage from image service
|
|
// and makes sure that only one size calculation is performed at the same time.
|
|
func (daemon *Daemon) layerDiskUsage(ctx context.Context) (int64, error) {
|
|
usage, _, err := daemon.usageLayer.Do(ctx, struct{}{}, func(ctx context.Context) (int64, error) {
|
|
usage, err := daemon.imageService.LayerDiskUsage(ctx)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
return usage, nil
|
|
})
|
|
return usage, err
|
|
}
|
|
|
|
// SystemDiskUsage returns information about the daemon data disk usage.
|
|
// Callers must not mutate contents of the returned fields.
|
|
func (daemon *Daemon) SystemDiskUsage(ctx context.Context, opts system.DiskUsageOptions) (*types.DiskUsage, error) {
|
|
eg, ctx := errgroup.WithContext(ctx)
|
|
|
|
var containers []*types.Container
|
|
if opts.Containers {
|
|
eg.Go(func() error {
|
|
var err error
|
|
containers, err = daemon.containerDiskUsage(ctx)
|
|
return err
|
|
})
|
|
}
|
|
|
|
var (
|
|
images []*image.Summary
|
|
layersSize int64
|
|
)
|
|
if opts.Images {
|
|
eg.Go(func() error {
|
|
var err error
|
|
images, err = daemon.imageDiskUsage(ctx)
|
|
return err
|
|
})
|
|
eg.Go(func() error {
|
|
var err error
|
|
layersSize, err = daemon.layerDiskUsage(ctx)
|
|
return err
|
|
})
|
|
}
|
|
|
|
var volumes []*volume.Volume
|
|
if opts.Volumes {
|
|
eg.Go(func() error {
|
|
var err error
|
|
volumes, err = daemon.localVolumesSize(ctx)
|
|
return err
|
|
})
|
|
}
|
|
|
|
if err := eg.Wait(); err != nil {
|
|
return nil, err
|
|
}
|
|
return &types.DiskUsage{
|
|
LayersSize: layersSize,
|
|
Containers: containers,
|
|
Volumes: volumes,
|
|
Images: images,
|
|
}, nil
|
|
}
|