Browse code

Check error returned by eng.Register() in TagStore.Install()

Signed-off-by: Tibor Vass <teabee89@gmail.com>

Tibor Vass authored on 2014/08/07 05:09:29
Showing 1 changed files
... ...
@@ -1,6 +1,7 @@
1 1
 package graph
2 2
 
3 3
 import (
4
+	"fmt"
4 5
 	"io"
5 6
 
6 7
 	"github.com/docker/docker/engine"
... ...
@@ -9,18 +10,24 @@ import (
9 9
 )
10 10
 
11 11
 func (s *TagStore) Install(eng *engine.Engine) error {
12
-	eng.Register("image_set", s.CmdSet)
13
-	eng.Register("image_tag", s.CmdTag)
14
-	eng.Register("tag", s.CmdTagLegacy) // FIXME merge with "image_tag"
15
-	eng.Register("image_get", s.CmdGet)
16
-	eng.Register("image_inspect", s.CmdLookup)
17
-	eng.Register("image_tarlayer", s.CmdTarLayer)
18
-	eng.Register("image_export", s.CmdImageExport)
19
-	eng.Register("history", s.CmdHistory)
20
-	eng.Register("images", s.CmdImages)
21
-	eng.Register("viz", s.CmdViz)
22
-	eng.Register("load", s.CmdLoad)
23
-	eng.Register("import", s.CmdImport)
12
+	for name, handler := range map[string]engine.Handler{
13
+		"image_set":      s.CmdSet,
14
+		"image_tag":      s.CmdTag,
15
+		"tag":            s.CmdTagLegacy, // FIXME merge with "image_tag"
16
+		"image_get":      s.CmdGet,
17
+		"image_inspect":  s.CmdLookup,
18
+		"image_tarlayer": s.CmdTarLayer,
19
+		"image_export":   s.CmdImageExport,
20
+		"history":        s.CmdHistory,
21
+		"images":         s.CmdImages,
22
+		"viz":            s.CmdViz,
23
+		"load":           s.CmdLoad,
24
+		"import":         s.CmdImport,
25
+	} {
26
+		if err := eng.Register(name, handler); err != nil {
27
+			return fmt.Errorf("Could not register %q: %v", name, err)
28
+		}
29
+	}
24 30
 	return nil
25 31
 }
26 32