Skip to content

Commit 34d1d74

Browse files
author
abel
committed
(fix) Changes to gas limit estimation to reduce a little bit gas consumption
1 parent 41be560 commit 34d1d74

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
@@ -343,7 +343,7 @@ async def get_account(self, address: str) -> Optional[account_pb2.EthAccount]:
343343
try:
344344
metadata = await self.load_cookie(type="chain")
345345
account_any = (await self.stubAuth.Account(
346-
auth_query.QueryAccountRequest.__call__(address=address), metadata=metadata
346+
auth_query.QueryAccountRequest(address=address), metadata=metadata
347347
)).account
348348
account = account_pb2.EthAccount()
349349
if account_any.Is(account.DESCRIPTOR):
@@ -380,7 +380,7 @@ async def simulate_tx(
380380
try:
381381
req = tx_service.SimulateRequest(tx_bytes=tx_byte)
382382
metadata = await self.load_cookie(type="chain")
383-
return await self.stubTx.Simulate.__call__(req, metadata=metadata), True
383+
return await self.stubTx.Simulate(request=req, metadata=metadata), True
384384
except grpc.RpcError as err:
385385
return err, False
386386

@@ -389,23 +389,23 @@ async def send_tx_sync_mode(self, tx_byte: bytes) -> abci_type.TxResponse:
389389
tx_bytes=tx_byte, mode=tx_service.BroadcastMode.BROADCAST_MODE_SYNC
390390
)
391391
metadata = await self.load_cookie(type="chain")
392-
result = await self.stubTx.BroadcastTx.__call__(req, metadata=metadata)
392+
result = await self.stubTx.BroadcastTx(request=req, metadata=metadata)
393393
return result.tx_response
394394

395395
async def send_tx_async_mode(self, tx_byte: bytes) -> abci_type.TxResponse:
396396
req = tx_service.BroadcastTxRequest(
397397
tx_bytes=tx_byte, mode=tx_service.BroadcastMode.BROADCAST_MODE_ASYNC
398398
)
399399
metadata = await self.load_cookie(type="chain")
400-
result = await self.stubTx.BroadcastTx.__call__(req, metadata=metadata)
400+
result = await self.stubTx.BroadcastTx(request=req, metadata=metadata)
401401
return result.tx_response
402402

403403
async def send_tx_block_mode(self, tx_byte: bytes) -> abci_type.TxResponse:
404404
req = tx_service.BroadcastTxRequest(
405405
tx_bytes=tx_byte, mode=tx_service.BroadcastMode.BROADCAST_MODE_BLOCK
406406
)
407407
metadata = await self.load_cookie(type="chain")
408-
result = await self.stubTx.BroadcastTx.__call__(req, metadata=metadata)
408+
result = await self.stubTx.BroadcastTx(request=req, metadata=metadata)
409409
return result.tx_response
410410

411411
async def get_chain_id(self) -> str:
@@ -673,7 +673,7 @@ async def stream_spot_markets(self, **kwargs):
673673
market_ids=kwargs.get("market_ids")
674674
)
675675
metadata = await self.load_cookie(type="exchange")
676-
return self.stubSpotExchange.StreamMarkets.__call__(req, metadata=metadata)
676+
return self.stubSpotExchange.StreamMarkets(request=req, metadata=metadata)
677677

678678
async def get_spot_orderbookV2(self, market_id: str):
679679
req = spot_exchange_rpc_pb.OrderbookV2Request(market_id=market_id)
@@ -730,12 +730,12 @@ async def get_spot_trades(self, **kwargs):
730730
async def stream_spot_orderbook_snapshot(self, market_ids: List[str]):
731731
req = spot_exchange_rpc_pb.StreamOrderbookV2Request(market_ids=market_ids)
732732
metadata = await self.load_cookie(type="exchange")
733-
return self.stubSpotExchange.StreamOrderbookV2.__call__(req, metadata=metadata)
733+
return self.stubSpotExchange.StreamOrderbookV2(request=req, metadata=metadata)
734734

735735
async def stream_spot_orderbook_update(self, market_ids: List[str]):
736736
req = spot_exchange_rpc_pb.StreamOrderbookUpdateRequest(market_ids=market_ids)
737737
metadata = await self.load_cookie(type="exchange")
738-
return self.stubSpotExchange.StreamOrderbookUpdate.__call__(req, metadata=metadata)
738+
return self.stubSpotExchange.StreamOrderbookUpdate(request=req, metadata=metadata)
739739

740740
async def stream_spot_orders(self, market_id: str, **kwargs):
741741
req = spot_exchange_rpc_pb.StreamOrdersRequest(
@@ -744,7 +744,7 @@ async def stream_spot_orders(self, market_id: str, **kwargs):
744744
subaccount_id=kwargs.get("subaccount_id"),
745745
)
746746
metadata = await self.load_cookie(type="exchange")
747-
return self.stubSpotExchange.StreamOrders.__call__(req, metadata=metadata)
747+
return self.stubSpotExchange.StreamOrders(request=req, metadata=metadata)
748748

749749
async def stream_historical_spot_orders(self, market_id: str, **kwargs):
750750
req = spot_exchange_rpc_pb.StreamOrdersHistoryRequest(
@@ -756,7 +756,7 @@ async def stream_historical_spot_orders(self, market_id: str, **kwargs):
756756
execution_types=kwargs.get("execution_types")
757757
)
758758
metadata = await self.load_cookie(type="exchange")
759-
return self.stubSpotExchange.StreamOrdersHistory.__call__(req, metadata=metadata)
759+
return self.stubSpotExchange.StreamOrdersHistory(request=req, metadata=metadata)
760760

