add reboot queue backoff reset command

Signed-off-by: YZ775 <[email protected]>

pkg/ckecli/cmd/reboot_queue_backoff_reset.go

@@ -0,0 +1,44 @@
+package cmd
+
+import (
+	"context"
+	"time"
+
+	"github.com/cybozu-go/cke"
+	"github.com/cybozu-go/well"
+	"github.com/spf13/cobra"
+)
+
+var rebootQueueBackoffResetCmd = &cobra.Command{
+	Use:   "reset-backoff",
+	Short: "reset backoff",
+	Long:  `reset backoff`,
+	RunE: func(cmd *cobra.Command, args []string) error {
+		well.Go(func(ctx context.Context) error {
+			entries, err := storage.GetRebootsEntries(ctx)
+			if err != nil {
+				return err
+			}
+
+			for _, entry := range entries {
+				entry.DrainBackOffCount = 0
+				entry.DrainBackOffExpire = time.Time{}
+				err := storage.UpdateRebootsEntry(ctx, entry)
+				if err == cke.ErrNotFound {
+					// The entry has just finished
+					continue
+				}
+				if err != nil {
+					return err
+				}
+			}
+			return nil
+		})
+		well.Stop()
+		return well.Wait()
+	},
+}
+
+func init() {
+	rebootQueueCmd.AddCommand(rebootQueueBackoffResetCmd)
+}