mirror of
https://github.com/ethereum/go-ethereum.git
synced 2026-08-18 18:02:24 +00:00
521 lines
No EOL
13 KiB
Go
521 lines
No EOL
13 KiB
Go
package shyftdb
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"math/big"
|
|
"github.com/ethereum/go-ethereum/common"
|
|
"github.com/ethereum/go-ethereum/core/types"
|
|
|
|
|
|
"database/sql"
|
|
|
|
"log"
|
|
|
|
_ "github.com/lib/pq"
|
|
)
|
|
|
|
//SBlock type
|
|
type SBlock struct {
|
|
Hash string
|
|
Coinbase string
|
|
Number string
|
|
}
|
|
|
|
//blockRes struct
|
|
type blockRes struct {
|
|
hash string
|
|
coinbase string
|
|
number string
|
|
Blocks []SBlock
|
|
}
|
|
|
|
type SAccounts struct {
|
|
Addr string
|
|
Balance string
|
|
}
|
|
|
|
type accountRes struct {
|
|
addr string
|
|
balance string
|
|
AllAccounts []SAccounts
|
|
}
|
|
|
|
//ShyftTxEntry structure
|
|
type ShyftTxEntry struct {
|
|
TxHash common.Hash
|
|
To *common.Address
|
|
From *common.Address
|
|
BlockHash string
|
|
Amount *big.Int
|
|
GasPrice *big.Int
|
|
Gas uint64
|
|
Nonce uint64
|
|
Data []byte
|
|
}
|
|
|
|
type txRes struct {
|
|
TxEntry []ShyftTxEntryPretty
|
|
}
|
|
|
|
type ShyftTxEntryPretty struct {
|
|
TxHash string
|
|
To string
|
|
From string
|
|
BlockHash string
|
|
Amount uint64
|
|
GasPrice uint64
|
|
Gas uint64
|
|
Nonce uint64
|
|
Data []byte
|
|
}
|
|
|
|
type ShyftAccountEntry struct {
|
|
Balance string
|
|
Txs []string
|
|
}
|
|
|
|
type SendAndReceive struct {
|
|
To string
|
|
From string
|
|
Amount string
|
|
Address string
|
|
Balance string
|
|
}
|
|
|
|
//WriteBlock writes to block info to sql db
|
|
func WriteBlock(sqldb *sql.DB, block *types.Block) error {
|
|
coinbase := block.Header().Coinbase.String()
|
|
number := block.Header().Number.String()
|
|
gasUsed := block.Header().GasUsed
|
|
gasLimit := block.Header().GasLimit
|
|
txCount := block.Transactions().Len()
|
|
uncleCount := len(block.Uncles())
|
|
age := block.Time()
|
|
fmt.Println("TIMESTAMP+++++++", age)
|
|
|
|
sqlStatement := `INSERT INTO blocks(hash, coinbase, number, gasUsed, gasLimit, txcount, uncleCount) VALUES(($1), ($2), ($3), ($4), ($5), ($6), ($7)) RETURNING number`
|
|
qerr := sqldb.QueryRow(sqlStatement, block.Header().Hash().Hex(), coinbase, number, gasUsed, gasLimit, txCount, uncleCount).Scan(&number)
|
|
if qerr != nil {
|
|
panic(qerr)
|
|
}
|
|
|
|
if block.Transactions().Len() > 0 && block.Transactions()[0].To() != nil {
|
|
for _, tx := range block.Transactions() {
|
|
//WriteMinerRewards(sqldb, block)
|
|
WriteTransactions(sqldb, tx, block.Header().Hash())
|
|
WriteFromBalance(sqldb, tx)
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
//WriteTransactions writes to sqldb
|
|
func WriteTransactions(sqldb *sql.DB, tx *types.Transaction, blockHash common.Hash) error {
|
|
txData := ShyftTxEntry{
|
|
TxHash: tx.Hash(),
|
|
From: tx.From(),
|
|
To: tx.To(),
|
|
BlockHash: blockHash.Hex(),
|
|
Amount: tx.Value(),
|
|
GasPrice: tx.GasPrice(),
|
|
Gas: tx.Gas(),
|
|
Nonce: tx.Nonce(),
|
|
Data: tx.Data(),
|
|
}
|
|
|
|
txHash := txData.TxHash.Hex()
|
|
from := txData.From.Hex()
|
|
blockHasher := txData.BlockHash
|
|
amount := txData.Amount.String()
|
|
gasPrice := txData.GasPrice.String()
|
|
nonce := txData.Nonce
|
|
gas := txData.Gas
|
|
data := txData.Data
|
|
to := txData.To
|
|
if to == nil{
|
|
var retNonce string
|
|
sqlStatement := `INSERT INTO txs(txhash, from_addr, blockhash, amount, gasprice, gas, nonce, data) VALUES(($1), ($2), ($3), ($4), ($5), ($6), ($7), ($8)) RETURNING nonce`
|
|
qerr := sqldb.QueryRow(sqlStatement, txHash, from, blockHasher, amount, gasPrice, gas, nonce, data).Scan(&retNonce)
|
|
|
|
if qerr != nil {
|
|
panic(qerr)
|
|
}
|
|
} else {
|
|
var retNonce string
|
|
sqlStatement := `INSERT INTO txs(txhash, from_addr, to_addr, blockhash, amount, gasprice, gas, nonce, data) VALUES(($1), ($2), ($3), ($4), ($5), ($6), ($7), ($8), ($9)) RETURNING nonce`
|
|
qerr := sqldb.QueryRow(sqlStatement, txHash, from, to.Hex(), blockHasher, amount, gasPrice, gas, nonce, data).Scan(&retNonce)
|
|
|
|
if qerr != nil {
|
|
panic(qerr)
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
//WriteFromBalance writes senders balance to accounts db
|
|
func WriteFromBalance(sqldb *sql.DB, tx *types.Transaction) error {
|
|
sendAndReceiveData, balanceRec, balanceSen := WriteBalanceHelper(sqldb, tx)
|
|
toAddr := sendAndReceiveData.To
|
|
fromAddr := sendAndReceiveData.From
|
|
amount := sendAndReceiveData.Amount
|
|
balanceReceiver := balanceRec
|
|
balanceSender := balanceSen
|
|
|
|
var response string
|
|
sqlExistsStatement := `SELECT balance from accounts WHERE addr = ($1)`
|
|
err := sqldb.QueryRow(sqlExistsStatement, toAddr).Scan(&response)
|
|
switch {
|
|
case err == sql.ErrNoRows:
|
|
|
|
sqlStatement := `INSERT INTO accounts(addr, balance) VALUES(($1), ($2)) RETURNING addr`
|
|
insertErr := sqldb.QueryRow(sqlStatement, toAddr, amount).Scan(&toAddr)
|
|
if insertErr != nil {
|
|
panic(insertErr)
|
|
}
|
|
case err != nil:
|
|
log.Fatal(err)
|
|
default:
|
|
|
|
var newBalanceReceiver big.Int
|
|
var newBalanceSender big.Int
|
|
updateSQLStatement := `UPDATE accounts SET balance = ($2) WHERE addr = ($1)`
|
|
|
|
r := new(big.Int)
|
|
_, err := fmt.Sscan(balanceReceiver, r)
|
|
if err != nil {
|
|
log.Println("error scanning value:", err)
|
|
}
|
|
|
|
s := new(big.Int)
|
|
_, error := fmt.Sscan(balanceSender, s)
|
|
if error != nil {
|
|
log.Println("error scanning value:", error)
|
|
}
|
|
|
|
newBalanceReceiver.Add(r, tx.Value())
|
|
newBalanceSender.Sub(s, tx.Value())
|
|
|
|
_, err = sqldb.Exec(updateSQLStatement, toAddr, newBalanceReceiver.String())
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
_, err = sqldb.Exec(updateSQLStatement, fromAddr, newBalanceSender.String())
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func WriteBalanceHelper(sqldb *sql.DB, tx *types.Transaction) (SendAndReceive, string, string) {
|
|
sendAndReceiveData := SendAndReceive{
|
|
To: tx.To().Hex(),
|
|
From: tx.From().Hex(),
|
|
Amount: tx.Value().String(),
|
|
}
|
|
|
|
toAddr := sendAndReceiveData.To
|
|
fromAddr := sendAndReceiveData.From
|
|
|
|
getAccountBalanceReceiver := GetAccount(sqldb, toAddr)
|
|
getAccountBalanceSender:= GetAccount(sqldb, fromAddr)
|
|
|
|
var receiverBalance SendAndReceive
|
|
if err := json.Unmarshal([]byte(getAccountBalanceReceiver), &receiverBalance); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
var senderBalance SendAndReceive
|
|
if err := json.Unmarshal([]byte(getAccountBalanceSender), &senderBalance); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
balanceReceiver := receiverBalance.Balance
|
|
balanceSender := senderBalance.Balance
|
|
|
|
return sendAndReceiveData, balanceReceiver, balanceSender
|
|
}
|
|
|
|
//func WriteMinerRewards(sqldb *sql.DB, block *types.Block) {
|
|
// var totalGas big.Int
|
|
// //var txs []string
|
|
//
|
|
// fmt.Println("this is BLOCK.UNCLE", block.Uncles())
|
|
// fmt.Println("this is BLOCK UNCLE HASH", block.UncleHash().String())
|
|
// fmt.Println("this is BLOCK.TRANSACTIONS", block.Transactions())
|
|
// fmt.Println("this is BLOCK TOTAL GAS", block.GasUsed())
|
|
//
|
|
// for _, tx := range block.Transactions() {
|
|
// totalGas.Add(&totalGas, new(big.Int).Mul(tx.GasPrice(), new(big.Int).SetUint64(tx.Gas())))
|
|
// }
|
|
//}
|
|
|
|
// @NOTE: This function is extremely complex and requires heavy testing and knowdlege of edge cases:
|
|
// uncle blocks, account balance updates based on reorgs, diverges that get dropped.
|
|
// Reason for this is because the accounts are not deterministic like the block and tx hashes.
|
|
// @TODO: Calculate reward if there are uncles
|
|
// @TODO: Calculate mining reward (most likely retrieve higher up in the operations)
|
|
// @TODO: Calculate reorg
|
|
//func WriteMinerReward(db *leveldb.DB, block *types.Block) {
|
|
// var totalGas *big.Int
|
|
// var txs []string
|
|
// key := append([]byte("acc-")[:], block.Coinbase().Hash().Bytes()[:]...)
|
|
// for _, tx := range block.Transactions() {
|
|
// totalGas.Add(totalGas, new(big.Int).Mul(tx.GasPrice(), new(big.Int).SetUint64(tx.Gas())))
|
|
// }
|
|
//// retrievedData, err := db.Get(key, nil)
|
|
// if err != nil {
|
|
// // Assume time this account has had a tx
|
|
// // Balacne is exclusively minerreward + total gas from the block b/c no prior evm activity
|
|
// // Txs would be empty because they have not had any transactions on the EVM
|
|
// // @TODO: Calc mining reward
|
|
// //balance := totalGas.Add(totalGas, MINING_REWARD)
|
|
// balance := totalGas
|
|
// accData := ShyftAccountEntry{
|
|
// Balance: balance,
|
|
// Txs: txs,
|
|
// }
|
|
// var encodedData bytes.Buffer
|
|
// encoder := gob.NewEncoder(&encodedData)
|
|
// if err := encoder.Encode(accData); err != nil {
|
|
// log.Crit("Faild to encode Miner Account data", "err", err)
|
|
// }
|
|
// if err := db.Put(key, encodedData.Bytes(), nil); err != nil {
|
|
// log.Crit("Could not write the miner's first tx", "err", err)
|
|
// }
|
|
// } else {
|
|
// // The account has already have previous data stored due to activity in the EVM
|
|
// // Decode the data to update balance
|
|
// var decodedData ShyftAccountEntry
|
|
// d := gob.NewDecoder(bytes.NewBuffer(retrievedData))
|
|
// if err := d.Decode(&decodedData); err != nil {
|
|
// log.Crit("Failed to decode miner data:", "err", err)
|
|
// }
|
|
// // Write new balance
|
|
// // @TODO: Calc mining reward
|
|
// // decodedData.Balance.Add(decodedData.Balance, totalGas.Add(totalGas, MINING_REWARD)))
|
|
// decodedData.Balance.Add(decodedData.Balance, totalGas)
|
|
// // Encode the data to be written back to the db
|
|
// var encodedData bytes.Buffer
|
|
// encoder := gob.NewEncoder(&encodedData)
|
|
// if err := encoder.Encode(decodedData); err != nil {
|
|
// log.Crit("Faild to encode Miner Account data", "err", err)
|
|
// }
|
|
// // Write newly encoded data back to the db
|
|
// if err := db.Put(key, encodedData.Bytes(), nil); err != nil {
|
|
// log.Crit("Could not update miner account data", "err", err)
|
|
// }
|
|
// }
|
|
//}
|
|
|
|
|
|
|
|
///////////
|
|
// Getters
|
|
//////////
|
|
//GetAllBlocks returns []SBlock blocks for API
|
|
func GetAllBlocks(sqldb *sql.DB) string {
|
|
var arr blockRes
|
|
var blockArr string
|
|
rows, err := sqldb.Query(`
|
|
SELECT
|
|
number,
|
|
hash,
|
|
coinbase
|
|
FROM blocks`)
|
|
if err != nil {
|
|
fmt.Println("err")
|
|
}
|
|
defer rows.Close()
|
|
|
|
for rows.Next() {
|
|
var num string
|
|
var hash string
|
|
var coinbase string
|
|
err = rows.Scan(
|
|
&num,
|
|
&hash,
|
|
&coinbase,
|
|
)
|
|
|
|
arr.Blocks = append(arr.Blocks, SBlock{
|
|
Hash: hash,
|
|
Number: num,
|
|
Coinbase: coinbase,
|
|
})
|
|
|
|
blocks, _ := json.Marshal(arr.Blocks)
|
|
blocksFmt := string(blocks)
|
|
blockArr = blocksFmt
|
|
}
|
|
return blockArr
|
|
}
|
|
|
|
//GetBlock queries to send single block info
|
|
//TODO provide blockHash arg passed from handler.go
|
|
func GetBlock(sqldb *sql.DB) string {
|
|
sqlStatement := `SELECT * FROM blocks WHERE number=$1;`
|
|
row := sqldb.QueryRow(sqlStatement, 3)
|
|
var num string
|
|
var hash string
|
|
var coinbase string
|
|
row.Scan(&num, &hash, &coinbase)
|
|
|
|
block := SBlock{
|
|
Hash: hash,
|
|
Number: num,
|
|
Coinbase: coinbase,
|
|
}
|
|
json, _ := json.Marshal(block)
|
|
return string(json)
|
|
}
|
|
|
|
//GetAllTransactions getter fn for API
|
|
func GetAllTransactions(sqldb *sql.DB) string {
|
|
var arr txRes
|
|
var txx string
|
|
rows, err := sqldb.Query(`
|
|
SELECT
|
|
txhash,
|
|
to_addr,
|
|
from_addr,
|
|
blockhash,
|
|
amount,
|
|
gasprice,
|
|
gas,
|
|
nonce
|
|
FROM txs`)
|
|
if err != nil {
|
|
fmt.Println("err")
|
|
}
|
|
defer rows.Close()
|
|
for rows.Next() {
|
|
var txhash string
|
|
var to_addr string
|
|
var from_addr string
|
|
var blockhash string
|
|
var amount uint64
|
|
var gasprice uint64
|
|
var gas uint64
|
|
var nonce uint64
|
|
err = rows.Scan(
|
|
&txhash,
|
|
&to_addr,
|
|
&from_addr,
|
|
&blockhash,
|
|
&amount,
|
|
&gasprice,
|
|
&gas,
|
|
&nonce,
|
|
)
|
|
|
|
arr.TxEntry = append(arr.TxEntry, ShyftTxEntryPretty{
|
|
TxHash: txhash,
|
|
To: to_addr,
|
|
From: from_addr,
|
|
BlockHash: blockhash,
|
|
Amount: amount,
|
|
GasPrice: gasprice,
|
|
Gas: gas,
|
|
Nonce: nonce,
|
|
})
|
|
|
|
tx, _ := json.Marshal(arr.TxEntry)
|
|
newtx := string(tx)
|
|
txx = newtx
|
|
}
|
|
return txx
|
|
}
|
|
|
|
//GetTransaction fn returns single tx
|
|
func GetTransaction(sqldb *sql.DB) string {
|
|
sqlStatement := `SELECT
|
|
txhash,
|
|
to_addr,
|
|
from_addr,
|
|
blockhash,
|
|
amount,
|
|
gasprice,
|
|
gas,
|
|
nonce
|
|
FROM txs WHERE nonce=$1;`
|
|
row := sqldb.QueryRow(sqlStatement, 1)
|
|
var txhash string
|
|
var to_addr string
|
|
var from_addr string
|
|
var blockhash string
|
|
var amount uint64
|
|
var gasprice uint64
|
|
var gas uint64
|
|
var nonce uint64
|
|
row.Scan(&txhash, &to_addr, &from_addr, &blockhash, &amount, &gasprice, &gas, &nonce)
|
|
|
|
tx := ShyftTxEntryPretty{
|
|
TxHash: txhash,
|
|
To: to_addr,
|
|
From: from_addr,
|
|
BlockHash: blockhash,
|
|
Amount: amount,
|
|
GasPrice: gasprice,
|
|
Gas: gas,
|
|
Nonce: nonce,
|
|
}
|
|
json, _ := json.Marshal(tx)
|
|
|
|
return string(json)
|
|
}
|
|
|
|
//GetAccount returns account balances
|
|
func GetAccount(sqldb *sql.DB, address string) string {
|
|
sqlStatement := `SELECT * FROM accounts WHERE addr=$1;`
|
|
row := sqldb.QueryRow(sqlStatement, address)
|
|
var addr string
|
|
var balance string
|
|
|
|
row.Scan(&addr, &balance)
|
|
|
|
account := SAccounts{
|
|
Addr: addr,
|
|
Balance: balance,
|
|
}
|
|
json, _ := json.Marshal(account)
|
|
return string(json)
|
|
}
|
|
|
|
//GetAllAccounts returns all accounts and balances
|
|
func GetAllAccounts(sqldb *sql.DB) string {
|
|
var array accountRes
|
|
var accountsArr string
|
|
accs, err := sqldb.Query(`
|
|
SELECT
|
|
addr,
|
|
balance
|
|
FROM accounts`)
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
}
|
|
|
|
defer accs.Close()
|
|
|
|
for accs.Next() {
|
|
var addr string
|
|
var balance string
|
|
err = accs.Scan(
|
|
&addr,
|
|
&balance,
|
|
)
|
|
|
|
array.AllAccounts = append(array.AllAccounts, SAccounts{
|
|
Addr: addr,
|
|
Balance: balance,
|
|
})
|
|
|
|
accounts, _ := json.Marshal(array.AllAccounts)
|
|
accountsFmt := string(accounts)
|
|
accountsArr = accountsFmt
|
|
}
|
|
return accountsArr
|
|
} |