From 9584e908fb4e2de4faff96c6dedbae910bf76233 Mon Sep 17 00:00:00 2001 From: Julie Qiu Date: Tue, 15 Jun 2021 18:23:21 -0400 Subject: devtools/cmd/seeddb/seeddb: fix script Since seeddb is moved to devtools/cmd/seeddb/seeddb, cmd/internal/cmdconfig can no longer be used. Instead, use database.Open directly. Change-Id: I2262d36f2199a275dbc62306528788bdf21f0727 Reviewed-on: https://go-review.googlesource.com/c/pkgsite/+/327932 Trust: Julie Qiu Run-TryBot: Julie Qiu TryBot-Result: kokoro Reviewed-by: Jonathan Amsterdam --- devtools/cmd/seeddb/seeddb/main.go | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) (limited to 'devtools/cmd') diff --git a/devtools/cmd/seeddb/seeddb/main.go b/devtools/cmd/seeddb/seeddb/main.go index 57173908..5815df7b 100644 --- a/devtools/cmd/seeddb/seeddb/main.go +++ b/devtools/cmd/seeddb/seeddb/main.go @@ -19,8 +19,8 @@ import ( "time" _ "github.com/jackc/pgx/v4/stdlib" // for pgx driver - "golang.org/x/pkgsite/cmd/internal/cmdconfig" "golang.org/x/pkgsite/internal/config" + "golang.org/x/pkgsite/internal/database" "golang.org/x/pkgsite/internal/derrors" "golang.org/x/pkgsite/internal/log" "golang.org/x/pkgsite/internal/postgres" @@ -31,7 +31,7 @@ import ( ) var ( - seedfile = flag.String("static", "cmd/internal/seeddb/seed.txt", "filename containing modules for seeding the database") + seedfile = flag.String("static", "devtools/cmd/seeddb/seeddb/seed.txt", "filename containing modules for seeding the database") ) func main() { @@ -42,14 +42,14 @@ func main() { if err != nil { log.Fatal(ctx, err) } - cfg.Dump(os.Stdout) - log.SetLevel(cfg.LogLevel) - db, err := cmdconfig.OpenDB(ctx, cfg, false) + // Wrap the postgres driver with our own wrapper, which adds OpenCensus instrumentation. + ddb, err := database.Open("pgx", cfg.DBConnInfo(), "seeddb") if err != nil { - log.Fatal(ctx, err) + log.Fatalf(ctx, "database.Open for host %s failed with %v", cfg.DBHost, err) } + db := postgres.New(ddb) defer db.Close() if err := run(ctx, db, cfg.ProxyURL); err != nil { -- cgit v1.3