package monitor
|
|
import (
|
"basic.com/valib/logger.git"
|
"bee-config/config"
|
"bee-config/models"
|
"bee-config/utils"
|
"context"
|
"os"
|
"strings"
|
"sync"
|
"time"
|
)
|
|
func Init(ctx context.Context, wg *sync.WaitGroup) {
|
wg.Add(1)
|
go monitorLoop(ctx, wg)
|
}
|
|
func monitorLoop(ctx context.Context, wg *sync.WaitGroup) {
|
t := time.NewTicker(time.Second * 2)
|
for {
|
select {
|
case <-ctx.Done():
|
t.Stop()
|
wg.Done()
|
return
|
case <- t.C:
|
syncStatus()
|
}
|
}
|
}
|
|
func syncStatus() {
|
var node models.BeeNode
|
arr, err := node.FindAll()
|
if nil != err {
|
logger.Error("syncStatus failed to read all data, err:", err)
|
return
|
}
|
|
nodes := make([]*models.BeeNode, len(arr))
|
|
for _, v := range arr {
|
nodes = append(nodes, &v)
|
|
id := uint8(v.Id)
|
dir := config.GetNodeDir(id)
|
if _, err := os.Stat(dir); os.IsNotExist(err) {
|
continue
|
}
|
|
output, err := utils.ShellExecute(dir, "docker-compose", "ps", "-a")
|
if nil != err {
|
logger.Error("syncStatus failed to invoke docker-compose, err:", err)
|
continue
|
}
|
|
containers := parseDockerCompose(output)
|
if len(containers) < 2 {
|
logger.Error("syncStatus no enough containers")
|
continue
|
}
|
|
var beeStatus, clefStatus string
|
for _, v := range containers {
|
words := strings.Split(v, " ")
|
if len(words) < 5 {
|
continue
|
}
|
|
if strings.Contains(words[0], "bee-1") {
|
beeStatus = words[3]
|
}
|
|
if strings.Contains(words[0], "clef-1") {
|
clefStatus = words[3]
|
}
|
}
|
|
if len(beeStatus) < 1 {
|
logger.Error("syncStatus failed to find bee container")
|
continue
|
}
|
|
if len(clefStatus) < 1 {
|
logger.Error("syncStatus failed to find clefStatus container")
|
continue
|
}
|
}
|
}
|
|
func parseDockerCompose(output string) []string {
|
lines := strings.Split(output, "\n")
|
for i, v := range lines {
|
if strings.HasPrefix(v, "------") {
|
index := i + 1
|
return lines[index:]
|
}
|
}
|
|
return []string{}
|
}
|
