2023-02-15 08:05:43 +01:00
|
|
|
package main
|
|
|
|
|
|
|
|
import (
|
|
|
|
"context"
|
|
|
|
"flag"
|
|
|
|
"strings"
|
|
|
|
"sync"
|
|
|
|
|
2023-02-20 10:56:30 +01:00
|
|
|
"github.com/grassrootseconomics/cic-custodial/internal/custodial"
|
2023-03-02 16:46:02 +01:00
|
|
|
"github.com/grassrootseconomics/cic-custodial/internal/events"
|
2023-02-15 08:05:43 +01:00
|
|
|
"github.com/grassrootseconomics/cic-custodial/internal/tasker"
|
2023-03-01 12:19:57 +01:00
|
|
|
"github.com/knadh/koanf/v2"
|
2023-02-15 08:05:43 +01:00
|
|
|
"github.com/labstack/echo/v4"
|
|
|
|
"github.com/zerodha/logf"
|
|
|
|
)
|
|
|
|
|
2023-03-02 16:46:02 +01:00
|
|
|
type (
|
|
|
|
internalServiceContainer struct {
|
|
|
|
apiService *echo.Echo
|
|
|
|
jetstreamSub *events.JetStream
|
|
|
|
taskerService *tasker.TaskerServer
|
|
|
|
}
|
|
|
|
)
|
|
|
|
|
2023-02-15 08:05:43 +01:00
|
|
|
var (
|
2023-03-01 12:19:57 +01:00
|
|
|
confFlag string
|
|
|
|
debugFlag bool
|
|
|
|
migrationsFolderFlag string
|
|
|
|
queriesFlag string
|
2023-02-15 08:05:43 +01:00
|
|
|
|
|
|
|
lo logf.Logger
|
|
|
|
ko *koanf.Koanf
|
|
|
|
)
|
|
|
|
|
|
|
|
func init() {
|
|
|
|
flag.StringVar(&confFlag, "config", "config.toml", "Config file location")
|
|
|
|
flag.BoolVar(&debugFlag, "log", false, "Enable debug logging")
|
2023-03-01 12:19:57 +01:00
|
|
|
flag.StringVar(&migrationsFolderFlag, "migrations", "migrations/", "Migrations folder location")
|
|
|
|
flag.StringVar(&queriesFlag, "queries", "queries.sql", "Queries file location")
|
2023-02-15 08:05:43 +01:00
|
|
|
flag.Parse()
|
|
|
|
|
2023-03-02 16:46:02 +01:00
|
|
|
lo = initLogger()
|
|
|
|
ko = initConfig()
|
2023-02-15 08:05:43 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
func main() {
|
2023-03-02 16:46:02 +01:00
|
|
|
parsedQueries := initQueries()
|
|
|
|
celoProvider := initCeloProvider()
|
|
|
|
postgresPool := initPostgresPool()
|
|
|
|
asynqRedisPool := initAsynqRedisPool()
|
|
|
|
redisPool := initCommonRedisPool()
|
|
|
|
|
|
|
|
postgresKeystore := initPostgresKeystore(postgresPool, parsedQueries)
|
|
|
|
pgStore := initPostgresStore(postgresPool, parsedQueries)
|
2023-02-15 08:05:43 +01:00
|
|
|
redisNoncestore := initRedisNoncestore(redisPool, celoProvider)
|
|
|
|
lockProvider := initLockProvider(redisPool.Client)
|
|
|
|
taskerClient := initTaskerClient(asynqRedisPool)
|
2023-03-02 16:46:02 +01:00
|
|
|
systemContainer := initSystemContainer(context.Background(), redisNoncestore)
|
2023-02-15 08:05:43 +01:00
|
|
|
|
2023-03-02 16:46:02 +01:00
|
|
|
jsEventEmitter := initJetStream(pgStore)
|
2023-02-15 08:05:43 +01:00
|
|
|
|
2023-02-20 10:56:30 +01:00
|
|
|
custodial := &custodial.Custodial{
|
|
|
|
CeloProvider: celoProvider,
|
|
|
|
EventEmitter: jsEventEmitter,
|
|
|
|
Keystore: postgresKeystore,
|
|
|
|
LockProvider: lockProvider,
|
|
|
|
Noncestore: redisNoncestore,
|
|
|
|
PgStore: pgStore,
|
|
|
|
SystemContainer: systemContainer,
|
|
|
|
TaskerClient: taskerClient,
|
2023-02-15 08:05:43 +01:00
|
|
|
}
|
|
|
|
|
2023-03-02 16:46:02 +01:00
|
|
|
internalServices := &internalServiceContainer{}
|
2023-02-20 10:56:30 +01:00
|
|
|
wg := &sync.WaitGroup{}
|
|
|
|
|
2023-03-02 16:46:02 +01:00
|
|
|
signalCh, closeCh := createSigChannel()
|
|
|
|
defer closeCh()
|
|
|
|
|
|
|
|
internalServices.apiService = initApiServer(custodial)
|
2023-02-15 08:05:43 +01:00
|
|
|
wg.Add(1)
|
|
|
|
go func() {
|
|
|
|
defer wg.Done()
|
|
|
|
lo.Info("main: starting API server")
|
2023-03-02 16:46:02 +01:00
|
|
|
if err := internalServices.apiService.Start(ko.MustString("service.address")); err != nil {
|
2023-02-15 08:05:43 +01:00
|
|
|
if strings.Contains(err.Error(), "Server closed") {
|
|
|
|
lo.Info("main: shutting down server")
|
|
|
|
} else {
|
|
|
|
lo.Fatal("main: critical error shutting down server", "err", err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}()
|
|
|
|
|
2023-03-02 16:46:02 +01:00
|
|
|
internalServices.taskerService = initTasker(custodial, asynqRedisPool)
|
2023-02-15 08:05:43 +01:00
|
|
|
wg.Add(1)
|
|
|
|
go func() {
|
|
|
|
defer wg.Done()
|
|
|
|
lo.Info("Starting tasker")
|
2023-03-02 16:46:02 +01:00
|
|
|
if err := internalServices.taskerService.Start(); err != nil {
|
2023-02-15 08:05:43 +01:00
|
|
|
lo.Fatal("main: could not start task server", "err", err)
|
|
|
|
}
|
|
|
|
}()
|
|
|
|
|
2023-03-02 16:46:02 +01:00
|
|
|
internalServices.jetstreamSub = jsEventEmitter
|
2023-03-02 10:12:39 +01:00
|
|
|
wg.Add(1)
|
|
|
|
go func() {
|
|
|
|
defer wg.Done()
|
2023-03-02 16:46:02 +01:00
|
|
|
lo.Info("Starting jetstream sub")
|
|
|
|
if err := internalServices.jetstreamSub.Subscriber(); err != nil {
|
|
|
|
lo.Fatal("main: error running jetstream sub", "err", err)
|
2023-03-02 10:12:39 +01:00
|
|
|
}
|
|
|
|
}()
|
|
|
|
|
2023-03-02 16:46:02 +01:00
|
|
|
<-signalCh
|
|
|
|
startGracefulShutdown(context.Background(), internalServices)
|
2023-02-15 08:05:43 +01:00
|
|
|
|
|
|
|
wg.Wait()
|
|
|
|
}
|