mirror of https://github.com/poanetwork/gecko.git
114 lines
3.1 KiB
Go
114 lines
3.1 KiB
Go
// (c) 2019-2020, Ava Labs, Inc. All rights reserved.
|
|
// See the file LICENSE for licensing terms.
|
|
|
|
package main
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/ava-labs/salticidae-go"
|
|
|
|
"github.com/ava-labs/gecko/genesis"
|
|
"github.com/ava-labs/gecko/ids"
|
|
"github.com/ava-labs/gecko/networking"
|
|
"github.com/ava-labs/gecko/utils/crypto"
|
|
"github.com/ava-labs/gecko/utils/formatting"
|
|
"github.com/ava-labs/gecko/utils/timer"
|
|
"github.com/ava-labs/gecko/vms/avm"
|
|
"github.com/ava-labs/gecko/vms/platformvm"
|
|
"github.com/ava-labs/gecko/xputtest/avmwallet"
|
|
)
|
|
|
|
// benchmark an instance of the avm
|
|
func (n *network) benchmarkAVM(chain *platformvm.CreateChainTx) {
|
|
genesisBytes := chain.GenesisData
|
|
wallet, err := avmwallet.NewWallet(n.log, n.networkID, chain.ID(), config.AvaTxFee)
|
|
n.log.AssertNoError(err)
|
|
|
|
cb58 := formatting.CB58{}
|
|
keyStr := genesis.Keys[config.Key]
|
|
n.log.AssertNoError(cb58.FromString(keyStr))
|
|
factory := crypto.FactorySECP256K1R{}
|
|
sk, err := factory.ToPrivateKey(cb58.Bytes)
|
|
n.log.AssertNoError(err)
|
|
wallet.ImportKey(sk.(*crypto.PrivateKeySECP256K1R))
|
|
|
|
codec := wallet.Codec()
|
|
|
|
genesis := avm.Genesis{}
|
|
n.log.AssertNoError(codec.Unmarshal(genesisBytes, &genesis))
|
|
|
|
genesisTx := genesis.Txs[0]
|
|
tx := avm.Tx{
|
|
UnsignedTx: &genesisTx.CreateAssetTx,
|
|
}
|
|
txBytes, err := codec.Marshal(&tx)
|
|
n.log.AssertNoError(err)
|
|
tx.Initialize(txBytes)
|
|
|
|
for _, utxo := range tx.UTXOs() {
|
|
wallet.AddUTXO(utxo)
|
|
}
|
|
|
|
assetID := genesisTx.ID()
|
|
|
|
n.log.AssertNoError(wallet.GenerateTxs(config.NumTxs, assetID))
|
|
|
|
go n.log.RecoverAndPanic(func() { n.IssueAVM(chain.ID(), assetID, wallet) })
|
|
}
|
|
|
|
// issue transactions to the instance of the avm funded by the provided wallet
|
|
func (n *network) IssueAVM(chainID ids.ID, assetID ids.ID, wallet *avmwallet.Wallet) {
|
|
n.log.Debug("Issuing with %d", wallet.Balance(assetID))
|
|
numAccepted := 0
|
|
numPending := 0
|
|
|
|
n.decided <- ids.ID{}
|
|
|
|
// track the last second of transactions
|
|
meter := timer.TimedMeter{Duration: time.Second}
|
|
for d := range n.decided {
|
|
// display the TPS every 1000 txs
|
|
if numAccepted%1000 == 0 {
|
|
n.log.Info("TPS: %d", meter.Ticks())
|
|
}
|
|
|
|
// d is the ID of the tx that was accepted
|
|
if !d.IsZero() {
|
|
meter.Tick()
|
|
n.log.Debug("Finalized %s", d)
|
|
numAccepted++
|
|
numPending--
|
|
}
|
|
|
|
// Issue all the txs that we can right now
|
|
for numPending < config.MaxOutstandingTxs && wallet.Balance(assetID) > 0 && numAccepted+numPending < config.NumTxs {
|
|
tx := wallet.NextTx()
|
|
n.log.AssertTrue(tx != nil, "Tx creation failed")
|
|
|
|
// send the IssueTx message
|
|
it, err := n.build.IssueTx(chainID, tx.Bytes())
|
|
n.log.AssertNoError(err)
|
|
ds := it.DataStream()
|
|
ba := salticidae.NewByteArrayMovedFromDataStream(ds, false)
|
|
newMsg := salticidae.NewMsgMovedFromByteArray(networking.IssueTx, ba, false)
|
|
|
|
n.conn.GetNet().SendMsg(newMsg, n.conn)
|
|
|
|
ds.Free()
|
|
ba.Free()
|
|
newMsg.Free()
|
|
|
|
numPending++
|
|
n.log.Debug("Sent tx, pending = %d, accepted = %d", numPending, numAccepted)
|
|
}
|
|
|
|
// If we are done issuing txs, return from the function
|
|
if numAccepted+numPending >= config.NumTxs {
|
|
n.log.Info("done with test")
|
|
net.ec.Stop()
|
|
return
|
|
}
|
|
}
|
|
}
|