mirror of
https://github.com/gravitl/netmaker.git
synced 2024-09-20 23:36:18 +08:00
only check for zombies on same network
This commit is contained in:
parent
63a8abf5cc
commit
cce7e30aeb
|
@ -213,7 +213,7 @@ func DeleteNodeByID(node *models.Node, exterminate bool) error {
|
|||
// ignoring for now, could hit a nil pointer if delete called twice
|
||||
logger.Log(2, "attempted to remove node ACL for node", node.Name, node.ID)
|
||||
}
|
||||
|
||||
removeZombie <- node.ID
|
||||
return removeLocalServer(node)
|
||||
}
|
||||
|
||||
|
@ -313,6 +313,7 @@ func CreateNode(node *models.Node) error {
|
|||
if err != nil {
|
||||
return err
|
||||
}
|
||||
CheckZombies(node)
|
||||
|
||||
nodebytes, err := json.Marshal(&node)
|
||||
if err != nil {
|
||||
|
|
96
logic/zombie.go
Normal file
96
logic/zombie.go
Normal file
|
@ -0,0 +1,96 @@
|
|||
package logic
|
||||
|
||||
import (
|
||||
"context"
|
||||
"time"
|
||||
|
||||
"github.com/gravitl/netmaker/logger"
|
||||
"github.com/gravitl/netmaker/models"
|
||||
)
|
||||
|
||||
const (
|
||||
ZOMBIE_TIMEOUT = 60 // timeout in seconds
|
||||
ZOMBIE_DELETE_TIME = 10 // minutes
|
||||
)
|
||||
|
||||
var (
|
||||
zombies []string
|
||||
removeZombie chan string = make(chan (string))
|
||||
newZombie chan string = make(chan (string))
|
||||
)
|
||||
|
||||
// CheckZombies - checks if new node has same macaddress as existing node
|
||||
// if so, existing node is added to zombie node quarantine list
|
||||
func CheckZombies(newnode *models.Node) {
|
||||
nodes, err := GetNetworkNodes(newnode.Network)
|
||||
if err != nil {
|
||||
logger.Log(1, "Failed to retrieve network nodes", newnode.Network, err.Error())
|
||||
return
|
||||
}
|
||||
for _, node := range nodes {
|
||||
if node.MacAddress == newnode.MacAddress {
|
||||
newZombie <- node.ID
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// ManageZombies - goroutine which adds/removes/deletes nodes from the zombie node quarantine list
|
||||
func ManageZombies(ctx context.Context) {
|
||||
for {
|
||||
select {
|
||||
case <-ctx.Done():
|
||||
return
|
||||
case id := <-newZombie:
|
||||
logger.Log(1, "adding", id, "to zombie quaratine list")
|
||||
zombies = append(zombies, id)
|
||||
case id := <-removeZombie:
|
||||
for i, zombie := range zombies {
|
||||
if zombie == id {
|
||||
logger.Log(1, "removing zombie from quaratine list", zombie)
|
||||
zombies = append(zombies[:i], zombies[i+1:]...)
|
||||
}
|
||||
}
|
||||
logger.Log(3, "no zombies found")
|
||||
case <-time.After(time.Second * ZOMBIE_TIMEOUT):
|
||||
for i, zombie := range zombies {
|
||||
node, err := GetNodeByID(zombie)
|
||||
if err != nil {
|
||||
logger.Log(1, "error retrieving zombie node", zombie, err.Error())
|
||||
continue
|
||||
}
|
||||
if time.Since(time.Unix(node.LastCheckIn, 0)) > time.Minute*ZOMBIE_DELETE_TIME {
|
||||
if err := DeleteNodeByID(&node, true); err != nil {
|
||||
logger.Log(1, "error deleting zombie node", zombie, err.Error())
|
||||
}
|
||||
logger.Log(1, "deleting zombie node", node.Name)
|
||||
zombies = append(zombies[:i], zombies[i+1:]...)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
//InitializeZombies - populates the zombie quarantine list (should be called from initialization)
|
||||
func InitalizeZombies() {
|
||||
nodes, err := GetAllNodes()
|
||||
if err != nil {
|
||||
logger.Log(1, "failed to retrieve nodes", err.Error())
|
||||
return
|
||||
}
|
||||
for _, node := range nodes {
|
||||
othernodes, err := GetNetworkNodes(node.Network)
|
||||
if err != nil {
|
||||
logger.Log(1, "failled to retrieve nodes for network", node.Network, err.Error())
|
||||
continue
|
||||
}
|
||||
for _, othernode := range othernodes {
|
||||
if node.MacAddress == othernode.MacAddress {
|
||||
if node.LastCheckIn > othernode.LastCheckIn {
|
||||
zombies = append(zombies, othernode.ID)
|
||||
} else {
|
||||
zombies = append(zombies, node.ID)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
2
main.go
2
main.go
|
@ -127,6 +127,7 @@ func initialize() { // Client Mode Prereq Check
|
|||
logger.Log(0, "error occurred when notifying nodes of startup", err.Error())
|
||||
}
|
||||
}
|
||||
logic.InitalizeZombies()
|
||||
}
|
||||
|
||||
func startControllers() {
|
||||
|
@ -169,6 +170,7 @@ func runMessageQueue(wg *sync.WaitGroup) {
|
|||
var client = mq.SetupMQTT(false) // Set up the subscription listener
|
||||
ctx, cancel := context.WithCancel(context.Background())
|
||||
go mq.Keepalive(ctx)
|
||||
go logic.ManageZombies(ctx)
|
||||
quit := make(chan os.Signal, 1)
|
||||
signal.Notify(quit, syscall.SIGTERM, os.Interrupt)
|
||||
<-quit
|
||||
|
|
Loading…
Reference in a new issue