imjasonh/kontain.me
1
0
Fork 0
mirror of https://github.com/imjasonh/kontain.me synced 2026-07-21 14:38:29 +00:00
Code Activity

thread contexts through

Browse source
This commit is contained in:
Jason Hall 2021-01-02 07:57:31 -05:00
parent 87374b6124
commit 06e4bad278
5 changed files with 17 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

13
cmd/viz/main.go
View file

@ -1,6 +1,7 @@
package main
import (
"context"
"fmt"
"io"
"log"
@ -37,6 +38,7 @@ type server struct {
}
func (s *server) ServeHTTP(w http.ResponseWriter, r *http.Request) {
ctx := r.Context()
s.info.Println("handler:", r.Method, r.URL)
if r.Method != http.MethodPost {
http.Error(w, "must be post", http.StatusMethodNotAllowed)
@ -44,14 +46,13 @@ func (s *server) ServeHTTP(w http.ResponseWriter, r *http.Request) {
}
r.ParseForm()
log.Println(r.Form) // TODO
refs, err := images(r.FormValue("images"))
if err != nil {
http.Error(w, err.Error(), http.StatusBadRequest)
return
}
d, err := genDot(refs)
d, err := genDot(ctx, refs)
if err != nil {
http.Error(w, err.Error(), http.StatusBadRequest)
return
@ -86,7 +87,7 @@ func images(s string) ([]name.Reference, error) {
return refs, nil
}
func genDot(refs []name.Reference) (string, error) {
func genDot(ctx context.Context, refs []name.Reference) (string, error) {
g := dot.NewGraph("images")
g.SetType(dot.DIGRAPH)
@ -98,7 +99,7 @@ func genDot(refs []name.Reference) (string, error) {
edges := map[string]bool{}
for _, ref := range refs {
layers, err := getLayers(ref)
layers, err := getLayers(ctx, ref)
if err != nil {
return "", fmt.Errorf("Failed to get layers for %q, ignoring: %v\n", ref, err)
}
@ -152,8 +153,8 @@ func short(layer v1.Descriptor) string {
return fmt.Sprintf("%s\n%s", layer.Digest.String()[7:19], humanize.Bytes(uint64(layer.Size)))
}
func getLayers(ref name.Reference) ([]v1.Descriptor, error) {
i, err := remote.Image(ref)
func getLayers(ctx context.Context, ref name.Reference) ([]v1.Descriptor, error) {
i, err := remote.Image(ref, remote.WithContext(ctx))
if err != nil {
return nil, fmt.Errorf("remote.Image: %v", err)
}