lbcd/database/ffldb/mockfile_test.go

164 lines
3.9 KiB
Go
Raw Normal View History

database: Major redesign of database package. This commit contains a complete redesign and rewrite of the database package that approaches things in a vastly different manner than the previous version. This is the first part of several stages that will be needed to ultimately make use of this new package. Some of the reason for this were discussed in #255, however a quick summary is as follows: - The previous database could only contain blocks on the main chain and reorgs required deleting the blocks from the database. This made it impossible to store orphans and could make external RPC calls for information about blocks during the middle of a reorg fail. - The previous database interface forced a high level of bitcoin-specific intelligence such as spend tracking into each backend driver. - The aforementioned point led to making it difficult to implement new backend drivers due to the need to repeat a lot of non-trivial logic which is better handled at a higher layer, such as the blockchain package. - The old database stored all blocks in leveldb. This made it extremely inefficient to do things such as lookup headers and individual transactions since the entire block had to be loaded from leveldb (which entails it doing data copies) to get access. In order to address all of these concerns, and others not mentioned, the database interface has been redesigned as follows: - Two main categories of functionality are provided: block storage and metadata storage - All block storage and metadata storage are done via read-only and read-write MVCC transactions with both manual and managed modes - Support for multiple concurrent readers and a single writer - Readers use a snapshot and therefore are not blocked by the writer - Some key properties of the block storage and retrieval API: - It is generic and does NOT contain additional bitcoin logic such spend tracking and block linking - Provides access to the raw serialized bytes so deserialization is not forced for callers that don't need it - Support for fetching headers via independent functions which allows implementations to provide significant optimizations - Ability to efficiently retrieve arbitrary regions of blocks (transactions, scripts, etc) - A rich metadata storage API is provided: - Key/value with arbitrary data - Support for buckets and nested buckets - Bucket iteration through a couple of different mechanisms - Cursors for efficient and direct key seeking - Supports registration of backend database implementations - Comprehensive test coverage - Provides strong documentation with example usage This commit also contains an implementation of the previously discussed interface named ffldb (flat file plus leveldb metadata backend). Here is a quick overview: - Highly optimized for read performance with consistent write performance regardless of database size - All blocks are stored in flat files on the file system - Bulk block region fetching is optimized to perform linear reads which improves performance on spindle disks - Anti-corruption mechanisms: - Flat files contain full block checksums to quickly an easily detect database corruption without needing to do expensive merkle root calculations - Metadata checksums - Open reconciliation - Extensive test coverage: - Comprehensive blackbox interface testing - Whitebox testing which uses intimate knowledge to exercise uncommon failure paths such as deleting files out from under the database - Corruption tests (replacing random data in the files) In addition, this commit also contains a new tool under the new database directory named dbtool which provides a few basic commands for testing the database. It is designed around commands, so it could be useful to expand on in the future. Finally, this commit addresses the following issues: - Adds support for and therefore closes #255 - Fixes #199 - Fixes #201 - Implements and closes #256 - Obsoletes and closes #257 - Closes #247 once the required chain and btcd modifications are in place to make use of this new code
2016-02-03 18:42:04 +01:00
// Copyright (c) 2015-2016 The btcsuite developers
// Use of this source code is governed by an ISC
// license that can be found in the LICENSE file.
// This file is part of the ffldb package rather than the ffldb_test package as
// it is part of the whitebox testing.
package ffldb
import (
"errors"
"io"
"sync"
)
// Errors used for the mock file.
var (
// errMockFileClosed is used to indicate a mock file is closed.
errMockFileClosed = errors.New("file closed")
// errInvalidOffset is used to indicate an offset that is out of range
// for the file was provided.
errInvalidOffset = errors.New("invalid offset")
// errSyncFail is used to indicate simulated sync failure.
errSyncFail = errors.New("simulated sync failure")
)
// mockFile implements the filer interface and used in order to force failures
// the database code related to reading and writing from the flat block files.
// A maxSize of -1 is unlimited.
type mockFile struct {
sync.RWMutex
maxSize int64
data []byte
forceSyncErr bool
closed bool
}
// Close closes the mock file without releasing any data associated with it.
// This allows it to be "reopened" without losing the data.
//
// This is part of the filer implementation.
func (f *mockFile) Close() error {
f.Lock()
defer f.Unlock()
if f.closed {
return errMockFileClosed
}
f.closed = true
return nil
}
// ReadAt reads len(b) bytes from the mock file starting at byte offset off. It
// returns the number of bytes read and the error, if any. ReadAt always
// returns a non-nil error when n < len(b). At end of file, that error is
// io.EOF.
//
// This is part of the filer implementation.
func (f *mockFile) ReadAt(b []byte, off int64) (int, error) {
f.RLock()
defer f.RUnlock()
if f.closed {
return 0, errMockFileClosed
}
maxSize := int64(len(f.data))
if f.maxSize > -1 && maxSize > f.maxSize {
maxSize = f.maxSize
}
if off < 0 || off > maxSize {
return 0, errInvalidOffset
}
// Limit to the max size field, if set.
numToRead := int64(len(b))
endOffset := off + numToRead
if endOffset > maxSize {
numToRead = maxSize - off
}
copy(b, f.data[off:off+numToRead])
if numToRead < int64(len(b)) {
return int(numToRead), io.EOF
}
return int(numToRead), nil
}
// Truncate changes the size of the mock file.
//
// This is part of the filer implementation.
func (f *mockFile) Truncate(size int64) error {
f.Lock()
defer f.Unlock()
if f.closed {
return errMockFileClosed
}
maxSize := int64(len(f.data))
if f.maxSize > -1 && maxSize > f.maxSize {
maxSize = f.maxSize
}
if size > maxSize {
return errInvalidOffset
}
f.data = f.data[:size]
return nil
}
// Write writes len(b) bytes to the mock file. It returns the number of bytes
// written and an error, if any. Write returns a non-nil error any time
// n != len(b).
//
// This is part of the filer implementation.
func (f *mockFile) WriteAt(b []byte, off int64) (int, error) {
f.Lock()
defer f.Unlock()
if f.closed {
return 0, errMockFileClosed
}
maxSize := f.maxSize
if maxSize < 0 {
maxSize = 100 * 1024 // 100KiB
}
if off < 0 || off > maxSize {
return 0, errInvalidOffset
}
// Limit to the max size field, if set, and grow the slice if needed.
numToWrite := int64(len(b))
if off+numToWrite > maxSize {
numToWrite = maxSize - off
}
if off+numToWrite > int64(len(f.data)) {
newData := make([]byte, off+numToWrite)
copy(newData, f.data)
f.data = newData
}
copy(f.data[off:], b[:numToWrite])
if numToWrite < int64(len(b)) {
return int(numToWrite), io.EOF
}
return int(numToWrite), nil
}
// Sync doesn't do anything for mock files. However, it will return an error if
// the mock file's forceSyncErr flag is set.
//
// This is part of the filer implementation.
func (f *mockFile) Sync() error {
if f.forceSyncErr {
return errSyncFail
}
return nil
}
// Ensure the mockFile type implements the filer interface.
var _ filer = (*mockFile)(nil)