Fix error state

keinenclue · Nov 11, 2021 · 941c28c · 941c28c
1 parent 5aa03e7
commit 941c28c
Showing 1 changed file with 10 additions and 6 deletions.
diff --git a/launcher/internal/container/container.go b/launcher/internal/container/container.go
@@ -149,30 +149,34 @@ func newContainer(image string, containerID string, containerEnv []string, conta
 
 // Launch launches the container
 func (c *LaunchableContainer) Launch() error {
-
+	var err error
 	c.setState(LaunchingState)
 	c.isLaunching = true
+
 	defer func() {
 		c.isLaunching = false
+		if err != nil {
+			c.setState(StoppingState)
+		}
 	}()
 
-	if err := c.Stop(); err != nil {
+	if err = c.Stop(); err != nil {
 		return err
 	}
 
-	if err := c.Remove(); err != nil {
+	if err = c.Remove(); err != nil {
 		return err
 	}
 
-	if err := c.Pull(); err != nil {
+	if err = c.Pull(); err != nil {
 		return err
 	}
 
-	if err := c.Start(); err != nil {
+	if err = c.Start(); err != nil {
 		return err
 	}
 
-	if err := c.Attach(); err != nil {
+	if err = c.Attach(); err != nil {
 		return err
 	}