diff --git a/cmd/pg-fill/main.go b/cmd/pg-fill/main.go index 204e697..7fc1b79 100644 --- a/cmd/pg-fill/main.go +++ b/cmd/pg-fill/main.go @@ -7,6 +7,7 @@ import ( "fmt" "log" "os" + "path" "time" "pg/src/with" @@ -30,8 +31,10 @@ func run(ctx context.Context) error { return with.PSQL(ctx, *c, func(pg *sql.DB) error { presently := func() int { + dbname := path.Base(*c) + var n2 int - if row := pg.QueryRowContext(ctx, `SELECT pg_total_relation_size('fill_with_data')/1024/1024`); row.Err() != nil { + if row := pg.QueryRowContext(ctx, `SELECT pg_database_size('`+dbname+`')/1024/1024`); row.Err() != nil { return 0 } else if err := row.Scan(&n2); err != nil { return 0