86 lines
3.3 KiB
Go
86 lines
3.3 KiB
Go
// Copyright © 2018 Rob Mulholand <rmulholand@8thlight.com>
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package cmd
|
|
|
|
import (
|
|
"log"
|
|
|
|
"github.com/spf13/cobra"
|
|
|
|
"github.com/vulcanize/eth-block-extractor/pkg/db"
|
|
"github.com/vulcanize/eth-block-extractor/pkg/ipfs"
|
|
"github.com/vulcanize/eth-block-extractor/pkg/ipfs/eth_state_trie"
|
|
"github.com/vulcanize/eth-block-extractor/pkg/ipfs/eth_storage_trie"
|
|
"github.com/vulcanize/eth-block-extractor/pkg/transformers"
|
|
)
|
|
|
|
// createIpldsForStateTrieCmd represents the createIpldsForStateTrie command
|
|
var createIpldsForStateTrieCmd = &cobra.Command{
|
|
Use: "createIpldsForStateTrie",
|
|
Short: "Create iplds for every ethereum state trie node",
|
|
Long: `Create iplds for every ethereum state trie node. For example:
|
|
|
|
./eth-block-extractor createIpldsForStateTrie
|
|
|
|
Note that this operation is very expensive in terms of both cpu and disk,
|
|
as it is reconstructing the entire ethereum state trie in the same fashion
|
|
as an archive node.`,
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
createIpldsForStateTrie()
|
|
},
|
|
}
|
|
|
|
func init() {
|
|
rootCmd.AddCommand(createIpldsForStateTrieCmd)
|
|
createIpldsForStateTrieCmd.Flags().BoolVarP(&computeState, "compute-state", "c", false, "Flag indicating state must be computed (non-archive node).")
|
|
createIpldsForStateTrieCmd.Flags().Int64VarP(&startingBlockNumber, "starting-block-number", "s", 0, "First block number to create IPLD for.")
|
|
createIpldsForStateTrieCmd.Flags().Int64VarP(&endingBlockNumber, "ending-block-number", "e", 5900000, "Last block number to create IPLD for.")
|
|
}
|
|
|
|
func createIpldsForStateTrie() {
|
|
if computeState && startingBlockNumber != 0 {
|
|
log.Println("Computing state trie must begin at genesis block. Ignoring passed starting block number.")
|
|
}
|
|
|
|
// init eth db
|
|
databaseConfig := db.CreateDatabaseConfig(db.Level, levelDbPath)
|
|
database, err := db.CreateDatabase(databaseConfig)
|
|
if err != nil {
|
|
log.Fatal("Error connecting to the ethereum db: ", err)
|
|
}
|
|
|
|
// init ipfs publishers
|
|
adder, err := ipfs.InitIPFSNode(ipfsPath)
|
|
if err != nil {
|
|
log.Fatal("Error connecting to ipfs: ", err)
|
|
}
|
|
stateTrieDagPutter := eth_state_trie.NewStateTrieDagPutter(adder)
|
|
stateTriePublisher := ipfs.NewIpfsPublisher(stateTrieDagPutter)
|
|
storageTrieDagPutter := eth_storage_trie.NewStorageTrieDagPutter(adder)
|
|
storageTriePublisher := ipfs.NewIpfsPublisher(storageTrieDagPutter)
|
|
|
|
// init and execute transformer
|
|
if computeState {
|
|
transformer := transformers.NewComputeEthStateTrieTransformer(database, stateTriePublisher, storageTriePublisher)
|
|
err = transformer.Execute(endingBlockNumber)
|
|
} else {
|
|
transformer := transformers.NewEthStateTrieTransformer(database, stateTriePublisher, storageTriePublisher)
|
|
err = transformer.Execute(startingBlockNumber, endingBlockNumber)
|
|
}
|
|
if err != nil {
|
|
log.Fatal("Error executing transformer: ", err)
|
|
}
|
|
}
|