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