storj/storage/cockroachkv/schema/migrate.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)
}