client: remove unused printer
Signed-off-by: Andrea Luzzardi <aluzzardi@gmail.com>
This commit is contained in:
parent
167edf4a76
commit
c9e0d0854d
139
dagger/client.go
139
dagger/client.go
@ -3,7 +3,6 @@ package dagger
|
|||||||
import (
|
import (
|
||||||
"archive/tar"
|
"archive/tar"
|
||||||
"context"
|
"context"
|
||||||
"fmt"
|
|
||||||
"io"
|
"io"
|
||||||
"os"
|
"os"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
@ -15,7 +14,6 @@ import (
|
|||||||
"github.com/rs/zerolog/log"
|
"github.com/rs/zerolog/log"
|
||||||
|
|
||||||
// Cue
|
// Cue
|
||||||
"cuelang.org/go/cue"
|
|
||||||
|
|
||||||
// buildkit
|
// buildkit
|
||||||
bk "github.com/moby/buildkit/client"
|
bk "github.com/moby/buildkit/client"
|
||||||
@ -79,33 +77,11 @@ func (c *Client) Compute(ctx context.Context, env *Env) (o *Value, err error) {
|
|||||||
return c.buildfn(ctx, env, events, outw)
|
return c.buildfn(ctx, env, events, outw)
|
||||||
})
|
})
|
||||||
|
|
||||||
// Spawn print function(s)
|
// Spawn print function
|
||||||
dispCtx := context.TODO()
|
|
||||||
if os.Getenv("DOCKER_OUTPUT") != "" {
|
if os.Getenv("DOCKER_OUTPUT") != "" {
|
||||||
// Multiplex events
|
|
||||||
eventsPrint := make(chan *bk.SolveStatus)
|
|
||||||
eventsDockerPrint := make(chan *bk.SolveStatus)
|
|
||||||
eg.Go(func() error {
|
eg.Go(func() error {
|
||||||
defer close(eventsPrint)
|
dispCtx := context.TODO()
|
||||||
defer close(eventsDockerPrint)
|
return c.dockerprintfn(dispCtx, events, lg)
|
||||||
|
|
||||||
for e := range events {
|
|
||||||
eventsPrint <- e
|
|
||||||
eventsDockerPrint <- e
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
})
|
|
||||||
|
|
||||||
eg.Go(func() error {
|
|
||||||
return c.printfn(dispCtx, eventsPrint)
|
|
||||||
})
|
|
||||||
|
|
||||||
eg.Go(func() error {
|
|
||||||
return c.dockerprintfn(dispCtx, eventsDockerPrint, lg)
|
|
||||||
})
|
|
||||||
} else {
|
|
||||||
eg.Go(func() error {
|
|
||||||
return c.printfn(dispCtx, events)
|
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -223,115 +199,6 @@ func (c *Client) outputfn(ctx context.Context, r io.Reader, out *Value, cc *Comp
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// Status of a node in the config tree being computed
|
|
||||||
// Node may be a component, or a value within a component
|
|
||||||
// (eg. a script or individual operation in a script)
|
|
||||||
type Node struct {
|
|
||||||
Path cue.Path
|
|
||||||
*bk.Vertex
|
|
||||||
}
|
|
||||||
|
|
||||||
func (n Node) ComponentPath() cue.Path {
|
|
||||||
parts := []cue.Selector{}
|
|
||||||
for _, sel := range n.Path.Selectors() {
|
|
||||||
if strings.HasPrefix(sel.String(), "#") {
|
|
||||||
break
|
|
||||||
}
|
|
||||||
parts = append(parts, sel)
|
|
||||||
}
|
|
||||||
return cue.MakePath(parts...)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (n Node) Logf(ctx context.Context, msg string, args ...interface{}) {
|
|
||||||
componentPath := n.ComponentPath().String()
|
|
||||||
args = append([]interface{}{componentPath}, args...)
|
|
||||||
if msg != "" && !strings.HasSuffix(msg, "\n") {
|
|
||||||
msg += "\n"
|
|
||||||
}
|
|
||||||
fmt.Fprintf(os.Stderr, "[%s] "+msg, args...)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (n Node) LogStream(ctx context.Context, nStream int, data []byte) {
|
|
||||||
lg := log.
|
|
||||||
Ctx(ctx).
|
|
||||||
With().
|
|
||||||
Str("path", n.ComponentPath().String()).
|
|
||||||
Logger()
|
|
||||||
|
|
||||||
switch nStream {
|
|
||||||
case 1:
|
|
||||||
lg = lg.With().Str("stream", "stdout").Logger()
|
|
||||||
case 2:
|
|
||||||
lg = lg.With().Str("stream", "stderr").Logger()
|
|
||||||
default:
|
|
||||||
lg = lg.With().Str("stream", fmt.Sprintf("%d", nStream)).Logger()
|
|
||||||
}
|
|
||||||
|
|
||||||
lg.Debug().Msg(string(data))
|
|
||||||
}
|
|
||||||
|
|
||||||
func (n Node) LogError(ctx context.Context, errmsg string) {
|
|
||||||
log.
|
|
||||||
Ctx(ctx).
|
|
||||||
Error().
|
|
||||||
Str("path", n.ComponentPath().String()).
|
|
||||||
Msg(errmsg)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *Client) printfn(ctx context.Context, ch chan *bk.SolveStatus) error {
|
|
||||||
lg := log.Ctx(ctx)
|
|
||||||
|
|
||||||
// Node status mapped to buildkit vertex digest
|
|
||||||
nodesByDigest := map[string]*Node{}
|
|
||||||
// Node status mapped to cue path
|
|
||||||
nodesByPath := map[string]*Node{}
|
|
||||||
|
|
||||||
for {
|
|
||||||
select {
|
|
||||||
case <-ctx.Done():
|
|
||||||
return ctx.Err()
|
|
||||||
case status, ok := <-ch:
|
|
||||||
if !ok {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
lg.
|
|
||||||
Debug().
|
|
||||||
Int("vertexes", len(status.Vertexes)).
|
|
||||||
Int("statuses", len(status.Statuses)).
|
|
||||||
Int("logs", len(status.Logs)).
|
|
||||||
Msg("status event")
|
|
||||||
|
|
||||||
for _, v := range status.Vertexes {
|
|
||||||
// FIXME: insert raw buildkit telemetry here (ie for debugging, etc.)
|
|
||||||
|
|
||||||
// IF a buildkit vertex has a valid cue path as name, extract additional info:
|
|
||||||
p := cue.ParsePath(v.Name)
|
|
||||||
if err := p.Err(); err != nil {
|
|
||||||
// Not a valid cue path: skip.
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
n := &Node{
|
|
||||||
Path: p,
|
|
||||||
Vertex: v,
|
|
||||||
}
|
|
||||||
nodesByPath[n.Path.String()] = n
|
|
||||||
nodesByDigest[n.Digest.String()] = n
|
|
||||||
if n.Error != "" {
|
|
||||||
n.LogError(ctx, n.Error)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
for _, log := range status.Logs {
|
|
||||||
if n, ok := nodesByDigest[log.Vertex.String()]; ok {
|
|
||||||
n.LogStream(ctx, log.Stream, log.Data)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
// debugJSON(status)
|
|
||||||
// FIXME: callbacks for extracting stream/result
|
|
||||||
// see proto 67
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *Client) dockerprintfn(ctx context.Context, ch chan *bk.SolveStatus, out io.Writer) error {
|
func (c *Client) dockerprintfn(ctx context.Context, ch chan *bk.SolveStatus, out io.Writer) error {
|
||||||
var cons console.Console
|
var cons console.Console
|
||||||
// FIXME: use smarter writer from blr
|
// FIXME: use smarter writer from blr
|
||||||
|
Reference in New Issue
Block a user