move viz to a job

Docker-DCO-1.1-Signed-off-by: Victor Vieux <victor.vieux@docker.com> (github: vieux)
This commit is contained in:
Victor Vieux 2014-01-15 13:52:35 -08:00
parent 0113c087b2
commit 7fbc315aa9
3 changed files with 19 additions and 15 deletions

5
api.go
View File

@ -238,10 +238,7 @@ func getImagesViz(srv *Server, version float64, w http.ResponseWriter, r *http.R
w.WriteHeader(http.StatusNotFound) w.WriteHeader(http.StatusNotFound)
return fmt.Errorf("This is now implemented in the client.") return fmt.Errorf("This is now implemented in the client.")
} }
srv.Eng.ServeHTTP(w, r)
if err := srv.ImagesViz(w); err != nil {
return err
}
return nil return nil
} }

View File

@ -16,8 +16,10 @@ import (
// as the exit status. // as the exit status.
// //
func (eng *Engine) ServeHTTP(w http.ResponseWriter, r *http.Request) { func (eng *Engine) ServeHTTP(w http.ResponseWriter, r *http.Request) {
jobName := path.Base(r.URL.Path) var (
jobArgs, exists := r.URL.Query()["a"] jobName = path.Base(r.URL.Path)
jobArgs, exists = r.URL.Query()["a"]
)
if !exists { if !exists {
jobArgs = []string{} jobArgs = []string{}
} }

View File

@ -135,6 +135,10 @@ func jobInitApi(job *engine.Job) engine.Status {
job.Error(err) job.Error(err)
return engine.StatusErr return engine.StatusErr
} }
if err := job.Eng.Register("viz", srv.ImagesViz); err != nil {
job.Error(err)
return engine.StatusErr
}
return engine.StatusOK return engine.StatusOK
} }
@ -538,12 +542,12 @@ func (srv *Server) ImageInsert(name, url, path string, out io.Writer, sf *utils.
return nil return nil
} }
func (srv *Server) ImagesViz(out io.Writer) error { func (srv *Server) ImagesViz(job *engine.Job) engine.Status {
images, _ := srv.runtime.graph.Map() images, _ := srv.runtime.graph.Map()
if images == nil { if images == nil {
return nil return engine.StatusOK
} }
out.Write([]byte("digraph docker {\n")) job.Stdout.Write([]byte("digraph docker {\n"))
var ( var (
parentImage *Image parentImage *Image
@ -552,12 +556,13 @@ func (srv *Server) ImagesViz(out io.Writer) error {
for _, image := range images { for _, image := range images {
parentImage, err = image.GetParent() parentImage, err = image.GetParent()
if err != nil { if err != nil {
return fmt.Errorf("Error while getting parent image: %v", err) job.Errorf("Error while getting parent image: %v", err)
return engine.StatusErr
} }
if parentImage != nil { if parentImage != nil {
out.Write([]byte(" \"" + parentImage.ID + "\" -> \"" + image.ID + "\"\n")) job.Stdout.Write([]byte(" \"" + parentImage.ID + "\" -> \"" + image.ID + "\"\n"))
} else { } else {
out.Write([]byte(" base -> \"" + image.ID + "\" [style=invis]\n")) job.Stdout.Write([]byte(" base -> \"" + image.ID + "\" [style=invis]\n"))
} }
} }
@ -570,10 +575,10 @@ func (srv *Server) ImagesViz(out io.Writer) error {
} }
for id, repos := range reporefs { for id, repos := range reporefs {
out.Write([]byte(" \"" + id + "\" [label=\"" + id + "\\n" + strings.Join(repos, "\\n") + "\",shape=box,fillcolor=\"paleturquoise\",style=\"filled,rounded\"];\n")) job.Stdout.Write([]byte(" \"" + id + "\" [label=\"" + id + "\\n" + strings.Join(repos, "\\n") + "\",shape=box,fillcolor=\"paleturquoise\",style=\"filled,rounded\"];\n"))
} }
out.Write([]byte(" base [style=invisible]\n}\n")) job.Stdout.Write([]byte(" base [style=invisible]\n}\n"))
return nil return engine.StatusOK
} }
func (srv *Server) Images(job *engine.Job) engine.Status { func (srv *Server) Images(job *engine.Job) engine.Status {