95 lines
2.2 KiB
Go
95 lines
2.2 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"log/slog"
|
|
"net/http"
|
|
"os"
|
|
"os/signal"
|
|
"syscall"
|
|
"time"
|
|
|
|
"git.g3e.fr/H6N/account/internal/config"
|
|
"git.g3e.fr/H6N/account/internal/pipeline"
|
|
"git.g3e.fr/H6N/account/internal/scheduler"
|
|
"git.g3e.fr/H6N/account/internal/server"
|
|
"git.g3e.fr/H6N/account/internal/snapshot"
|
|
"git.g3e.fr/H6N/account/internal/store"
|
|
"github.com/jackc/pgx/v5/pgxpool"
|
|
)
|
|
|
|
func main() {
|
|
logger := slog.New(slog.NewJSONHandler(os.Stdout, nil))
|
|
|
|
cfg := config.Load()
|
|
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
|
|
pool, err := pgxpool.New(ctx, cfg.DatabaseURL)
|
|
if err != nil {
|
|
logger.Error("failed to create db pool", "error", err)
|
|
os.Exit(1)
|
|
}
|
|
defer pool.Close()
|
|
|
|
if err := pool.Ping(ctx); err != nil {
|
|
logger.Error("database ping failed", "error", err)
|
|
os.Exit(1)
|
|
}
|
|
logger.Info("database connected")
|
|
|
|
st := store.New(pool)
|
|
snap := snapshot.New(st, logger, cfg.SnapshotHorizonDays)
|
|
pl := pipeline.New(st, pipeline.Config{
|
|
OpenFIGIKey: cfg.OpenFIGIKey,
|
|
CoinGeckoKey: cfg.CoinGeckoKey,
|
|
}, logger)
|
|
|
|
sched := scheduler.New(logger)
|
|
sched.Add(scheduler.Job{
|
|
Name: "fetch-prices",
|
|
Interval: cfg.PriceFetchInterval,
|
|
Fn: pl.FetchAll,
|
|
})
|
|
sched.Add(scheduler.Job{
|
|
Name: "clean-price-history",
|
|
Interval: cfg.PriceCleanInterval,
|
|
Fn: pl.CleanHistory,
|
|
})
|
|
sched.Add(scheduler.Job{
|
|
Name: "daily-snapshot",
|
|
Interval: 24 * time.Hour,
|
|
Fn: snap.DailySnapshot,
|
|
})
|
|
sched.Start(ctx)
|
|
|
|
srv := server.New(cfg, pool, logger)
|
|
|
|
httpServer := &http.Server{
|
|
Addr: fmt.Sprintf(":%s", cfg.Port),
|
|
Handler: srv,
|
|
ReadTimeout: 10 * time.Second,
|
|
WriteTimeout: 10 * time.Second,
|
|
IdleTimeout: 60 * time.Second,
|
|
}
|
|
|
|
go func() {
|
|
logger.Info("server listening", "port", cfg.Port, "env", cfg.Env)
|
|
if err := httpServer.ListenAndServe(); err != nil && err != http.ErrServerClosed {
|
|
logger.Error("server error", "error", err)
|
|
os.Exit(1)
|
|
}
|
|
}()
|
|
|
|
quit := make(chan os.Signal, 1)
|
|
signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)
|
|
<-quit
|
|
|
|
logger.Info("shutting down")
|
|
cancel()
|
|
shutdownCtx, stop := context.WithTimeout(context.Background(), 10*time.Second)
|
|
defer stop()
|
|
httpServer.Shutdown(shutdownCtx)
|
|
}
|