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

refactor: database url config

Browse Source
This commit is contained in:
Kylesoda
2026-04-02 14:28:03 -05:00
parent 36af9378ad
commit ad89519ece
2 changed files with 12 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
cmd/go_migrate/main.go
View File

@@ -27,7 +27,7 @@ func main() {
ctxSource, cancel := context.WithTimeout(context.Background(), 20*time.Second) ctxSource, cancel := context.WithTimeout(context.Background(), 20*time.Second)
defer cancel() defer cancel()
sourcePool, err := db.Connect(ctxSource, config.App.FromDb.Url) sourcePool, err := db.Connect(ctxSource, config.App.SourceDbUrl)
defer db.Close(sourcePool) defer db.Close(sourcePool)
if err != nil { if err != nil {
log.Fatal(err) log.Fatal(err)
@@ -37,7 +37,7 @@ func main() {
ctxTarget, cancel := context.WithTimeout(context.Background(), 20*time.Second) ctxTarget, cancel := context.WithTimeout(context.Background(), 20*time.Second)
defer cancel() defer cancel()
targetPool, err := db.Connect(ctxTarget, config.App.ToDb.Url) targetPool, err := db.Connect(ctxTarget, config.App.TargetDbUrl)
defer db.Close(targetPool) defer db.Close(targetPool)
if err != nil { if err != nil {
log.Fatal(err) log.Fatal(err)

28
internal/app/config/main.go
View File

@@ -7,13 +7,9 @@ import (
log "github.com/sirupsen/logrus" log "github.com/sirupsen/logrus"
) )
type dbConfig struct {
Url string
}
type appConfig struct { type appConfig struct {
FromDb dbConfig SourceDbUrl string
ToDb dbConfig TargetDbUrl string
} }
func loadEnv() { func loadEnv() {
@@ -26,23 +22,19 @@ func loadEnv() {
func getAppConfig() appConfig { func getAppConfig() appConfig {
loadEnv() loadEnv()
fromDbUrl := os.Getenv("PG_FROM_DB_URL") sourceDbUrl := os.Getenv("SOURCE_DB_URL")
if fromDbUrl == "" { if sourceDbUrl == "" {
log.Fatal("PG_FROM_DB_URL environment variable not set") log.Fatal("SOURCE_DB_URL environment variable not set")
} }
toDbUrl := os.Getenv("PG_TO_DB_URL") targetDbUrl := os.Getenv("TARGET_DB_URL")
if toDbUrl == "" { if targetDbUrl == "" {
log.Fatal("PG_TO_DB_URL environment variable not set") log.Fatal("TARGET_DB_URL environment variable not set")
} }
return appConfig{ return appConfig{
FromDb: dbConfig{ SourceDbUrl: sourceDbUrl,
Url: fromDbUrl, TargetDbUrl: targetDbUrl,
},
ToDb: dbConfig{
Url: toDbUrl,
},
} }
} }