Ask Bjørn Hansen
c9481d12c6
All checks were successful
continuous-integration/drone/push Build is passing
Replace MySQL driver with pgx/v5 and pgxpool: - Update sqlc to use postgresql engine - Convert query.sql to PostgreSQL syntax ($1 params, CASE WHEN, ANY() arrays) - Replace sql.DB with pgxpool.Pool throughout - Change nullable types from sql.Null* to pgtype.* - Update ID types from uint32 to int64 for PostgreSQL compatibility - Delete MySQL-specific dynamic_connect.go - Add opentelemetry.gowrap template for tracing
47 lines
1.2 KiB
Go
47 lines
1.2 KiB
Go
package server
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"net/netip"
|
|
"strconv"
|
|
"time"
|
|
|
|
"github.com/jackc/pgx/v5"
|
|
"go.ntppool.org/common/logger"
|
|
"go.ntppool.org/common/tracing"
|
|
"go.ntppool.org/data-api/ntpdb"
|
|
)
|
|
|
|
func (srv *Server) FindServer(ctx context.Context, serverID string) (ntpdb.Server, error) {
|
|
log := logger.Setup()
|
|
ctx, span := tracing.Tracer().Start(ctx, "FindServer")
|
|
defer span.End()
|
|
q := ntpdb.NewWrappedQuerier(ntpdb.New(srv.db))
|
|
|
|
var serverData ntpdb.Server
|
|
var dberr error
|
|
if id, err := strconv.Atoi(serverID); id > 0 && err == nil {
|
|
serverData, dberr = q.GetServerByID(ctx, int64(id))
|
|
} else {
|
|
ip, err := netip.ParseAddr(serverID)
|
|
if err != nil || !ip.IsValid() {
|
|
return ntpdb.Server{}, nil // 404 error
|
|
}
|
|
serverData, dberr = q.GetServerByIP(ctx, ip.String())
|
|
}
|
|
if dberr != nil {
|
|
if !errors.Is(dberr, pgx.ErrNoRows) {
|
|
log.Error("could not query server id", "err", dberr)
|
|
return serverData, dberr
|
|
}
|
|
}
|
|
|
|
if serverData.ID == 0 || (serverData.DeletionOn.Valid && serverData.DeletionOn.Time.Before(time.Now().Add(-1*time.Hour*24*30*24))) {
|
|
// no data and no error to produce 404 errors
|
|
return ntpdb.Server{}, nil
|
|
}
|
|
|
|
return serverData, nil
|
|
}
|