erigon-pulse/cmd/devnettest/requests/requests.go
leonardchinonso 4ee0ae1610
adding eth_getLogs functionality for contract events for the devnet tool (#3680)
* adding eth_getLogs functionality for contract events for the devnet tool

* Made changes to fix double hashing

* Fixed lint errors

* Changed strings.Replace to strings.ReplaceAll for replacing strings

Co-authored-by: Alexey Sharp <alexeysharp@Alexeys-iMac.local>
2022-03-14 13:59:51 +00:00

93 lines
2.3 KiB
Go

package requests
import (
"bytes"
"encoding/json"
"fmt"
"github.com/ledgerwatch/erigon/cmd/rpctest/rpctest"
"github.com/ledgerwatch/erigon/common"
"github.com/ledgerwatch/erigon/core/types"
)
func parseResponse(resp interface{}) string {
result, err := json.Marshal(resp)
if err != nil {
panic(err)
}
return string(result)
}
func GetBalance(reqId int, address common.Address, blockNum string) {
reqGen := initialiseRequestGenerator(reqId)
var b rpctest.EthBalance
res := reqGen.Erigon("eth_getBalance", reqGen.getBalance(address, blockNum), &b)
if res.Err != nil {
fmt.Printf("Error getting balance: %v\n", res.Err)
return
}
fmt.Printf("Balance retrieved: %v\n", parseResponse(b))
}
func SendTx(reqId int, signedTx *types.Transaction) (*common.Hash, error) {
reqGen := initialiseRequestGenerator(reqId)
var b rpctest.EthSendRawTransaction
var buf bytes.Buffer
err := (*signedTx).MarshalBinary(&buf)
if err != nil {
return nil, err
}
res := reqGen.Erigon("eth_sendRawTransaction", reqGen.sendRawTransaction(buf.Bytes()), &b)
if res.Err != nil {
return nil, err
}
fmt.Printf("Submitted transaction successfully: %v\n", parseResponse(b))
return &b.TxnHash, nil
}
func TxpoolContent(reqId int) {
reqGen := initialiseRequestGenerator(reqId)
var b rpctest.EthTxPool
res := reqGen.Erigon("txpool_content", reqGen.txpoolContent(), &b)
if res.Err != nil {
fmt.Printf("Error fetching txpool: %v\n", res.Err)
return
}
fmt.Printf("Txpool content: %v\n", parseResponse(b))
}
func ParityList(reqId int, account common.Address, quantity int, offset []byte, blockNum string) {
reqGen := initialiseRequestGenerator(reqId)
var b rpctest.ParityListStorageKeysResult
res := reqGen.Erigon("parity_listStorageKeys", reqGen.parityStorageKeyListContent(account, quantity, offset, blockNum), &b)
if res.Err != nil {
fmt.Printf("Error fetching storage keys: %v\n", res.Err)
return
}
fmt.Printf("Storage keys: %v\n", parseResponse(b))
}
func GetLogs(reqId int, fromBlock, toBlock uint64, address common.Address) error {
reqGen := initialiseRequestGenerator(reqId)
var b rpctest.Log
reqq := reqGen.getLogs(fromBlock, toBlock, address)
res := reqGen.Erigon("eth_getLogs", reqq, &b)
if res.Err != nil {
return fmt.Errorf("Error fetching logs: %v\n", res.Err)
}
fmt.Printf("Logs and events: %v\n", parseResponse(b))
return nil
}