Skip to content

Commit aeeb1d3

Browse files
author
abel
committed
(fix) Changes to gas limit estimation to reduce a little bit gas consumption
1 parent 5cd4e06 commit aeeb1d3

File tree

2 files changed

+24
-30
lines changed

2 files changed

+24
-30
lines changed

pyinjective/async_client.py

Lines changed: 19 additions & 25 deletions
Original file line numberDiff line numberDiff line change
@@ -302,7 +302,7 @@ async def get_account(self, address: str) -> Optional[account_pb2.EthAccount]:
302302
try:
303303
metadata = await self.load_cookie(type="chain")
304304
account_any = (await self.stubAuth.Account(
305-
auth_query.QueryAccountRequest.__call__(address=address), metadata=metadata
305+
auth_query.QueryAccountRequest(address=address), metadata=metadata
306306
)).account
307307
account = account_pb2.EthAccount()
308308
if account_any.Is(account.DESCRIPTOR):
@@ -339,7 +339,7 @@ async def simulate_tx(
339339
try:
340340
req = tx_service.SimulateRequest(tx_bytes=tx_byte)
341341
metadata = await self.load_cookie(type="chain")
342-
return await self.stubTx.Simulate.__call__(req, metadata=metadata), True
342+
return await self.stubTx.Simulate(request=req, metadata=metadata), True
343343
except grpc.RpcError as err:
344344
return err, False
345345

@@ -348,23 +348,23 @@ async def send_tx_sync_mode(self, tx_byte: bytes) -> abci_type.TxResponse:
348348
tx_bytes=tx_byte, mode=tx_service.BroadcastMode.BROADCAST_MODE_SYNC
349349
)
350350
metadata = await self.load_cookie(type="chain")
351-
result = await self.stubTx.BroadcastTx.__call__(req, metadata=metadata)
351+
result = await self.stubTx.BroadcastTx(request=req, metadata=metadata)
352352
return result.tx_response
353353

354354
async def send_tx_async_mode(self, tx_byte: bytes) -> abci_type.TxResponse:
355355
req = tx_service.BroadcastTxRequest(
356356
tx_bytes=tx_byte, mode=tx_service.BroadcastMode.BROADCAST_MODE_ASYNC
357357
)
358358
metadata = await self.load_cookie(type="chain")
359-
result = await self.stubTx.BroadcastTx.__call__(req, metadata=metadata)
359+
result = await self.stubTx.BroadcastTx(request=req, metadata=metadata)
360360
return result.tx_response
361361

362362
async def send_tx_block_mode(self, tx_byte: bytes) -> abci_type.TxResponse:
363363
req = tx_service.BroadcastTxRequest(
364364
tx_bytes=tx_byte, mode=tx_service.BroadcastMode.BROADCAST_MODE_BLOCK
365365
)
366366
metadata = await self.load_cookie(type="chain")
367-
result = await self.stubTx.BroadcastTx.__call__(req, metadata=metadata)
367+
result = await self.stubTx.BroadcastTx(request=req, metadata=metadata)
368368
return result.tx_response
369369

370370
async def get_chain_id(self) -> str:
@@ -632,7 +632,7 @@ async def stream_spot_markets(self, **kwargs):
632632
market_ids=kwargs.get("market_ids")
633633
)
634634
metadata = await self.load_cookie(type="exchange")
635-
return self.stubSpotExchange.StreamMarkets.__call__(req, metadata=metadata)
635+
return self.stubSpotExchange.StreamMarkets(request=req, metadata=metadata)
636636

637637
async def get_spot_orderbookV2(self, market_id: str):
638638
req = spot_exchange_rpc_pb.OrderbookV2Request(market_id=market_id)
@@ -689,12 +689,12 @@ async def get_spot_trades(self, **kwargs):
689689
async def stream_spot_orderbook_snapshot(self, market_ids: List[str]):
690690
req = spot_exchange_rpc_pb.StreamOrderbookV2Request(market_ids=market_ids)
691691
metadata = await self.load_cookie(type="exchange")
692-
return self.stubSpotExchange.StreamOrderbookV2.__call__(req, metadata=metadata)
692+
return self.stubSpotExchange.StreamOrderbookV2(request=req, metadata=metadata)
693693

