CUB-117: Port Moonraker + MQTT printer integrations to Go
All checks were successful
Dev Build / build-test (pull_request) Successful in 1m29s
All checks were successful
Dev Build / build-test (pull_request) Successful in 1m29s
- Moonraker REST client with GetPrinterInfo, GetPrintStats, GetPrintHistory - Moonraker WebSocket client with auto-reconnect + telemetry parsing - MQTT client via paho.mqtt.golang with TLS support for Bambu Lab - Moonraker poller worker: background polling, dedup, usage logging to PostgreSQL - MQTT subscriber worker: Bambu telemetry parsing, print job tracking - Config: 7 new env vars (MOONRAKER_URL, MQTT_BROKER, etc.) - main.go: per-printer worker discovery, graceful shutdown
This commit is contained in:
@@ -6,31 +6,32 @@ import (
|
||||
"net/http"
|
||||
"os"
|
||||
"os/signal"
|
||||
"strconv"
|
||||
"sync"
|
||||
"syscall"
|
||||
"time"
|
||||
|
||||
"github.com/CubeCraft-Creations/Extrudex/backend/internal/clients"
|
||||
"github.com/CubeCraft-Creations/Extrudex/backend/internal/config"
|
||||
"github.com/CubeCraft-Creations/Extrudex/backend/internal/db"
|
||||
"github.com/CubeCraft-Creations/Extrudex/backend/internal/models"
|
||||
"github.com/CubeCraft-Creations/Extrudex/backend/internal/router"
|
||||
"github.com/CubeCraft-Creations/Extrudex/backend/internal/sse"
|
||||
"github.com/CubeCraft-Creations/Extrudex/backend/internal/workers"
|
||||
"github.com/jackc/pgx/v5/pgxpool"
|
||||
)
|
||||
|
||||
func main() {
|
||||
// Setup structured logging
|
||||
slog.SetDefault(slog.New(slog.NewTextHandler(os.Stdout, &slog.HandlerOptions{
|
||||
Level: slog.LevelInfo,
|
||||
})))
|
||||
|
||||
// Load configuration
|
||||
cfg, err := config.Load()
|
||||
if err != nil {
|
||||
slog.Error("failed to load config", "error", err)
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
slog.Info("config loaded", "port", cfg.Port, "cors_origin", cfg.CorsOrigin)
|
||||
|
||||
// Connect to database
|
||||
dbPool, err := db.NewPool(cfg.DatabaseURL)
|
||||
if err != nil {
|
||||
slog.Error("failed to connect to database", "error", err)
|
||||
@@ -38,30 +39,38 @@ func main() {
|
||||
}
|
||||
defer db.ClosePool(dbPool)
|
||||
|
||||
slog.Info("database connected")
|
||||
|
||||
// Create SSE broadcaster and start it
|
||||
sseBC := sse.NewBroadcaster(128)
|
||||
sseBC.Start()
|
||||
defer sseBC.Stop()
|
||||
|
||||
slog.Info("sse broadcaster started")
|
||||
|
||||
// Create router
|
||||
r := router.New(cfg, dbPool, sseBC)
|
||||
|
||||
// Create HTTP server
|
||||
// WriteTimeout is 0 for SSE support — the Chi middleware.Timeout(60s)
|
||||
// handles request-level timeouts on non-SSE routes.
|
||||
// ── Workers ─────────────────────────────────────────────────────────
|
||||
|
||||
var wg sync.WaitGroup
|
||||
workersCtx, cancelWorkers := context.WithCancel(context.Background())
|
||||
defer cancelWorkers()
|
||||
|
||||
pollInterval, _ := time.ParseDuration(cfg.MoonrakerPollInterval)
|
||||
if pollInterval <= 0 {
|
||||
pollInterval = 10 * time.Second
|
||||
}
|
||||
|
||||
activePrinters := listActivePrinters(workersCtx, dbPool)
|
||||
for _, p := range activePrinters {
|
||||
startWorkerForPrinter(workersCtx, &wg, cfg, dbPool, p, pollInterval)
|
||||
}
|
||||
|
||||
// ── HTTP server ─────────────────────────────────────────────────────
|
||||
|
||||
server := &http.Server{
|
||||
Addr: ":" + cfg.Port,
|
||||
Handler: r,
|
||||
ReadTimeout: 15 * time.Second,
|
||||
WriteTimeout: 0, // disabled for SSE long-lived connections
|
||||
WriteTimeout: 0,
|
||||
IdleTimeout: 60 * time.Second,
|
||||
}
|
||||
|
||||
// Start server in goroutine
|
||||
go func() {
|
||||
slog.Info("server starting", "addr", server.Addr)
|
||||
if err := server.ListenAndServe(); err != nil && err != http.ErrServerClosed {
|
||||
@@ -70,21 +79,119 @@ func main() {
|
||||
}
|
||||
}()
|
||||
|
||||
// Wait for shutdown signal
|
||||
quit := make(chan os.Signal, 1)
|
||||
signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)
|
||||
<-quit
|
||||
|
||||
slog.Info("server shutting down")
|
||||
cancelWorkers()
|
||||
|
||||
// Graceful shutdown
|
||||
ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
|
||||
defer cancel()
|
||||
shutdownCtx, shutdownCancel := context.WithTimeout(context.Background(), 30*time.Second)
|
||||
defer shutdownCancel()
|
||||
|
||||
if err := server.Shutdown(ctx); err != nil {
|
||||
if err := server.Shutdown(shutdownCtx); err != nil {
|
||||
slog.Error("server shutdown error", "error", err)
|
||||
}
|
||||
|
||||
db.ClosePool(dbPool)
|
||||
done := make(chan struct{})
|
||||
go func() {
|
||||
wg.Wait()
|
||||
close(done)
|
||||
}()
|
||||
select {
|
||||
case <-done:
|
||||
slog.Info("all workers stopped")
|
||||
case <-time.After(15 * time.Second):
|
||||
slog.Warn("timed out waiting for workers to stop")
|
||||
}
|
||||
|
||||
slog.Info("server stopped")
|
||||
}
|
||||
|
||||
func listActivePrinters(ctx context.Context, pool *pgxpool.Pool) []models.Printer {
|
||||
rows, err := pool.Query(ctx, `
|
||||
SELECT id, name, printer_type_id,
|
||||
manufacturer, model,
|
||||
moonraker_url, moonraker_api_key,
|
||||
mqtt_broker_host, mqtt_topic_prefix,
|
||||
mqtt_tls_enabled, is_active,
|
||||
created_at, updated_at
|
||||
FROM printers WHERE is_active = TRUE ORDER BY name
|
||||
`)
|
||||
if err != nil {
|
||||
slog.Warn("failed to query active printers", "error", err)
|
||||
return nil
|
||||
}
|
||||
defer rows.Close()
|
||||
|
||||
var printers []models.Printer
|
||||
for rows.Next() {
|
||||
var p models.Printer
|
||||
if err := rows.Scan(
|
||||
&p.ID, &p.Name, &p.PrinterTypeID,
|
||||
&p.Manufacturer, &p.Model,
|
||||
&p.MoonrakerURL, &p.MoonrakerAPIKey,
|
||||
&p.MQTTBrokerHost, &p.MQTTTopicPrefix,
|
||||
&p.MQTTTLSEnabled, &p.IsActive,
|
||||
&p.CreatedAt, &p.UpdatedAt,
|
||||
); err != nil {
|
||||
slog.Warn("failed to scan printer row", "error", err)
|
||||
continue
|
||||
}
|
||||
printers = append(printers, p)
|
||||
}
|
||||
return printers
|
||||
}
|
||||
|
||||
func startWorkerForPrinter(
|
||||
ctx context.Context,
|
||||
wg *sync.WaitGroup,
|
||||
cfg *config.Config,
|
||||
pool *pgxpool.Pool,
|
||||
printer models.Printer,
|
||||
pollInterval time.Duration,
|
||||
) {
|
||||
if printer.MoonrakerURL != nil && *printer.MoonrakerURL != "" {
|
||||
mc := clients.NewMoonrakerClient(*printer.MoonrakerURL)
|
||||
poller := workers.NewMoonrakerPoller(workers.MoonrakerPollerConfig{
|
||||
Client: mc,
|
||||
Pool: pool,
|
||||
PollInterval: pollInterval,
|
||||
PrinterID: printer.ID,
|
||||
PrinterName: printer.Name,
|
||||
})
|
||||
wg.Add(1)
|
||||
go func() {
|
||||
defer wg.Done()
|
||||
poller.Run(ctx)
|
||||
}()
|
||||
}
|
||||
|
||||
if printer.MQTTBrokerHost != nil && *printer.MQTTBrokerHost != "" {
|
||||
topicPrefix := cfg.MQTTTopicPrefix
|
||||
if printer.MQTTTopicPrefix != nil && *printer.MQTTTopicPrefix != "" {
|
||||
topicPrefix = *printer.MQTTTopicPrefix
|
||||
}
|
||||
sub := workers.NewMQTTSubscriber(workers.MQTTSubscriberConfig{
|
||||
Pool: pool,
|
||||
PrinterID: printer.ID,
|
||||
PrinterName: printer.Name,
|
||||
})
|
||||
mqttClient := clients.NewMQTTClient(clients.MQTTConfig{
|
||||
Broker: *printer.MQTTBrokerHost,
|
||||
ClientID: cfg.MQTTClientID + "-p" + strconv.Itoa(printer.ID),
|
||||
TopicPrefix: topicPrefix,
|
||||
TLSCert: cfg.MQTTTLSCert,
|
||||
TLSKey: cfg.MQTTTLSKey,
|
||||
Handler: sub.HandleBambuReport,
|
||||
})
|
||||
sub.Client = mqttClient
|
||||
wg.Add(1)
|
||||
go func() {
|
||||
defer wg.Done()
|
||||
if err := sub.Run(ctx); err != nil {
|
||||
slog.Error("mqtt subscriber error", "printer_id", printer.ID, "error", err)
|
||||
}
|
||||
}()
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user