60 lines
2.0 KiB
JavaScript
60 lines
2.0 KiB
JavaScript
"use strict";
|
|
import { ethers } from "ethers";
|
|
import { version } from "./_version";
|
|
const logger = new ethers.utils.Logger(version);
|
|
// @TODO: Keep a per-NonceManager pool of sent but unmined transactions for
|
|
// rebroadcasting, in case we overrun the transaction pool
|
|
export class NonceManager extends ethers.Signer {
|
|
constructor(signer) {
|
|
logger.checkNew(new.target, NonceManager);
|
|
super();
|
|
this._deltaCount = 0;
|
|
ethers.utils.defineReadOnly(this, "signer", signer);
|
|
}
|
|
connect(provider) {
|
|
return new NonceManager(this.signer.connect(provider));
|
|
}
|
|
getAddress() {
|
|
return this.signer.getAddress();
|
|
}
|
|
getTransactionCount(blockTag) {
|
|
if (blockTag === "pending") {
|
|
if (!this._initialPromise) {
|
|
this._initialPromise = this.signer.getTransactionCount("pending");
|
|
}
|
|
const deltaCount = this._deltaCount;
|
|
return this._initialPromise.then((initial) => (initial + deltaCount));
|
|
}
|
|
return this.signer.getTransactionCount(blockTag);
|
|
}
|
|
setTransactionCount(transactionCount) {
|
|
this._initialPromise = Promise.resolve(transactionCount).then((nonce) => {
|
|
return ethers.BigNumber.from(nonce).toNumber();
|
|
});
|
|
this._deltaCount = 0;
|
|
}
|
|
incrementTransactionCount(count) {
|
|
this._deltaCount += (count ? count : 1);
|
|
}
|
|
signMessage(message) {
|
|
return this.signer.signMessage(message);
|
|
;
|
|
}
|
|
signTransaction(transaction) {
|
|
return this.signer.signTransaction(transaction);
|
|
}
|
|
sendTransaction(transaction) {
|
|
if (transaction.nonce == null) {
|
|
transaction = ethers.utils.shallowCopy(transaction);
|
|
transaction.nonce = this.getTransactionCount("pending");
|
|
this.incrementTransactionCount();
|
|
}
|
|
else {
|
|
this.setTransactionCount(transaction.nonce);
|
|
}
|
|
return this.signer.sendTransaction(transaction).then((tx) => {
|
|
return tx;
|
|
});
|
|
}
|
|
}
|