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
106 lines
2.8 KiB
Go
106 lines
2.8 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 postgres
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/ethereum/go-ethereum/statediff/indexer/shared"
|
|
)
|
|
|
|
// DriverType to explicitly type the kind of sql driver we are using
|
|
type DriverType string
|
|
|
|
const (
|
|
PGX DriverType = "PGX"
|
|
SQLX DriverType = "SQLX"
|
|
Unknown DriverType = "Unknown"
|
|
)
|
|
|
|
// ResolveDriverType resolves a DriverType from a provided string
|
|
func ResolveDriverType(str string) (DriverType, error) {
|
|
switch strings.ToLower(str) {
|
|
case "pgx", "pgxpool":
|
|
return PGX, nil
|
|
case "sqlx":
|
|
return SQLX, nil
|
|
default:
|
|
return Unknown, fmt.Errorf("unrecognized driver type string: %s", str)
|
|
}
|
|
}
|
|
|
|
// DefaultConfig are default parameters for connecting to a Postgres sql
|
|
var DefaultConfig = Config{
|
|
Hostname: "localhost",
|
|
Port: 8077,
|
|
DatabaseName: "vulcanize_testing",
|
|
Username: "vdbm",
|
|
Password: "password",
|
|
}
|
|
|
|
// Config holds params for a Postgres db
|
|
type Config struct {
|
|
// conn string params
|
|
Hostname string
|
|
Port int
|
|
DatabaseName string
|
|
Username string
|
|
Password string
|
|
|
|
// conn settings
|
|
MaxConns int
|
|
MaxIdle int
|
|
MinConns int
|
|
MaxConnIdleTime time.Duration
|
|
MaxConnLifetime time.Duration
|
|
ConnTimeout time.Duration
|
|
LogStatements bool
|
|
|
|
// node info params
|
|
ID string
|
|
ClientName string
|
|
|
|
// driver type
|
|
Driver DriverType
|
|
|
|
// toggle on/off upserts
|
|
Upsert bool
|
|
|
|
// toggle on/off CopyFrom
|
|
CopyFrom bool
|
|
}
|
|
|
|
// Type satisfies interfaces.Config
|
|
func (c Config) Type() shared.DBType {
|
|
return shared.POSTGRES
|
|
}
|
|
|
|
// DbConnectionString constructs and returns the connection string from the config
|
|
func (c Config) DbConnectionString() string {
|
|
if len(c.Username) > 0 && len(c.Password) > 0 {
|
|
return fmt.Sprintf("postgresql://%s:%s@%s:%d/%s?sslmode=disable",
|
|
c.Username, c.Password, c.Hostname, c.Port, c.DatabaseName)
|
|
}
|
|
if len(c.Username) > 0 && len(c.Password) == 0 {
|
|
return fmt.Sprintf("postgresql://%s@%s:%d/%s?sslmode=disable",
|
|
c.Username, c.Hostname, c.Port, c.DatabaseName)
|
|
}
|
|
return fmt.Sprintf("postgresql://%s:%d/%s?sslmode=disable", c.Hostname, c.Port, c.DatabaseName)
|
|
}
|