kylefang/bitcoin-indexer
1
0
Fork 0
mirror of https://github.com/alexgo-io/bitcoin-indexer.git synced 2026-06-13 16:19:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: change forking behavior

Browse Source
This commit is contained in:
Ludo Galabru
2023-05-17 22:45:13 -04:00
parent d1c83e7ea7
commit 4c100147c2
1 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
components/chainhook-event-observer/src/indexer/stacks/blocks_pool.rs
View File

@@ -13,6 +13,7 @@ use std::collections::{hash_map::Entry, BTreeMap, BTreeSet, HashMap, HashSet};
pub struct StacksBlockPool {
canonical_fork_id: usize,
number_of_blocks_since_last_reorg: u16,
orphans: BTreeSet<BlockIdentifier>,
block_store: HashMap<BlockIdentifier, StacksBlockData>,
forks: BTreeMap<usize, ChainSegment>,
@@ -28,6 +29,7 @@ impl StacksBlockPool {
forks.insert(0, ChainSegment::new());
StacksBlockPool {
canonical_fork_id: 0,
number_of_blocks_since_last_reorg: 0,
block_store: HashMap::new(),
orphans: BTreeSet::new(),
forks,
@@ -214,7 +216,9 @@ impl StacksBlockPool {
_ => return Ok(None),
};
self.collect_and_prune_confirmed_blocks(&mut chain_event, ctx);
if self.number_of_blocks_since_last_reorg > 4 {
self.collect_and_prune_confirmed_blocks(&mut chain_event, ctx);
}
Ok(Some(chain_event))
}
@@ -844,6 +848,8 @@ impl StacksBlockPool {
if let Ok(divergence) = canonical_segment.try_identify_divergence(other_segment, false, ctx)
{
if divergence.block_ids_to_rollback.is_empty() {
self.number_of_blocks_since_last_reorg =
self.number_of_blocks_since_last_reorg.saturating_add(1);
let mut new_blocks = vec![];
for i in 0..divergence.block_ids_to_apply.len() {
let block_identifier = &divergence.block_ids_to_apply[i];
@@ -893,6 +899,7 @@ impl StacksBlockPool {
},
));
} else {
self.number_of_blocks_since_last_reorg = 0;
let blocks_to_rollback = divergence
.block_ids_to_rollback
.iter()