1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
use crate::*;
use std::mem::size_of;
use core::ops::Range;
use indexmap::IndexMap;
use bitcoin::{Block, Script};
use crate::db::utxo::UtxoEntry;
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct RichListEntry {
pub script_pubkey: Script,
pub value: u64,
}
#[derive(Debug, Clone)]
pub struct RichList {
map: IndexMap<Script, u64>,
}
impl Default for RichList {
fn default() -> Self {
Self::new()
}
}
impl RichList {
pub fn new() -> Self {
Self {
map: IndexMap::new(),
}
}
pub fn is_empty(&self) -> bool {
self.map.is_empty()
}
pub fn len(&self) -> usize {
self.map.len()
}
pub fn capacity(&self) -> usize {
self.map.capacity()
}
pub fn size(&self) -> usize {
self.map.iter().map(|(script, _val)| script.len() + size_of::<u64>()).sum()
}
pub fn shrink_to_fit(&mut self) {
self.map.shrink_to_fit();
}
pub fn push(&mut self, utxo: &UtxoEntry) {
let value = self.map.get(&utxo.script_pubkey).unwrap_or(&0u64) + utxo.value;
self.map.insert(utxo.script_pubkey.clone(), value);
}
pub fn remove(&mut self, script_pubkey: &Script, value: u64) {
if value != 0 {
let v = self.map.get_mut(script_pubkey).unwrap();
*v -= value;
if *v == 0 {
self.map.remove(script_pubkey);
}
}
}
pub fn get_index_of(&self, script_pubkey: &Script) -> Option<usize> {
self.map.get_index_of(script_pubkey)
}
pub fn get_in_range_as_rest(&self, range: Range<usize>, config: &Config) -> Vec<Option<chainseeker::RichListEntry>> {
range.map(|i| {
self.map.get_index(i).map(|(script_pubkey, value)| {
chainseeker::RichListEntry {
script_pub_key: create_script_pub_key(script_pubkey, config),
value: *value,
}
})
}).collect()
}
pub fn process_block(&mut self, block: &Block, previous_utxos: &[UtxoEntry]) {
for tx in block.txdata.iter() {
let txid = tx.txid();
for vout in 0..tx.output.len() {
let output = &tx.output[vout];
if output.value > 0 {
let vout = vout as u32;
let utxo = UtxoEntry {
script_pubkey: output.script_pubkey.clone(),
txid,
vout,
value: output.value,
};
self.push(&utxo);
}
}
}
let mut previous_utxo_index = 0;
for tx in block.txdata.iter() {
for vin in tx.input.iter() {
if !vin.previous_output.is_null() {
let previous_utxo = &previous_utxos[previous_utxo_index];
self.remove(&previous_utxo.script_pubkey, previous_utxo.value);
previous_utxo_index += 1;
}
}
}
self.finalize();
}
pub fn finalize(&mut self) {
self.map.par_sort_by(|_k1, v1, _k2, v2| v2.cmp(v1));
}
}
#[cfg(test)]
mod tests {
use crate::db::utxo::UtxoDB;
use super::*;
const JSON: &str = r#"[{"scriptPubKey":{"asm":"OP_0 OP_PUSHBYTES_20 97033ca70d45fe6d49310859e132a9df98f97625","hex":"001497033ca70d45fe6d49310859e132a9df98f97625","type":"witnesspubkeyhash","address":"bcrt1qjupnefcdghlx6jf3ppv7zv4fm7v0ja39dzzwvd"},"value":505000034900},{"scriptPubKey":{"asm":"OP_PUSHBYTES_65 04678afdb0fe5548271967f1a67130b7105cd6a828e03909a67962e0ea1f61deb649f6bc3f4cef38c4f35504e51ec112de5c384df7ba0b8d578a4c702b6bf11d5f OP_CHECKSIG","hex":"4104678afdb0fe5548271967f1a67130b7105cd6a828e03909a67962e0ea1f61deb649f6bc3f4cef38c4f35504e51ec112de5c384df7ba0b8d578a4c702b6bf11d5fac","type":"unknown","address":null},"value":5000000000},{"scriptPubKey":{"asm":"OP_0 OP_PUSHBYTES_20 13bb0bcb776f3e15fa8800877552044d4db27b96","hex":"001413bb0bcb776f3e15fa8800877552044d4db27b96","type":"witnesspubkeyhash","address":"bcrt1qzwashjmhdulpt75gqzrh25syf4xmy7uk6clm0p"},"value":4900000000},{"scriptPubKey":{"asm":"OP_0 OP_PUSHBYTES_20 261b6555a3cb5d3593c0275ff46f31c36e42a4c7","hex":"0014261b6555a3cb5d3593c0275ff46f31c36e42a4c7","type":"witnesspubkeyhash","address":"bcrt1qycdk24dredwnty7qya0lgme3cdhy9fx83qc9wd"},"value":99965100}]"#;
#[allow(dead_code)]
fn print_rich_list(rich_list: &RichList) {
let entries = rich_list.get_in_range_as_rest(0..rich_list.len(), &config_example("rbtc"));
println!("{}", serde_json::to_string(&entries).unwrap());
}
#[test]
fn rich_list() {
let mut rich_list: RichList = Default::default();
assert!(rich_list.is_empty());
let mut utxo_db = UtxoDB::new("test/rich_list", true);
for block in fixtures::regtest_blocks().iter() {
let prev_utxos = utxo_db.process_block(&block, false);
rich_list.process_block(&block, &prev_utxos);
}
rich_list.finalize();
rich_list.shrink_to_fit();
print_rich_list(&rich_list);
let entries: Vec<Option<chainseeker::RichListEntry>> = serde_json::from_str(JSON).unwrap();
assert!(!rich_list.is_empty());
assert_eq!(rich_list.len(), entries.len());
assert_eq!(rich_list.capacity(), entries.len());
assert_eq!(rich_list.size(), 165);
assert_eq!(rich_list.get_in_range_as_rest(0..entries.len(), &config_example("rbtc")), entries);
for (i, entry) in entries.iter().enumerate() {
let script_pubkey = Script::from_hex(&entry.as_ref().unwrap().script_pub_key.hex).unwrap();
assert_eq!(rich_list.get_index_of(&script_pubkey), Some(i));
}
}
}