Allow exporting subset of chain

This commit is contained in:
Taylor Gerring 2015-06-05 23:01:54 -04:00
parent 43ceb0f5c7
commit 89c9320d80

View File

@ -341,13 +341,25 @@ func (bc *ChainManager) ResetWithGenesisBlock(gb *types.Block) {
// Export writes the active chain to the given writer. // Export writes the active chain to the given writer.
func (self *ChainManager) Export(w io.Writer) error { func (self *ChainManager) Export(w io.Writer) error {
if err := self.ExportN(w, uint64(1), self.currentBlock.NumberU64()); err != nil {
return err
}
return nil
}
// ExportN writes a subset of the active chain to the given writer.
func (self *ChainManager) ExportN(w io.Writer, first uint64, last uint64) error {
self.mu.RLock() self.mu.RLock()
defer self.mu.RUnlock() defer self.mu.RUnlock()
glog.V(logger.Info).Infof("exporting %v blocks...\n", self.currentBlock.Header().Number) glog.V(logger.Info).Infof("exporting %v blocks...\n", self.currentBlock.Header().Number)
last := self.currentBlock.NumberU64() if first > last {
return fmt.Errorf("export failed: first (%d) is greater than last (%d)", first, last)
}
for nr := uint64(1); nr <= last; nr++ { glog.V(logger.Info).Infof("exporting %d blocks...\n", last-first)
for nr := first; nr <= last; nr++ {
block := self.GetBlockByNumber(nr) block := self.GetBlockByNumber(nr)
if block == nil { if block == nil {
return fmt.Errorf("export failed on #%d: not found", nr) return fmt.Errorf("export failed on #%d: not found", nr)