main.go

package main

import (
	"flag"
	"log"

	"ivanj26/sonic/constant"
	"ivanj26/sonic/lib/migrator"
	"ivanj26/sonic/util"
	"ivanj26/sonic/util/logger"
)

func main() {
	var sourceIp string
	flag.StringVar(&sourceIp, "s", "", "Redis source master ip address")

	var destIp string
	flag.StringVar(&destIp, "d", "", "Redis dest master ip address")

	var password string
	flag.StringVar(&password, "a", "", "Redis password")

	var slotRaw string
	flag.StringVar(&slotRaw, "l", "", "Slots. Could be range 6500,6600 or single slot 6600")

	var nbOfSlot int
	flag.IntVar(&nbOfSlot, "n", 0, "Number of slots. Must be positive integer >0.")

	var maxConcurrent int
	flag.IntVar(&maxConcurrent, "p", 5, "Number of parallel slot migration (equivalent to number of go routine)")

	var logFilePath string
	flag.StringVar(&logFilePath, "log-path", constant.DEFAULT_LOG_FILENAME, "Path to log file")

	var enableInfoLog bool
	flag.BoolVar(&enableInfoLog, "enable-info-log", true, "To disable info log, set as false")

	flag.Parse()

	// Initialize the logger
	err := logger.New().
		SetFilePath(logFilePath).
		SetInfoEnabled(enableInfoLog).
		Initialize()

	if err != nil {
		log.Fatalf("Failed to initialize logger: %v", err)
	}

	if util.IsEmpty(sourceIp) ||
		util.IsEmpty(destIp) ||
		util.IsEmpty(password) {
		logger.Fatal("Please fill the arguments to execute the reshard")
	}

	sonicMigrator, closeConn := migrator.New().
		SetSource(sourceIp, constant.DEFAULT_REDIS_PORT, password).
		SetDestination(destIp, constant.DEFAULT_REDIS_PORT, password).
		SetMaxConcurrency(maxConcurrent).
		Build()
	defer closeConn()

	// Perform resharding
	var slotI interface{}
	if util.IsEmpty(slotRaw) {
		slotI = nbOfSlot
	} else {
		slotI = slotRaw
	}

	err = sonicMigrator.Reshard(slotI)
	if err != nil {
		logger.Error(err.Error())
	}
}