status.go

package node

import (
	"fmt"
	entity "github.com/eolinker/goku/server/entity/console-entity"
	"sync"
	"time"
)

const EXPIRE  = time.Second * 10
var(
	manager = _StatusManager{
		locker:sync.RWMutex{},
		lastHeartBeat:make(map[string]time.Time),
	}
)
type _StatusManager struct {
	locker sync.RWMutex
	lastHeartBeat map[string]time.Time
}

func (m *_StatusManager)refresh(id string)  {
	t:=time.Now()
	m.locker.Lock()

	m.lastHeartBeat[id]=t

	m.locker.Unlock()
}

func (m *_StatusManager)stop(id string)  {

	m.locker.Lock()

	delete(m.lastHeartBeat,id)

	m.locker.Unlock()
}
func  (m *_StatusManager)get(id string) (time.Time, bool)  {
	m.locker.RLock()
	t,b:=m.lastHeartBeat[id]
	m.locker.RUnlock()
	return t,b
}

func Refresh(ip string,port string)  {
	id:=fmt.Sprintf("%s:%d",ip,port)
	manager.refresh(id)
}

func NodeStop(ip,port string)  {
	id:=fmt.Sprintf("%s:%d",ip,port)
	manager.stop(id)
}

func IsLive(ip string,port string) bool  {
	id:=fmt.Sprintf("%s:%d",ip,port)
	t,has:=manager.get(id)
	if !has{
		return false
	}

	if  time.Now().Sub(t) > EXPIRE{
		return false
	}
	return true
}
func ResetNodeStatus(nodes... *entity.Node)  {
	for _, node:=range nodes{

		if IsLive(node.NodeIP,node.NodePort){
			node.NodeStatus = 1
		}else{
			node.NodeStatus = 0
		}
	}
}