Roy Crihfield
589f8a3977
- Exports IPLD encoders. - Refactors `ClearDB` to source list of tables from `schema` package, and to allow downstream usage. Reviewed-on: #30
106 lines
2.9 KiB
Go
106 lines
2.9 KiB
Go
// VulcanizeDB
|
|
// Copyright © 2019 Vulcanize
|
|
|
|
// This program is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU Affero General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
|
|
// This program 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 Affero General Public License for more details.
|
|
|
|
// You should have received a copy of the GNU Affero General Public License
|
|
// along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
package ipld
|
|
|
|
import (
|
|
"github.com/ethereum/go-ethereum/core/types"
|
|
)
|
|
|
|
// FromBlockAndReceipts takes a block and processes it
|
|
// to return it a set of IPLD nodes for further processing.
|
|
func FromBlockAndReceipts(block *types.Block, receipts []*types.Receipt) ([]IPLD, []IPLD, [][]IPLD, []IPLD, error) {
|
|
// Process the txs
|
|
txNodes, err := processTransactions(block.Transactions())
|
|
if err != nil {
|
|
return nil, nil, nil, nil, err
|
|
}
|
|
withdrawalNodes, err := processWithdrawals(block.Withdrawals())
|
|
if err != nil {
|
|
return nil, nil, nil, nil, err
|
|
}
|
|
|
|
// Process the receipts and logs
|
|
rctNodes, logNodes, err := processReceiptsAndLogs(receipts)
|
|
|
|
return txNodes, rctNodes, logNodes, withdrawalNodes, err
|
|
}
|
|
|
|
// processTransactions will take the found transactions in a parsed block body
|
|
// to return IPLD node slices for eth-tx
|
|
func processTransactions(txs []*types.Transaction) ([]IPLD, error) {
|
|
var ethTxNodes []IPLD
|
|
for _, tx := range txs {
|
|
ethTx, err := EncodeTx(tx)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
ethTxNodes = append(ethTxNodes, ethTx)
|
|
}
|
|
|
|
return ethTxNodes, nil
|
|
}
|
|
|
|
func processWithdrawals(withdrawals []*types.Withdrawal) ([]IPLD, error) {
|
|
var withdrawalNodes []IPLD
|
|
for _, withdrawal := range withdrawals {
|
|
ethW, err := EncodeWithdrawal(withdrawal)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
withdrawalNodes = append(withdrawalNodes, ethW)
|
|
}
|
|
|
|
return withdrawalNodes, nil
|
|
}
|
|
|
|
// processReceiptsAndLogs will take in receipts
|
|
// to return IPLD node slices for eth-rct and eth-log
|
|
func processReceiptsAndLogs(rcts []*types.Receipt) ([]IPLD, [][]IPLD, error) {
|
|
// Pre allocating memory.
|
|
ethRctNodes := make([]IPLD, len(rcts))
|
|
ethLogNodes := make([][]IPLD, len(rcts))
|
|
|
|
for idx, rct := range rcts {
|
|
logNodes, err := processLogs(rct.Logs)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
|
|
ethRct, err := EncodeReceipt(rct)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
|
|
ethRctNodes[idx] = ethRct
|
|
ethLogNodes[idx] = logNodes
|
|
}
|
|
|
|
return ethRctNodes, ethLogNodes, nil
|
|
}
|
|
|
|
func processLogs(logs []*types.Log) ([]IPLD, error) {
|
|
logNodes := make([]IPLD, len(logs))
|
|
for idx, log := range logs {
|
|
logNode, err := EncodeLog(log)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
logNodes[idx] = logNode
|
|
}
|
|
return logNodes, nil
|
|
}
|