Skip to content

Commit

Permalink
compensate for overhead in storing remembered_txs by adding 100 bytes…
Browse files Browse the repository at this point in the history
… to each transaction
  • Loading branch information
forrestv committed Nov 5, 2012
1 parent 62c10bd commit e1ea409
Showing 1 changed file with 8 additions and 8 deletions.
16 changes: 8 additions & 8 deletions p2pool/p2p.py
Expand Up @@ -183,16 +183,16 @@ def update_remote_view_of_my_mining_txs(before, after):
added = set(after) - set(before)
removed = set(before) - set(after)
if added:
self.remote_remembered_txs_size += sum(bitcoin_data.tx_type.packed_size(after[x]) for x in added)
self.remote_remembered_txs_size += sum(100 + bitcoin_data.tx_type.packed_size(after[x]) for x in added)
assert self.remote_remembered_txs_size <= self.max_remembered_txs_size
fragment(self.send_remember_tx, tx_hashes=[x for x in added if x in self.remote_tx_hashes], txs=[after[x] for x in added if x not in self.remote_tx_hashes])
if removed:
self.send_forget_tx(tx_hashes=list(removed))
self.remote_remembered_txs_size -= sum(bitcoin_data.tx_type.packed_size(before[x]) for x in removed)
self.remote_remembered_txs_size -= sum(100 + bitcoin_data.tx_type.packed_size(before[x]) for x in removed)
watch_id2 = self.node.mining_txs_var.transitioned.watch(update_remote_view_of_my_mining_txs)
self.connection_lost_event.watch(lambda: self.node.mining_txs_var.transitioned.unwatch(watch_id2))

self.remote_remembered_txs_size += sum(bitcoin_data.tx_type.packed_size(x) for x in self.node.mining_txs_var.value.values())
self.remote_remembered_txs_size += sum(100 + bitcoin_data.tx_type.packed_size(x) for x in self.node.mining_txs_var.value.values())
assert self.remote_remembered_txs_size <= self.max_remembered_txs_size
fragment(self.send_remember_tx, tx_hashes=[], txs=self.node.mining_txs_var.value.values())

Expand Down Expand Up @@ -271,7 +271,7 @@ def sendShares(self, shares, tracker, known_txs, include_txs_with=[]):

hashes_to_send = [x for x in tx_hashes if x not in self.node.mining_txs_var.value and x in known_txs]

new_remote_remembered_txs_size = self.remote_remembered_txs_size + sum(bitcoin_data.tx_type.packed_size(known_txs[x]) for x in hashes_to_send)
new_remote_remembered_txs_size = self.remote_remembered_txs_size + sum(100 + bitcoin_data.tx_type.packed_size(known_txs[x]) for x in hashes_to_send)
if new_remote_remembered_txs_size > self.max_remembered_txs_size:
raise ValueError('shares have too many txs')
self.remote_remembered_txs_size = new_remote_remembered_txs_size
Expand All @@ -283,7 +283,7 @@ def sendShares(self, shares, tracker, known_txs, include_txs_with=[]):
if self.other_version >= 8:
res = self.send_forget_tx(tx_hashes=hashes_to_send)

self.remote_remembered_txs_size -= sum(bitcoin_data.tx_type.packed_size(known_txs[x]) for x in hashes_to_send)
self.remote_remembered_txs_size -= sum(100 + bitcoin_data.tx_type.packed_size(known_txs[x]) for x in hashes_to_send)

return res

Expand Down Expand Up @@ -362,7 +362,7 @@ def handle_remember_tx(self, tx_hashes, txs):
return

self.remembered_txs[tx_hash] = tx
self.remembered_txs_size += bitcoin_data.tx_type.packed_size(tx)
self.remembered_txs_size += 100 + bitcoin_data.tx_type.packed_size(tx)
new_known_txs = dict(self.node.known_txs_var.value)
warned = False
for tx in txs:
Expand All @@ -377,7 +377,7 @@ def handle_remember_tx(self, tx_hashes, txs):
warned = True

self.remembered_txs[tx_hash] = tx
self.remembered_txs_size += bitcoin_data.tx_type.packed_size(tx)
self.remembered_txs_size += 100 + bitcoin_data.tx_type.packed_size(tx)
new_known_txs[tx_hash] = tx
self.node.known_txs_var.set(new_known_txs)
if self.remembered_txs_size >= self.max_remembered_txs_size:
Expand All @@ -387,7 +387,7 @@ def handle_remember_tx(self, tx_hashes, txs):
])
def handle_forget_tx(self, tx_hashes):
for tx_hash in tx_hashes:
self.remembered_txs_size -= bitcoin_data.tx_type.packed_size(self.remembered_txs[tx_hash])
self.remembered_txs_size -= 100 + bitcoin_data.tx_type.packed_size(self.remembered_txs[tx_hash])
assert self.remembered_txs_size >= 0
del self.remembered_txs[tx_hash]

Expand Down

0 comments on commit e1ea409

Please sign in to comment.