761761
async def stream_historical_derivative_orders(self, market_id: str, **kwargs):
762762
req = derivative_exchange_rpc_pb.StreamOrdersHistoryRequest(
@@ -768,7 +768,7 @@ async def stream_historical_derivative_orders(self, market_id: str, **kwargs):
768768
execution_types=kwargs.get("execution_types")
769769
)
770770
metadata = await self.load_cookie(type="exchange")
771-
return self.stubDerivativeExchange.StreamOrdersHistory.__call__(req, metadata=metadata)
771+
return self.stubDerivativeExchange.StreamOrdersHistory(request=req, metadata=metadata)
772772

773773
async def stream_spot_trades(self, **kwargs):
774774
req = spot_exchange_rpc_pb.StreamTradesRequest(
@@ -781,7 +781,7 @@ async def stream_spot_trades(self, **kwargs):
781781
execution_types=kwargs.get("execution_types"),
782782
)
783783
metadata = await self.load_cookie(type="exchange")
784-
return self.stubSpotExchange.StreamTrades.__call__(req, metadata=metadata)
784+
return self.stubSpotExchange.StreamTrades(request=req, metadata=metadata)
785785

786786
async def get_spot_subaccount_orders(self, subaccount_id: str, **kwargs):
787787
req = spot_exchange_rpc_pb.SubaccountOrdersListRequest(
@@ -821,7 +821,7 @@ async def stream_derivative_markets(self, **kwargs):
821821
market_ids=kwargs.get("market_ids")
822822
)
823823
metadata = await self.load_cookie(type="exchange")
824-
return self.stubDerivativeExchange.StreamMarket.__call__(req, metadata=metadata)
824+
return self.stubDerivativeExchange.StreamMarket(request=req, metadata=metadata)
825825

826826
async def get_derivative_orderbook(self, market_id: str):
827827
req = derivative_exchange_rpc_pb.OrderbookV2Request(market_id=market_id)
@@ -887,16 +887,12 @@ async def get_derivative_trades(self, **kwargs):
887887
async def stream_derivative_orderbook_snapshot(self, market_ids: List[str]):
888888
req = derivative_exchange_rpc_pb.StreamOrderbookV2Request(market_ids=market_ids)
889889
metadata = await self.load_cookie(type="exchange")
890-
return self.stubDerivativeExchange.StreamOrderbookV2.__call__(
891-
req, metadata=metadata
892-
)
890+
return self.stubDerivativeExchange.StreamOrderbookV2(request=req, metadata=metadata)
893891

894892
async def stream_derivative_orderbook_update(self, market_ids: List[str]):
895893
req = derivative_exchange_rpc_pb.StreamOrderbookUpdateRequest(market_ids=market_ids)
896894
metadata = await self.load_cookie(type="exchange")
897-
return self.stubDerivativeExchange.StreamOrderbookUpdate.__call__(
898-
req, metadata=metadata
899-
)
895+
return self.stubDerivativeExchange.StreamOrderbookUpdate(request=req, metadata=metadata)
900896

901897
async def stream_derivative_orders(self, market_id: str, **kwargs):
902898
req = derivative_exchange_rpc_pb.StreamOrdersRequest(
@@ -905,7 +901,7 @@ async def stream_derivative_orders(self, market_id: str, **kwargs):
905901
subaccount_id=kwargs.get("subaccount_id"),
906902
)
907903
metadata = await self.load_cookie(type="exchange")
908-
return self.stubDerivativeExchange.StreamOrders.__call__(req, metadata=metadata)
904+
return self.stubDerivativeExchange.StreamOrders(request=req, metadata=metadata)
909905

910906
async def stream_derivative_trades(self, **kwargs):
911907
req = derivative_exchange_rpc_pb.StreamTradesRequest(
@@ -920,7 +916,7 @@ async def stream_derivative_trades(self, **kwargs):
920916
execution_types=kwargs.get("execution_types"),
921917
)
922918
metadata = await self.load_cookie(type="exchange")
923-
return self.stubDerivativeExchange.StreamTrades.__call__(req, metadata=metadata)
919+
return self.stubDerivativeExchange.StreamTrades(request=req, metadata=metadata)
924920

925921
async def get_derivative_positions(self, **kwargs):
926922
req = derivative_exchange_rpc_pb.PositionsRequest(
@@ -942,9 +938,7 @@ async def stream_derivative_positions(self, **kwargs):
942938
subaccount_ids=kwargs.get("subaccount_ids"),
943939
)
944940
metadata = await self.load_cookie(type="exchange")
945-
return self.stubDerivativeExchange.StreamPositions.__call__(
946-
req, metadata=metadata
947-
)
941+
return self.stubDerivativeExchange.StreamPositions(request=req, metadata=metadata)
948942

949943
async def get_derivative_liquidable_positions(self, **kwargs):
950944
req = derivative_exchange_rpc_pb.LiquidablePositionsRequest(
@@ -1020,7 +1014,7 @@ async def stream_account_portfolio(self, account_address: str, **kwargs):
10201014
type=kwargs.get("type")
10211015
)
10221016
metadata = await self.load_cookie(type="exchange")
1023-
return self.stubPortfolio.StreamAccountPortfolio.__call__(req, metadata=metadata)
1017+
return self.stubPortfolio.StreamAccountPortfolio(request=req, metadata=metadata)
10241018

10251019
async def composer(self):
10261020
return Composer(

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)