0858c3797a
Currently we were duplicating code for AS OF SYSTEM TIME in several places. This replaces the code with using a method on dbutil.Implementation. As a consequence it's more useful to use a shorter name for implementation - 'impl' should be sufficiently clear in the context. Similarly, using AsOfSystemInterval and AsOfSystemTime to distinguish between the two modes is useful and slightly shorter without causing confusion. Change-Id: Idefe55528efa758b6176591017b6572a8d443e3d
123 lines
2.7 KiB
Go
123 lines
2.7 KiB
Go
// Copyright (C) 2020 Storj Labs, Inc.
|
|
// See LICENSE for copying information.
|
|
|
|
package multinodedb
|
|
|
|
import (
|
|
"context"
|
|
"strings"
|
|
|
|
"github.com/spacemonkeygo/monkit/v3"
|
|
"github.com/zeebo/errs"
|
|
"go.uber.org/zap"
|
|
|
|
"storj.io/private/dbutil"
|
|
"storj.io/private/dbutil/pgutil"
|
|
"storj.io/storj/multinode"
|
|
"storj.io/storj/multinode/console"
|
|
"storj.io/storj/multinode/multinodedb/dbx"
|
|
"storj.io/storj/multinode/nodes"
|
|
)
|
|
|
|
var (
|
|
// ensures that multinodeDB implements multinode.DB.
|
|
_ multinode.DB = (*multinodeDB)(nil)
|
|
|
|
mon = monkit.Package()
|
|
|
|
// Error is the default multinodedb errs class.
|
|
Error = errs.Class("multinodedb")
|
|
)
|
|
|
|
// multinodeDB combines access to different database tables with a record
|
|
// of the db driver, db implementation, and db source URL.
|
|
// Implementation of multinode.DB interface.
|
|
//
|
|
// architecture: Master Database
|
|
type multinodeDB struct {
|
|
*dbx.DB
|
|
|
|
log *zap.Logger
|
|
driver string
|
|
impl dbutil.Implementation
|
|
source string
|
|
}
|
|
|
|
// Open creates instance of database supports postgres.
|
|
func Open(ctx context.Context, log *zap.Logger, databaseURL string) (multinode.DB, error) {
|
|
driver, source, impl, err := dbutil.SplitConnStr(databaseURL)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
switch impl {
|
|
case dbutil.SQLite3:
|
|
source = sqlite3SetDefaultOptions(source)
|
|
case dbutil.Postgres:
|
|
source, err = pgutil.CheckApplicationName(source, "multinode")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
default:
|
|
return nil, Error.New("unsupported driver %q", driver)
|
|
}
|
|
|
|
dbxDB, err := dbx.Open(driver, source)
|
|
if err != nil {
|
|
return nil, Error.New("failed opening database via DBX at %q: %v",
|
|
source, err)
|
|
}
|
|
|
|
log.Debug("Connected to:", zap.String("db source", source))
|
|
|
|
dbutil.Configure(ctx, dbxDB.DB, "multinodedb", mon)
|
|
|
|
core := &multinodeDB{
|
|
DB: dbxDB,
|
|
|
|
log: log,
|
|
driver: driver,
|
|
impl: impl,
|
|
source: source,
|
|
}
|
|
|
|
return core, nil
|
|
}
|
|
|
|
// Nodes returns nodes database.
|
|
func (db *multinodeDB) Nodes() nodes.DB {
|
|
return &nodesdb{
|
|
methods: db,
|
|
}
|
|
}
|
|
|
|
// Members returns members database.
|
|
func (db *multinodeDB) Members() console.Members {
|
|
return &members{
|
|
methods: db,
|
|
}
|
|
}
|
|
|
|
// CreateSchema creates schema.
|
|
func (db *multinodeDB) CreateSchema(ctx context.Context) error {
|
|
_, err := db.ExecContext(ctx, db.DB.Schema())
|
|
return err
|
|
}
|
|
|
|
// sqlite3SetDefaultOptions sets default options for disk-based db with URI filename source string
|
|
// if no options were set.
|
|
func sqlite3SetDefaultOptions(source string) string {
|
|
if !strings.HasPrefix(source, "file:") {
|
|
return source
|
|
}
|
|
// do not set anything for in-memory db
|
|
if strings.HasPrefix(source, "file::memory:") {
|
|
return source
|
|
}
|
|
if strings.Contains(source, "?") {
|
|
return source
|
|
}
|
|
|
|
return source + "?_journal=WAL&_busy_timeout=10000"
|
|
}
|