refactor: 拆分 exporter、forecast fetch 、radar fetch 功能

This commit is contained in:
yarnom 2025-10-14 11:09:01 +08:00
parent 2556ccf351
commit b0086a984f
3 changed files with 125 additions and 0 deletions

View File

@ -0,0 +1,33 @@
package main
import (
"context"
"errors"
"log"
"os"
"os/signal"
"syscall"
"weatherstation/internal/server"
"weatherstation/internal/tools"
)
func main() {
server.SetupLogger()
// If CAIYUN_TOKEN is provided, enable wind override automatically.
opts := tools.ExporterOptions{}
if token := os.Getenv("CAIYUN_TOKEN"); token != "" {
opts.OverrideWindWithCaiyun = true
opts.CaiyunToken = token
log.Printf("[service-exporter] wind override enabled via CAIYUN_TOKEN")
}
exp := tools.NewExporterWithOptions(opts)
ctx, stop := signal.NotifyContext(context.Background(), os.Interrupt, syscall.SIGTERM)
defer stop()
if err := exp.Start(ctx); err != nil && !errors.Is(err, context.Canceled) {
log.Fatalf("service-exporter failed: %v", err)
}
}

View File

@ -0,0 +1,65 @@
package main
import (
"context"
"log"
"os"
"os/signal"
"syscall"
"time"
"weatherstation/internal/config"
"weatherstation/internal/forecast"
"weatherstation/internal/server"
)
func hourlyLoop(ctx context.Context, fn func() error, name string) {
for {
select {
case <-ctx.Done():
return
default:
}
now := time.Now()
next := now.Truncate(time.Hour).Add(time.Hour)
t := time.NewTimer(time.Until(next))
select {
case <-ctx.Done():
t.Stop()
return
case <-t.C:
}
if err := fn(); err != nil {
log.Printf("[%s] scheduled run failed: %v", name, err)
} else {
log.Printf("[%s] scheduled run completed", name)
}
}
}
func main() {
server.SetupLogger()
ctx, stop := signal.NotifyContext(context.Background(), os.Interrupt, syscall.SIGTERM)
defer stop()
// Open-Meteo hourly fetch
go hourlyLoop(ctx, func() error { return forecast.RunOpenMeteoFetch(context.Background()) }, "open-meteo")
// Caiyun hourly fetch (if token configured)
token := os.Getenv("CAIYUN_TOKEN")
if token == "" {
token = config.GetConfig().Forecast.CaiyunToken
}
if token == "" {
log.Printf("[caiyun] token not set; caiyun scheduler disabled")
} else {
t := token
go hourlyLoop(ctx, func() error { return forecast.RunCaiyunFetch(context.Background(), t) }, "caiyun")
}
// CMA hourly fetch
// go hourlyLoop(ctx, func() error { return forecast.RunCMAFetch(context.Background()) }, "cma")
<-ctx.Done()
log.Println("service-forecast shutting down")
}

27
cmd/service-radar/main.go Normal file
View File

@ -0,0 +1,27 @@
package main
import (
"context"
"log"
"os"
"os/signal"
"syscall"
"weatherstation/internal/radar"
"weatherstation/internal/server"
)
func main() {
server.SetupLogger()
ctx, stop := signal.NotifyContext(context.Background(), os.Interrupt, syscall.SIGTERM)
defer stop()
// Start radar scheduler with defaults (StoreToDB=true; tile defaults read inside package)
if err := radar.Start(ctx, radar.Options{StoreToDB: true}); err != nil {
log.Fatalf("service-radar start error: %v", err)
}
// Keep process alive until signal
<-ctx.Done()
log.Println("service-radar shutting down")
}