138 lines
3.7 KiB
Go
138 lines
3.7 KiB
Go
package publisher
|
|
|
|
import (
|
|
"encoding/csv"
|
|
"github.com/ethereum/go-ethereum/statediff/builder"
|
|
"os"
|
|
"path/filepath"
|
|
"strconv"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
var (
|
|
Headers = []string{
|
|
"blockNumber", "blockHash", "accountAction", "codeHash",
|
|
"nonceValue", "balanceValue", "contractRoot", "storageDiffPaths",
|
|
}
|
|
|
|
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 accumulateCreatedAccountRows(sd) {
|
|
data = append(data, row)
|
|
}
|
|
for _, row := range accumulateUpdatedAccountRows(sd) {
|
|
data = append(data, row)
|
|
}
|
|
for _, row := range accumulateDeletedAccountRows(sd) {
|
|
data = append(data, row)
|
|
}
|
|
|
|
for _, value := range data {
|
|
err := writer.Write(value)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
}
|
|
|
|
return filePath, nil
|
|
}
|
|
|
|
func accumulateUpdatedAccountRows(sd builder.StateDiff) [][]string {
|
|
var updatedAccountRows [][]string
|
|
for _, accountDiff := range sd.UpdatedAccounts {
|
|
formattedAccountData := formatAccountDiffIncremental(accountDiff, sd, updatedAccountAction)
|
|
|
|
updatedAccountRows = append(updatedAccountRows, formattedAccountData)
|
|
}
|
|
|
|
return updatedAccountRows
|
|
}
|
|
|
|
func accumulateDeletedAccountRows(sd builder.StateDiff) [][]string {
|
|
var deletedAccountRows [][]string
|
|
for _, accountDiff := range sd.DeletedAccounts {
|
|
formattedAccountData := formatAccountDiffEventual(accountDiff, sd, deletedAccountAction)
|
|
|
|
deletedAccountRows = append(deletedAccountRows, formattedAccountData)
|
|
}
|
|
|
|
return deletedAccountRows
|
|
}
|
|
|
|
func accumulateCreatedAccountRows(sd builder.StateDiff) [][]string {
|
|
var createdAccountRows [][]string
|
|
for _, accountDiff := range sd.CreatedAccounts {
|
|
formattedAccountData := formatAccountDiffEventual(accountDiff, sd, createdAccountAction)
|
|
|
|
createdAccountRows = append(createdAccountRows, formattedAccountData)
|
|
}
|
|
|
|
return createdAccountRows
|
|
}
|
|
|
|
func formatAccountDiffEventual(accountDiff builder.AccountDiff, sd builder.StateDiff, accountAction string) []string {
|
|
newContractRoot := accountDiff.ContractRoot.Value
|
|
var storageDiffPaths []string
|
|
for k := range accountDiff.Storage {
|
|
storageDiffPaths = append(storageDiffPaths, k)
|
|
}
|
|
formattedAccountData := []string{
|
|
strconv.FormatInt(sd.BlockNumber, 10),
|
|
sd.BlockHash.String(),
|
|
accountAction,
|
|
accountDiff.CodeHash,
|
|
strconv.FormatUint(*accountDiff.Nonce.Value, 10),
|
|
accountDiff.Balance.Value.String(),
|
|
*newContractRoot,
|
|
strings.Join(storageDiffPaths, ","),
|
|
}
|
|
return formattedAccountData
|
|
}
|
|
|
|
func formatAccountDiffIncremental(accountDiff builder.AccountDiff, sd builder.StateDiff, accountAction string) []string {
|
|
newContractRoot := accountDiff.ContractRoot.Value
|
|
var storageDiffPaths []string
|
|
for k := range accountDiff.Storage {
|
|
storageDiffPaths = append(storageDiffPaths, k)
|
|
}
|
|
formattedAccountData := []string{
|
|
strconv.FormatInt(sd.BlockNumber, 10),
|
|
sd.BlockHash.String(),
|
|
accountAction,
|
|
accountDiff.CodeHash,
|
|
strconv.FormatUint(*accountDiff.Nonce.Value, 10),
|
|
accountDiff.Balance.Value.String(),
|
|
*newContractRoot,
|
|
strings.Join(storageDiffPaths, ","),
|
|
}
|
|
return formattedAccountData
|
|
}
|