84 lines
1.6 KiB
Go
84 lines
1.6 KiB
Go
package client
|
|
|
|
import (
|
|
"context"
|
|
"io"
|
|
"time"
|
|
|
|
controlapi "github.com/moby/buildkit/api/services/control"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
func (c *Client) Prune(ctx context.Context, ch chan UsageInfo, opts ...PruneOption) error {
|
|
info := &PruneInfo{}
|
|
for _, o := range opts {
|
|
o.SetPruneOption(info)
|
|
}
|
|
|
|
req := &controlapi.PruneRequest{
|
|
Filter: info.Filter,
|
|
KeepDuration: int64(info.KeepDuration),
|
|
KeepBytes: int64(info.KeepBytes),
|
|
}
|
|
if info.All {
|
|
req.All = true
|
|
}
|
|
cl, err := c.controlClient().Prune(ctx, req)
|
|
if err != nil {
|
|
return errors.Wrap(err, "failed to call prune")
|
|
}
|
|
|
|
for {
|
|
d, err := cl.Recv()
|
|
if err != nil {
|
|
if err == io.EOF {
|
|
return nil
|
|
}
|
|
return err
|
|
}
|
|
if ch != nil {
|
|
ch <- UsageInfo{
|
|
ID: d.ID,
|
|
Mutable: d.Mutable,
|
|
InUse: d.InUse,
|
|
Size: d.Size_,
|
|
Parent: d.Parent,
|
|
CreatedAt: d.CreatedAt,
|
|
Description: d.Description,
|
|
UsageCount: int(d.UsageCount),
|
|
LastUsedAt: d.LastUsedAt,
|
|
RecordType: UsageRecordType(d.RecordType),
|
|
Shared: d.Shared,
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
type PruneOption interface {
|
|
SetPruneOption(*PruneInfo)
|
|
}
|
|
|
|
type PruneInfo struct {
|
|
Filter []string
|
|
All bool
|
|
KeepDuration time.Duration
|
|
KeepBytes int64
|
|
}
|
|
|
|
type pruneOptionFunc func(*PruneInfo)
|
|
|
|
func (f pruneOptionFunc) SetPruneOption(pi *PruneInfo) {
|
|
f(pi)
|
|
}
|
|
|
|
var PruneAll = pruneOptionFunc(func(pi *PruneInfo) {
|
|
pi.All = true
|
|
})
|
|
|
|
func WithKeepOpt(duration time.Duration, bytes int64) PruneOption {
|
|
return pruneOptionFunc(func(pi *PruneInfo) {
|
|
pi.KeepDuration = duration
|
|
pi.KeepBytes = bytes
|
|
})
|
|
}
|