William Valentin
256b841cbf
Adds Go microservices (ingest-gateway, event-processor, query-api, web-ui), NATS+Postgres wiring, initial schema/init job, ingress manifests for LAN+tailnet, and a multi-arch image build script.
110 lines
2.4 KiB
Go
110 lines
2.4 KiB
Go
package main
|
|
|
|
import (
|
|
"encoding/json"
|
|
"log"
|
|
"net/http"
|
|
"os"
|
|
|
|
"agentmon/internal/httpx"
|
|
qnats "agentmon/internal/queue/nats"
|
|
|
|
"github.com/go-chi/chi/v5"
|
|
"github.com/go-chi/chi/v5/middleware"
|
|
"github.com/gorilla/websocket"
|
|
)
|
|
|
|
func main() {
|
|
addr := envDefault("AGENTMON_ADDR", ":8080")
|
|
natsURL := envDefault("NATS_URL", "nats://nats:4222")
|
|
natsTopic := envDefault("NATS_TOPIC", "agentmon.events.v1")
|
|
|
|
pub, err := qnats.NewPublisher(natsURL, natsTopic)
|
|
if err != nil {
|
|
log.Fatalf("failed to connect to NATS: %v", err)
|
|
}
|
|
defer pub.Close()
|
|
|
|
r := chi.NewRouter()
|
|
r.Use(middleware.RequestID)
|
|
r.Use(middleware.RealIP)
|
|
r.Use(middleware.Logger)
|
|
r.Use(middleware.Recoverer)
|
|
|
|
r.Get("/healthz", func(w http.ResponseWriter, r *http.Request) {
|
|
w.WriteHeader(http.StatusOK)
|
|
_, _ = w.Write([]byte("ok"))
|
|
})
|
|
|
|
r.Post("/v1/events", func(w http.ResponseWriter, r *http.Request) {
|
|
var events []json.RawMessage
|
|
if err := json.NewDecoder(r.Body).Decode(&events); err != nil {
|
|
httpx.WriteJSON(w, http.StatusBadRequest, map[string]any{"error": "invalid_json"})
|
|
return
|
|
}
|
|
|
|
accepted := 0
|
|
rejected := 0
|
|
for _, raw := range events {
|
|
if len(raw) == 0 {
|
|
rejected++
|
|
continue
|
|
}
|
|
if err := pub.Publish(r.Context(), raw); err != nil {
|
|
rejected++
|
|
continue
|
|
}
|
|
accepted++
|
|
}
|
|
|
|
httpx.WriteJSON(w, http.StatusAccepted, map[string]any{"accepted": accepted, "rejected": rejected})
|
|
})
|
|
|
|
r.Get("/v1/ws", wsHandler(pub))
|
|
|
|
log.Printf("ingest-gateway listening on %s", addr)
|
|
log.Fatal(http.ListenAndServe(addr, r))
|
|
}
|
|
|
|
func wsHandler(pub *qnats.Publisher) http.HandlerFunc {
|
|
upgrader := websocket.Upgrader{
|
|
CheckOrigin: func(r *http.Request) bool { return true },
|
|
}
|
|
|
|
return func(w http.ResponseWriter, r *http.Request) {
|
|
conn, err := upgrader.Upgrade(w, r, nil)
|
|
if err != nil {
|
|
return
|
|
}
|
|
defer conn.Close()
|
|
|
|
for {
|
|
_, msg, err := conn.ReadMessage()
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
var v map[string]any
|
|
if err := json.Unmarshal(msg, &v); err != nil {
|
|
_ = conn.WriteJSON(map[string]any{"error": "invalid_json"})
|
|
continue
|
|
}
|
|
|
|
if err := pub.Publish(r.Context(), msg); err != nil {
|
|
_ = conn.WriteJSON(map[string]any{"error": "publish_failed"})
|
|
continue
|
|
}
|
|
|
|
_ = v
|
|
_ = conn.WriteJSON(map[string]any{"ack": map[string]any{"up_to_seq": nil}})
|
|
}
|
|
}
|
|
}
|
|
|
|
func envDefault(key, def string) string {
|
|
if v := os.Getenv(key); v != "" {
|
|
return v
|
|
}
|
|
return def
|
|
}
|