fix(iot): run() int вместо os.Exit — финальный flush не теряется (находка #1)

os.Exit обходит defer-ы: на fatal-пути и при ошибке HTTP-сервера iotStore.Close()
(финальный flushAll последнего батча телеметрии) не выполнялся. Логика вынесена в
run() int, main = os.Exit(run()); все выходы через return — defer-ы отрабатывают.
This commit is contained in:
Repinoid
2026-10-10 19:09:33 +03:00
parent 946600238b
commit dc56a9c2e2
+18 -6
View File
@@ -33,11 +33,18 @@ import (
// version — подставляется через ldflags при сборке.
var version = "dev"
// main — тонкая обёртка: os.Exit обходит defer-ы, поэтому вся логика в run(),
// которая возвращает код выхода и выполняет свои defer-ы (devices/iotStore
// Close, финальный flush) ДО выхода (ревью 🔴5: «os.Exit обходит defer-ы»).
func main() {
os.Exit(run())
}
func run() int {
cfg, err := config.Load()
if err != nil {
slog.Error("config", "err", err)
os.Exit(1)
return 1
}
log := slog.New(slog.NewTextHandler(os.Stdout, &slog.HandlerOptions{Level: cfg.LogLevel}))
@@ -54,7 +61,7 @@ func main() {
storeCancel()
if err != nil {
log.Error("devices store", "err", err)
os.Exit(1)
return 1
}
defer devices.Close()
log.Info("devices store ready")
@@ -65,14 +72,14 @@ func main() {
ensureCancel()
if err != nil {
log.Error("ensure bridge device", "err", err)
os.Exit(1)
return 1
}
log.Info("bridge device ensured", "username", cfg.MQTTUsername)
iotStore, err := iotpg.New(cfg.IOTPGDSN, log)
if err != nil {
log.Error("iotpg store", "err", err)
os.Exit(1)
return 1
}
defer iotStore.Close()
log.Info("telemetry store ready")
@@ -146,13 +153,18 @@ func main() {
log.Info("iot-service listening", "addr", server.Addr)
if err := server.ListenAndServe(); err != nil && !errors.Is(err, http.ErrServerClosed) {
log.Error("http server", "err", err)
os.Exit(1)
cancel() // останавливаем bridge/consumer
wg.Wait()
return 1
}
wg.Wait()
if fatal.Load() {
// Не os.Exit: возвращаем код, чтобы отработали defer-ы (iotStore.Close →
// финальный flush последнего батча). os.Exit их бы обошёл.
log.Error("iot-service: подсистема завершилась с ошибкой — выход ≠ 0")
os.Exit(1)
return 1
}
log.Info("iot-service stopped")
return 0
}