Merge pull request #1319 from denniswon/r3_txs_throttle
[r3 patch] throttle transactions with configuration per-network levelpull/1360/head r3-20190814.0
Leo Chen
5 years ago
committed by
GitHub
commit
060181b0ef
@ -0,0 +1 @@ |
|||||||
|
The wallet program is the demo wallet which talks to Harmony devnet for various kinds of operations. For detail, please compile and execute ./bin/wallet. |
||||||
@ -0,0 +1,86 @@ |
|||||||
|
package main |
||||||
|
|
||||||
|
const ( |
||||||
|
defaultWalletIni = `[default] |
||||||
|
bootnode = /ip4/100.26.90.187/tcp/9874/p2p/Qmdfjtk6hPoyrH1zVD9PEH4zfWLo38dP2mDvvKXfh3tnEv |
||||||
|
bootnode = /ip4/54.213.43.194/tcp/9874/p2p/QmZJJx6AdaoEkGLrYG4JeLCKeCKDjnFz2wfHNHxAqFSGA9 |
||||||
|
bootnode = /ip4/13.113.101.219/tcp/12019/p2p/QmQayinFSgMMw5cSpDUiD9pQ2WeP6WNmGxpZ6ou3mdVFJX |
||||||
|
bootnode = /ip4/99.81.170.167/tcp/12019/p2p/QmRVbTpEYup8dSaURZfF6ByrMTSKa4UyUzJhSjahFzRqNj |
||||||
|
shards = 4 |
||||||
|
|
||||||
|
[default.shard0.rpc] |
||||||
|
rpc = l0.t.hmny.io:14555 |
||||||
|
rpc = s0.t.hmny.io:14555 |
||||||
|
|
||||||
|
[default.shard1.rpc] |
||||||
|
rpc = l1.t.hmny.io:14555 |
||||||
|
rpc = s1.t.hmny.io:14555 |
||||||
|
|
||||||
|
[default.shard2.rpc] |
||||||
|
rpc = l2.t.hmny.io:14555 |
||||||
|
rpc = s2.t.hmny.io:14555 |
||||||
|
|
||||||
|
[default.shard3.rpc] |
||||||
|
rpc = l3.t.hmny.io:14555 |
||||||
|
rpc = s3.t.hmny.io:14555 |
||||||
|
|
||||||
|
[local] |
||||||
|
bootnode = /ip4/127.0.0.1/tcp/19876/p2p/Qmc1V6W7BwX8Ugb42Ti8RnXF1rY5PF7nnZ6bKBryCgi6cv |
||||||
|
shards = 2 |
||||||
|
|
||||||
|
[local.shard0.rpc] |
||||||
|
rpc = 127.0.0.1:14555 |
||||||
|
rpc = 127.0.0.1:14557 |
||||||
|
rpc = 127.0.0.1:14559 |
||||||
|
|
||||||
|
[local.shard1.rpc] |
||||||
|
rpc = 127.0.0.1:14556 |
||||||
|
rpc = 127.0.0.1:14558 |
||||||
|
rpc = 127.0.0.1:14560 |
||||||
|
|
||||||
|
[beta] |
||||||
|
bootnode = /ip4/54.213.43.194/tcp/9868/p2p/QmZJJx6AdaoEkGLrYG4JeLCKeCKDjnFz2wfHNHxAqFSGA9 |
||||||
|
bootnode = /ip4/100.26.90.187/tcp/9868/p2p/Qmdfjtk6hPoyrH1zVD9PEH4zfWLo38dP2mDvvKXfh3tnEv |
||||||
|
bootnode = /ip4/13.113.101.219/tcp/12018/p2p/QmQayinFSgMMw5cSpDUiD9pQ2WeP6WNmGxpZ6ou3mdVFJX |
||||||
|
shards = 4 |
||||||
|
|
||||||
|
[beta.shard0.rpc] |
||||||
|
rpc = l0.b.hmny.io:14555 |
||||||
|
rpc = s0.b.hmny.io:14555 |
||||||
|
|
||||||
|
[beta.shard1.rpc] |
||||||
|
rpc = l1.b.hmny.io:14555 |
||||||
|
rpc = s1.b.hmny.io:14555 |
||||||
|
|
||||||
|
[beta.shard2.rpc] |
||||||
|
rpc = l2.b.hmny.io:14555 |
||||||
|
rpc = s2.b.hmny.io:14555 |
||||||
|
|
||||||
|
[beta.shard3.rpc] |
||||||
|
rpc = l3.b.hmny.io:14555 |
||||||
|
rpc = s3.b.hmny.io:14555 |
||||||
|
|
||||||
|
[pangaea] |
||||||
|
bootnode = /ip4/100.26.90.187/tcp/9867/p2p/Qmdfjtk6hPoyrH1zVD9PEH4zfWLo38dP2mDvvKXfh3tnEv |
||||||
|
bootnode = /ip4/54.213.43.194/tcp/9867/p2p/QmZJJx6AdaoEkGLrYG4JeLCKeCKDjnFz2wfHNHxAqFSGA9 |
||||||
|
bootnode = /ip4/13.113.101.219/tcp/9867/p2p/QmQayinFSgMMw5cSpDUiD9pQ2WeP6WNmGxpZ6ou3mdVFJX |
||||||
|
bootnode = /ip4/99.81.170.167/tcp/9867/p2p/QmRVbTpEYup8dSaURZfF6ByrMTSKa4UyUzJhSjahFzRqNj |
||||||
|
shards = 4 |
||||||
|
|
||||||
|
[pangaea.shard0.rpc] |
||||||
|
rpc = l0.p.hmny.io:14555 |
||||||
|
rpc = s0.p.hmny.io:14555 |
||||||
|
|
||||||
|
[pangaea.shard1.rpc] |
||||||
|
rpc = l1.p.hmny.io:14555 |
||||||
|
rpc = s1.p.hmny.io:14555 |
||||||
|
|
||||||
|
[pangaea.shard2.rpc] |
||||||
|
rpc = l2.p.hmny.io:14555 |
||||||
|
rpc = s2.p.hmny.io:14555 |
||||||
|
|
||||||
|
[pangaea.shard3.rpc] |
||||||
|
rpc = l3.p.hmny.io:14555 |
||||||
|
rpc = s3.p.hmny.io:14555 |
||||||
|
` |
||||||
|
) |
||||||
@ -0,0 +1,465 @@ |
|||||||
|
package main |
||||||
|
|
||||||
|
import ( |
||||||
|
"flag" |
||||||
|
"fmt" |
||||||
|
"io/ioutil" |
||||||
|
"math/big" |
||||||
|
"math/rand" |
||||||
|
"os" |
||||||
|
"path" |
||||||
|
"sync" |
||||||
|
"time" |
||||||
|
|
||||||
|
"github.com/ethereum/go-ethereum/common" |
||||||
|
"github.com/ethereum/go-ethereum/log" |
||||||
|
"github.com/harmony-one/harmony/accounts" |
||||||
|
"github.com/harmony-one/harmony/accounts/keystore" |
||||||
|
"github.com/harmony-one/harmony/api/client" |
||||||
|
clientService "github.com/harmony-one/harmony/api/client/service" |
||||||
|
proto_node "github.com/harmony-one/harmony/api/proto/node" |
||||||
|
"github.com/harmony-one/harmony/common/denominations" |
||||||
|
"github.com/harmony-one/harmony/core" |
||||||
|
"github.com/harmony-one/harmony/core/types" |
||||||
|
common2 "github.com/harmony-one/harmony/internal/common" |
||||||
|
nodeconfig "github.com/harmony-one/harmony/internal/configs/node" |
||||||
|
"github.com/harmony-one/harmony/internal/ctxerror" |
||||||
|
"github.com/harmony-one/harmony/internal/shardchain" |
||||||
|
"github.com/harmony-one/harmony/internal/utils" |
||||||
|
"github.com/harmony-one/harmony/node" |
||||||
|
"github.com/harmony-one/harmony/p2p" |
||||||
|
p2p_host "github.com/harmony-one/harmony/p2p/host" |
||||||
|
"github.com/harmony-one/harmony/p2p/p2pimpl" |
||||||
|
) |
||||||
|
|
||||||
|
var ( |
||||||
|
version string |
||||||
|
builtBy string |
||||||
|
builtAt string |
||||||
|
commit string |
||||||
|
nextNonce uint64 |
||||||
|
) |
||||||
|
|
||||||
|
func printVersion(me string) { |
||||||
|
fmt.Fprintf(os.Stderr, "Harmony (C) 2019. %v, version %v-%v (%v %v)\n", path.Base(me), version, commit, builtBy, builtAt) |
||||||
|
os.Exit(0) |
||||||
|
} |
||||||
|
|
||||||
|
// AccountState includes the balance and nonce of an account
|
||||||
|
type AccountState struct { |
||||||
|
balance *big.Int |
||||||
|
nonce uint64 |
||||||
|
} |
||||||
|
|
||||||
|
const ( |
||||||
|
rpcRetry = 3 |
||||||
|
defaultConfigFile = ".hmy/wallet.ini" |
||||||
|
defaultProfile = "default" |
||||||
|
keystoreDir = ".hmy/keystore" |
||||||
|
) |
||||||
|
|
||||||
|
var ( |
||||||
|
// Transfer subcommands
|
||||||
|
transferCommand = flag.NewFlagSet("transfer", flag.ExitOnError) |
||||||
|
transferSenderPtr = transferCommand.String("from", "0", "Specify the sender account address or index") |
||||||
|
transferReceiverPtr = transferCommand.String("to", "", "Specify the receiver account") |
||||||
|
transferAmountPtr = transferCommand.Float64("amount", 0, "Specify the amount to transfer") |
||||||
|
transferGasPricePtr = transferCommand.Uint64("gasPrice", 0, "Specify the gas price amount. Unit is Nano.") |
||||||
|
transferShardIDPtr = transferCommand.Int("shardID", 0, "Specify the shard ID for the transfer") |
||||||
|
transferInputDataPtr = transferCommand.String("inputData", "", "Base64-encoded input data to embed in the transaction") |
||||||
|
transferSenderPassPtr = transferCommand.String("pass", "", "Passphrase of the sender's private key") |
||||||
|
) |
||||||
|
|
||||||
|
var ( |
||||||
|
walletProfile *utils.WalletProfile |
||||||
|
ks *keystore.KeyStore |
||||||
|
) |
||||||
|
|
||||||
|
// setupLog setup log for verbose output
|
||||||
|
func setupLog() { |
||||||
|
// enable logging for wallet
|
||||||
|
h := log.StreamHandler(os.Stdout, log.TerminalFormat(true)) |
||||||
|
log.Root().SetHandler(h) |
||||||
|
} |
||||||
|
|
||||||
|
// The main wallet program entrance. Note the this wallet program is for demo-purpose only. It does not implement
|
||||||
|
// the secure storage of keys.
|
||||||
|
func main() { |
||||||
|
rand.Seed(int64(time.Now().Nanosecond())) |
||||||
|
|
||||||
|
// Verify that a subcommand has been provided
|
||||||
|
// os.Arg[0] is the main command
|
||||||
|
// os.Arg[1] will be the subcommand
|
||||||
|
if len(os.Args) < 2 { |
||||||
|
fmt.Println("Usage:") |
||||||
|
fmt.Println(" wallet -p profile <action> <params>") |
||||||
|
fmt.Println(" -p profile - Specify the profile of the wallet, either testnet/devnet or others configured. Default is: testnet") |
||||||
|
fmt.Println(" The profile is in file:", defaultConfigFile) |
||||||
|
fmt.Println() |
||||||
|
fmt.Println("Actions:") |
||||||
|
fmt.Println(" 1. stressTest - Stress test transactions with corner cases.") |
||||||
|
os.Exit(1) |
||||||
|
} |
||||||
|
|
||||||
|
ARG: |
||||||
|
for { |
||||||
|
lastArg := os.Args[len(os.Args)-1] |
||||||
|
switch lastArg { |
||||||
|
case "--verbose": |
||||||
|
setupLog() |
||||||
|
os.Args = os.Args[:len(os.Args)-1] |
||||||
|
default: |
||||||
|
break ARG |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
var profile string |
||||||
|
if os.Args[1] == "-p" { |
||||||
|
profile = os.Args[2] |
||||||
|
os.Args = os.Args[2:] |
||||||
|
} else { |
||||||
|
profile = defaultProfile |
||||||
|
} |
||||||
|
if len(os.Args) == 1 { |
||||||
|
fmt.Println("Missing action") |
||||||
|
flag.PrintDefaults() |
||||||
|
os.Exit(1) |
||||||
|
} |
||||||
|
|
||||||
|
// create new keystore backend
|
||||||
|
scryptN := keystore.StandardScryptN |
||||||
|
scryptP := keystore.StandardScryptP |
||||||
|
ks = keystore.NewKeyStore(keystoreDir, scryptN, scryptP) |
||||||
|
|
||||||
|
// Switch on the subcommand
|
||||||
|
switch os.Args[1] { |
||||||
|
case "-version": |
||||||
|
printVersion(os.Args[0]) |
||||||
|
case "stressTest": |
||||||
|
readProfile(profile) |
||||||
|
processStressTestCommand() |
||||||
|
default: |
||||||
|
fmt.Printf("Unknown action: %s\n", os.Args[1]) |
||||||
|
flag.PrintDefaults() |
||||||
|
os.Exit(1) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
//go:generate go run ../../../scripts/wallet_embed_ini_files.go
|
||||||
|
|
||||||
|
func readProfile(profile string) { |
||||||
|
fmt.Printf("Using %s profile for wallet\n", profile) |
||||||
|
|
||||||
|
// try to load .hmy/wallet.ini from filesystem
|
||||||
|
// use default_wallet_ini if .hmy/wallet.ini doesn't exist
|
||||||
|
var err error |
||||||
|
var iniBytes []byte |
||||||
|
|
||||||
|
iniBytes, err = ioutil.ReadFile(defaultConfigFile) |
||||||
|
if err != nil { |
||||||
|
log.Debug(fmt.Sprintf("%s doesn't exist, using default ini\n", defaultConfigFile)) |
||||||
|
iniBytes = []byte(defaultWalletIni) |
||||||
|
} |
||||||
|
|
||||||
|
walletProfile, err = utils.ReadWalletProfile(iniBytes, profile) |
||||||
|
if err != nil { |
||||||
|
fmt.Printf("Read wallet profile error: %v\nExiting ...\n", err) |
||||||
|
os.Exit(2) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// createWalletNode creates wallet server node.
|
||||||
|
func createWalletNode() *node.Node { |
||||||
|
bootNodeAddrs, err := utils.StringsToAddrs(walletProfile.Bootnodes) |
||||||
|
if err != nil { |
||||||
|
panic(err) |
||||||
|
} |
||||||
|
utils.BootNodes = bootNodeAddrs |
||||||
|
shardID := 0 |
||||||
|
// dummy host for wallet
|
||||||
|
// TODO: potentially, too many dummy IP may flush out good IP address from our bootnode DHT
|
||||||
|
// we need to understand the impact to bootnode DHT with this dummy host ip added
|
||||||
|
self := p2p.Peer{IP: "127.0.0.1", Port: "6999"} |
||||||
|
priKey, _, _ := utils.GenKeyP2P("127.0.0.1", "6999") |
||||||
|
host, err := p2pimpl.NewHost(&self, priKey) |
||||||
|
if err != nil { |
||||||
|
panic(err) |
||||||
|
} |
||||||
|
chainDBFactory := &shardchain.MemDBFactory{} |
||||||
|
w := node.New(host, nil, chainDBFactory, false) |
||||||
|
w.Client = client.NewClient(w.GetHost(), uint32(shardID)) |
||||||
|
|
||||||
|
w.NodeConfig.SetRole(nodeconfig.ClientNode) |
||||||
|
w.ServiceManagerSetup() |
||||||
|
w.RunServices() |
||||||
|
return w |
||||||
|
} |
||||||
|
|
||||||
|
// ./bin/wallet -p local transfer
|
||||||
|
// --from one1uyshu2jgv8w465yc8kkny36thlt2wvel89tcmg
|
||||||
|
// --to one1spshr72utf6rwxseaz339j09ed8p6f8ke370zj
|
||||||
|
// --amount 1 --shardID 1
|
||||||
|
|
||||||
|
func processStressTestCommand() { |
||||||
|
/* |
||||||
|
|
||||||
|
Account 17: |
||||||
|
Address: one1spshr72utf6rwxseaz339j09ed8p6f8ke370zj |
||||||
|
Balance in Shard 0: x.xxx, nonce: 0 |
||||||
|
Balance in Shard 1: 0.0000, nonce: 0 |
||||||
|
Account 18: |
||||||
|
Address: one1uyshu2jgv8w465yc8kkny36thlt2wvel89tcmg |
||||||
|
Balance in Shard 0: 0.0000, nonce: 0 |
||||||
|
Balance in Shard 1: x.xxx, nonce: 0 |
||||||
|
|
||||||
|
*/ |
||||||
|
|
||||||
|
fmt.Println("Creating wallet node") |
||||||
|
walletNode := createWalletNode() |
||||||
|
|
||||||
|
senderAddress := common2.ParseAddr("one1uyshu2jgv8w465yc8kkny36thlt2wvel89tcmg") |
||||||
|
receiverAddress := common2.ParseAddr("one1spshr72utf6rwxseaz339j09ed8p6f8ke370zj") |
||||||
|
shardID := 1 |
||||||
|
|
||||||
|
fmt.Printf("Sender account: %s:\n", common2.MustAddressToBech32(senderAddress)) |
||||||
|
|
||||||
|
// default inputData
|
||||||
|
data := make([]byte, 0) |
||||||
|
|
||||||
|
gasLimit, _ := core.IntrinsicGas(data, false, true) |
||||||
|
|
||||||
|
gasPrice := 0 |
||||||
|
gasPriceBigInt := big.NewInt(int64(gasPrice)) |
||||||
|
gasPriceBigInt = gasPriceBigInt.Mul(gasPriceBigInt, big.NewInt(denominations.Nano)) |
||||||
|
|
||||||
|
fmt.Printf("gas limit: %d, gas price: %d", gasLimit, gasPriceBigInt.Uint64()) |
||||||
|
|
||||||
|
senderPass := "" |
||||||
|
|
||||||
|
var shardIDToAccountStateSender []*AccountState |
||||||
|
var shardIDToAccountStateReceiver []*AccountState |
||||||
|
var senderState *AccountState |
||||||
|
var receiverState *AccountState |
||||||
|
var retry uint32 |
||||||
|
|
||||||
|
for i := 0; ; i++ { |
||||||
|
for retry = 0; retry < 10; retry++ { |
||||||
|
shardIDToAccountStateSender = FetchBalance(senderAddress) |
||||||
|
shardIDToAccountStateReceiver = FetchBalance(receiverAddress) |
||||||
|
|
||||||
|
senderState = shardIDToAccountStateSender[shardID] |
||||||
|
receiverState = shardIDToAccountStateReceiver[shardID] |
||||||
|
|
||||||
|
if senderState.nonce == nextNonce { |
||||||
|
break |
||||||
|
} |
||||||
|
time.Sleep(3 * time.Second) |
||||||
|
fmt.Printf(".") |
||||||
|
} |
||||||
|
|
||||||
|
if retry == 10 { |
||||||
|
fmt.Printf("\nRetry expired. Num txs made: %d\n", i) |
||||||
|
break |
||||||
|
} |
||||||
|
|
||||||
|
nextNonce++ |
||||||
|
|
||||||
|
senderBalance := senderState.balance |
||||||
|
receiverBalance := receiverState.balance |
||||||
|
|
||||||
|
// amount 1/10th of the balance
|
||||||
|
amountBigInt := senderBalance.Div(senderBalance, big.NewInt(10)) |
||||||
|
|
||||||
|
fmt.Printf("\nsender: balance (shard %d: %s, nonce: %v)\n", shardID, convertBalanceIntoReadableFormat(senderBalance), senderState.nonce) |
||||||
|
fmt.Printf("receiver balance (shard %d: %s, nonce: %v)\n", shardID, convertBalanceIntoReadableFormat(receiverBalance), receiverState.nonce) |
||||||
|
|
||||||
|
// stop stress testing here after printing out the final balance
|
||||||
|
if i == 10 { |
||||||
|
break |
||||||
|
} |
||||||
|
|
||||||
|
tx := types.NewTransaction( |
||||||
|
senderState.nonce, receiverAddress, uint32(shardID), amountBigInt, |
||||||
|
gasLimit, gasPriceBigInt, data) |
||||||
|
|
||||||
|
account, _ := ks.Find(accounts.Account{Address: senderAddress}) |
||||||
|
|
||||||
|
ks.Unlock(account, senderPass) |
||||||
|
|
||||||
|
tx, _ = ks.SignTx(account, tx, nil) |
||||||
|
|
||||||
|
if err := submitTransaction(tx, walletNode, uint32(shardID)); err != nil { |
||||||
|
fmt.Println(ctxerror.New("submitTransaction failed", |
||||||
|
"tx", tx, "shardID", shardID).WithCause(err)) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
for shardID, balanceNonce := range FetchBalance(senderAddress) { |
||||||
|
fmt.Printf(" Final: Balance in Shard %d: %s, nonce: %v \n", shardID, convertBalanceIntoReadableFormat(balanceNonce.balance), balanceNonce.nonce) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
func convertBalanceIntoReadableFormat(balance *big.Int) string { |
||||||
|
balance = balance.Div(balance, big.NewInt(denominations.Nano)) |
||||||
|
strBalance := fmt.Sprintf("%d", balance.Uint64()) |
||||||
|
|
||||||
|
bytes := []byte(strBalance) |
||||||
|
hasDecimal := false |
||||||
|
for i := 0; i < 11; i++ { |
||||||
|
if len(bytes)-1-i < 0 { |
||||||
|
bytes = append([]byte{'0'}, bytes...) |
||||||
|
} |
||||||
|
if bytes[len(bytes)-1-i] != '0' && i < 9 { |
||||||
|
hasDecimal = true |
||||||
|
} |
||||||
|
if i == 9 { |
||||||
|
newBytes := append([]byte{'.'}, bytes[len(bytes)-i:]...) |
||||||
|
bytes = append(bytes[:len(bytes)-i], newBytes...) |
||||||
|
} |
||||||
|
} |
||||||
|
zerosToRemove := 0 |
||||||
|
for i := 0; i < len(bytes); i++ { |
||||||
|
if hasDecimal { |
||||||
|
if bytes[len(bytes)-1-i] == '0' { |
||||||
|
bytes = bytes[:len(bytes)-1-i] |
||||||
|
i-- |
||||||
|
} else { |
||||||
|
break |
||||||
|
} |
||||||
|
} else { |
||||||
|
if zerosToRemove < 5 { |
||||||
|
bytes = bytes[:len(bytes)-1-i] |
||||||
|
i-- |
||||||
|
zerosToRemove++ |
||||||
|
} else { |
||||||
|
break |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
return string(bytes) |
||||||
|
} |
||||||
|
|
||||||
|
// FetchBalance fetches account balance of specified address from the Harmony network
|
||||||
|
func FetchBalance(address common.Address) []*AccountState { |
||||||
|
result := []*AccountState{} |
||||||
|
for shardID := 0; shardID < walletProfile.Shards; shardID++ { |
||||||
|
// Fill in nil pointers for each shard; nil represent failed balance fetch.
|
||||||
|
result = append(result, nil) |
||||||
|
} |
||||||
|
|
||||||
|
var wg sync.WaitGroup |
||||||
|
wg.Add(walletProfile.Shards) |
||||||
|
|
||||||
|
for shardID := 0; shardID < walletProfile.Shards; shardID++ { |
||||||
|
go func(shardID int) { |
||||||
|
defer wg.Done() |
||||||
|
balance := big.NewInt(0) |
||||||
|
var nonce uint64 |
||||||
|
result[uint32(shardID)] = &AccountState{balance, 0} |
||||||
|
|
||||||
|
var wgShard sync.WaitGroup |
||||||
|
wgShard.Add(len(walletProfile.RPCServer[shardID])) |
||||||
|
|
||||||
|
var mutexAccountState = &sync.Mutex{} |
||||||
|
|
||||||
|
for rpcServerID := 0; rpcServerID < len(walletProfile.RPCServer[shardID]); rpcServerID++ { |
||||||
|
go func(rpcServerID int) { |
||||||
|
for retry := 0; retry < rpcRetry; retry++ { |
||||||
|
|
||||||
|
server := walletProfile.RPCServer[shardID][rpcServerID] |
||||||
|
client, err := clientService.NewClient(server.IP, server.Port) |
||||||
|
if err != nil { |
||||||
|
continue |
||||||
|
} |
||||||
|
|
||||||
|
log.Debug("FetchBalance", "server", server) |
||||||
|
response, err := client.GetBalance(address) |
||||||
|
if err != nil { |
||||||
|
log.Info("failed to get balance, retrying ...") |
||||||
|
time.Sleep(200 * time.Millisecond) |
||||||
|
continue |
||||||
|
} |
||||||
|
log.Debug("FetchBalance", "response", response) |
||||||
|
respBalance := big.NewInt(0) |
||||||
|
respBalance.SetBytes(response.Balance) |
||||||
|
|
||||||
|
mutexAccountState.Lock() |
||||||
|
if balance.Cmp(respBalance) < 0 { |
||||||
|
balance.SetBytes(response.Balance) |
||||||
|
nonce = response.Nonce |
||||||
|
} |
||||||
|
mutexAccountState.Unlock() |
||||||
|
break |
||||||
|
} |
||||||
|
wgShard.Done() |
||||||
|
}(rpcServerID) |
||||||
|
} |
||||||
|
wgShard.Wait() |
||||||
|
|
||||||
|
result[shardID] = &AccountState{balance, nonce} |
||||||
|
}(shardID) |
||||||
|
} |
||||||
|
wg.Wait() |
||||||
|
return result |
||||||
|
} |
||||||
|
|
||||||
|
// GetFreeToken requests for token test token on each shard
|
||||||
|
func GetFreeToken(address common.Address) { |
||||||
|
for i := 0; i < walletProfile.Shards; i++ { |
||||||
|
// use the 1st server (leader) to make the getFreeToken call
|
||||||
|
server := walletProfile.RPCServer[i][0] |
||||||
|
client, err := clientService.NewClient(server.IP, server.Port) |
||||||
|
if err != nil { |
||||||
|
continue |
||||||
|
} |
||||||
|
|
||||||
|
log.Debug("GetFreeToken", "server", server) |
||||||
|
|
||||||
|
for retry := 0; retry < rpcRetry; retry++ { |
||||||
|
response, err := client.GetFreeToken(address) |
||||||
|
if err != nil { |
||||||
|
log.Info("failed to get free token, retrying ...") |
||||||
|
time.Sleep(200 * time.Millisecond) |
||||||
|
continue |
||||||
|
} |
||||||
|
log.Debug("GetFreeToken", "response", response) |
||||||
|
txID := common.Hash{} |
||||||
|
txID.SetBytes(response.TxId) |
||||||
|
fmt.Printf("Transaction Id requesting free token in shard %d: %s\n", i, txID.Hex()) |
||||||
|
break |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// clearKeystore deletes all data in the local keystore
|
||||||
|
func clearKeystore() { |
||||||
|
dir, err := ioutil.ReadDir(keystoreDir) |
||||||
|
if err != nil { |
||||||
|
panic("Failed to read keystore directory") |
||||||
|
} |
||||||
|
for _, d := range dir { |
||||||
|
subdir := path.Join([]string{keystoreDir, d.Name()}...) |
||||||
|
if err := os.RemoveAll(subdir); err != nil { |
||||||
|
fmt.Println(ctxerror.New("cannot remove directory", |
||||||
|
"path", subdir).WithCause(err)) |
||||||
|
} |
||||||
|
} |
||||||
|
fmt.Println("All existing accounts deleted...") |
||||||
|
} |
||||||
|
|
||||||
|
// submitTransaction submits the transaction to the Harmony network
|
||||||
|
func submitTransaction(tx *types.Transaction, walletNode *node.Node, shardID uint32) error { |
||||||
|
msg := proto_node.ConstructTransactionListMessageAccount(types.Transactions{tx}) |
||||||
|
clientGroup := p2p.NewClientGroupIDByShardID(p2p.ShardID(shardID)) |
||||||
|
|
||||||
|
err := walletNode.GetHost().SendMessageToGroups([]p2p.GroupID{clientGroup}, p2p_host.ConstructP2pMessage(byte(0), msg)) |
||||||
|
if err != nil { |
||||||
|
fmt.Printf("Error in SubmitTransaction: %v\n", err) |
||||||
|
return err |
||||||
|
} |
||||||
|
fmt.Printf("Transaction Id for shard %d: %s\n", int(shardID), tx.Hash().Hex()) |
||||||
|
// FIXME (leo): how to we know the tx was successful sent to the network
|
||||||
|
// this is a hacky way to wait for sometime
|
||||||
|
time.Sleep(3 * time.Second) |
||||||
|
return nil |
||||||
|
} |
||||||
Loading…
Reference in new issue