kylefang/gaze-indexer
1
0
Fork 0
mirror of https://github.com/alexgo-io/gaze-indexer.git synced 2026-04-25 20:35:17 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat(cli): support db selection

Browse Source 
Co-authored-by: Gaze <dev@gaze.network>
This commit is contained in:
Gaze
2024-04-16 00:07:03 +07:00
parent d2b4a614e5
commit f92dbbbd78
5 changed files with 16 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
cmd/run.go
View File

@@ -3,6 +3,7 @@ package cmd
import (
"os"
"os/signal"
"strings"
"syscall"
"github.com/btcsuite/btcd/rpcclient"
@@ -11,6 +12,7 @@ import (
"github.com/gaze-network/indexer-network/internal/config"
"github.com/gaze-network/indexer-network/internal/postgres"
"github.com/gaze-network/indexer-network/modules/bitcoin"
btcdatagateway "github.com/gaze-network/indexer-network/modules/bitcoin/datagateway"
btcpostgres "github.com/gaze-network/indexer-network/modules/bitcoin/repository/postgres"
"github.com/gaze-network/indexer-network/pkg/logger"
"github.com/gaze-network/indexer-network/pkg/logger/slogx"
@@ -88,13 +90,19 @@ func runHandler(opts *runCmdOptions, cmd *cobra.Command, _ []string) {
// Initialize Bitcoin Indexer
if opts.Bitcoin.Enabled {
pg, err := postgres.NewPool(ctx, conf.Modules["bitcoin"].Postgres)
if err != nil {
logger.PanicContext(ctx, "Failed to create Postgres connection pool", slogx.Error(err))
var db btcdatagateway.BitcoinDataGateway
switch strings.ToLower(opts.Bitcoin.Database) {
case "postgres", "pg":
pg, err := postgres.NewPool(ctx, conf.Modules["bitcoin"].Postgres)
if err != nil {
logger.PanicContext(ctx, "Failed to create Postgres connection pool", slogx.Error(err))
}
defer pg.Close()
db = btcpostgres.NewRepository(pg)
default:
logger.PanicContext(ctx, "Unsupported database", slogx.String("database", opts.Bitcoin.Database))
}
defer pg.Close()
bitcoinRepository := btcpostgres.NewRepository(pg)
bitcoinProcessor := bitcoin.NewProcessor(bitcoinRepository)
bitcoinProcessor := bitcoin.NewProcessor(db)
bitcoinNodeDatasource := datasources.NewBitcoinNode(client)
bitcoinIndexer := indexers.NewBitcoinIndexer(bitcoinProcessor, bitcoinNodeDatasource)

0
modules/bitcoin/internal/datagateway/bitcoin_data.go → modules/bitcoin/datagateway/bitcoin_data.go
View File

0
modules/bitcoin/internal/datagateway/indexer_info.go → modules/bitcoin/datagateway/indexer_info.go
View File

2
modules/bitcoin/processor.go
View File

@@ -9,7 +9,7 @@ import (
"github.com/cockroachdb/errors"
"github.com/gaze-network/indexer-network/core/indexers"
"github.com/gaze-network/indexer-network/core/types"
"github.com/gaze-network/indexer-network/modules/bitcoin/internal/datagateway"
"github.com/gaze-network/indexer-network/modules/bitcoin/datagateway"
"github.com/gaze-network/indexer-network/pkg/logger"
"github.com/gaze-network/indexer-network/pkg/logger/slogx"
)

2
modules/bitcoin/repository/postgres/postgres.go
View File

@@ -2,7 +2,7 @@ package postgres
import (
"github.com/gaze-network/indexer-network/internal/postgres"
"github.com/gaze-network/indexer-network/modules/bitcoin/internal/datagateway"
"github.com/gaze-network/indexer-network/modules/bitcoin/datagateway"
"github.com/gaze-network/indexer-network/modules/bitcoin/repository/postgres/gen"
)