88 lines
2.4 KiB
Go
88 lines
2.4 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"time"
|
|
|
|
"git.milar.in/milarin/gmath"
|
|
"git.tordarus.net/tordarus/channel"
|
|
"git.tordarus.net/tordarus/envvars"
|
|
omadaapi "git.tordarus.net/tordarus/omada-api"
|
|
omadamodel "git.tordarus.net/tordarus/omada-api/model"
|
|
"git.tordarus.net/tordarus/slices"
|
|
)
|
|
|
|
var ( // flags
|
|
FlagSiteNames = envvars.StringSlice("SITES", ",", []string{})
|
|
FlagOmadaURL = envvars.String("OMADA_URL", "http://localhost:8088")
|
|
FlagOmadaID = envvars.String("OMADA_ID", "")
|
|
FlagOmadaClientID = envvars.String("OMADA_CLIENT_ID", "")
|
|
FlagOmadaClientSecret = envvars.String("OMADA_CLIENT_SECRET", "")
|
|
FlagOmadaUsername = envvars.String("OMADA_USERNAME", "")
|
|
FlagOmadaPassword = envvars.String("OMADA_PASSWORD", "")
|
|
FlagRefreshInterval = envvars.Duration("REFRESH_INTERVAL", 30*time.Second)
|
|
)
|
|
|
|
var MinRefreshInterval = 30 * time.Second
|
|
|
|
func main() {
|
|
FlagRefreshInterval = gmath.Max(FlagRefreshInterval, MinRefreshInterval)
|
|
|
|
api, err := omadaapi.NewApi(omadaapi.ApiConfig{
|
|
BasePath: FlagOmadaURL,
|
|
OmadaID: FlagOmadaID,
|
|
ClientID: FlagOmadaClientID,
|
|
ClientSecret: FlagOmadaClientSecret,
|
|
Username: FlagOmadaUsername,
|
|
Password: FlagOmadaPassword,
|
|
})
|
|
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
sites := slices.Filter(channel.ToSlice(api.GetSites()), FilterSitesByName(FlagSiteNames...))
|
|
|
|
ticker := time.NewTicker(FlagRefreshInterval)
|
|
|
|
CalculateSiteTraffic(api, sites, 0)
|
|
|
|
lastTick := time.Now()
|
|
for now := range ticker.C {
|
|
trafficStats := CalculateSiteTraffic(api, sites, now.Sub(lastTick))
|
|
fmt.Println(trafficStats)
|
|
lastTick = now
|
|
}
|
|
}
|
|
|
|
func CalculateSiteTraffic(api *omadaapi.Api, sites []*omadamodel.Site, duration time.Duration) *TrafficStats {
|
|
siteTraffics := map[string]*SiteTraffic{}
|
|
|
|
for _, site := range sites {
|
|
trafficByClient := map[string]TrafficRate{}
|
|
|
|
clients := channel.ToSlice(api.GetClients(site.ID))
|
|
for _, client := range clients {
|
|
traffic := CalculateClientTraffic(site, client, duration)
|
|
trafficByClient[client.Name] = traffic
|
|
}
|
|
|
|
siteTraffics[site.Name] = NewSiteTraffic(trafficByClient)
|
|
}
|
|
|
|
return NewTrafficStats(siteTraffics)
|
|
}
|
|
|
|
func FilterSitesByName(allowedSiteNames ...string) func(site *omadamodel.Site) bool {
|
|
if len(allowedSiteNames) == 0 {
|
|
return func(site *omadamodel.Site) bool { return true }
|
|
}
|
|
|
|
siteNames := slices.ToStructMap(allowedSiteNames)
|
|
|
|
return func(site *omadamodel.Site) bool {
|
|
_, ok := siteNames[site.Name]
|
|
return ok
|
|
}
|
|
}
|