This PR ports a few changes from PBSS: - Fix the snapshot generator waiter in case the generation is not even initialized - Refactor db inspector for ancient store
		
			
				
	
	
		
			122 lines
		
	
	
		
			3.7 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			122 lines
		
	
	
		
			3.7 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
| // Copyright 2022 The go-ethereum Authors
 | |
| // This file is part of the go-ethereum library.
 | |
| //
 | |
| // The go-ethereum library is free software: you can redistribute it and/or modify
 | |
| // it under the terms of the GNU Lesser General Public License as published by
 | |
| // the Free Software Foundation, either version 3 of the License, or
 | |
| // (at your option) any later version.
 | |
| //
 | |
| // The go-ethereum library is distributed in the hope that it will be useful,
 | |
| // but WITHOUT ANY WARRANTY; without even the implied warranty of
 | |
| // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 | |
| // GNU Lesser General Public License for more details.
 | |
| //
 | |
| // You should have received a copy of the GNU Lesser General Public License
 | |
| // along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
 | |
| 
 | |
| package rawdb
 | |
| 
 | |
| import (
 | |
| 	"fmt"
 | |
| 
 | |
| 	"github.com/ethereum/go-ethereum/common"
 | |
| 	"github.com/ethereum/go-ethereum/ethdb"
 | |
| )
 | |
| 
 | |
| type tableSize struct {
 | |
| 	name string
 | |
| 	size common.StorageSize
 | |
| }
 | |
| 
 | |
| // freezerInfo contains the basic information of the freezer.
 | |
| type freezerInfo struct {
 | |
| 	name  string      // The identifier of freezer
 | |
| 	head  uint64      // The number of last stored item in the freezer
 | |
| 	tail  uint64      // The number of first stored item in the freezer
 | |
| 	sizes []tableSize // The storage size per table
 | |
| }
 | |
| 
 | |
| // count returns the number of stored items in the freezer.
 | |
| func (info *freezerInfo) count() uint64 {
 | |
| 	return info.head - info.tail + 1
 | |
| }
 | |
| 
 | |
| // size returns the storage size of the entire freezer.
 | |
| func (info *freezerInfo) size() common.StorageSize {
 | |
| 	var total common.StorageSize
 | |
| 	for _, table := range info.sizes {
 | |
| 		total += table.size
 | |
| 	}
 | |
| 	return total
 | |
| }
 | |
| 
 | |
| // inspectFreezers inspects all freezers registered in the system.
 | |
| func inspectFreezers(db ethdb.Database) ([]freezerInfo, error) {
 | |
| 	var infos []freezerInfo
 | |
| 	for _, freezer := range freezers {
 | |
| 		switch freezer {
 | |
| 		case chainFreezerName:
 | |
| 			// Chain ancient store is a bit special. It's always opened along
 | |
| 			// with the key-value store, inspect the chain store directly.
 | |
| 			info := freezerInfo{name: freezer}
 | |
| 			// Retrieve storage size of every contained table.
 | |
| 			for table := range chainFreezerNoSnappy {
 | |
| 				size, err := db.AncientSize(table)
 | |
| 				if err != nil {
 | |
| 					return nil, err
 | |
| 				}
 | |
| 				info.sizes = append(info.sizes, tableSize{name: table, size: common.StorageSize(size)})
 | |
| 			}
 | |
| 			// Retrieve the number of last stored item
 | |
| 			ancients, err := db.Ancients()
 | |
| 			if err != nil {
 | |
| 				return nil, err
 | |
| 			}
 | |
| 			info.head = ancients - 1
 | |
| 
 | |
| 			// Retrieve the number of first stored item
 | |
| 			tail, err := db.Tail()
 | |
| 			if err != nil {
 | |
| 				return nil, err
 | |
| 			}
 | |
| 			info.tail = tail
 | |
| 			infos = append(infos, info)
 | |
| 
 | |
| 		default:
 | |
| 			return nil, fmt.Errorf("unknown freezer, supported ones: %v", freezers)
 | |
| 		}
 | |
| 	}
 | |
| 	return infos, nil
 | |
| }
 | |
| 
 | |
| // InspectFreezerTable dumps out the index of a specific freezer table. The passed
 | |
| // ancient indicates the path of root ancient directory where the chain freezer can
 | |
| // be opened. Start and end specify the range for dumping out indexes.
 | |
| // Note this function can only be used for debugging purposes.
 | |
| func InspectFreezerTable(ancient string, freezerName string, tableName string, start, end int64) error {
 | |
| 	var (
 | |
| 		path   string
 | |
| 		tables map[string]bool
 | |
| 	)
 | |
| 	switch freezerName {
 | |
| 	case chainFreezerName:
 | |
| 		path, tables = resolveChainFreezerDir(ancient), chainFreezerNoSnappy
 | |
| 	default:
 | |
| 		return fmt.Errorf("unknown freezer, supported ones: %v", freezers)
 | |
| 	}
 | |
| 	noSnappy, exist := tables[tableName]
 | |
| 	if !exist {
 | |
| 		var names []string
 | |
| 		for name := range tables {
 | |
| 			names = append(names, name)
 | |
| 		}
 | |
| 		return fmt.Errorf("unknown table, supported ones: %v", names)
 | |
| 	}
 | |
| 	table, err := newFreezerTable(path, tableName, noSnappy, true)
 | |
| 	if err != nil {
 | |
| 		return err
 | |
| 	}
 | |
| 	table.dumpIndexStdout(start, end)
 | |
| 	return nil
 | |
| }
 |