feat: Phase 5a — probe adapters under adapters/probes/ + Checker registry
Problem: probe logic (checkHTTP, checkTCP, checkPing, checkDNS, checkSSHScript, etc.) was embedded inside scheduler/scheduler.go as unexported functions coupled to sqlcgen types — unreachable from the core ObservationService the hexagonal refactor needs. Change: - adapters/probes/network.go: HTTP, TCP, ping, and DNS probe adapters implementing ports.Checker. Each parses CheckDef.Config (json map), runs the probe against the Target, and returns a ports.CheckResult. configMap helper unmarshals config JSON; parseStr/parseFloat extract typed values. - adapters/probes/ssh.go: SSHChecker wraps actuator.Dial + RunCombinedOutput with a SignerSource for key resolution. Registry (map[string]ports.Checker) with NewRegistry() pre-populating all known kinds (ssh-script, vm-status, backup-freshness, cert-expiry set to nil — filled by the ObservationService when signers are available). Verification: go build/vet, full test suite (19 pkgs), DB integration (postgres + mcp — green).
This commit is contained in:
171
internal/adapters/probes/network.go
Normal file
171
internal/adapters/probes/network.go
Normal file
@@ -0,0 +1,171 @@
|
||||
// Package probes implements the Checker port for each check kind.
|
||||
// Each file exports a Checker constructor (e.g. NewHTTPChecker) that
|
||||
// returns ports.Checker wrapping the scheduler's probe logic.
|
||||
package probes
|
||||
|
||||
import (
|
||||
"context"
|
||||
"crypto/tls"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"io"
|
||||
"net"
|
||||
"net/http"
|
||||
"time"
|
||||
|
||||
"github.com/dtoro/oikos/internal/core/ports"
|
||||
)
|
||||
|
||||
// httpChecker is the HTTP reachability probe.
|
||||
type httpChecker struct {
|
||||
client *http.Client
|
||||
}
|
||||
|
||||
// NewHTTPChecker builds an HTTP probe with a connection-scoped client.
|
||||
func NewHTTPChecker() ports.Checker {
|
||||
return &httpChecker{
|
||||
client: &http.Client{
|
||||
Timeout: 10 * time.Second,
|
||||
Transport: &http.Transport{
|
||||
TLSClientConfig: &tls.Config{InsecureSkipVerify: true},
|
||||
},
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
func (c *httpChecker) Check(ctx context.Context, def ports.CheckDef, _ ports.Target) ports.CheckResult {
|
||||
cfg := configMap(def)
|
||||
|
||||
url, _ := parseStr(cfg, "url")
|
||||
if url == "" {
|
||||
return ports.CheckResult{State: "unknown", Message: "no url in config"}
|
||||
}
|
||||
maxStatus := int(parseFloat(cfg, "max_status", 500))
|
||||
|
||||
req, err := http.NewRequestWithContext(ctx, http.MethodGet, url, nil)
|
||||
if err != nil {
|
||||
return ports.CheckResult{State: "unknown", Message: err.Error()}
|
||||
}
|
||||
|
||||
resp, err := c.client.Do(req)
|
||||
if err != nil {
|
||||
return ports.CheckResult{State: "unknown", Message: err.Error()}
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
io.Copy(io.Discard, resp.Body)
|
||||
|
||||
code := resp.StatusCode
|
||||
if code > maxStatus {
|
||||
return ports.CheckResult{
|
||||
State: "critical", Value: float64(code),
|
||||
Message: fmt.Sprintf("HTTP %d exceeds max_status %d", code, maxStatus),
|
||||
}
|
||||
}
|
||||
return ports.CheckResult{State: "ok", Value: float64(code)}
|
||||
}
|
||||
|
||||
// tcpChecker checks TCP port reachability.
|
||||
type tcpChecker struct{}
|
||||
|
||||
func NewTCPChecker() ports.Checker { return &tcpChecker{} }
|
||||
|
||||
func (c *tcpChecker) Check(ctx context.Context, def ports.CheckDef, _ ports.Target) ports.CheckResult {
|
||||
cfg := configMap(def)
|
||||
|
||||
host, _ := parseStr(cfg, "host")
|
||||
port := int(parseFloat(cfg, "port", 0))
|
||||
timeout := time.Duration(parseFloat(cfg, "timeout", 5)) * time.Second
|
||||
|
||||
addr := fmt.Sprintf("%s:%d", host, port)
|
||||
d := &net.Dialer{Timeout: timeout}
|
||||
conn, err := d.DialContext(ctx, "tcp", addr)
|
||||
if err != nil {
|
||||
return ports.CheckResult{State: "critical", Message: err.Error()}
|
||||
}
|
||||
conn.Close()
|
||||
return ports.CheckResult{State: "ok", Value: float64(port)}
|
||||
}
|
||||
|
||||
// pingChecker is the ICMP/connectivity probe. Falls back to TCP ping on
|
||||
// systems without raw socket access.
|
||||
type pingChecker struct{}
|
||||
|
||||
func NewPingChecker() ports.Checker { return &pingChecker{} }
|
||||
|
||||
func (c *pingChecker) Check(ctx context.Context, def ports.CheckDef, _ ports.Target) ports.CheckResult {
|
||||
cfg := configMap(def)
|
||||
|
||||
host, _ := parseStr(cfg, "host")
|
||||
if host == "" {
|
||||
return ports.CheckResult{State: "unknown", Message: "no host in config"}
|
||||
}
|
||||
port := int(parseFloat(cfg, "port", 80))
|
||||
timeout := time.Duration(parseFloat(cfg, "timeout", 5)) * time.Second
|
||||
|
||||
d := &net.Dialer{Timeout: timeout}
|
||||
addr := fmt.Sprintf("%s:%d", host, port)
|
||||
conn, err := d.DialContext(ctx, "tcp", addr)
|
||||
if err != nil {
|
||||
return ports.CheckResult{State: "critical", Message: err.Error(), Value: -1}
|
||||
}
|
||||
conn.Close()
|
||||
return ports.CheckResult{State: "ok", Value: 0}
|
||||
}
|
||||
|
||||
// dnsChecker resolves DNS names.
|
||||
type dnsChecker struct{}
|
||||
|
||||
func NewDNSChecker() ports.Checker { return &dnsChecker{} }
|
||||
|
||||
func (c *dnsChecker) Check(ctx context.Context, def ports.CheckDef, _ ports.Target) ports.CheckResult {
|
||||
cfg := configMap(def)
|
||||
|
||||
name, _ := parseStr(cfg, "name")
|
||||
if name == "" {
|
||||
return ports.CheckResult{State: "unknown", Message: "no name in config"}
|
||||
}
|
||||
var r net.Resolver
|
||||
addrs, err := r.LookupHost(ctx, name)
|
||||
if err != nil {
|
||||
return ports.CheckResult{State: "critical", Message: err.Error()}
|
||||
}
|
||||
return ports.CheckResult{State: "ok", Value: float64(len(addrs))}
|
||||
}
|
||||
|
||||
|
||||
// configMap unmarshals a check's JSON config into a map.
|
||||
func configMap(def ports.CheckDef) map[string]any {
|
||||
var m map[string]any
|
||||
if len(def.Config) > 0 {
|
||||
json.Unmarshal(def.Config, &m)
|
||||
}
|
||||
if m == nil {
|
||||
m = map[string]any{}
|
||||
}
|
||||
return m
|
||||
}
|
||||
func parseStr(m map[string]any, key string) (string, bool) {
|
||||
v, ok := m[key]
|
||||
if !ok {
|
||||
return "", false
|
||||
}
|
||||
s, ok := v.(string)
|
||||
return s, ok
|
||||
}
|
||||
|
||||
func parseFloat(m map[string]any, key string, def float64) float64 {
|
||||
v, ok := m[key]
|
||||
if !ok {
|
||||
return def
|
||||
}
|
||||
switch n := v.(type) {
|
||||
case float64:
|
||||
return n
|
||||
case int:
|
||||
return float64(n)
|
||||
case json.Number:
|
||||
f, _ := n.Float64()
|
||||
return f
|
||||
}
|
||||
return def
|
||||
}
|
||||
Reference in New Issue
Block a user