Add estimate gas with block overrides (#98)

KyberNetwork · Nov 26, 2024 · 3e4764f · 3e4764f
1 parent 5f3cf28
commit 3e4764f
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/pkg/eth/simulator.go b/pkg/eth/simulator.go
@@ -45,6 +45,20 @@ func (s *Simulator) CallContract(
 	return s.gethClient.CallContract(ctx, msg, blockNumber, overrides)
 }
 
+func (s *Simulator) EstimateGasWithBlockOverrides(
+	ctx context.Context, msg ethereum.CallMsg, blockNumber *big.Int,
+	overrides *map[common.Address]gethclient.OverrideAccount,
+	blockOverrides gethclient.BlockOverrides,
+) (uint64, error) {
+	var gas hexutil.Uint64
+	err := s.c.CallContext(
+		ctx, &gas, "eth_estimateGas", mev.ToCallArg(msg),
+		toBlockNumArg(blockNumber), overrides, blockOverrides,
+	)
+
+	return uint64(gas), err
+}
+
 func toBlockNumArg(number *big.Int) string {
 	if number == nil {
 		return "latest"