ntppool/data-api
Private
Public Access
1
0
Fork 0
Code Activity

Calculate netspeed points, too
All checks were successful
continuous-integration/drone/push Build is passing
Details

Browse Source
This commit is contained in:
Ask Bjørn Hansen
2023-07-16 02:40:50 -07:00
parent 1f7973e885
commit 27c8bc4776
11 changed files with 281 additions and 123 deletions
Download Patch File Download Diff File Expand all files Collapse all files

109
server/dnsanswers.go
View File

@@ -6,7 +6,10 @@ import (
"github.com/labstack/echo/v4"
"go.ntppool.org/common/logger"
chdb "go.ntppool.org/data-api/chdb"
"go.ntppool.org/data-api/ntpdb"
"golang.org/x/exp/slog"
"golang.org/x/sync/errgroup"
)
const pointBasis float64 = 10000
@@ -23,11 +26,11 @@ func (srv *Server) dnsAnswers(c echo.Context) error {
c.Response().Header().Set("Cache-Control", "max-age=20")
conn, err := srv.chConn(ctx)
if err != nil {
slog.Error("could not connect to clickhouse", "err", err)
return c.String(http.StatusInternalServerError, "clickhouse error")
}
// conn, err := srv.chConn(ctx)
// if err != nil {
// slog.Error("could not connect to clickhouse", "err", err)
// return c.String(http.StatusInternalServerError, "clickhouse error")
// }
ip, err := netip.ParseAddr(c.Param("server"))
if err != nil {
@@ -35,37 +38,93 @@ func (srv *Server) dnsAnswers(c echo.Context) error {
return c.NoContent(http.StatusNotFound)
}
// q := ntpdb.New(srv.db)
// zoneStats, err := q.GetZoneStats(ctx)
// if err != nil {
// slog.Error("GetZoneStats", "err", err)
// return c.String(http.StatusInternalServerError, err.Error())
// }
// if zoneStats == nil {
// slog.Info("didn't get zoneStats")
// }
if ip.String() != c.Param("server") || len(c.QueryString()) > 0 {
return c.Redirect(http.StatusPermanentRedirect, "https://www.ntppool.org/api/data/server/dns/answers/"+ip.String())
}
days := 4
queryGroup, ctx := errgroup.WithContext(ctx)
serverData, err := srv.ch.ServerAnswerCounts(c.Request().Context(), conn, ip.String(), days)
var zoneStats []ntpdb.GetZoneStatsV2Row
var serverNetspeed uint32
queryGroup.Go(func() error {
var err error
q := ntpdb.New(srv.db)
zoneStats, err = q.GetZoneStatsV2(ctx, ip.String())
if err != nil {
slog.Error("GetZoneStatsV2", "err", err)
return err
}
if zoneStats == nil {
slog.Info("didn't get zoneStats")
}
serverNetspeed, err = q.GetServerNetspeed(ctx, ip.String())
if err != nil {
slog.Error("GetServerNetspeed", "err", err)
return err
}
return nil
})
days := 3
var serverData chdb.ServerQueries
queryGroup.Go(func() error {
var err error
serverData, err = srv.ch.ServerAnswerCounts(c.Request().Context(), ip.String(), days)
if err != nil {
slog.Error("ServerUserCCData", "err", err)
return err
}
return nil
})
var totalData chdb.ServerTotals
queryGroup.Go(func() error {
var err error
qtype := "A"
if ip.Is6() {
qtype = "AAAA"
}
totalData, err = srv.ch.AnswerTotals(c.Request().Context(), qtype, days)
if err != nil {
slog.Error("AnswerTotals", "err", err)
}
return err
})
err = queryGroup.Wait()
if err != nil {
slog.Error("ServerUserCCData", "err", err)
slog.Error("query error", "err", err)
return c.String(http.StatusInternalServerError, err.Error())
}
qtype := "A"
if ip.Is6() {
qtype = "AAAA"
}
zoneTotals := map[string]int32{}
totalData, err := srv.ch.AnswerTotals(c.Request().Context(), conn, qtype, days)
if err != nil {
slog.Error("AnswerTotals", "err", err)
return c.String(http.StatusInternalServerError, err.Error())
for _, z := range zoneStats {
zn := z.ZoneName
if zn == "@" {
zn = ""
}
zoneTotals[zn] = z.NetspeedActive // binary.BigEndian.Uint64(...)
// slog.Info("zone netspeed", "cc", z.ZoneName, "speed", z.NetspeedActive)
}
for _, cc := range serverData {
cc.Points = (pointBasis / float64(totalData[cc.CC])) * float64(cc.Count)
totalName := cc.CC
if totalName == "gb" {
totalName = "uk"
}
if zt, ok := zoneTotals[totalName]; ok {
cc.Netspeed = (pointBasis / float64(zt)) * float64(serverNetspeed)
}
// log.Info("points", "cc", cc.CC, "points", cc.Points)
}