Merge pull request #579 from q384566678/hook-add-fix

validation: Increase err judgment

Author: liangchenye

validation/poststart.go

@@ -27,14 +27,15 @@ func main() {
 			r.SetID(uuid.NewV4().String())
 			g := util.GetDefaultGenerator()
 			output = filepath.Join(r.BundleDir, g.Spec().Root.Path, "output")
-			poststart := rspec.Hook{
+			err := g.AddPostStartHook(rspec.Hook{
 				Path: filepath.Join(r.BundleDir, g.Spec().Root.Path, "/bin/sh"),
 				Args: []string{
 					"sh", "-c", fmt.Sprintf("echo 'post-start called' >> %s", output),
 				},
+			})
+			if err != nil {
+				return err
 			}
-
-			g.AddPostStartHook(poststart)
 			g.SetProcessArgs([]string{"sh", "-c", fmt.Sprintf("echo 'process called' >> %s", "/output")})
 			r.SetConfig(g)
 			return nil

validation/poststop.go

@@ -28,14 +28,15 @@ func main() {
 			r.SetID(uuid.NewV4().String())
 			g := util.GetDefaultGenerator()
 			output = filepath.Join(r.BundleDir, g.Spec().Root.Path, "output")
-			poststop := rspec.Hook{
+			err := g.AddPostStopHook(rspec.Hook{
 				Path: filepath.Join(r.BundleDir, g.Spec().Root.Path, "/bin/sh"),
 				Args: []string{
 					"sh", "-c", fmt.Sprintf("echo 'post-stop called' >> %s", output),
 				},
+			})
+			if err != nil {
+				return err
 			}
-
-			g.AddPostStopHook(poststop)
 			g.SetProcessArgs([]string{"sh", "-c", fmt.Sprintf("echo 'process called' >> %s", "/output")})
 			r.SetConfig(g)
 			return nil

validation/prestart.go

@@ -26,14 +26,15 @@ func main() {
 			r.SetID(uuid.NewV4().String())
 			g := util.GetDefaultGenerator()
 			output = filepath.Join(r.BundleDir, g.Spec().Root.Path, "output")
-			prestart := rspec.Hook{
+			err := g.AddPreStartHook(rspec.Hook{
 				Path: filepath.Join(r.BundleDir, g.Spec().Root.Path, "/bin/sh"),
 				Args: []string{
 					"sh", "-c", fmt.Sprintf("echo 'pre-start called' >> %s", output),
 				},
+			})
+			if err != nil {
+				return err
 			}
-
-			g.AddPreStartHook(prestart)
 			g.SetProcessArgs([]string{"sh", "-c", fmt.Sprintf("echo 'process called' >> %s", "/output")})
 			r.SetConfig(g)
 			return nil