diff options
author | Wei-Ning Huang <w@dexon.org> | 2018-10-30 20:00:24 +0800 |
---|---|---|
committer | Wei-Ning Huang <w@dexon.org> | 2019-03-12 12:19:09 +0800 |
commit | 886d114e4f10d9737367c1e4f3e91b50e3b310e7 (patch) | |
tree | 324a1c97db8d456b68f283fc4cc5a92c117a3c5c | |
parent | be36e03b7ae08fc6993881a66773ad20a48be4fc (diff) | |
download | dexon-886d114e4f10d9737367c1e4f3e91b50e3b310e7.tar dexon-886d114e4f10d9737367c1e4f3e91b50e3b310e7.tar.gz dexon-886d114e4f10d9737367c1e4f3e91b50e3b310e7.tar.bz2 dexon-886d114e4f10d9737367c1e4f3e91b50e3b310e7.tar.lz dexon-886d114e4f10d9737367c1e4f3e91b50e3b310e7.tar.xz dexon-886d114e4f10d9737367c1e4f3e91b50e3b310e7.tar.zst dexon-886d114e4f10d9737367c1e4f3e91b50e3b310e7.zip |
cmd: monkey: create random accounts to simulate network traffic
-rw-r--r-- | cmd/monkey/monkey.go | 130 |
1 files changed, 96 insertions, 34 deletions
diff --git a/cmd/monkey/monkey.go b/cmd/monkey/monkey.go index c155e5cf0..76a306c7a 100644 --- a/cmd/monkey/monkey.go +++ b/cmd/monkey/monkey.go @@ -21,8 +21,10 @@ package main import ( "context" + "crypto/ecdsa" "flag" "fmt" + "math" "math/big" "math/rand" "time" @@ -34,60 +36,120 @@ import ( ) var key = flag.String("key", "", "private key path") -var endpoint = flag.String("endpoint", "http://localhost:8545", "JSON RPC endpoint") -var tps = flag.Int("tps", 10, "transactions per second") +var endpoint = flag.String("endpoint", "http://127.0.0.1:8545", "JSON RPC endpoint") +var n = flag.Int("n", 100, "number of random accounts") + +type Monkey struct { + client *ethclient.Client + source *ecdsa.PrivateKey + keys []*ecdsa.PrivateKey + networkID *big.Int +} -func randomAddress() common.Address { - x := common.Address{} - for i := 0; i < 20; i++ { - x[i] = byte(rand.Int31() % 16) +func New(ep string, source *ecdsa.PrivateKey, num int) *Monkey { + client, err := ethclient.Dial(ep) + if err != nil { + panic(err) } - return x -} -func main() { - flag.Parse() + var keys []*ecdsa.PrivateKey - privKey, err := crypto.LoadECDSA(*key) + for i := 0; i < num; i++ { + key, err := crypto.GenerateKey() + if err != nil { + panic(err) + } + keys = append(keys, key) + } + + networkID, err := client.NetworkID(context.Background()) if err != nil { panic(err) } - client, err := ethclient.Dial(*endpoint) + return &Monkey{ + client: client, + source: source, + keys: keys, + networkID: networkID, + } +} + +func (m *Monkey) transfer( + key *ecdsa.PrivateKey, toAddress common.Address, amount *big.Int, nonce uint64) { + + if nonce == math.MaxUint64 { + var err error + address := crypto.PubkeyToAddress(key.PublicKey) + nonce, err = m.client.PendingNonceAt(context.Background(), address) + if err != nil { + panic(err) + } + } + + tx := types.NewTransaction( + uint64(nonce), + toAddress, + amount, + uint64(21000), + big.NewInt(1e9), + nil) + + signer := types.NewEIP155Signer(m.networkID) + tx, err := types.SignTx(tx, signer, key) if err != nil { panic(err) } - address := crypto.PubkeyToAddress(privKey.PublicKey) + fmt.Println("Sending TX", "fullhash", tx.Hash().String()) - nonce, err := client.PendingNonceAt(context.Background(), address) + err = m.client.SendTransaction(context.Background(), tx) if err != nil { panic(err) } +} - for { - tx := types.NewTransaction( - nonce, - randomAddress(), - big.NewInt(10), - uint64(21000), - big.NewInt(1e9), - nil) - - signer := types.NewEIP155Signer(big.NewInt(237)) - tx, err = types.SignTx(tx, signer, privKey) - if err != nil { - panic(err) - } +func (m *Monkey) Distribute() { + fmt.Println("Distributing DEX to random accounts ...") + address := crypto.PubkeyToAddress(m.source.PublicKey) + nonce, err := m.client.PendingNonceAt(context.Background(), address) + if err != nil { + panic(err) + } - err := client.SendTransaction(context.Background(), tx) - if err != nil { - panic(err) + for _, key := range m.keys { + address := crypto.PubkeyToAddress(key.PublicKey) + amount := new(big.Int) + amount.SetString("1000000000000000000", 10) + m.transfer(m.source, address, amount, nonce) + nonce += 1 + } + time.Sleep(60 * time.Second) +} + +func (m *Monkey) Crazy() { + fmt.Println("Performing random transfers ...") + nonce := uint64(0) + for { + fmt.Println("nonce", nonce) + for _, key := range m.keys { + to := crypto.PubkeyToAddress(m.keys[rand.Int()%len(m.keys)].PublicKey) + m.transfer(key, to, big.NewInt(1), nonce) } + nonce += 1 + time.Sleep(500 * time.Millisecond) + } +} - fmt.Println("Sending TX", "fullhash", tx.Hash().String()) +func main() { + flag.Parse() - nonce++ - time.Sleep(time.Duration(float32(time.Second) / float32(*tps))) + privKey, err := crypto.LoadECDSA(*key) + if err != nil { + panic(err) } + + m := New(*endpoint, privKey, *n) + m.Distribute() + m.Crazy() } |