feat: add cmds

internal/cmd_channel_export.go

@@ -0,0 +1,69 @@
+package internal
+
+import (
+	"context"
+	"log"
+	"math"
+	"time"
+
+	"exporter/config"
+
+	"github.com/pkg/errors"
+	"github.com/spf13/cobra"
+	"go.uber.org/zap"
+)
+
+var dbChannelID int64
+
+func ChannelExportCmd() *cobra.Command {
+	cmd := &cobra.Command{
+		Use:     "export",
+		Aliases: []string{"e"},
+		Short:   "export channels",
+		PreRunE: func(cmd *cobra.Command, args []string) error {
+			log.Println("init client")
+			defer log.Println("init client done")
+			return InitClient(config.C)
+		},
+		RunE: wrapE(channelExportCmd),
+	}
+
+	cmd.Flags().Int64Var(&dbChannelID, "channel", 0, "channel id")
+
+	return cmd
+}
+
+func channelExportCmd(ctx context.Context) error {
+	if dbChannelID == 0 {
+		return errors.New("db channel id required")
+	}
+
+	channel, err := client.ChannelInfoByID(ctx, channelID)
+	if err != nil {
+		return err
+	}
+
+	if channel.GetID() == 0 {
+		return errors.New("channel not found")
+	}
+
+	cfg := NewDBChannel(channel.GetID(), channel.Username, channel.Title)
+	if err := cfg.GetOrCreate(ctx); err != nil {
+		return err
+	}
+
+	var continueRetryTimes int
+	for {
+		if err := client.Channel(ctx, channel, cfg, false); err != nil {
+			logger.Error("failed to export channel", zap.Error(err), zap.Int64("uuid", channel.GetID()))
+			continueRetryTimes++
+			sleepSeconds := math.Pow(2, float64(continueRetryTimes))
+			if sleepSeconds > 128 {
+				return err
+			}
+
+			logger.Info("retry after", zap.Int("seconds", int(sleepSeconds)))
+			time.Sleep(time.Second * time.Duration(sleepSeconds))
+		}
+	}
+}