Gilles Soulier
99 lines
2.7 KiB
Go
99 lines
2.7 KiB
Go
package main
|
|
|
|
import (
|
|
"log"
|
|
"net/http"
|
|
"time"
|
|
|
|
"github.com/prometheus/client_golang/prometheus/promhttp"
|
|
"github.com/user/nanometrics/server/config"
|
|
"github.com/user/nanometrics/server/db"
|
|
"github.com/user/nanometrics/server/handlers"
|
|
"github.com/user/nanometrics/server/models"
|
|
prom "github.com/user/nanometrics/server/prometheus"
|
|
"github.com/user/nanometrics/server/transport"
|
|
ws "github.com/user/nanometrics/server/websocket"
|
|
)
|
|
|
|
func main() {
|
|
cfg := config.Load()
|
|
|
|
database, err := db.Open(cfg.DBPath)
|
|
if err != nil {
|
|
log.Fatalf("DB: %v", err)
|
|
}
|
|
|
|
hub := ws.NewHub()
|
|
|
|
onMetrics := func(m *models.AgentMetrics) {
|
|
if err := database.UpsertAgent(m); err != nil {
|
|
log.Printf("[ingest] upsert agent: %v", err)
|
|
}
|
|
if err := database.InsertMetrics(m); err != nil {
|
|
log.Printf("[ingest] insert metrics: %v", err)
|
|
}
|
|
prom.Update(m)
|
|
hub.Broadcast(models.WSMessage{
|
|
Type: "metrics_update",
|
|
AgentID: m.Hostname,
|
|
Data: m,
|
|
})
|
|
}
|
|
|
|
if err := transport.StartUDP(cfg.UDPAddr, onMetrics); err != nil {
|
|
log.Fatalf("UDP: %v", err)
|
|
}
|
|
|
|
var mqttClient *transport.MQTTClient
|
|
if mc, err := transport.StartMQTT(cfg.MQTTBroker, cfg.MQTTTopicBase, onMetrics); err != nil {
|
|
log.Printf("[mqtt] non disponible: %v", err)
|
|
} else {
|
|
mqttClient = mc
|
|
}
|
|
|
|
pushConfig := func(agentID string, agentCfg *models.AgentConfig) {
|
|
if mqttClient != nil {
|
|
if err := mqttClient.PushConfig(agentID, agentCfg); err != nil {
|
|
log.Printf("[mqtt] push config to %s: %v", agentID, err)
|
|
}
|
|
}
|
|
}
|
|
|
|
go func() {
|
|
ticker := time.NewTicker(time.Minute)
|
|
defer ticker.Stop()
|
|
for range ticker.C {
|
|
srvCfg, _ := database.GetServerConfig()
|
|
_ = database.PruneOldMetrics(srvCfg.RetentionDays)
|
|
_ = database.MarkOffline(30)
|
|
}
|
|
}()
|
|
|
|
mux := http.NewServeMux()
|
|
mux.Handle("/metrics", promhttp.Handler())
|
|
mux.Handle("/ws", ws.Handler(hub))
|
|
mux.HandleFunc("/api/agents", handlers.AgentsHandler(database))
|
|
mux.HandleFunc("/api/agents/", func(w http.ResponseWriter, r *http.Request) {
|
|
switch {
|
|
case endsWith(r.URL.Path, "/history"):
|
|
handlers.MetricsHistoryHandler(database)(w, r)
|
|
case endsWith(r.URL.Path, "/config"):
|
|
handlers.AgentConfigHandler(database, pushConfig)(w, r)
|
|
case endsWith(r.URL.Path, "/icon") && r.Method == http.MethodPost:
|
|
handlers.IconUploadHandler(database)(w, r)
|
|
case endsWith(r.URL.Path, "/icon") && r.Method == http.MethodGet:
|
|
handlers.IconGetHandler(database)(w, r)
|
|
default:
|
|
http.NotFound(w, r)
|
|
}
|
|
})
|
|
mux.HandleFunc("/api/config", handlers.ServerConfigHandler(database))
|
|
|
|
log.Printf("[http] écoute sur %s", cfg.HTTPAddr)
|
|
log.Fatal(http.ListenAndServe(cfg.HTTPAddr, mux))
|
|
}
|
|
|
|
func endsWith(path, suffix string) bool {
|
|
return len(path) >= len(suffix) && path[len(path)-len(suffix):] == suffix
|
|
}
|