storj/storagenode/storagenodedb/database.go

109 lines
2.3 KiB
Go
Raw Normal View History

2019-01-10 13:13:27 +00:00
// Copyright (C) 2019 Storj Labs, Inc.
// See LICENSE for copying information.
package storagenodedb
import (
"github.com/zeebo/errs"
"go.uber.org/zap"
2019-01-10 13:13:27 +00:00
"storj.io/storj/pkg/kademlia"
pstore "storj.io/storj/pkg/piecestore"
2019-01-10 13:13:27 +00:00
"storj.io/storj/pkg/piecestore/psserver/psdb"
"storj.io/storj/storage"
"storj.io/storj/storage/boltdb"
2019-01-10 13:13:27 +00:00
"storj.io/storj/storage/teststore"
"storj.io/storj/storagenode"
)
var _ storagenode.DB = (*DB)(nil)
// Config configures storage node database
type Config struct {
// TODO: figure out better names
Storage string
Info string
Kademlia string
}
2019-01-10 13:13:27 +00:00
// DB contains access to different database tables
type DB struct {
log *zap.Logger
storage *pstore.Storage
2019-01-10 13:13:27 +00:00
psdb *psdb.DB
kdb, ndb storage.KeyValueStore
}
// New creates a new master database for storage node
func New(log *zap.Logger, config Config) (*DB, error) {
storage := pstore.NewStorage(config.Storage)
psdb, err := psdb.Open(config.Info)
if err != nil {
return nil, err
}
dbs, err := boltdb.NewShared(config.Kademlia, kademlia.KademliaBucket, kademlia.NodeBucket)
if err != nil {
return nil, err
}
return &DB{
log: log,
storage: storage,
psdb: psdb,
kdb: dbs[0],
ndb: dbs[1],
}, nil
}
// NewInMemory creates new inmemory master database for storage node
2019-01-10 13:13:27 +00:00
// TODO: still stores data on disk
func NewInMemory(log *zap.Logger, storageDir string) (*DB, error) {
storage := pstore.NewStorage(storageDir)
psdb, err := psdb.OpenInMemory()
2019-01-10 13:13:27 +00:00
if err != nil {
return nil, err
}
return &DB{
log: log,
storage: storage,
psdb: psdb,
kdb: teststore.New(),
ndb: teststore.New(),
2019-01-10 13:13:27 +00:00
}, nil
}
2019-01-24 20:28:06 +00:00
// CreateTables creates any necessary tables.
func (db *DB) CreateTables() error {
migration := db.psdb.Migration()
return migration.Run(db.log.Named("migration"), db.psdb)
2019-01-24 20:28:06 +00:00
}
2019-01-10 13:13:27 +00:00
// Close closes any resources.
func (db *DB) Close() error {
return errs.Combine(
db.psdb.Close(),
db.kdb.Close(),
db.ndb.Close(),
db.storage.Close(),
2019-01-10 13:13:27 +00:00
)
}
// Storage returns piecestore location
func (db *DB) Storage() *pstore.Storage {
return db.storage
2019-01-10 13:13:27 +00:00
}
// PSDB returns piecestore database
func (db *DB) PSDB() *psdb.DB {
return db.psdb
}
// RoutingTable returns kademlia routing table
func (db *DB) RoutingTable() (kdb, ndb storage.KeyValueStore) {
return db.kdb, db.ndb
}