-
Notifications
You must be signed in to change notification settings - Fork 174
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #1125 from livepeer/nv/nonce-gap
gomod, eth: fix nonce gap issues
- Loading branch information
Showing
9 changed files
with
178 additions
and
84 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,141 @@ | ||
| package eth | ||
|
|
||
| import ( | ||
| "context" | ||
| "math/big" | ||
| "strings" | ||
|
|
||
| ethereum "github.com/ethereum/go-ethereum" | ||
| "github.com/ethereum/go-ethereum/accounts/abi" | ||
| "github.com/ethereum/go-ethereum/common" | ||
| "github.com/ethereum/go-ethereum/core/types" | ||
| "github.com/ethereum/go-ethereum/ethclient" | ||
| "github.com/golang/glog" | ||
| "github.com/livepeer/go-livepeer/eth/contracts" | ||
| ) | ||
|
|
||
| var abis = []string{ | ||
| contracts.BondingManagerABI, | ||
| contracts.ControllerABI, | ||
| contracts.LivepeerTokenABI, | ||
| contracts.LivepeerTokenFaucetABI, | ||
| contracts.MinterABI, | ||
| contracts.RoundsManagerABI, | ||
| contracts.ServiceRegistryABI, | ||
| contracts.TicketBrokerABI, | ||
| } | ||
|
|
||
| type Backend interface { | ||
| ethereum.ChainStateReader | ||
| ethereum.TransactionReader | ||
| ethereum.TransactionSender | ||
| ethereum.ContractCaller | ||
| ethereum.PendingContractCaller | ||
| ethereum.PendingStateReader | ||
| ethereum.GasEstimator | ||
| ethereum.GasPricer | ||
| ethereum.LogFilterer | ||
|
|
||
| ChainID(ctx context.Context) (*big.Int, error) | ||
| } | ||
|
|
||
| type backend struct { | ||
| *ethclient.Client | ||
| methods map[string]string | ||
| nonceManager *NonceManager | ||
| } | ||
|
|
||
| func NewBackend(client *ethclient.Client) (Backend, error) { | ||
| methods, err := makeMethodsMap() | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
|
|
||
| return &backend{ | ||
| client, | ||
| methods, | ||
| NewNonceManager(client), | ||
| }, nil | ||
| } | ||
|
|
||
| func (b *backend) PendingNonceAt(ctx context.Context, account common.Address) (uint64, error) { | ||
| b.nonceManager.Lock(account) | ||
| defer b.nonceManager.Unlock(account) | ||
|
|
||
| return b.nonceManager.Next(account) | ||
| } | ||
|
|
||
| func (b *backend) SendTransaction(ctx context.Context, tx *types.Transaction) error { | ||
| err := b.Client.SendTransaction(ctx, tx) | ||
| if err != nil { | ||
| return err | ||
| } | ||
|
|
||
| // update local nonce | ||
| msg, err := tx.AsMessage(types.HomesteadSigner{}) | ||
| if err != nil { | ||
| return err | ||
| } | ||
| sender := msg.From() | ||
| b.nonceManager.Lock(sender) | ||
| b.nonceManager.Update(sender, tx.Nonce()) | ||
| b.nonceManager.Unlock(sender) | ||
|
|
||
| data := tx.Data() | ||
| method, ok := b.methods[string(data[:4])] | ||
| if !ok { | ||
| method = "unknown" | ||
| } | ||
|
|
||
| if err != nil { | ||
| glog.Infof("\n%vEth Transaction%v\n\nInvoking transaction: \"%v\". \nTransaction Failed: %v\n\n%v\n", strings.Repeat("*", 30), strings.Repeat("*", 30), method, err, strings.Repeat("*", 75)) | ||
| return err | ||
| } | ||
|
|
||
| glog.Infof("\n%vEth Transaction%v\n\nInvoking transaction: \"%v\". Hash: \"%v\". \n\n%v\n", strings.Repeat("*", 30), strings.Repeat("*", 30), method, tx.Hash().String(), strings.Repeat("*", 75)) | ||
|
|
||
| return nil | ||
| } | ||
|
|
||
| func (b *backend) CallContract(ctx context.Context, msg ethereum.CallMsg, blockNumber *big.Int) ([]byte, error) { | ||
| return b.retryRemoteCall(func() ([]byte, error) { | ||
| return b.Client.CallContract(ctx, msg, blockNumber) | ||
| }) | ||
| } | ||
|
|
||
| func (b *backend) PendingCallContract(ctx context.Context, msg ethereum.CallMsg) ([]byte, error) { | ||
| return b.retryRemoteCall(func() ([]byte, error) { | ||
| return b.Client.PendingCallContract(ctx, msg) | ||
| }) | ||
| } | ||
|
|
||
| func (b *backend) retryRemoteCall(remoteCall func() ([]byte, error)) (out []byte, err error) { | ||
| count := 3 // consider making this a package-level global constant | ||
| retry := true // consider making this a package-level global constant | ||
|
|
||
| for i := 0; i < count && retry; i++ { | ||
| out, err = remoteCall() | ||
| if err != nil && (err.Error() == "EOF" || err.Error() == "tls: use of closed connection") { | ||
| glog.V(4).Infof("Retrying call to remote ethereum node") | ||
| } else { | ||
| retry = false | ||
| } | ||
| } | ||
|
|
||
| return out, err | ||
| } | ||
|
|
||
| func makeMethodsMap() (map[string]string, error) { | ||
| methods := make(map[string]string) | ||
| for _, ABI := range abis { | ||
| parsedAbi, err := abi.JSON(strings.NewReader(ABI)) | ||
| if err != nil { | ||
| return map[string]string{}, err | ||
| } | ||
| for _, m := range parsedAbi.Methods { | ||
| methods[string(m.ID())] = m.Name | ||
| } | ||
| } | ||
|
|
||
| return methods, nil | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.