93 lines
2.8 KiB
Go
93 lines
2.8 KiB
Go
package httpapi
|
|
|
|
import (
|
|
"net/http"
|
|
"time"
|
|
|
|
"cloudipvalidator/internal/db"
|
|
)
|
|
|
|
func (s *Server) handleProberRegister(w http.ResponseWriter, r *http.Request) {
|
|
var req registerProberRequest
|
|
if err := readJSON(r, &req); err != nil {
|
|
writeError(w, http.StatusBadRequest, "invalid body: "+err.Error())
|
|
return
|
|
}
|
|
if s.Orch.SiteIndexForID(req.SiteID) == 0 {
|
|
writeError(w, http.StatusBadRequest, "unknown site_id: "+req.SiteID)
|
|
return
|
|
}
|
|
s.Orch.RecordEvent(r.Context(), "prober", req.SiteID, nil, "registered", "")
|
|
writeJSON(w, http.StatusOK, registerAgentResponse{OK: true, PollIntervalSeconds: s.Orch.Cfg.PollIntervalSeconds})
|
|
}
|
|
|
|
// handleProberAssignments returns every IP currently in the checking
|
|
// state — probers work the whole active set each poll, not one IP at a
|
|
// time, since multiple validators run in parallel.
|
|
func (s *Server) handleProberAssignments(w http.ResponseWriter, r *http.Request) {
|
|
siteID := r.PathValue("site_id")
|
|
if s.Orch.SiteIndexForID(siteID) == 0 {
|
|
writeError(w, http.StatusNotFound, "unknown site_id: "+siteID)
|
|
return
|
|
}
|
|
items, err := s.DB.ListChecking(r.Context())
|
|
if err != nil {
|
|
writeError(w, http.StatusInternalServerError, err.Error())
|
|
return
|
|
}
|
|
out := make([]proberAssignment, 0, len(items))
|
|
for _, item := range items {
|
|
out = append(out, proberAssignment{
|
|
IPID: item.ID, IPAddress: item.IPAddress,
|
|
Ports: s.Orch.Inbound.Ports, ICMP: s.Orch.Inbound.ICMP,
|
|
})
|
|
}
|
|
writeJSON(w, http.StatusOK, out)
|
|
}
|
|
|
|
func (s *Server) handleProberResults(w http.ResponseWriter, r *http.Request) {
|
|
siteID := r.PathValue("site_id")
|
|
siteIndex := s.Orch.SiteIndexForID(siteID)
|
|
if siteIndex == 0 {
|
|
writeError(w, http.StatusNotFound, "unknown site_id: "+siteID)
|
|
return
|
|
}
|
|
var req proberResultsRequest
|
|
if err := readJSON(r, &req); err != nil {
|
|
writeError(w, http.StatusBadRequest, "invalid body: "+err.Error())
|
|
return
|
|
}
|
|
|
|
completed := map[int64]bool{}
|
|
for _, res := range req.Results {
|
|
item, err := s.DB.GetIP(r.Context(), res.IPID)
|
|
if err != nil {
|
|
writeError(w, http.StatusNotFound, "unknown ip_id")
|
|
return
|
|
}
|
|
checkedAt, err := time.Parse(time.RFC3339Nano, res.CheckedAt)
|
|
if err != nil {
|
|
checkedAt = db.Now()
|
|
}
|
|
err = s.Orch.RecordCheck(r.Context(), db.Check{
|
|
IPID: item.ID, IPAddress: item.IPAddress, AttemptNumber: item.AttemptNumber,
|
|
Source: db.InboundSource(siteIndex), CheckType: res.CheckType, Target: res.IPAddress,
|
|
Success: res.Success, LatencyMS: res.LatencyMS, Detail: res.Detail, CheckedAt: checkedAt,
|
|
})
|
|
if err != nil {
|
|
writeError(w, http.StatusInternalServerError, err.Error())
|
|
return
|
|
}
|
|
if res.Complete {
|
|
completed[res.IPID] = true
|
|
}
|
|
}
|
|
for ipID := range completed {
|
|
if err := s.Orch.MarkSiteComplete(r.Context(), ipID, siteIndex); err != nil {
|
|
writeError(w, http.StatusInternalServerError, err.Error())
|
|
return
|
|
}
|
|
}
|
|
writeJSON(w, http.StatusOK, okResponse{OK: true})
|
|
}
|