atmost working example

This commit is contained in:
Aleksandr Trushkin
2024-08-11 21:44:19 +03:00
parent 0046755c7d
commit 880f67aa73
16 changed files with 566 additions and 107 deletions

View File

@ -9,12 +9,14 @@ import (
"os"
"os/signal"
"golang.org/x/sync/errgroup"
"git.loyso.art/frx/devsim"
"git.loyso.art/frx/devsim/internal/api/http"
"git.loyso.art/frx/devsim/internal/store"
"git.loyso.art/frx/devsim/internal/store/mongo"
"git.loyso.art/frx/devsim/internal/store/pg"
"git.loyso.art/frx/devsim/pkg/collections"
"golang.org/x/sync/errgroup"
)
var availableStoreTypes = collections.NewSet([]string{
@ -25,20 +27,33 @@ func main() {
ctx, cancel := signal.NotifyContext(context.Background(), os.Interrupt)
defer cancel()
log := slog.New(slog.NewJSONHandler(os.Stdout, &slog.HandlerOptions{}))
log := slog.New(slog.NewJSONHandler(io.Discard, &slog.HandlerOptions{
Level: slog.LevelInfo,
}))
var settings applicationSettings
settings.fromEnv()
log.InfoContext(
ctx, "running application",
slog.Int("pid", os.Getpid()),
slog.String("version", devsim.Version()),
slog.String("revision", devsim.Revision()),
slog.String("buildtime", devsim.BuildTime()),
)
settings := loadConfigFromEnv()
err := settings.validate()
if err != nil {
log.ErrorContext(ctx, "unable to validate settings", slog.Any("err", err))
log.ErrorContext(ctx, "unable to validate settings", slog.String("err", err.Error()))
os.Exit(1)
}
log.InfoContext(
ctx, "config parsed",
slog.Any("settings", settings),
)
err = app(ctx, settings, log)
if err != nil {
log.ErrorContext(ctx, "unable to run app", slog.Any("err", err))
log.ErrorContext(ctx, "unable to run app", slog.String("err", err.Error()))
os.Exit(1)
}
}
@ -71,19 +86,20 @@ type applicationSettings struct {
mongo mongoSettings
}
func (s *applicationSettings) fromEnv() {
func loadConfigFromEnv() applicationSettings {
const webaddr = ":9123"
*s = applicationSettings{
listenAddr: valueOrDefault(os.Getenv("DEVSIM_HTTP_ADDR"), webaddr),
storeType: os.Getenv("DEVSIM_STORE_TYPE"),
}
var cfg applicationSettings
cfg.listenAddr = valueOrDefault(os.Getenv("DEVSIM_HTTP_ADDR"), webaddr)
cfg.storeType = os.Getenv("DEVSIM_STORE_TYPE")
s.pg.fromEnv()
s.mongo.fromEnv()
cfg.pg.fromEnv()
cfg.mongo.fromEnv()
return cfg
}
func (s *applicationSettings) validate() (err error) {
func (s applicationSettings) validate() (err error) {
if !availableStoreTypes.Contains(s.storeType) {
err = errors.Join(err, errors.New("store_type value is unsupported"))
}
@ -117,9 +133,9 @@ func app(ctx context.Context, settings applicationSettings, log *slog.Logger) (e
switch settings.storeType {
case "pg":
pgconn, err := pg.Dial(ctx, settings.pg.DSN)
if err != nil {
return fmt.Errorf("connecting to postgres: %w", err)
pgconn, errDial := pg.Dial(ctx, settings.pg.DSN)
if errDial != nil {
return fmt.Errorf("connecting to postgres: %w", errDial)
}
repo = pgconn.StatsRepository()
@ -128,9 +144,9 @@ func app(ctx context.Context, settings applicationSettings, log *slog.Logger) (e
closer: pgconn,
})
case "mongo":
mongoconn, err := mongo.Dial(ctx, settings.mongo.DSN)
if err != nil {
return fmt.Errorf("connecting to mongo: %w", err)
mongoconn, errDial := mongo.Dial(ctx, settings.mongo.DSN)
if errDial != nil {
return fmt.Errorf("connecting to mongo: %w", errDial)
}
repo = mongoconn.StatsRepository()