handle update error

arvancloud · Sep 20, 2022 · d4740ce · d4740ce
1 parent c3b783a
commit d4740ce
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/pkg/cli/cli.go b/pkg/cli/cli.go
@@ -110,6 +110,11 @@ func updateCmd() *cobra.Command {
 			reader, err := os.Open(filepath.Join(os.TempDir(), cliName))
 			utl.CheckErr(err)
 			err = update.Apply(reader, update.Options{})
+			if err != nil {
+				update.RollbackError(err)
+				fmt.Println(err)
+				utl.CheckErr(fmt.Errorf("update failed :("))
+			}
 			fmt.Println("update finished successfully :)")
 		},
 	}