-
Notifications
You must be signed in to change notification settings - Fork 21
/
rolllback.go
40 lines (34 loc) · 989 Bytes
/
rolllback.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
package cmd
import (
"log"
"github.com/0chain/gosdk/zboxcore/sdk"
"github.com/spf13/cobra"
)
// NOTE: This is for testing purpose only.
var rollbackCmd = &cobra.Command{
Use: "rollback",
Short: "rollback file to previous version",
Long: `rollback file to previous version`,
Args: cobra.MinimumNArgs(0),
Run: func(cmd *cobra.Command, args []string) {
fflags := cmd.Flags()
if !fflags.Changed("allocation") {
log.Fatal("Error: allocation flag is missing")
}
allocationID := cmd.Flag("allocation").Value.String()
allocationObj, err := sdk.GetAllocation(allocationID)
if err != nil {
log.Fatal("Error fetching the allocation.", err)
}
_, err = allocationObj.GetCurrentVersion()
if err != nil {
log.Fatal("Error rolling back", err)
}
log.Println("Rollback successful")
},
}
func init() {
rootCmd.AddCommand(rollbackCmd)
rollbackCmd.PersistentFlags().String("allocation", "", "Allocation ID")
_ = rollbackCmd.MarkFlagRequired("allocation")
}