synced/coder
1
0
Fork 0
mirror of https://github.com/coder/coder.git synced 2025-07-03 16:13:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore(coderd/database/gen/dump): add optional DB_DUMP_CONNECTION_URL (#16243)

Browse Source
This commit is contained in:
Mathias Fredriksson
2025-01-23 18:53:24 +02:00
committed by GitHub
parent 5f4ff58f84
commit f27e73d21a
2 changed files with 21 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
coderd/database/gen/dump/main.go
View File

@ -7,6 +7,8 @@ import (
"path/filepath"
"runtime"
"golang.org/x/xerrors"
"github.com/coder/coder/v2/coderd/database/dbtestutil"
"github.com/coder/coder/v2/coderd/database/migrations"
)
@ -37,25 +39,34 @@ func main() {
}
}()
connection, cleanup, err := dbtestutil.OpenContainerized(t, dbtestutil.DBContainerOptions{})
if err != nil {
panic(err)
connection := os.Getenv("DB_DUMP_CONNECTION_URL")
if connection == "" {
var cleanup func()
var err error
connection, cleanup, err = dbtestutil.OpenContainerized(t, dbtestutil.DBContainerOptions{})
if err != nil {
err = xerrors.Errorf("open containerized database failed: %w", err)
panic(err)
}
defer cleanup()
}
defer cleanup()
db, err := sql.Open("postgres", connection)
if err != nil {
err = xerrors.Errorf("open database failed: %w", err)
panic(err)
}
defer db.Close()
err = migrations.Up(db)
if err != nil {
err = xerrors.Errorf("run migrations failed: %w", err)
panic(err)
}
dumpBytes, err := dbtestutil.PGDumpSchemaOnly(connection)
if err != nil {
err = xerrors.Errorf("dump schema failed: %w", err)
panic(err)
}
@ -65,6 +76,7 @@ func main() {
}
err = os.WriteFile(filepath.Join(mainPath, "..", "..", "..", "dump.sql"), append(preamble, dumpBytes...), 0o600)
if err != nil {
err = xerrors.Errorf("write dump failed: %w", err)
panic(err)
}
}