package nodes

import (
	"net/rpc"
	"strconv"

	"go.uber.org/zap"
)

type State = uint8

const (
	Follower State = iota + 1
	Candidate
	Leader
)

type Public_node_info struct {
	connect bool
	address string
}

type Node struct {
	// 当前节点id
	selfId string

	// 除当前节点外其他节点信息
	nodes map[string]*Public_node_info

	//管道名
	pipeAddr string

	// 当前节点状态
	state State

	// 简单的kv存储
	log map[int]LogEntry

	// leader用来标记新log
	maxLogId int 
}

func (node *Node) BroadCastKV(logId int, kv LogEntry) {
	// 遍历所有节点
	for id, _ := range node.nodes {
		go func(id string, kv LogEntry) {
			var reply KVReply
			node.sendKV(id, logId, kv, &reply)
		}(id, kv)
	}
}

func (node *Node) sendKV(id string, logId int, kv LogEntry, reply *KVReply) {
	client, err := rpc.DialHTTP("tcp", node.nodes[id].address)
	if err != nil {
		log.Error("dialing: ", zap.Error(err))
		return
	}

	defer func(client *rpc.Client) {
		err := client.Close()
		if err != nil {
			log.Error("client close err: ", zap.Error(err))
		}
	}(client)

	arg := LogIdAndEntry{logId, kv}
	callErr := client.Call("Node.ReceiveKV", arg, reply) // RPC
	if callErr != nil {
		log.Error("dialing: ", zap.Error(callErr))
	}

	if reply.Reply { // 发送成功

	} else { // 失败

	}
}

// RPC call
func (node *Node) ReceiveKV(arg LogIdAndEntry, reply *KVReply) error {
	log.Info("node_" + node.selfId + " receive: logId = "+ strconv.Itoa(arg.LogId) + ", key = " + arg.Entry.Key)
	entry, ok := node.log[arg.LogId]
	if !ok {
		node.log[arg.LogId] = entry
	}
	// 持久化
	reply.Reply = true
	return nil
}

// func (node *Node) broadcastHeartbeat() {
// 	// 遍历所有节点
// 	for i := range raft.nodes {
// 		// request 参数
// 		hb := Heartbeat{
// 			Term:        raft.currTerm,
// 			LeaderId:    raft.self,
// 			CommitIndex: raft.commitIndex,
// 		}

// 		prevLogIndex := raft.nextIndex[i] - 1

// 		// 如果有日志未同步则发送
// 		if raft.getLastIndex() > prevLogIndex {
// 			hb.PrevLogIndex = prevLogIndex
// 			hb.PrevLogTerm = raft.log[prevLogIndex].CurrTerm
// 			hb.Entries = raft.log[prevLogIndex:]
// 			// log.Info("will send log entries", zap.Any("logEntries", hb.Entries))
// 		}

// 		go func(index int, hb Heartbeat) {
// 			var reply HeartbeatReply
// 			// 向某一个节点发送 heartbeat
// 			raft.sendHeartbeat(index, hb, &reply)
// 		}(i, hb)
// 	}
// }