mirror of
https://github.com/d0zingcat/gocryptotrader.git
synced 2026-05-14 07:26:47 +00:00
0c76789b0d
* added audit manager * Basic database DOA setup * Added base config file * added sqlite support and creation of schema * added basic tests and config entry * corrected issues of database is disabled * fixed path for test * WIP * Added tests fixed config checking * reverted files back to upstream * reverted go.mod files * no more test test test * removed local testing details for psql * hello * added comments * increased ping to 30 seconds * renamed database table and added additional condition around test * removed database test details * goimport ran on all files * WIP * first attempt at migration * fixes for migration system * Migration system logger interface implemented * fixes to print functions * added write pooling pass * gofmt :D * formatted imports correctly * removed old code * added creation of migration * gofmt * :D Hello * ❌ 🏎️ * maybe one day i will remember to revert go mod files * checked err return condition correctly * first changes for PR feedback * code clean up * protect Connected with RWmutex & event with mutex * : D * we can just pretend like it never happened * MOved migrations back to source directory and added README * readme formatting update * Addd command line override for datadir * use correct var when creating a migration and confirm folder is created * Check if database version is newer than latest migration and also you know make migrations work..... * uses filepath instead of manual path to use correct path seperator * Add connection message and lower timeout * Added support for sslmode for psql * no longer force Close of database instead allow driver to maage * Added closer func to test output * sslmode added to example config
42 lines
970 B
Go
42 lines
970 B
Go
package postgres
|
|
|
|
import (
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/jackc/pgx"
|
|
"github.com/jackc/pgx/stdlib"
|
|
"github.com/jmoiron/sqlx"
|
|
"github.com/thrasher-corp/gocryptotrader/database"
|
|
)
|
|
|
|
// Connect establishes a connection pool to the database
|
|
func Connect() (*database.Database, error) {
|
|
configDSN := fmt.Sprintf("host=%s port=%d user=%s password=%s database=%s sslmode=%s",
|
|
database.Conn.Config.Host,
|
|
database.Conn.Config.Port,
|
|
database.Conn.Config.Username,
|
|
database.Conn.Config.Password,
|
|
database.Conn.Config.Database,
|
|
database.Conn.Config.SSLMode)
|
|
|
|
connConfig, err := pgx.ParseDSN(configDSN)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
connPool, err := pgx.NewConnPool(pgx.ConnPoolConfig{
|
|
ConnConfig: connConfig,
|
|
AfterConnect: nil,
|
|
MaxConnections: 20,
|
|
AcquireTimeout: 30 * time.Second,
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
sqlxDB := stdlib.OpenDBFromPool(connPool)
|
|
database.Conn.SQL = sqlx.NewDb(sqlxDB, "pgx")
|
|
return database.Conn, nil
|
|
}
|