87 lines
2.3 KiB
Go
87 lines
2.3 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"log/slog"
|
|
|
|
"gitea.dwysokinski.me/twhelp/corev3/internal/adapter"
|
|
"gitea.dwysokinski.me/twhelp/corev3/internal/app"
|
|
"github.com/urfave/cli/v2"
|
|
)
|
|
|
|
var (
|
|
cmdJob = &cli.Command{
|
|
Name: "job",
|
|
Usage: "Run various jobs (e.g. data sync)",
|
|
Subcommands: []*cli.Command{
|
|
{
|
|
Name: "sync",
|
|
Usage: "Trigger various data syncs",
|
|
Subcommands: []*cli.Command{
|
|
{
|
|
Name: "data",
|
|
Usage: "Trigger data sync (servers, players, tribes, villages)",
|
|
Flags: concatSlices(dbFlags, rmqFlags),
|
|
Action: func(c *cli.Context) error {
|
|
logger := loggerFromCtx(c.Context)
|
|
watermillLogger := newWatermillLogger(logger)
|
|
|
|
amqpConn, err := newAMQPConnectionFromFlags(c, watermillLogger)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer func() {
|
|
if closeErr := amqpConn.Close(); closeErr != nil {
|
|
logger.Warn("couldn't close amqp connection", slog.Any("error", err))
|
|
}
|
|
}()
|
|
|
|
publisher, err := newAMQPPublisher(amqpConn, watermillLogger)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer func() {
|
|
if closeErr := publisher.Close(); closeErr != nil {
|
|
logger.Warn("couldn't close amqp publisher", slog.Any("error", err))
|
|
}
|
|
}()
|
|
|
|
bunDB, err := newBunDBFromFlags(c)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer func() {
|
|
logger.Debug("closing db connections...", slog.Int("db.openConnections", bunDB.Stats().OpenConnections))
|
|
if dbCloseErr := bunDB.Close(); dbCloseErr != nil {
|
|
logger.Warn("couldn't close db connections", slog.Any("error", dbCloseErr))
|
|
} else {
|
|
logger.Debug("db connections closed")
|
|
}
|
|
}()
|
|
|
|
serverPublisher := adapter.NewServerWatermillPublisher(
|
|
publisher,
|
|
newWatermillMarshaler(),
|
|
c.String(rmqFlagTopicSyncServersCmd.Name),
|
|
)
|
|
|
|
versionSvc := app.NewVersionService(adapter.NewVersionBunRepository(bunDB))
|
|
|
|
shutdownSignalCtx, stop := newShutdownSignalContext(c.Context)
|
|
defer stop()
|
|
|
|
if err = app.NewDataSyncService(versionSvc, serverPublisher).TriggerDataSync(shutdownSignalCtx); err != nil {
|
|
return fmt.Errorf("couldn't trigger data sync: %w", err)
|
|
}
|
|
|
|
logger.Info("data sync triggered")
|
|
|
|
return nil
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
)
|