package nodeService

import (
	"time"
	"vamicro/config"
	"vamicro/extend/util"

	"basic.com/pubsub/protomsg.git"
	"basic.com/valib/bhomedbapi.git"
	"basic.com/valib/logger.git"
)

type ClusterNode struct {
	// 当前节点是否为主节点
	IsMaster bool `json:"is_master"`
	// 当前服务状态
	ServerStatus bool `json:"server_status"`
	// 集群ID
	ClusterId string `json:"cluster_id"`
	// 集群名称
	ClusterName string `json:"cluster_name"`
	// 当前节点ID
	NodeId string `json:"node_id"`
	// 集群密码
	Password string `json:"password"`
	// 漂移IP
	VirtualIp string `json:"virtual_ip"`
	// 所有节点列表
	NodeLists []*protomsg.Node `json:"Nodes"`
}

// 节点属性
var Node = &ClusterNode{}

// 初始化节点属性
func InitNode() {

	// 当前节点ID
	Node.NodeId = config.Server.AnalyServerId
	// 初始化集群信息
	var clusterApi bhomedbapi.ClusterApi
	b, c := clusterApi.FindCluster()
	logger.Debugf("InitClusterNodes..., c=%v", c)
	if b && c.Nodes != nil && len(c.Nodes) > 0 {
		logger.Debug("Is cluster...")
		// 是集群
		Node.NodeLists = c.Nodes
		Node.ClusterId = c.ClusterId
		Node.ClusterName = c.ClusterName
		Node.Password = c.Password
		Node.VirtualIp = c.VirtualIp
		for _, n := range c.Nodes {
			if n.NodeId == Node.NodeId {
				// 当前节点是否为主节点
				if n.DriftState == "master" {
					logger.Debug("Is master...")
					Node.IsMaster = true
				} else {
					logger.Debug("Not master...")
					Node.IsMaster = false
				}
			}
		}
	} else {
		// 不是集群，只有当前节点
		logger.Debug("Not cluster...")

		Node.IsMaster = true
		Node.NodeLists = make([]*protomsg.Node, 0)
		ipv4, _, _ := util.GetLocalIP(config.Server.NetworkAdapter)
		Node.NodeLists = append(Node.NodeLists, &protomsg.Node{
			Id:         config.Server.AnalyServerId,
			ClusterId:  "",
			NodeName:   config.Server.ServerName,
			NodeId:     config.Server.AnalyServerId,
			NodeIp:     ipv4,
			IsDelete:   false,
			DriftState: "master",
			DeviceType: config.Server.DeviceType,
		})
	}
	logger.Debug("InitClusterNodes success...", Node)
}

// 监测节点变化
func WatchNode() {
	go func() {
		tk := time.NewTicker(15 * time.Second)
		for {
			select {
			case <-tk.C:
				InitNode()
			}
		}
	}()
}