mirror of
https://github.com/kaspanet/kaspad.git
synced 2025-05-21 06:16:45 +00:00
41647fd488
* [DEV-376] Changed any instance of %v in format strings with a more specific format token * [DEV-376] Fixed some more wrong formatting strings + removed redundant cast * [DEV-376] Added fmt.Sprintf where it was missing * [DEV-376] use %s for util.Amount, to invoke .String() * [DEV-376] Some more fixes in format strings * [DEV-376] fixed mruinvmap_test to expect the correct behaviour
94 lines
2.3 KiB
Go
94 lines
2.3 KiB
Go
// 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.
|
|
|
|
package main
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/daglabs/btcd/dagconfig/daghash"
|
|
"github.com/daglabs/btcd/database"
|
|
)
|
|
|
|
// headersCmd defines the configuration options for the loadheaders command.
|
|
type headersCmd struct {
|
|
Bulk bool `long:"bulk" description:"Use bulk loading of headers instead of one at a time"`
|
|
}
|
|
|
|
var (
|
|
// headersCfg defines the configuration options for the command.
|
|
headersCfg = headersCmd{
|
|
Bulk: false,
|
|
}
|
|
)
|
|
|
|
// Execute is the main entry point for the command. It's invoked by the parser.
|
|
func (cmd *headersCmd) Execute(args []string) error {
|
|
// Setup the global config options and ensure they are valid.
|
|
if err := setupGlobalConfig(); err != nil {
|
|
return err
|
|
}
|
|
|
|
// Load the block database.
|
|
db, err := loadBlockDB()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer db.Close()
|
|
|
|
// NOTE: This code will only work for ffldb. Ideally the package using
|
|
// the database would keep a metadata index of its own.
|
|
blockIdxName := []byte("ffldb-blockidx")
|
|
if !headersCfg.Bulk {
|
|
err = db.View(func(dbTx database.Tx) error {
|
|
totalHdrs := 0
|
|
blockIdxBucket := dbTx.Metadata().Bucket(blockIdxName)
|
|
blockIdxBucket.ForEach(func(k, v []byte) error {
|
|
totalHdrs++
|
|
return nil
|
|
})
|
|
log.Infof("Loading headers for %d blocks...", totalHdrs)
|
|
numLoaded := 0
|
|
startTime := time.Now()
|
|
blockIdxBucket.ForEach(func(k, v []byte) error {
|
|
var hash daghash.Hash
|
|
copy(hash[:], k)
|
|
_, err := dbTx.FetchBlockHeader(&hash)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
numLoaded++
|
|
return nil
|
|
})
|
|
log.Infof("Loaded %d headers in %s", numLoaded,
|
|
time.Since(startTime))
|
|
return nil
|
|
})
|
|
return err
|
|
}
|
|
|
|
// Bulk load headers.
|
|
err = db.View(func(dbTx database.Tx) error {
|
|
blockIdxBucket := dbTx.Metadata().Bucket(blockIdxName)
|
|
hashes := make([]daghash.Hash, 0, 500000)
|
|
blockIdxBucket.ForEach(func(k, v []byte) error {
|
|
var hash daghash.Hash
|
|
copy(hash[:], k)
|
|
hashes = append(hashes, hash)
|
|
return nil
|
|
})
|
|
|
|
log.Infof("Loading headers for %d blocks...", len(hashes))
|
|
startTime := time.Now()
|
|
hdrs, err := dbTx.FetchBlockHeaders(hashes)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
log.Infof("Loaded %d headers in %s", len(hdrs),
|
|
time.Since(startTime))
|
|
return nil
|
|
})
|
|
return err
|
|
}
|