bbdb351e5e
What: Use the github.com/jackc/pgx postgresql driver in place of github.com/lib/pq. Why: github.com/lib/pq has some problems with error handling and context cancellations (i.e. it might even issue queries or DML statements more than once! see https://github.com/lib/pq/issues/939). The github.com/jackx/pgx library appears not to have these problems, and also appears to be better engineered and implemented (in particular, it doesn't use "exceptions by panic"). It should also give us some performance improvements in some cases, and even more so if we can use it directly instead of going through the database/sql layer. Change-Id: Ia696d220f340a097dee9550a312d37de14ed2044
40 lines
1.0 KiB
Go
40 lines
1.0 KiB
Go
// Copyright (C) 2019 Storj Labs, Inc.
|
|
// See LICENSE for copying information.
|
|
|
|
package schema
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/zeebo/errs"
|
|
|
|
"storj.io/storj/private/dbutil/pgutil"
|
|
"storj.io/storj/private/tagsql"
|
|
)
|
|
|
|
// PrepareDB creates the pathdata tables if they don't already exist.
|
|
func PrepareDB(ctx context.Context, db tagsql.DB) (err error) {
|
|
var dbName string
|
|
if err := db.QueryRow(ctx, `SELECT current_database();`).Scan(&dbName); err != nil {
|
|
return errs.Wrap(err)
|
|
}
|
|
|
|
// Note: the buckets table is unused. It exists here to ease importing
|
|
// backups from postgres. Similarly, the bucket column in pathdata is
|
|
// also unused and exists to ease imports.
|
|
_, err = db.Exec(ctx, fmt.Sprintf(`
|
|
CREATE DATABASE IF NOT EXISTS %s;
|
|
CREATE TABLE IF NOT EXISTS buckets (
|
|
bucketname BYTES PRIMARY KEY,
|
|
delim INT8 NOT NULL
|
|
);
|
|
CREATE TABLE IF NOT EXISTS pathdata (
|
|
fullpath BYTEA PRIMARY KEY,
|
|
metadata BYTEA NOT NULL,
|
|
bucket BYTEA
|
|
);
|
|
`, pgutil.QuoteIdentifier(dbName)))
|
|
return errs.Wrap(err)
|
|
}
|