11a8a0f99c
* Update cobra to require `head` or `historic` when using `capture`. * A very generic package for implementing PGX driver I copied most of the code from the `statediff` service from within geth. The idea is that I can create formal DB packages, that can be utilized in other projects down the road. * Put PGX and other future Postgres Drivers behind SQL package This PR makes allows users to provide a config, along with a driver string. It will then provide the respective driver. * Add DB Connection and Logging * Utilize LogRus * Create a DB connection using PGX. * Create an internal boot package for starting the application. * Code clean up + Beacon Chain Connection This concludes all the code needed to connect to the DB and beacon node. We will no longer reference the lighthouse client because this application should work interchangeably with any beacon node. I have also standardized logging. * Last second clean ups * Utilize Ginkgo toreplace `testing` library, and add CI/CD. (#9) * Utilize Ginkgo and replace `testing` library. * Add TOC to docs * Add Docker specific files * Remove -e * Update on-pr-manual.yml * Add depth * Add repositories * Remove $ from path * Update path for make * Setup Go and Ginkgo * Use go mod download * Use go install * Update on-pr-manual.yml * Use latest * Remove install of GINKGO * Add explicit gopath * Explicitly specify the gopath * Update on-pr-manual.yml * Update on-pr-manual.yml * Update on-pr-manual.yml * Update on-pr-manual.yml * Update on-pr-manual.yml * Update on-pr-manual.yml * Use which ginkgo * Try with make now * Final working Make
73 lines
1.8 KiB
Go
73 lines
1.8 KiB
Go
package postgres
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
// 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"
|
|
)
|
|
|
|
// DefaultConfig are default parameters for connecting to a Postgres sql
|
|
var DefaultConfig = Config{
|
|
Hostname: "localhost",
|
|
Port: 8077,
|
|
DatabaseName: "vulcanize_testing",
|
|
Username: "vdbm",
|
|
Password: "password",
|
|
Driver: "PGX",
|
|
}
|
|
|
|
// 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)
|
|
}
|
|
}
|
|
|
|
// 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
|
|
|
|
// driver type
|
|
Driver DriverType
|
|
}
|
|
|
|
// 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)
|
|
}
|