694694
async def stream_spot_orderbook_update(self, market_ids: List[str]):
695695
req = spot_exchange_rpc_pb.StreamOrderbookUpdateRequest(market_ids=market_ids)
696696
metadata = await self.load_cookie(type="exchange")
697-
return self.stubSpotExchange.StreamOrderbookUpdate.__call__(req, metadata=metadata)
697+
return self.stubSpotExchange.StreamOrderbookUpdate(request=req, metadata=metadata)
698698

699699
async def stream_spot_orders(self, market_id: str, **kwargs):
700700
req = spot_exchange_rpc_pb.StreamOrdersRequest(
@@ -703,7 +703,7 @@ async def stream_spot_orders(self, market_id: str, **kwargs):
703703
subaccount_id=kwargs.get("subaccount_id"),
704704
)
705705
metadata = await self.load_cookie(type="exchange")
706-
return self.stubSpotExchange.StreamOrders.__call__(req, metadata=metadata)
706+
return self.stubSpotExchange.StreamOrders(request=req, metadata=metadata)
707707

708708
async def stream_historical_spot_orders(self, market_id: str, **kwargs):
709709
req = spot_exchange_rpc_pb.StreamOrdersHistoryRequest(
@@ -715,7 +715,7 @@ async def stream_historical_spot_orders(self, market_id: str, **kwargs):
715715
execution_types=kwargs.get("execution_types")
716716
)
717717
metadata = await self.load_cookie(type="exchange")
718-
return self.stubSpotExchange.StreamOrdersHistory.__call__(req, metadata=metadata)
718+
return self.stubSpotExchange.StreamOrdersHistory(request=req, metadata=metadata)
719719

720720
async def stream_historical_derivative_orders(self, market_id: str, **kwargs):
721721
req = derivative_exchange_rpc_pb.StreamOrdersHistoryRequest(
@@ -727,7 +727,7 @@ async def stream_historical_derivative_orders(self, market_id: str, **kwargs):
727727
execution_types=kwargs.get("execution_types")
728728
)
729729
metadata = await self.load_cookie(type="exchange")
730-
return self.stubDerivativeExchange.StreamOrdersHistory.__call__(req, metadata=metadata)
730+
return self.stubDerivativeExchange.StreamOrdersHistory(request=req, metadata=metadata)
731731

732732
async def stream_spot_trades(self, **kwargs):
733733
req = spot_exchange_rpc_pb.StreamTradesRequest(
@@ -740,7 +740,7 @@ async def stream_spot_trades(self, **kwargs):
740740
execution_types=kwargs.get("execution_types"),
741741
)
742742
metadata = await self.load_cookie(type="exchange")
743-
return self.stubSpotExchange.StreamTrades.__call__(req, metadata=metadata)
743+
return self.stubSpotExchange.StreamTrades(request=req, metadata=metadata)
744744

745745
async def get_spot_subaccount_orders(self, subaccount_id: str, **kwargs):
746746
req = spot_exchange_rpc_pb.SubaccountOrdersListRequest(
@@ -780,7 +780,7 @@ async def stream_derivative_markets(self, **kwargs):
780780
market_ids=kwargs.get("market_ids")
781781
)
782782
metadata = await self.load_cookie(type="exchange")
783-
return self.stubDerivativeExchange.StreamMarket.__call__(req, metadata=metadata)
783+
return self.stubDerivativeExchange.StreamMarket(request=req, metadata=metadata)
784784

785785
async def get_derivative_orderbook(self, market_id: str):
786786
req = derivative_exchange_rpc_pb.OrderbookV2Request(market_id=market_id)
@@ -846,16 +846,12 @@ async def get_derivative_trades(self, **kwargs):
846846
async def stream_derivative_orderbook_snapshot(self, market_ids: List[str]):
847847
req = derivative_exchange_rpc_pb.StreamOrderbookV2Request(market_ids=market_ids)
848848
metadata = await self.load_cookie(type="exchange")
849-
return self.stubDerivativeExchange.StreamOrderbookV2.__call__(
850-
req, metadata=metadata
851-
)
849+
return self.stubDerivativeExchange.StreamOrderbookV2(request=req, metadata=metadata)
852850

853851
async def stream_derivative_orderbook_update(self, market_ids: List[str]):
854852
req = derivative_exchange_rpc_pb.StreamOrderbookUpdateRequest(market_ids=market_ids)
855853
metadata = await self.load_cookie(type="exchange")
856-
return self.stubDerivativeExchange.StreamOrderbookUpdate.__call__(
857-
req, metadata=metadata
858-
)
854+
return self.stubDerivativeExchange.StreamOrderbookUpdate(request=req, metadata=metadata)
859855

