7be0033d12
update tests, helper methods, etc for changed interfaces linted and some tests updated... statediff tests failing on filesystem call locally undo changes to go.mod from rebase changed ref and repo to try old stack-orch with miner.etherbase arg turn off new tests yml for old tests with hack for old stack-orchestrator cicd cleanup to trigger PR and testing (#324) publish step using broken tests switched (#325) Publish with old tests and no vulcanize publish (#326) * publish step using broken tests switched * rebase inserted old vulcanize publish steps run tests in Jenkins CICD (#327) * run race tests in CICD * set HOME env for .ethereum mkdir permission denied * use same homeDir method as other places in code * unused variable in test removed * do NOT run race tests Unit test inconsistencies (#330) * run race tests in CICD * set HOME env for .ethereum mkdir permission denied * use same homeDir method as other places in code * unused variable in test removed * do NOT run race tests * add statediffing test to Jenkinsfile Add COPY support for inserting multiple rows in a single operation. (#328) * Add COPY support for inserting multiple rows in a single command. Fix CI tests by using specific version of Foundry (#333) * Fix CI tests by using specific version of Foundry --------- Co-authored-by: Michael Shaw <michael@abastionofsanity.com> Add timers/counters for LevelDB Get, Put, Has, and Delete. (#332) * Add timers/counters for LevelDB Get, Put, Has, and Delete. * Test for null metrics (the unit tests don't initialize them). Add timer and counter for batched write operations. (#337) * Add timer and counter for batched write operations. * Tweak comment
82 lines
3.1 KiB
Go
82 lines
3.1 KiB
Go
// VulcanizeDB
|
|
// Copyright © 2021 Vulcanize
|
|
|
|
// This program is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU Affero General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
|
|
// This program is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU Affero General Public License for more details.
|
|
|
|
// You should have received a copy of the GNU Affero General Public License
|
|
// along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
package indexer
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/ethereum/go-ethereum/log"
|
|
"github.com/ethereum/go-ethereum/params"
|
|
"github.com/ethereum/go-ethereum/statediff/indexer/database/dump"
|
|
"github.com/ethereum/go-ethereum/statediff/indexer/database/file"
|
|
"github.com/ethereum/go-ethereum/statediff/indexer/database/sql"
|
|
"github.com/ethereum/go-ethereum/statediff/indexer/database/sql/postgres"
|
|
"github.com/ethereum/go-ethereum/statediff/indexer/interfaces"
|
|
"github.com/ethereum/go-ethereum/statediff/indexer/node"
|
|
"github.com/ethereum/go-ethereum/statediff/indexer/shared"
|
|
)
|
|
|
|
// NewStateDiffIndexer creates and returns an implementation of the StateDiffIndexer interface.
|
|
func NewStateDiffIndexer(ctx context.Context, chainConfig *params.ChainConfig, nodeInfo node.Info, config interfaces.Config) (sql.Database, interfaces.StateDiffIndexer, error) {
|
|
switch config.Type() {
|
|
case shared.FILE:
|
|
log.Info("Starting statediff service in SQL file writing mode")
|
|
fc, ok := config.(file.Config)
|
|
if !ok {
|
|
return nil, nil, fmt.Errorf("file config is not the correct type: got %T, expected %T", config, file.Config{})
|
|
}
|
|
fc.NodeInfo = nodeInfo
|
|
ind, err := file.NewStateDiffIndexer(ctx, chainConfig, fc)
|
|
return nil, ind, err
|
|
case shared.POSTGRES:
|
|
log.Info("Starting statediff service in Postgres writing mode")
|
|
pgc, ok := config.(postgres.Config)
|
|
if !ok {
|
|
return nil, nil, fmt.Errorf("postgres config is not the correct type: got %T, expected %T", config, postgres.Config{})
|
|
}
|
|
var err error
|
|
var driver sql.Driver
|
|
switch pgc.Driver {
|
|
case postgres.PGX:
|
|
driver, err = postgres.NewPGXDriver(ctx, pgc, nodeInfo)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
case postgres.SQLX:
|
|
driver, err = postgres.NewSQLXDriver(ctx, pgc, nodeInfo)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
default:
|
|
return nil, nil, fmt.Errorf("unrecognized Postgres driver type: %s", pgc.Driver)
|
|
}
|
|
db := postgres.NewPostgresDB(driver, pgc.Upsert)
|
|
ind, err := sql.NewStateDiffIndexer(ctx, chainConfig, db)
|
|
return db, ind, err
|
|
case shared.DUMP:
|
|
log.Info("Starting statediff service in data dump mode")
|
|
dumpc, ok := config.(dump.Config)
|
|
if !ok {
|
|
return nil, nil, fmt.Errorf("dump config is not the correct type: got %T, expected %T", config, dump.Config{})
|
|
}
|
|
return nil, dump.NewStateDiffIndexer(chainConfig, dumpc), nil
|
|
default:
|
|
return nil, nil, fmt.Errorf("unrecognized database type: %s", config.Type())
|
|
}
|
|
}
|