138 lines
3.4 KiB
Go
138 lines
3.4 KiB
Go
package publisher
|
|
|
|
import (
|
|
"encoding/csv"
|
|
"github.com/ethereum/go-ethereum/common"
|
|
"github.com/ethereum/go-ethereum/statediff/builder"
|
|
"os"
|
|
"path/filepath"
|
|
"strconv"
|
|
"time"
|
|
)
|
|
|
|
var (
|
|
Headers = []string{
|
|
"blockNumber", "blockHash", "accountAction", "codeHash",
|
|
"nonceValue", "balanceValue", "contractRoot", "storageDiffPaths",
|
|
"accountAddress", "storageKey", "storageValue",
|
|
}
|
|
|
|
timeStampFormat = "20060102150405.00000"
|
|
deletedAccountAction = "deleted"
|
|
createdAccountAction = "created"
|
|
updatedAccountAction = "updated"
|
|
)
|
|
|
|
func createCSVFilePath(path, blockNumber string) string {
|
|
now := time.Now()
|
|
timeStamp := now.Format(timeStampFormat)
|
|
suffix := timeStamp + "-" + blockNumber
|
|
filePath := filepath.Join(path, suffix)
|
|
filePath = filePath + ".csv"
|
|
return filePath
|
|
}
|
|
|
|
func (p *publisher) publishStateDiffToCSV(sd builder.StateDiff) (string, error) {
|
|
filePath := createCSVFilePath(p.Config.Path, strconv.FormatInt(sd.BlockNumber, 10))
|
|
|
|
file, err := os.OpenFile(filePath, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
defer file.Close()
|
|
|
|
writer := csv.NewWriter(file)
|
|
defer writer.Flush()
|
|
|
|
var data [][]string
|
|
data = append(data, Headers)
|
|
for _, row := range accumulateAccountRows(sd) {
|
|
data = append(data, row)
|
|
}
|
|
for _, value := range data {
|
|
err := writer.Write(value)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
}
|
|
|
|
return filePath, nil
|
|
}
|
|
|
|
func accumulateAccountRows(sd builder.StateDiff) [][]string {
|
|
var accountRows [][]string
|
|
for accountAddr, accountDiff := range sd.CreatedAccounts {
|
|
formattedAccountData := formatAccountData(accountAddr, accountDiff, sd, createdAccountAction)
|
|
|
|
for _, accountData := range formattedAccountData {
|
|
accountRows = append(accountRows, accountData)
|
|
}
|
|
}
|
|
|
|
for accountAddr, accountDiff := range sd.UpdatedAccounts {
|
|
formattedAccountData := formatAccountData(accountAddr, accountDiff, sd, updatedAccountAction)
|
|
|
|
for _, accountData := range formattedAccountData {
|
|
accountRows = append(accountRows, accountData)
|
|
}
|
|
}
|
|
|
|
for accountAddr, accountDiff := range sd.DeletedAccounts {
|
|
formattedAccountData := formatAccountData(accountAddr, accountDiff, sd, deletedAccountAction)
|
|
|
|
for _, accountData := range formattedAccountData {
|
|
accountRows = append(accountRows, accountData)
|
|
}
|
|
}
|
|
|
|
return accountRows
|
|
}
|
|
|
|
func formatAccountData(accountAddr common.Address, accountDiff builder.AccountDiff, sd builder.StateDiff, accountAction string) [][]string {
|
|
blockNumberString := strconv.FormatInt(sd.BlockNumber, 10)
|
|
blockHash := sd.BlockHash.String()
|
|
codeHash := accountDiff.CodeHash
|
|
nonce := strconv.FormatUint(*accountDiff.Nonce.Value, 10)
|
|
balance := accountDiff.Balance.Value.String()
|
|
newContractRoot := accountDiff.ContractRoot.Value
|
|
address := accountAddr.String()
|
|
var result [][]string
|
|
|
|
if len(accountDiff.Storage) > 0 {
|
|
for storagePath, storage := range accountDiff.Storage {
|
|
formattedAccountData := []string{
|
|
blockNumberString,
|
|
blockHash,
|
|
accountAction,
|
|
codeHash,
|
|
nonce,
|
|
balance,
|
|
*newContractRoot,
|
|
storagePath,
|
|
address,
|
|
*storage.Key,
|
|
*storage.Value,
|
|
}
|
|
|
|
result = append(result, formattedAccountData)
|
|
}
|
|
} else {
|
|
formattedAccountData := []string{
|
|
blockNumberString,
|
|
blockHash,
|
|
accountAction,
|
|
codeHash,
|
|
nonce,
|
|
balance,
|
|
*newContractRoot,
|
|
"",
|
|
address,
|
|
"",
|
|
"",
|
|
}
|
|
result = append(result, formattedAccountData)
|
|
}
|
|
|
|
return result
|
|
}
|