860856
async def stream_derivative_orders(self, market_id: str, **kwargs):
861857
req = derivative_exchange_rpc_pb.StreamOrdersRequest(
@@ -864,7 +860,7 @@ async def stream_derivative_orders(self, market_id: str, **kwargs):
864860
subaccount_id=kwargs.get("subaccount_id"),
865861
)
866862
metadata = await self.load_cookie(type="exchange")
867-
return self.stubDerivativeExchange.StreamOrders.__call__(req, metadata=metadata)
863+
return self.stubDerivativeExchange.StreamOrders(request=req, metadata=metadata)
868864

869865
async def stream_derivative_trades(self, **kwargs):
870866
req = derivative_exchange_rpc_pb.StreamTradesRequest(
@@ -879,7 +875,7 @@ async def stream_derivative_trades(self, **kwargs):
879875
execution_types=kwargs.get("execution_types"),
880876
)
881877
metadata = await self.load_cookie(type="exchange")
882-
return self.stubDerivativeExchange.StreamTrades.__call__(req, metadata=metadata)
878+
return self.stubDerivativeExchange.StreamTrades(request=req, metadata=metadata)
883879

884880
async def get_derivative_positions(self, **kwargs):
885881
req = derivative_exchange_rpc_pb.PositionsRequest(
@@ -901,9 +897,7 @@ async def stream_derivative_positions(self, **kwargs):
901897
subaccount_ids=kwargs.get("subaccount_ids"),
902898
)
903899
metadata = await self.load_cookie(type="exchange")
904-
return self.stubDerivativeExchange.StreamPositions.__call__(
905-
req, metadata=metadata
906-
)
900+
return self.stubDerivativeExchange.StreamPositions(request=req, metadata=metadata)
907901

908902
async def get_derivative_liquidable_positions(self, **kwargs):
909903
req = derivative_exchange_rpc_pb.LiquidablePositionsRequest(
@@ -979,4 +973,4 @@ async def stream_account_portfolio(self, account_address: str, **kwargs):
979973
type=kwargs.get("type")
980974
)
981975
metadata = await self.load_cookie(type="exchange")
982-
return self.stubPortfolio.StreamAccountPortfolio.__call__(req, metadata=metadata)
976+
return self.stubPortfolio.StreamAccountPortfolio(request=req, metadata=metadata)

pyinjective/core/broadcaster.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -287,8 +287,8 @@ async def configure_gas_fee_for_transaction(
287287

288288

289289
class MessageBasedTransactionFeeCalculator(TransactionFeeCalculator):
290-
DEFAULT_GAS_LIMIT = 400_000
291-
DEFAULT_EXCHANGE_GAS_LIMIT = 200_000
290+
DEFAULT_GAS_LIMIT = 150_000
291+
DEFAULT_EXCHANGE_GAS_LIMIT = 100_000
292292

293293
def __init__(
294294
self,
@@ -313,11 +313,11 @@ def __init__(
313313
Decimal("2.5") * self._base_gas_limit
314314
),
315315
(
316-
lambda message: "wasm" in self._message_type(message=message),
316+
lambda message: "wasm." in self._message_type(message=message),
317317
Decimal("1.5") * self._base_gas_limit
318318
),
319319
(
320-
lambda message: "exchange" in self._message_type(message=message),
320+
lambda message: "exchange." in self._message_type(message=message),
321321
Decimal("1") * self._base_exchange_gas_limit
322322
),
323323
(
@@ -353,7 +353,7 @@ def _message_type(self, message: any_pb2.Any) -> str:
353353

354354
def _is_governance_message(self, message: any_pb2.Any) -> bool:
355355
message_type = self._message_type(message=message)
356-
return "gov" in message_type and ("MsgDeposit" in message_type or "MsgSubmitProposal" in message_type)
356+
return "gov." in message_type and ("MsgDeposit" in message_type or "MsgSubmitProposal" in message_type)
357357

358358
def _calculate_gas_limit(self, messages: List[any_pb2.Any]) -> int:
359359
total_gas_limit = Decimal("0")

0 commit comments

Comments
 (0)