aceberg/miniboard

package web

import (
    // "log"
    "fmt"
    "strings"
    "sync"
    "time"

    "github.com/aceberg/miniboard/internal/check"
    "github.com/aceberg/miniboard/internal/models"
    "github.com/aceberg/miniboard/internal/notify"
)

var (
    // RetriesSyncMap - count retries to send notifications
    RetriesSyncMap sync.Map
    // MuUptime - mutex for AllLinks.Uptime
    MuUptime sync.Mutex
)

func appendUptimeMon(panelName string, host models.Host, notif bool) {
    mon := models.MonData{}
    mon.Panel = panelName
    mon.Host = host.Name
    mon.Addr = host.Addr
    mon.Port = host.Port
    t := time.Now()
    mon.Date = t.Format("2006-01-02")
    mon.Time = t.Format("15:04:05")
    mon.Color = check.Color(mon.Date)
    mon.State = host.State
    if notif {
        MuUptime.Lock()
        mon.Notify = strings.Join(AllLinks.Uptime.Panels[panelName].Notify, " ")
        MuUptime.Unlock()
    }
    UptimeMon = append(UptimeMon, mon)

    newUptimeMutex.Lock()
    NewUptimeMon = append(NewUptimeMon, mon)
    newUptimeMutex.Unlock()
}

func scanUptime(panelName string, host models.Host, oldState bool) {
    var retries int
    var notifEnabled, notif bool

    MuUptime.Lock()
    panel, exists := AllLinks.Uptime.Panels[panelName]
    MuUptime.Unlock()

    if exists {
        if len(panel.Notify) > 0 {
            retriesAny, ok := RetriesSyncMap.LoadOrStore(panelName+host.Name, 0)
            if ok {
                retries = retriesAny.(int)
            }
            notifEnabled = true
        }

        if oldState != host.State {

            if notifEnabled && host.State && (retries > panel.Retries) {
                notify.Notify(panelName, host.Name, "is up", AllLinks.Uptime)
                RetriesSyncMap.Store(panelName+host.Name, 0)
                notif = true
            }
            appendUptimeMon(panelName, host, notif)
        }
        if notifEnabled && !host.State {
            if retries == panel.Retries {
                msg := fmt.Sprintf("is down (retries: %d)", panel.Retries)
                notify.Notify(panelName, host.Name, msg, AllLinks.Uptime)

                appendUptimeMon(panelName, host, true)
            }
            RetriesSyncMap.Store(panelName+host.Name, retries+1)
        }
    }
}