kylesoda/go-migrate
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: refactor database connection logic and add column type querying functions

Browse Source
This commit is contained in:
Kylesoda
2026-04-04 00:23:28 -05:00
parent 5ebaac482d
commit b05e566fca
2 changed files with 111 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
cmd/go_migrate/connect.go
View File

@@ -46,29 +46,31 @@ func connectToPostgres() (*pgxpool.Pool, error) {
return pool, nil
}
func connectToDatabases(sourceDb *sql.DB, targetDb *pgxpool.Pool) error {
func connectToDatabases() (*sql.DB, *pgxpool.Pool, error) {
var sourceDbErr, targetDbErr error
var sourceDb *sql.DB
var targetDb *pgxpool.Pool
var wg sync.WaitGroup
wg.Go(func() {
sourceDb, sourceDbErr = connectToSqlServer()
if sourceDbErr != nil {
log.Error("Unable to connect to source db: ", sourceDbErr)
}
sourceDb, sourceDbErr = connectToSqlServer()
if sourceDbErr != nil {
log.Error("Unable to connect to source db: ", sourceDbErr)
}
})
wg.Go(func() {
targetDb, targetDbErr = connectToPostgres()
if targetDbErr != nil {
log.Error("Unable to connect to target db: ", targetDbErr)
}
targetDb, targetDbErr = connectToPostgres()
if targetDbErr != nil {
log.Error("Unable to connect to target db: ", targetDbErr)
}
})
wg.Wait()
if sourceDbErr != nil || targetDbErr != nil {
return fmt.Errorf("Unable to connect to databases: %w (source), %w (target)", sourceDbErr, targetDbErr)
return nil, nil, fmt.Errorf("Unable to connect to databases: %w (source), %w (target)", sourceDbErr, targetDbErr)
}
return nil
return sourceDb, targetDb, nil
}