a_bennassar
4 years ago
committed by
GitHub
GitHub
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
12 changed files with 926 additions and 176 deletions
Unified View
Diff Options
-
+6 -1README.md
-
+7 -5common/account.go
-
+1 -1common/block.go
-
+1 -1common/coordinator.go
-
+38 -20common/l1tx.go
-
+24 -16common/l2tx.go
-
+8 -7common/token.go
-
+28 -8common/tx.go
-
+159 -9db/historydb/historydb.go
-
+202 -76db/historydb/historydb_test.go
-
+109 -32db/historydb/migrations/001_init.sql
-
+343 -0test/historydb.go
@ -0,0 +1,343 @@ |
|||||
|
package test |
||||
|
|
||||
|
import ( |
||||
|
"errors" |
||||
|
"fmt" |
||||
|
"math/big" |
||||
|
"strconv" |
||||
|
"time" |
||||
|
|
||||
|
ethCommon "github.com/ethereum/go-ethereum/common" |
||||
|
"github.com/hermeznetwork/hermez-node/common" |
||||
|
"github.com/iden3/go-iden3-crypto/babyjub" |
||||
|
) |
||||
|
|
||||
|
// WARNING: the generators in this file doesn't necessary follow the protocol
|
||||
|
// they are intended to check that the parsers between struct <==> DB are correct
|
||||
|
|
||||
|
// GenBlocks generates block from, to block numbers. WARNING: This is meant for DB/API testing, and may not be fully consistent with the protocol.
|
||||
|
func GenBlocks(from, to int64) []common.Block { |
||||
|
var blocks []common.Block |
||||
|
for i := from; i < to; i++ { |
||||
|
blocks = append(blocks, common.Block{ |
||||
|
EthBlockNum: i, |
||||
|
//nolint:gomnd
|
||||
|
Timestamp: time.Now().Add(time.Second * 13).UTC(), |
||||
|
Hash: ethCommon.BigToHash(big.NewInt(int64(i))), |
||||
|
}) |
||||
|
} |
||||
|
return blocks |
||||
|
} |
||||
|
|
||||
|
// GenTokens generates tokens. WARNING: This is meant for DB/API testing, and may not be fully consistent with the protocol.
|
||||
|
func GenTokens(nTokens int, blocks []common.Block) []common.Token { |
||||
|
tokens := []common.Token{} |
||||
|
for i := 0; i < nTokens; i++ { |
||||
|
token := common.Token{ |
||||
|
TokenID: common.TokenID(i), |
||||
|
Name: fmt.Sprint(i), |
||||
|
Symbol: fmt.Sprint(i), |
||||
|
Decimals: uint64(i), |
||||
|
EthBlockNum: blocks[i%len(blocks)].EthBlockNum, |
||||
|
EthAddr: ethCommon.BigToAddress(big.NewInt(int64(i))), |
||||
|
} |
||||
|
if i%2 == 0 { |
||||
|
token.USD = 3 |
||||
|
token.USDUpdate = time.Now() |
||||
|
} |
||||
|
tokens = append(tokens, token) |
||||
|
} |
||||
|
return tokens |
||||
|
} |
||||
|
|
||||
|
// GenBatches generates batches. WARNING: This is meant for DB/API testing, and may not be fully consistent with the protocol.
|
||||
|
func GenBatches(nBatches int, blocks []common.Block) []common.Batch { |
||||
|
batches := []common.Batch{} |
||||
|
collectedFees := make(map[common.TokenID]*big.Int) |
||||
|
for i := 0; i < 64; i++ { |
||||
|
collectedFees[common.TokenID(i)] = big.NewInt(int64(i)) |
||||
|
} |
||||
|
for i := 0; i < nBatches; i++ { |
||||
|
batch := common.Batch{ |
||||
|
BatchNum: common.BatchNum(i + 1), |
||||
|
EthBlockNum: blocks[i%len(blocks)].EthBlockNum, |
||||
|
//nolint:gomnd
|
||||
|
ForgerAddr: ethCommon.BigToAddress(big.NewInt(6886723)), |
||||
|
CollectedFees: collectedFees, |
||||
|
StateRoot: common.Hash([]byte("duhdqlwiucgwqeiu")), |
||||
|
//nolint:gomnd
|
||||
|
NumAccounts: 30, |
||||
|
ExitRoot: common.Hash([]byte("tykertheuhtgenuer3iuw3b")), |
||||
|
SlotNum: common.SlotNum(i), |
||||
|
} |
||||
|
if i%2 == 0 { |
||||
|
batch.ForgeL1TxsNum = uint32(i) |
||||
|
} |
||||
|
batches = append(batches, batch) |
||||
|
} |
||||
|
return batches |
||||
|
} |
||||
|
|
||||
|
// GenAccounts generates accounts. WARNING: This is meant for DB/API testing, and may not be fully consistent with the protocol.
|
||||
|
func GenAccounts(totalAccounts, userAccounts int, tokens []common.Token, userAddr *ethCommon.Address, batches []common.Batch) []common.Account { |
||||
|
if totalAccounts < userAccounts { |
||||
|
panic("totalAccounts must be greater than userAccounts") |
||||
|
} |
||||
|
privK := babyjub.NewRandPrivKey() |
||||
|
pubK := privK.Public() |
||||
|
accs := []common.Account{} |
||||
|
for i := 0; i < totalAccounts; i++ { |
||||
|
var addr ethCommon.Address |
||||
|
if i < userAccounts { |
||||
|
addr = *userAddr |
||||
|
} else { |
||||
|
addr = ethCommon.BigToAddress(big.NewInt(int64(i))) |
||||
|
} |
||||
|
accs = append(accs, common.Account{ |
||||
|
Idx: common.Idx(i), |
||||
|
TokenID: tokens[i%len(tokens)].TokenID, |
||||
|
EthAddr: addr, |
||||
|
BatchNum: batches[i%len(batches)].BatchNum, |
||||
|
PublicKey: pubK, |
||||
|
}) |
||||
|
} |
||||
|
return accs |
||||
|
} |
||||
|
|
||||
|
// GenL1Txs generates L1 txs. WARNING: This is meant for DB/API testing, and may not be fully consistent with the protocol.
|
||||
|
func GenL1Txs( |
||||
|
fromIdx int, |
||||
|
totalTxs, nUserTxs int, |
||||
|
userAddr *ethCommon.Address, |
||||
|
accounts []common.Account, |
||||
|
tokens []common.Token, |
||||
|
blocks []common.Block, |
||||
|
batches []common.Batch, |
||||
|
) ([]common.L1Tx, []common.L1Tx) { |
||||
|
if totalTxs < nUserTxs { |
||||
|
panic("totalTxs must be greater than userTxs") |
||||
|
} |
||||
|
userTxs := []common.L1Tx{} |
||||
|
othersTxs := []common.L1Tx{} |
||||
|
for i := 0; i < totalTxs; i++ { |
||||
|
var tx common.L1Tx |
||||
|
if batches[i%len(batches)].ForgeL1TxsNum != 0 { |
||||
|
tx = common.L1Tx{ |
||||
|
TxID: common.TxID(common.Hash([]byte("L1_" + strconv.Itoa(fromIdx+i)))), |
||||
|
ToForgeL1TxsNum: batches[i%len(batches)].ForgeL1TxsNum, |
||||
|
Position: i, |
||||
|
UserOrigin: i%2 == 0, |
||||
|
TokenID: tokens[i%len(tokens)].TokenID, |
||||
|
Amount: big.NewInt(int64(i + 1)), |
||||
|
LoadAmount: big.NewInt(int64(i + 1)), |
||||
|
EthBlockNum: blocks[i%len(blocks)].EthBlockNum, |
||||
|
Type: randomTxType(i), |
||||
|
} |
||||
|
if i%4 == 0 { |
||||
|
tx.BatchNum = batches[i%len(batches)].BatchNum |
||||
|
} |
||||
|
} else { |
||||
|
continue |
||||
|
} |
||||
|
if i < nUserTxs { |
||||
|
var from, to common.Account |
||||
|
var err error |
||||
|
if i%2 == 0 { |
||||
|
from, err = randomAccount(i, true, userAddr, accounts) |
||||
|
if err != nil { |
||||
|
panic(err) |
||||
|
} |
||||
|
to, err = randomAccount(i, false, userAddr, accounts) |
||||
|
if err != nil { |
||||
|
panic(err) |
||||
|
} |
||||
|
} else { |
||||
|
from, err = randomAccount(i, false, userAddr, accounts) |
||||
|
if err != nil { |
||||
|
panic(err) |
||||
|
} |
||||
|
to, err = randomAccount(i, true, userAddr, accounts) |
||||
|
if err != nil { |
||||
|
panic(err) |
||||
|
} |
||||
|
} |
||||
|
tx.FromIdx = from.Idx |
||||
|
tx.FromEthAddr = from.EthAddr |
||||
|
tx.FromBJJ = from.PublicKey |
||||
|
tx.ToIdx = to.Idx |
||||
|
userTxs = append(userTxs, tx) |
||||
|
} else { |
||||
|
from, err := randomAccount(i, false, userAddr, accounts) |
||||
|
if err != nil { |
||||
|
panic(err) |
||||
|
} |
||||
|
to, err := randomAccount(i, false, userAddr, accounts) |
||||
|
if err != nil { |
||||
|
panic(err) |
||||
|
} |
||||
|
tx.FromIdx = from.Idx |
||||
|
tx.FromEthAddr = from.EthAddr |
||||
|
tx.FromBJJ = from.PublicKey |
||||
|
tx.ToIdx = to.Idx |
||||
|
othersTxs = append(othersTxs, tx) |
||||
|
} |
||||
|
} |
||||
|
return userTxs, othersTxs |
||||
|
} |
||||
|
|
||||
|
// GenL2Txs generates L2 txs. WARNING: This is meant for DB/API testing, and may not be fully consistent with the protocol.
|
||||
|
func GenL2Txs( |
||||
|
fromIdx int, |
||||
|
totalTxs, nUserTxs int, |
||||
|
userAddr *ethCommon.Address, |
||||
|
accounts []common.Account, |
||||
|
tokens []common.Token, |
||||
|
blocks []common.Block, |
||||
|
batches []common.Batch, |
||||
|
) ([]common.L2Tx, []common.L2Tx) { |
||||
|
if totalTxs < nUserTxs { |
||||
|
panic("totalTxs must be greater than userTxs") |
||||
|
} |
||||
|
userTxs := []common.L2Tx{} |
||||
|
othersTxs := []common.L2Tx{} |
||||
|
for i := 0; i < totalTxs; i++ { |
||||
|
tx := common.L2Tx{ |
||||
|
TxID: common.TxID(common.Hash([]byte("L2_" + strconv.Itoa(fromIdx+i)))), |
||||
|
BatchNum: batches[i%len(batches)].BatchNum, |
||||
|
Position: i, |
||||
|
//nolint:gomnd
|
||||
|
Amount: big.NewInt(int64(i + 1)), |
||||
|
//nolint:gomnd
|
||||
|
Fee: common.FeeSelector(i % 256), |
||||
|
Nonce: common.Nonce(i + 1), |
||||
|
EthBlockNum: blocks[i%len(blocks)].EthBlockNum, |
||||
|
Type: randomTxType(i), |
||||
|
} |
||||
|
if i < nUserTxs { |
||||
|
var from, to common.Account |
||||
|
var err error |
||||
|
if i%2 == 0 { |
||||
|
from, err = randomAccount(i, true, userAddr, accounts) |
||||
|
if err != nil { |
||||
|
panic(err) |
||||
|
} |
||||
|
to, err = randomAccount(i, false, userAddr, accounts) |
||||
|
if err != nil { |
||||
|
panic(err) |
||||
|
} |
||||
|
} else { |
||||
|
from, err = randomAccount(i, false, userAddr, accounts) |
||||
|
if err != nil { |
||||
|
panic(err) |
||||
|
} |
||||
|
to, err = randomAccount(i, true, userAddr, accounts) |
||||
|
if err != nil { |
||||
|
panic(err) |
||||
|
} |
||||
|
} |
||||
|
tx.FromIdx = from.Idx |
||||
|
tx.ToIdx = to.Idx |
||||
|
userTxs = append(userTxs, tx) |
||||
|
} else { |
||||
|
from, err := randomAccount(i, false, userAddr, accounts) |
||||
|
if err != nil { |
||||
|
panic(err) |
||||
|
} |
||||
|
to, err := randomAccount(i, false, userAddr, accounts) |
||||
|
if err != nil { |
||||
|
panic(err) |
||||
|
} |
||||
|
tx.FromIdx = from.Idx |
||||
|
tx.ToIdx = to.Idx |
||||
|
othersTxs = append(othersTxs, tx) |
||||
|
} |
||||
|
} |
||||
|
return userTxs, othersTxs |
||||
|
} |
||||
|
|
||||
|
// GenCoordinators generates coordinators. WARNING: This is meant for DB/API testing, and may not be fully consistent with the protocol.
|
||||
|
func GenCoordinators(nCoords int, blocks []common.Block) []common.Coordinator { |
||||
|
coords := []common.Coordinator{} |
||||
|
for i := 0; i < nCoords; i++ { |
||||
|
coords = append(coords, common.Coordinator{ |
||||
|
EthBlockNum: blocks[i%len(blocks)].EthBlockNum, |
||||
|
Forger: ethCommon.BigToAddress(big.NewInt(int64(i))), |
||||
|
Withdraw: ethCommon.BigToAddress(big.NewInt(int64(i))), |
||||
|
URL: "https://foo.bar", |
||||
|
}) |
||||
|
} |
||||
|
return coords |
||||
|
} |
||||
|
|
||||
|
// GenBids generates bids. WARNING: This is meant for DB/API testing, and may not be fully consistent with the protocol.
|
||||
|
func GenBids(nBids int, blocks []common.Block, coords []common.Coordinator) []common.Bid { |
||||
|
bids := []common.Bid{} |
||||
|
for i := 0; i < nBids; i++ { |
||||
|
bids = append(bids, common.Bid{ |
||||
|
SlotNum: common.SlotNum(i), |
||||
|
BidValue: big.NewInt(int64(i)), |
||||
|
EthBlockNum: blocks[i%len(blocks)].EthBlockNum, |
||||
|
ForgerAddr: coords[i%len(blocks)].Forger, |
||||
|
}) |
||||
|
} |
||||
|
return bids |
||||
|
} |
||||
|
|
||||
|
func randomAccount(seed int, userAccount bool, userAddr *ethCommon.Address, accs []common.Account) (common.Account, error) { |
||||
|
i := seed % len(accs) |
||||
|
firstI := i |
||||
|
for { |
||||
|
acc := accs[i] |
||||
|
if userAccount && *userAddr == acc.EthAddr { |
||||
|
return acc, nil |
||||
|
} |
||||
|
if !userAccount && (userAddr == nil || *userAddr != acc.EthAddr) { |
||||
|
return acc, nil |
||||
|
} |
||||
|
i++ |
||||
|
i = i % len(accs) |
||||
|
if i == firstI { |
||||
|
return acc, errors.New("Didnt found any account matchinng the criteria") |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
func randomTxType(seed int) common.TxType { |
||||
|
//nolint:gomnd
|
||||
|
switch seed % 11 { |
||||
|
case 0: |
||||
|
return common.TxTypeExit |
||||
|
//nolint:gomnd
|
||||
|
case 1: |
||||
|
return common.TxTypeWithdrawn |
||||
|
//nolint:gomnd
|
||||
|
case 2: |
||||
|
return common.TxTypeTransfer |
||||
|
//nolint:gomnd
|
||||
|
case 3: |
||||
|
return common.TxTypeDeposit |
||||
|
//nolint:gomnd
|
||||
|
case 4: |
||||
|
return common.TxTypeCreateAccountDeposit |
||||
|
//nolint:gomnd
|
||||
|
case 5: |
||||
|
return common.TxTypeCreateAccountDepositTransfer |
||||
|
//nolint:gomnd
|
||||
|
case 6: |
||||
|
return common.TxTypeDepositTransfer |
||||
|
//nolint:gomnd
|
||||
|
case 7: |
||||
|
return common.TxTypeForceTransfer |
||||
|
//nolint:gomnd
|
||||
|
case 8: |
||||
|
return common.TxTypeForceExit |
||||
|
//nolint:gomnd
|
||||
|
case 9: |
||||
|
return common.TxTypeTransferToEthAddr |
||||
|
//nolint:gomnd
|
||||
|
case 10: |
||||
|
return common.TxTypeTransferToBJJ |
||||
|
default: |
||||
|
return common.TxTypeTransfer |
||||
|
} |
||||
|
} |
||||