path: root/dex/blockdb/db.go

package blockdb

import (
    coreCommon "github.com/dexon-foundation/dexon-consensus/common"
    coreBlockdb "github.com/dexon-foundation/dexon-consensus/core/blockdb"
    coreTypes "github.com/dexon-foundation/dexon-consensus/core/types"

    "github.com/dexon-foundation/dexon/common"
    "github.com/dexon-foundation/dexon/core/rawdb"
    "github.com/dexon-foundation/dexon/ethdb"
)

// BlockDB implement dexon-consensus BlockDatabase interface.
type BlockDB struct {
    db ethdb.Database
}

func NewDatabase(db ethdb.Database) *BlockDB {
    return &BlockDB{db}
}

func (d *BlockDB) Has(hash coreCommon.Hash) bool {
    return rawdb.HasCoreBlock(d.db, common.Hash(hash))
}

func (d *BlockDB) Get(hash coreCommon.Hash) (coreTypes.Block, error) {
    block := rawdb.ReadCoreBlock(d.db, common.Hash(hash))
    if block == nil {
        return coreTypes.Block{}, coreBlockdb.ErrBlockDoesNotExist
    }
    return *block, nil
}

func (d *BlockDB) GetAll() (coreBlockdb.BlockIterator, error) {
    return nil, coreBlockdb.ErrNotImplemented
}

func (d *BlockDB) Update(block coreTypes.Block) error {
    if !d.Has(block.Hash) {
        return coreBlockdb.ErrBlockDoesNotExist
    }
    rawdb.WriteCoreBlock(d.db, common.Hash(block.Hash), &block)
    return nil
}

func (d *BlockDB) Put(block coreTypes.Block) error {
    if d.Has(block.Hash) {
        return coreBlockdb.ErrBlockExists
    }
    rawdb.WriteCoreBlock(d.db, common.Hash(block.Hash), &block)
    return nil
}

func (d *BlockDB) Close() error { return nil }