refactor: move prepare command to dataloss sub command

go-chaos/cmd/dataloss_sim.go

@@ -22,6 +22,7 @@ import (
 
 func init() {
 	rootCmd.AddCommand(datalossCmd)
+	rootCmd.AddCommand(prepareCmd)
 	datalossCmd.AddCommand(datalossDelete)
 	datalossCmd.AddCommand(datalossRecover)
 
@@ -35,6 +36,27 @@ var datalossCmd = &cobra.Command{
 	Long:  `Simulate dataloss of a broker, and restart from it. "zbchaos prepare" must be executed once on this cluster before this command`,
 }
 
+var prepareCmd = &cobra.Command{
+	Use:   "prepare",
+	Short: "Prepare the k8s deployment for dataloss test",
+	Long:  `Prepares the k8s deployment - such as applying patches to statefulsets - to enable applying dataloss commands.`,
+	Run: func(cmd *cobra.Command, args []string) {
+		k8Client, err := internal.CreateK8Client()
+		if err != nil {
+			panic(err)
+		}
+
+		// Add Init container for dataloss simulation test
+		err = k8Client.ApplyInitContainerPatch()
+
+		if err != nil {
+			panic(err)
+		}
+
+		fmt.Printf("Prepared cluster in namesapce %s\n", k8Client.GetCurrentNamespace())
+	},
+}
+
 var datalossDelete = &cobra.Command{
 	Use:   "delete",
 	Short: "Delete data of a broker",