cic-dw/cmd/main.go

65 lines
1.3 KiB
Go
Raw Normal View History

package main
import (
"cic-dw/pkg/cicnet"
"github.com/hibiken/asynq"
"github.com/jackc/pgx/v4/pgxpool"
"github.com/knadh/koanf"
"github.com/lmittmann/w3"
"github.com/nleof/goyesql"
"github.com/rs/zerolog"
"github.com/rs/zerolog/log"
"os"
"sync"
)
type App struct {
db *pgxpool.Pool
queries goyesql.Queries
rClient asynq.RedisClientOpt
cicnetClient *cicnet.CicNet
sigChan chan os.Signal
}
const (
confEnvOverridePrefix = ""
)
var (
conf = koanf.New(".")
db *pgxpool.Pool
queries goyesql.Queries
redisConn asynq.RedisClientOpt
cicnetClient *cicnet.CicNet
)
func init() {
log.Logger = log.Output(zerolog.ConsoleWriter{Out: os.Stderr})
if err := loadConfig("config.toml", confEnvOverridePrefix, conf); err != nil {
log.Fatal().Err(err).Msg("failed to load config")
}
db = connectDb(conf.String("db.dsn"))
queries = loadQueries("queries.sql")
redisConn = connectQueue(conf.String("redis.dsn"))
cicnetClient = cicnet.NewCicNet(conf.String("chain.rpc"), w3.A(conf.String("chain.registry")))
}
func main() {
// TODO: Graceful shutdown of go routines (handle SIG INT/TERM)
var wg sync.WaitGroup
app := &App{
db: db,
queries: queries,
rClient: redisConn,
cicnetClient: cicnetClient,
}
wg.Add(2)
go runScheduler(app)
go runProcessor(app)
wg.Wait()
}