108 lines
2.3 KiB
Go
108 lines
2.3 KiB
Go
|
package solver
|
||
|
|
||
|
import (
|
||
|
"context"
|
||
|
"io"
|
||
|
"time"
|
||
|
|
||
|
"github.com/moby/buildkit/client"
|
||
|
"github.com/moby/buildkit/util/progress"
|
||
|
digest "github.com/opencontainers/go-digest"
|
||
|
"github.com/sirupsen/logrus"
|
||
|
)
|
||
|
|
||
|
func (j *Job) Status(ctx context.Context, ch chan *client.SolveStatus) error {
|
||
|
vs := &vertexStream{cache: map[digest.Digest]*client.Vertex{}}
|
||
|
pr := j.pr.Reader(ctx)
|
||
|
defer func() {
|
||
|
if enc := vs.encore(); len(enc) > 0 {
|
||
|
ch <- &client.SolveStatus{Vertexes: enc}
|
||
|
}
|
||
|
}()
|
||
|
for {
|
||
|
p, err := pr.Read(ctx)
|
||
|
if err != nil {
|
||
|
if err == io.EOF {
|
||
|
return nil
|
||
|
}
|
||
|
return err
|
||
|
}
|
||
|
ss := &client.SolveStatus{}
|
||
|
for _, p := range p {
|
||
|
switch v := p.Sys.(type) {
|
||
|
case client.Vertex:
|
||
|
ss.Vertexes = append(ss.Vertexes, vs.append(v)...)
|
||
|
|
||
|
case progress.Status:
|
||
|
vtx, ok := p.Meta("vertex")
|
||
|
if !ok {
|
||
|
logrus.Warnf("progress %s status without vertex info", p.ID)
|
||
|
continue
|
||
|
}
|
||
|
vs := &client.VertexStatus{
|
||
|
ID: p.ID,
|
||
|
Vertex: vtx.(digest.Digest),
|
||
|
Name: v.Action,
|
||
|
Total: int64(v.Total),
|
||
|
Current: int64(v.Current),
|
||
|
Timestamp: p.Timestamp,
|
||
|
Started: v.Started,
|
||
|
Completed: v.Completed,
|
||
|
}
|
||
|
ss.Statuses = append(ss.Statuses, vs)
|
||
|
case client.VertexLog:
|
||
|
vtx, ok := p.Meta("vertex")
|
||
|
if !ok {
|
||
|
logrus.Warnf("progress %s log without vertex info", p.ID)
|
||
|
continue
|
||
|
}
|
||
|
v.Vertex = vtx.(digest.Digest)
|
||
|
v.Timestamp = p.Timestamp
|
||
|
ss.Logs = append(ss.Logs, &v)
|
||
|
}
|
||
|
}
|
||
|
select {
|
||
|
case <-ctx.Done():
|
||
|
return ctx.Err()
|
||
|
case ch <- ss:
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
type vertexStream struct {
|
||
|
cache map[digest.Digest]*client.Vertex
|
||
|
}
|
||
|
|
||
|
func (vs *vertexStream) append(v client.Vertex) []*client.Vertex {
|
||
|
var out []*client.Vertex
|
||
|
vs.cache[v.Digest] = &v
|
||
|
if v.Cached {
|
||
|
for _, inp := range v.Inputs {
|
||
|
if inpv, ok := vs.cache[inp]; ok {
|
||
|
if !inpv.Cached && inpv.Completed == nil {
|
||
|
inpv.Cached = true
|
||
|
inpv.Started = v.Completed
|
||
|
inpv.Completed = v.Completed
|
||
|
}
|
||
|
delete(vs.cache, inp)
|
||
|
out = append(out, vs.append(*inpv)...)
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
vcopy := v
|
||
|
return append(out, &vcopy)
|
||
|
}
|
||
|
|
||
|
func (vs *vertexStream) encore() []*client.Vertex {
|
||
|
var out []*client.Vertex
|
||
|
for _, v := range vs.cache {
|
||
|
if v.Started != nil && v.Completed == nil {
|
||
|
now := time.Now()
|
||
|
v.Completed = &now
|
||
|
v.Error = context.Canceled.Error()
|
||
|
out = append(out, v)
|
||
|
}
|
||
|
}
|
||
|
return out
|
||
|
}
|