db.go

package lnbolt

import (
	"fmt"
	"github.com/boltdb/bolt"
	"github.com/mit-dci/lit/lncore"
	"os"
	"path"
)

// LitBoltDB uses Bolt as a storage backend.
type LitBoltDB struct {
	open     bool
	walletdb *bolt.DB
	peerdb   *bolt.DB
	chandb   *bolt.DB
}

// Open .
func (db *LitBoltDB) Open(dbpath string) error {

	// Sanity check.
	if db.open {
		return fmt.Errorf("tried to open an open BoltDB database")
	}

	os.Mkdir(dbpath, os.ModeDir|0700)

	// Figure out file paths.
	wpath := path.Join(dbpath, "wallet.db")
	ppath := path.Join(dbpath, "peer.db")
	cpath := path.Join(dbpath, "channels.db")

	var err error

	walletdb, err := bolt.Open(wpath, 0644, nil)
	if err != nil {
		return err
	}

	peerdb, err := bolt.Open(ppath, 0644, nil)
	if err != nil {
		walletdb.Close()
		return err
	}

	chandb, err := bolt.Open(cpath, 0644, nil)
	if err != nil {
		walletdb.Close()
		peerdb.Close()
		return err
	}

	// Actually set the new databases in ourselves.
	db.walletdb = walletdb
	db.peerdb = peerdb
	db.chandb = chandb
	db.open = true

	return nil

}

// IsSingleFile .
func (*LitBoltDB) IsSingleFile() bool {
	return false
}

// Close .
func (db *LitBoltDB) Close() error {

	// Sanity check.
	if !db.open {
		return fmt.Errorf("tried to close a closed BoltDB database")
	}

	var err error

	err = db.walletdb.Close()
	if err != nil {
		return err
	}

	err = db.peerdb.Close()
	if err != nil {
		return err
	}

	err = db.chandb.Close()
	if err != nil {
		return err
	}

	db.open = false
	return nil

}

// Check .
func (db *LitBoltDB) Check() error {

	if db.walletdb == nil {
		return fmt.Errorf("Wallet DB is nil")
	}

	if db.peerdb == nil {
		return fmt.Errorf("Peer DB is nil")
	}

	if db.chandb == nil {
		return fmt.Errorf("Channel DB is nil")
	}

	return nil

}

// GetWalletDB .
func (db *LitBoltDB) GetWalletDB(cointype uint32) lncore.LitWalletStorage {
	return nil
}

// GetPeerDB .
func (db *LitBoltDB) GetPeerDB() lncore.LitPeerStorage {
	w := peerboltdb{}
	w.db = db.peerdb

	err := w.init()
	if err != nil {
		panic(err)
	}

	var w2 lncore.LitPeerStorage
	w2 = &w
	return w2
}

// GetChannelDB .
func (db *LitBoltDB) GetChannelDB() lncore.LitChannelStorage {
	return nil
}