cic-internal-integration/apps/cic-eth/cic_eth/eth/util.py

# standard imports
import logging

# external imports
from chainlib.eth.gas import RPCGasOracle
from hexathon import strip_0x

# local imports
from cic_eth.db.models.gas_cache import GasCache
from cic_eth.encode import tx_normalize
from cic_eth.db.models.base import SessionBase

MAXIMUM_FEE_UNITS = 8000000

logg = logging.getLogger(__name__)


class MaxGasOracle(RPCGasOracle):

    def get_fee_units(self, code=None):
        return MAXIMUM_FEE_UNITS


class CacheGasOracle(MaxGasOracle):
    """Returns a previously recorded value for fee unit expenditure for a contract call, if it exists. Otherwise returns max units.

    :todo: instead of max units, connect a pluggable gas heuristics engine.
    """

    def __init__(self, conn, address, method=None, session=None, min_price=None, id_generator=None):
        super(CacheGasOracle, self).__init__(conn, code_callback=self.get_fee_units, min_price=min_price, id_generator=id_generator)
        self.value = None
        self.address = address
        self.method = method

        address = tx_normalize.executable_address(address)
        session = SessionBase.bind_session(session)
        q = session.query(GasCache)
        q = q.filter(GasCache.address==address)
        if method != None:
            method = strip_0x(method)
            q = q.filter(GasCache.method==method)
        o = q.first()
        if o != None:
            self.value = int(o.value)

        SessionBase.release_session(session)


    def get_fee_units(self, code=None):
        if self.value != None:
            logg.debug('found stored gas unit value {} for address {} method {}'.format(self.value, self.address, self.method))
            return self.value
        return super(CacheGasOracle, self).get_fee_units(code=code)