storj/storagenode/storagenodedb/database.go

128 lines
2.4 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"
2019-01-10 13:13:27 +00:00
"storj.io/storj/storage"
"storj.io/storj/storage/boltdb"
"storj.io/storj/storage/filestore"
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
Info2 string
Kademlia string
Pieces string
}
2019-01-10 13:13:27 +00:00
// DB contains access to different database tables
type DB struct {
2019-04-26 06:17:18 +01:00
log *zap.Logger
pieces interface {
storage.Blobs
Close() error
}
info *InfoDB
kdb, ndb, adb storage.KeyValueStore
2019-01-10 13:13:27 +00:00
}
// New creates a new master database for storage node
func New(log *zap.Logger, config Config) (*DB, error) {
piecesDir, err := filestore.NewDir(config.Pieces)
if err != nil {
return nil, err
}
pieces := filestore.New(piecesDir)
infodb, err := newInfo(config.Info2)
if err != nil {
return nil, err
}
dbs, err := boltdb.NewShared(config.Kademlia, kademlia.KademliaBucket, kademlia.NodeBucket, kademlia.AntechamberBucket)
if err != nil {
return nil, err
}
return &DB{
2019-04-26 06:17:18 +01:00
log: log,
pieces: pieces,
info: infodb,
kdb: dbs[0],
ndb: dbs[1],
adb: dbs[2],
}, 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) {
piecesDir, err := filestore.NewDir(storageDir)
if err != nil {
return nil, err
}
pieces := filestore.New(piecesDir)
infodb, err := NewInfoInMemory()
if err != nil {
return nil, err
}
2019-01-10 13:13:27 +00:00
return &DB{
2019-04-26 06:17:18 +01:00
log: log,
pieces: pieces,
2019-04-26 06:17:18 +01:00
info: infodb,
kdb: teststore.New(),
ndb: teststore.New(),
adb: 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 {
2019-04-26 06:17:18 +01:00
return db.info.CreateTables(db.log)
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.kdb.Close(),
db.ndb.Close(),
db.adb.Close(),
db.pieces.Close(),
db.info.Close(),
2019-01-10 13:13:27 +00:00
)
}
// Pieces returns blob storage for pieces
func (db *DB) Pieces() storage.Blobs {
return db.pieces
}
2019-01-10 13:13:27 +00:00
// RoutingTable returns kademlia routing table
func (db *DB) RoutingTable() (kdb, ndb, adb storage.KeyValueStore) {
return db.kdb, db.ndb, db.adb
2019-01-10 13:13:27 +00:00
}