package main

import (
	"flag"
	"fmt"
	"os"
	"os/signal"
	"simple-kv-store/internal/logprovider"
	"simple-kv-store/internal/nodes"
	"strconv"
	"strings"
	"syscall"
	"github.com/syndtr/goleveldb/leveldb"

	"go.uber.org/zap"
)

var log, _ = logprovider.CreateDefaultZapLogger(zap.InfoLevel)

func main() {
	defer func() {
		if err := recover(); err != nil {
			log.Info("i get a panic", zap.Any("panic error", err))
		}
	}()

	// 设置一个通道来捕获中断信号
	sigs := make(chan os.Signal, 1)
	signal.Notify(sigs, syscall.SIGTERM, syscall.SIGINT)

	port := flag.String("port", ":9091", "rpc listen port")
	cluster := flag.String("cluster", "127.0.0.1:9092,127.0.0.1:9093", "comma sep")
	id := flag.String("id", "1", "node ID")
	pipe := flag.String("pipe", "", "input from scripts")
	isLeader := flag.Bool("isleader", false, "init node state")
	isNewDb := flag.Bool("isNewDb", true, "new test or restart")

	// 参数解析
	flag.Parse()
	clusters := strings.Split(*cluster, ",")
	idClusterPairs := make(map[string]string)
	idCnt := 1
	selfi, err := strconv.Atoi(*id)
	if err != nil {
		log.Error("figure id only")
	}
	for _, addr := range clusters {
		if idCnt == selfi {
			idCnt++ // 命令行cluster按id排序传入，记录时跳过自己的id，先保证所有节点互相记录的id一致
		}
		idClusterPairs[strconv.Itoa(idCnt)] = addr 
		idCnt++
	}

	if *isNewDb {
		os.RemoveAll("leveldb/simple-kv-store" + *id)
	}
	// 打开或创建每个结点自己的数据库
	db, err := leveldb.OpenFile("leveldb/simple-kv-store" + *id, nil)
	if err != nil {
		log.Fatal("Failed to open database: ", zap.Error(err))
	}
	defer db.Close() // 确保数据库在使用完毕后关闭
	iter := db.NewIterator(nil, nil)
	defer iter.Release()

	// 计数
	count := 0
	for iter.Next() {
		count++
	}
	fmt.Printf(*id + "结点目前有数据：%d\n", count)

	node := nodes.Init(*id, idClusterPairs, *pipe, db)
	log.Info("id: " + *id + "节点开始监听: " + *port + "端口")
	// 监听rpc
	node.Rpc(*port)
	// 开启 raft
	nodes.Start(node, *isLeader)

	sig := <-sigs
	fmt.Println("node_" + *id + "接收到信号:", sig)

}