sol_parser_sdk/logs/
optimized_matcher.rs

1//! Optimized log matcher with early discriminator filtering
2//!
3//! Performance strategy:
4//! 1. SIMD-based log type detection (~50ns)
5//! 2. Extract discriminator BEFORE full parsing (~50ns)
6//! 3. Check filter at discriminator level - skip parsing if not needed
7//! 4. Only parse events user actually configured
8//! 5. Compiler-optimized base64 decoding (auto-vectorized with target-cpu=native)
9
10use super::perf_hints::{likely, unlikely};
11use crate::core::events::{DexEvent, EventMetadata};
12use crate::grpc::types::{EventType, EventTypeFilter};
13use memchr::memmem;
14use once_cell::sync::Lazy;
15use solana_sdk::signature::Signature;
16
17/// SIMD 优化的字符串查找器 - 预编译一次，重复使用
18static PUMPFUN_FINDER: Lazy<memmem::Finder> =
19    Lazy::new(|| memmem::Finder::new(b"6EF8rrecthR5Dkzon8Nwu78hRvfCKubJ14M5uBEwF6P"));
20static RAYDIUM_AMM_FINDER: Lazy<memmem::Finder> =
21    Lazy::new(|| memmem::Finder::new(b"675kPX9MHTjS2zt1qfr1NYHuzeLXfQM9H24wFSUt1Mp8"));
22static RAYDIUM_CLMM_FINDER: Lazy<memmem::Finder> =
23    Lazy::new(|| memmem::Finder::new(b"CAMMCzo5YL8w4VFF8KVHrK22GGUQpMdRBFSzKNT3t4ivN6"));
24static RAYDIUM_CPMM_FINDER: Lazy<memmem::Finder> =
25    Lazy::new(|| memmem::Finder::new(b"CPMDWBwJDtYax9qKcQP3CtKz7tHjJsN3H8hGrYVD9mZD"));
26static BONK_FINDER: Lazy<memmem::Finder> =
27    Lazy::new(|| memmem::Finder::new(b"Bxby5A7E8xPDGGc3FyJw7m5eK5aqNVLU83H2zLTQDH1b"));
28static PROGRAM_FINDER: Lazy<memmem::Finder> = Lazy::new(|| memmem::Finder::new(b"Program"));
29static PROGRAM_DATA_FINDER: Lazy<memmem::Finder> =
30    Lazy::new(|| memmem::Finder::new(b"Program data: "));
31static PUMPFUN_CREATE_FINDER: Lazy<memmem::Finder> =
32    Lazy::new(|| memmem::Finder::new(b"Program data: G3KpTd7rY3Y"));
33    static WHIRL_FINDER: Lazy<memmem::Finder> = Lazy::new(|| memmem::Finder::new(b"whirL"));
34static METEORA_FINDER: Lazy<memmem::Finder> = Lazy::new(|| memmem::Finder::new(b"meteora"));
35static METEORA_LB_FINDER: Lazy<memmem::Finder> = Lazy::new(|| memmem::Finder::new(b"LB"));
36static METEORA_DLMM_FINDER: Lazy<memmem::Finder> = Lazy::new(|| memmem::Finder::new(b"DLMM"));
37static PUMPSWAP_LOWER_FINDER: Lazy<memmem::Finder> = Lazy::new(|| memmem::Finder::new(b"pumpswap"));
38static PUMPSWAP_UPPER_FINDER: Lazy<memmem::Finder> = Lazy::new(|| memmem::Finder::new(b"PumpSwap"));
39
40/// 预计算的程序 ID 字符串常量
41pub mod program_id_strings {
42    pub const PUMPFUN_INVOKE: &str = "Program 6EF8rrecthR5Dkzon8Nwu78hRvfCKubJ14M5uBEwF6P invoke";
43    pub const PUMPFUN_SUCCESS: &str = "Program 6EF8rrecthR5Dkzon8Nwu78hRvfCKubJ14M5uBEwF6P success";
44    pub const PUMPFUN_ID: &str = "6EF8rrecthR5Dkzon8Nwu78hRvfCKubJ14M5uBEwF6P";
45
46    pub const BONK_INVOKE: &str = "Program Bxby5A7E8xPDGGc3FyJw7m5eK5aqNVLU83H2zLTQDH1b invoke";
47    pub const BONK_SUCCESS: &str = "Program Bxby5A7E8xPDGGc3FyJw7m5eK5aqNVLU83H2zLTQDH1b success";
48    pub const BONK_ID: &str = "Bxby5A7E8xPDGGc3FyJw7m5eK5aqNVLU83H2zLTQDH1b";
49
50    pub const RAYDIUM_CLMM_INVOKE: &str =
51        "Program CAMMCzo5YL8w4VFF8KVHrK22GGUQpMdRBFSzKNT3t4ivN6 invoke";
52    pub const RAYDIUM_CLMM_SUCCESS: &str =
53        "Program CAMMCzo5YL8w4VFF8KVHrK22GGUQpMdRBFSzKNT3t4ivN6 success";
54    pub const RAYDIUM_CLMM_ID: &str = "CAMMCzo5YL8w4VFF8KVHrK22GGUQpMdRBFSzKNT3t4ivN6";
55
56    pub const RAYDIUM_CPMM_INVOKE: &str =
57        "Program CPMDWBwJDtYax9qKcQP3CtKz7tHjJsN3H8hGrYVD9mZD invoke";
58    pub const RAYDIUM_CPMM_SUCCESS: &str =
59        "Program CPMDWBwJDtYax9qKcQP3CtKz7tHjJsN3H8hGrYVD9mZD success";
60    pub const RAYDIUM_CPMM_ID: &str = "CPMDWBwJDtYax9qKcQP3CtKz7tHjJsN3H8hGrYVD9mZD";
61
62    pub const RAYDIUM_AMM_V4_ID: &str = "675kPX9MHTjS2zt1qfr1NYHuzeLXfQM9H24wFSUt1Mp8";
63
64    // 常用的日志模式
65    pub const PROGRAM_DATA: &str = "Program data: ";
66    pub const PROGRAM_LOG: &str = "Program log: ";
67
68    // PumpFun 事件 discriminator (base64)
69    pub const PUMPFUN_CREATE_DISCRIMINATOR: &str = "GB7IKAUcB3c"; // [24, 30, 200, 40, 5, 28, 7, 119]
70}
71
72/// 快速日志类型枚举
73#[derive(Debug, Copy, Clone, PartialEq)]
74pub enum LogType {
75    PumpFun,
76    RaydiumLaunchpad,
77    PumpAmm,
78    RaydiumClmm,
79    RaydiumCpmm,
80    RaydiumAmm,
81    OrcaWhirlpool,
82    MeteoraAmm,
83    MeteoraDamm,
84    MeteoraDlmm,
85    Unknown,
86}
87
88/// SIMD 优化的日志类型检测器 - 激进早期退出
89#[inline(always)]
90pub fn detect_log_type(log: &str) -> LogType {
91    let log_bytes = log.as_bytes();
92
93    // 第一步：快速长度检查 - 太短的日志直接跳过
94    if log_bytes.len() < 20 {
95        return LogType::Unknown;
96    }
97
98    // 第二步：检查是否有 "Program data:" - 这是事件日志的标志
99    let has_program_data = PROGRAM_DATA_FINDER.find(log_bytes).is_some();
100
101    // 只有 "Program data:" 日志才可能是交易事件
102    if unlikely(!has_program_data) {
103        return LogType::Unknown;
104    }
105
106    // 第三步：使用 SIMD 快速检测具体协议
107    // Raydium AMM - 高频，有明确程序ID（最常见）
108    if likely(RAYDIUM_AMM_FINDER.find(log_bytes).is_some()) {
109        return LogType::RaydiumAmm;
110    }
111
112    // Raydium CLMM
113    if RAYDIUM_CLMM_FINDER.find(log_bytes).is_some() {
114        return LogType::RaydiumClmm;
115    }
116
117    // Raydium CPMM
118    if RAYDIUM_CPMM_FINDER.find(log_bytes).is_some() {
119        return LogType::RaydiumCpmm;
120    }
121
122    // Raydium Launchpad (Bonk)
123    if BONK_FINDER.find(log_bytes).is_some() {
124        return LogType::RaydiumLaunchpad;
125    }
126
127    // Orca Whirlpool
128    if WHIRL_FINDER.find(log_bytes).is_some() {
129        return LogType::OrcaWhirlpool;
130    }
131
132    // Meteora - SIMD 优化
133    if let Some(pos) = METEORA_FINDER.find(log_bytes) {
134        let rest = &log_bytes[pos..];
135        if METEORA_LB_FINDER.find(rest).is_some() {
136            return LogType::MeteoraDamm;
137        } else if METEORA_DLMM_FINDER.find(rest).is_some() {
138            return LogType::MeteoraDlmm;
139        } else {
140            return LogType::MeteoraAmm;
141        }
142    }
143
144    // Pump AMM
145    if PUMPSWAP_LOWER_FINDER.find(log_bytes).is_some()
146        || PUMPSWAP_UPPER_FINDER.find(log_bytes).is_some()
147    {
148        return LogType::PumpAmm;
149    }
150
151    // PumpFun - 特殊处理：可能有程序ID，也可能直接是base64数据
152    // 1. 先检查是否包含程序ID（高频事件）
153    if likely(PUMPFUN_FINDER.find(log_bytes).is_some()) {
154        return LogType::PumpFun;
155    }
156
157    // 2. 兜底：有 "Program data:" 但无法识别协议的，尝试作为 PumpFun 解析
158    // PumpFun的日志格式：Program data: <base64>
159    // 只要日志够长且包含Program data，就认为可能是PumpFun
160    if log.len() > 30 {
161        return LogType::PumpFun;
162    }
163
164    LogType::Unknown
165}
166
167// ============================================================================
168// Discriminator constants (compile-time computed) - All protocols
169// ============================================================================
170mod discriminators {
171    // PumpFun discriminators
172    pub const PUMPFUN_CREATE: u64 = u64::from_le_bytes([27, 114, 169, 77, 222, 235, 99, 118]);
173    pub const PUMPFUN_TRADE: u64 = u64::from_le_bytes([189, 219, 127, 211, 78, 230, 97, 238]);
174    pub const PUMPFUN_MIGRATE: u64 = u64::from_le_bytes([189, 233, 93, 185, 92, 148, 234, 148]);
175    
176    // PumpSwap discriminators
177    pub const PUMPSWAP_BUY: u64 = u64::from_le_bytes([103, 244, 82, 31, 44, 245, 119, 119]);
178    pub const PUMPSWAP_SELL: u64 = u64::from_le_bytes([62, 47, 55, 10, 165, 3, 220, 42]);
179    pub const PUMPSWAP_CREATE_POOL: u64 = u64::from_le_bytes([177, 49, 12, 210, 160, 118, 167, 116]);
180    pub const PUMPSWAP_ADD_LIQUIDITY: u64 = u64::from_le_bytes([120, 248, 61, 83, 31, 142, 107, 144]);
181    pub const PUMPSWAP_REMOVE_LIQUIDITY: u64 = u64::from_le_bytes([22, 9, 133, 26, 160, 44, 71, 192]);
182    
183    // Raydium CLMM discriminators
184    pub const RAYDIUM_CLMM_SWAP: u64 = u64::from_le_bytes([248, 198, 158, 145, 225, 117, 135, 200]);
185    pub const RAYDIUM_CLMM_INCREASE_LIQUIDITY: u64 = u64::from_le_bytes([133, 29, 89, 223, 69, 238, 176, 10]);
186    pub const RAYDIUM_CLMM_DECREASE_LIQUIDITY: u64 = u64::from_le_bytes([160, 38, 208, 111, 104, 91, 44, 1]);
187    pub const RAYDIUM_CLMM_CREATE_POOL: u64 = u64::from_le_bytes([233, 146, 209, 142, 207, 104, 64, 188]);
188    pub const RAYDIUM_CLMM_COLLECT_FEE: u64 = u64::from_le_bytes([164, 152, 207, 99, 187, 104, 171, 119]);
189    
190    // Raydium CPMM discriminators
191    pub const RAYDIUM_CPMM_SWAP_BASE_IN: u64 = u64::from_le_bytes([143, 190, 90, 218, 196, 30, 51, 222]);
192    pub const RAYDIUM_CPMM_SWAP_BASE_OUT: u64 = u64::from_le_bytes([55, 217, 98, 86, 163, 74, 180, 173]);
193    pub const RAYDIUM_CPMM_CREATE_POOL: u64 = u64::from_le_bytes([233, 146, 209, 142, 207, 104, 64, 188]);
194    pub const RAYDIUM_CPMM_DEPOSIT: u64 = u64::from_le_bytes([242, 35, 198, 137, 82, 225, 242, 182]);
195    pub const RAYDIUM_CPMM_WITHDRAW: u64 = u64::from_le_bytes([183, 18, 70, 156, 148, 109, 161, 34]);
196    
197    // Raydium AMM V4 discriminators  
198    pub const RAYDIUM_AMM_SWAP_BASE_IN: u64 = u64::from_le_bytes([0, 0, 0, 0, 0, 0, 0, 9]);
199    pub const RAYDIUM_AMM_SWAP_BASE_OUT: u64 = u64::from_le_bytes([0, 0, 0, 0, 0, 0, 0, 11]);
200    pub const RAYDIUM_AMM_DEPOSIT: u64 = u64::from_le_bytes([0, 0, 0, 0, 0, 0, 0, 3]);
201    pub const RAYDIUM_AMM_WITHDRAW: u64 = u64::from_le_bytes([0, 0, 0, 0, 0, 0, 0, 4]);
202    pub const RAYDIUM_AMM_INITIALIZE2: u64 = u64::from_le_bytes([0, 0, 0, 0, 0, 0, 0, 1]);
203    
204    // Orca Whirlpool discriminators
205    pub const ORCA_TRADED: u64 = u64::from_le_bytes([225, 202, 73, 175, 147, 43, 160, 150]);
206    pub const ORCA_LIQUIDITY_INCREASED: u64 = u64::from_le_bytes([30, 7, 144, 181, 102, 254, 155, 161]);
207    pub const ORCA_LIQUIDITY_DECREASED: u64 = u64::from_le_bytes([166, 1, 36, 71, 112, 202, 181, 171]);
208    pub const ORCA_POOL_INITIALIZED: u64 = u64::from_le_bytes([100, 118, 173, 87, 12, 198, 254, 229]);
209    
210    // Meteora AMM discriminators
211    pub const METEORA_AMM_SWAP: u64 = u64::from_le_bytes([81, 108, 227, 190, 205, 208, 10, 196]);
212    pub const METEORA_AMM_ADD_LIQUIDITY: u64 = u64::from_le_bytes([31, 94, 125, 90, 227, 52, 61, 186]);
213    pub const METEORA_AMM_REMOVE_LIQUIDITY: u64 = u64::from_le_bytes([116, 244, 97, 232, 103, 31, 152, 58]);
214    pub const METEORA_AMM_BOOTSTRAP_LIQUIDITY: u64 = u64::from_le_bytes([121, 127, 38, 136, 92, 55, 14, 247]);
215    pub const METEORA_AMM_POOL_CREATED: u64 = u64::from_le_bytes([202, 44, 41, 88, 104, 220, 157, 82]);
216    
217    // Meteora DAMM V2 discriminators
218    pub const METEORA_DAMM_SWAP: u64 = u64::from_le_bytes([27, 60, 21, 213, 138, 170, 187, 147]);
219    pub const METEORA_DAMM_SWAP2: u64 = u64::from_le_bytes([189, 66, 51, 168, 38, 80, 117, 153]);
220    pub const METEORA_DAMM_ADD_LIQUIDITY: u64 = u64::from_le_bytes([175, 242, 8, 157, 30, 247, 185, 169]);
221    pub const METEORA_DAMM_REMOVE_LIQUIDITY: u64 = u64::from_le_bytes([87, 46, 88, 98, 175, 96, 34, 91]);
222    pub const METEORA_DAMM_INITIALIZE_POOL: u64 = u64::from_le_bytes([228, 50, 246, 85, 203, 66, 134, 37]);
223    pub const METEORA_DAMM_CREATE_POSITION: u64 = u64::from_le_bytes([156, 15, 119, 198, 29, 181, 221, 55]);
224    pub const METEORA_DAMM_CLOSE_POSITION: u64 = u64::from_le_bytes([20, 145, 144, 68, 143, 142, 214, 178]);
225    
226    // Meteora DLMM discriminators
227    pub const METEORA_DLMM_SWAP: u64 = u64::from_le_bytes([143, 190, 90, 218, 196, 30, 51, 222]);
228    pub const METEORA_DLMM_ADD_LIQUIDITY: u64 = u64::from_le_bytes([181, 157, 89, 67, 143, 182, 52, 72]);
229    pub const METEORA_DLMM_REMOVE_LIQUIDITY: u64 = u64::from_le_bytes([80, 85, 209, 72, 24, 206, 35, 178]);
230    pub const METEORA_DLMM_INITIALIZE_POOL: u64 = u64::from_le_bytes([95, 180, 10, 172, 84, 174, 232, 40]);
231    pub const METEORA_DLMM_CREATE_POSITION: u64 = u64::from_le_bytes([123, 233, 11, 43, 146, 180, 97, 119]);
232    pub const METEORA_DLMM_CLOSE_POSITION: u64 = u64::from_le_bytes([94, 168, 102, 45, 59, 122, 137, 54]);
233}
234
235/// Optimized unified log parser with **single-decode, early-filter** strategy
236/// 
237/// **Performance Strategy**:
238/// 1. Decode base64 ONCE to stack buffer (~100ns)
239/// 2. Extract discriminator from decoded data (~5ns)
240/// 3. Check filter BEFORE parsing fields - return None if not wanted
241/// 4. Parse only the specific event type requested
242/// 
243/// **Key optimization**: NO double base64 decoding!
244/// Old: extract_discriminator(decode) -> parser(decode again) = 2x decode
245/// New: decode once -> check filter -> parse from buffer = 1x decode
246#[inline(always)]
247pub fn parse_log_optimized(
248    log: &str,
249    signature: Signature,
250    slot: u64,
251    tx_index: u64,
252    block_time_us: Option<i64>,
253    grpc_recv_us: i64,
254    event_type_filter: Option<&EventTypeFilter>,
255    is_created_buy: bool,
256) -> Option<DexEvent> {
257    // Step 1: Find "Program data: " prefix using SIMD
258    let log_bytes = log.as_bytes();
259    let pos = PROGRAM_DATA_FINDER.find(log_bytes)?;
260    let data_start = pos + 14; // "Program data: " length
261    
262    if log_bytes.len() <= data_start {
263        return None;
264    }
265    
266    // Step 2: Decode base64 ONCE to stack buffer (compiler auto-vectorized, zero heap allocation)
267    let mut buf = [0u8; 2048];  // Increased back to 2048 to prevent buffer overflow panics
268    let data_part = &log[data_start..];
269    let trimmed = data_part.trim();
270
271    // Validate input size before decoding (base64: 4 chars -> 3 bytes, so max input = (2048/3)*4 = ~2730 chars)
272    // Add safety margin to prevent base64-simd assertion failures
273    if trimmed.len() > 2700 {
274        return None;
275    }
276
277    // SIMD-accelerated base64 decoding (AVX2/SSE4/NEON)
278    use base64_simd::AsOut;
279    let decoded_slice = base64_simd::STANDARD
280        .decode(trimmed.as_bytes(), buf.as_mut().as_out())
281        .ok()?;
282    let decoded_len = decoded_slice.len();
283    
284    if decoded_len < 8 {
285        return None;
286    }
287    
288    let program_data = &buf[..decoded_len];
289    
290    // Step 3: Extract discriminator (~5ns, just read 8 bytes)
291    let discriminator = unsafe {
292        let ptr = program_data.as_ptr() as *const u64;
293        ptr.read_unaligned()
294    };
295    
296    // Step 4: Map discriminator to EventType for early filtering
297    let event_type = discriminator_to_event_type(discriminator);
298    
299    // Step 5: Early filter check - BEFORE parsing any fields!
300    if let Some(filter) = event_type_filter {
301        if let Some(et) = event_type {
302            if !filter.should_include(et) {
303                return None; // Skip ALL parsing - saves ~200-500ns
304            }
305        } else {
306            // Unknown discriminator - check if any supported protocol is wanted
307            if let Some(ref include_only) = filter.include_only {
308                let wants_supported = include_only.iter().any(|t| matches!(t,
309                    EventType::PumpFunTrade | EventType::PumpFunCreate | EventType::PumpFunMigrate |
310                    EventType::PumpFunBuy | EventType::PumpFunSell | EventType::PumpFunBuyExactSolIn |
311                    EventType::PumpSwapBuy | EventType::PumpSwapSell | EventType::PumpSwapCreatePool |
312                    EventType::PumpSwapLiquidityAdded | EventType::PumpSwapLiquidityRemoved
313                ));
314                if !wants_supported {
315                    return None;
316                }
317            }
318        }
319    }
320    
321    // Step 6: Parse the specific event type (data already decoded!)
322    let data = &program_data[8..]; // Skip discriminator
323
324    use crate::core::events::*;
325
326    let metadata = EventMetadata {
327        signature,
328        slot,
329        tx_index,
330        block_time_us: block_time_us.unwrap_or(0),
331        grpc_recv_us,
332    };
333
334    // ========================================================================
335    // Hot-path optimization: Fast check for top 5 most common discriminators
336    // This avoids the large match statement for ~80% of events
337    // Expected savings: 5-20ns per hot event
338    // ========================================================================
339
340    // Check hot-path discriminators first (ordered by frequency)
341    if likely(discriminator == discriminators::PUMPFUN_TRADE) {
342        // PumpFun Trade - Most common (~40% of all events)
343        let event = crate::logs::pump::parse_trade_from_data(data, metadata, is_created_buy)?;
344        // Secondary filter check
345        if let Some(filter) = event_type_filter {
346            if let Some(ref include_only) = filter.include_only {
347                let has_specific_filter = include_only.iter().any(|t| matches!(t,
348                    EventType::PumpFunBuy | EventType::PumpFunSell | EventType::PumpFunBuyExactSolIn
349                ));
350                if has_specific_filter {
351                    let event_type_matches = match &event {
352                        DexEvent::PumpFunBuy(_) => include_only.contains(&EventType::PumpFunBuy),
353                        DexEvent::PumpFunSell(_) => include_only.contains(&EventType::PumpFunSell),
354                        DexEvent::PumpFunBuyExactSolIn(_) => include_only.contains(&EventType::PumpFunBuyExactSolIn),
355                        DexEvent::PumpFunTrade(_) => include_only.contains(&EventType::PumpFunTrade),
356                        _ => false,
357                    };
358                    if !event_type_matches {
359                        return None;
360                    }
361                }
362            }
363        }
364        return Some(event);
365    }
366
367    if likely(discriminator == discriminators::RAYDIUM_CLMM_SWAP) {
368        // Raydium CLMM Swap - High frequency (~20% of events)
369        return crate::logs::raydium_clmm::parse_swap_from_data(data, metadata);
370    }
371
372    if likely(discriminator == discriminators::RAYDIUM_AMM_SWAP_BASE_IN) {
373        // Raydium AMM Swap Base In - High frequency (~15% of events)
374        return crate::logs::raydium_amm::parse_swap_base_in_from_data(data, metadata);
375    }
376
377    if likely(discriminator == discriminators::PUMPSWAP_BUY) {
378        // PumpSwap Buy - Medium frequency (~10% of events)
379        return crate::logs::pump_amm::parse_buy_from_data(data, metadata);
380    }
381
382    if discriminator == discriminators::PUMPSWAP_SELL {
383        // PumpSwap Sell - Medium frequency (~5% of events)
384        return crate::logs::pump_amm::parse_sell_from_data(data, metadata);
385    }
386
387    // ========================================================================
388    // Cold path: Handle remaining ~10% of events via match statement
389    // ========================================================================
390
391    match discriminator {
392        // Note: Hot-path discriminators (PUMPFUN_TRADE, RAYDIUM_CLMM_SWAP, RAYDIUM_AMM_SWAP_BASE_IN,
393        // PUMPSWAP_BUY, PUMPSWAP_SELL) are handled above and never reach this match statement
394
395        // PumpFun events (cold path)
396        discriminators::PUMPFUN_CREATE => {
397            crate::logs::pump::parse_create_from_data(data, metadata)
398        }
399        discriminators::PUMPFUN_MIGRATE => {
400            crate::logs::pump::parse_migrate_from_data(data, metadata)
401        }
402
403        // PumpSwap events (cold path)
404        discriminators::PUMPSWAP_CREATE_POOL => {
405            crate::logs::pump_amm::parse_create_pool_from_data(data, metadata)
406        }
407        discriminators::PUMPSWAP_ADD_LIQUIDITY => {
408            crate::logs::pump_amm::parse_add_liquidity_from_data(data, metadata)
409        }
410        discriminators::PUMPSWAP_REMOVE_LIQUIDITY => {
411            crate::logs::pump_amm::parse_remove_liquidity_from_data(data, metadata)
412        }
413
414        // ========== Other protocols - route by discriminator ==========
415        // Raydium CLMM - use from_data functions (cold path)
416        discriminators::RAYDIUM_CLMM_INCREASE_LIQUIDITY => {
417            crate::logs::raydium_clmm::parse_increase_liquidity_from_data(data, metadata)
418        }
419        discriminators::RAYDIUM_CLMM_DECREASE_LIQUIDITY => {
420            crate::logs::raydium_clmm::parse_decrease_liquidity_from_data(data, metadata)
421        }
422        discriminators::RAYDIUM_CLMM_CREATE_POOL => {
423            crate::logs::raydium_clmm::parse_create_pool_from_data(data, metadata)
424        }
425        discriminators::RAYDIUM_CLMM_COLLECT_FEE => {
426            crate::logs::raydium_clmm::parse_collect_fee_from_data(data, metadata)
427        }
428        
429        // Raydium CPMM - use from_data functions (single decode)
430        discriminators::RAYDIUM_CPMM_SWAP_BASE_IN => {
431            crate::logs::raydium_cpmm::parse_swap_base_in_from_data(data, metadata)
432        }
433        discriminators::RAYDIUM_CPMM_SWAP_BASE_OUT => {
434            crate::logs::raydium_cpmm::parse_swap_base_out_from_data(data, metadata)
435        }
436        // Note: RAYDIUM_CPMM_CREATE_POOL discriminator conflicts with RAYDIUM_CLMM_CREATE_POOL
437        // CPMM create pool is rare, handled via log content detection if needed
438        discriminators::RAYDIUM_CPMM_DEPOSIT => {
439            crate::logs::raydium_cpmm::parse_deposit_from_data(data, metadata)
440        }
441        discriminators::RAYDIUM_CPMM_WITHDRAW => {
442            crate::logs::raydium_cpmm::parse_withdraw_from_data(data, metadata)
443        }
444        
445        // Raydium AMM V4 - use from_data functions (single decode)
446        discriminators::RAYDIUM_AMM_SWAP_BASE_IN => {
447            crate::logs::raydium_amm::parse_swap_base_in_from_data(data, metadata)
448        }
449        discriminators::RAYDIUM_AMM_SWAP_BASE_OUT => {
450            crate::logs::raydium_amm::parse_swap_base_out_from_data(data, metadata)
451        }
452        discriminators::RAYDIUM_AMM_DEPOSIT => {
453            crate::logs::raydium_amm::parse_deposit_from_data(data, metadata)
454        }
455        discriminators::RAYDIUM_AMM_WITHDRAW => {
456            crate::logs::raydium_amm::parse_withdraw_from_data(data, metadata)
457        }
458        discriminators::RAYDIUM_AMM_INITIALIZE2 => {
459            crate::logs::raydium_amm::parse_initialize2_from_data(data, metadata)
460        }
461        
462        // Orca Whirlpool - use from_data functions (single decode)
463        discriminators::ORCA_TRADED => {
464            crate::logs::orca_whirlpool::parse_traded_from_data(data, metadata)
465        }
466        discriminators::ORCA_LIQUIDITY_INCREASED => {
467            crate::logs::orca_whirlpool::parse_liquidity_increased_from_data(data, metadata)
468        }
469        discriminators::ORCA_LIQUIDITY_DECREASED => {
470            crate::logs::orca_whirlpool::parse_liquidity_decreased_from_data(data, metadata)
471        }
472        discriminators::ORCA_POOL_INITIALIZED => {
473            crate::logs::orca_whirlpool::parse_pool_initialized_from_data(data, metadata)
474        }
475        
476        // Meteora AMM - use from_data functions (single decode)
477        discriminators::METEORA_AMM_SWAP => {
478            crate::logs::meteora_amm::parse_swap_from_data(data, metadata)
479        }
480        discriminators::METEORA_AMM_ADD_LIQUIDITY => {
481            crate::logs::meteora_amm::parse_add_liquidity_from_data(data, metadata)
482        }
483        discriminators::METEORA_AMM_REMOVE_LIQUIDITY => {
484            crate::logs::meteora_amm::parse_remove_liquidity_from_data(data, metadata)
485        }
486        discriminators::METEORA_AMM_BOOTSTRAP_LIQUIDITY => {
487            crate::logs::meteora_amm::parse_bootstrap_liquidity_from_data(data, metadata)
488        }
489        discriminators::METEORA_AMM_POOL_CREATED => {
490            crate::logs::meteora_amm::parse_pool_created_from_data(data, metadata)
491        }
492        
493        // Meteora DAMM V2
494        discriminators::METEORA_DAMM_SWAP |
495        discriminators::METEORA_DAMM_SWAP2 |
496        discriminators::METEORA_DAMM_ADD_LIQUIDITY |
497        discriminators::METEORA_DAMM_REMOVE_LIQUIDITY |
498        discriminators::METEORA_DAMM_INITIALIZE_POOL |
499        discriminators::METEORA_DAMM_CREATE_POSITION |
500        discriminators::METEORA_DAMM_CLOSE_POSITION => {
501            crate::logs::parse_meteora_damm_log(log, signature, slot, tx_index, block_time_us, grpc_recv_us)
502        }
503        
504        // NOTE: Meteora DLMM discriminators conflict with Raydium CPMM!
505        // METEORA_DLMM_SWAP == RAYDIUM_CPMM_SWAP_BASE_IN
506        // Handle DLMM in fallback using log content detection
507        
508        // Unknown discriminator - try fallback protocols
509        _ => {
510            // Try Meteora DLMM (has discriminator conflict with Raydium CPMM)
511            if let Some(event) = crate::logs::parse_meteora_dlmm_log(log, signature, slot, tx_index, block_time_us, grpc_recv_us) {
512                return Some(event);
513            }
514            None
515        }
516    }
517}
518
519/// Map discriminator to EventType (compile-time optimized match)
520#[inline(always)]
521fn discriminator_to_event_type(discriminator: u64) -> Option<EventType> {
522    match discriminator {
523        discriminators::PUMPFUN_CREATE => Some(EventType::PumpFunCreate),
524        discriminators::PUMPFUN_TRADE => Some(EventType::PumpFunTrade),
525        discriminators::PUMPFUN_MIGRATE => Some(EventType::PumpFunMigrate),
526        discriminators::PUMPSWAP_BUY => Some(EventType::PumpSwapBuy),
527        discriminators::PUMPSWAP_SELL => Some(EventType::PumpSwapSell),
528        discriminators::PUMPSWAP_CREATE_POOL => Some(EventType::PumpSwapCreatePool),
529        discriminators::PUMPSWAP_ADD_LIQUIDITY => Some(EventType::PumpSwapLiquidityAdded),
530        discriminators::PUMPSWAP_REMOVE_LIQUIDITY => Some(EventType::PumpSwapLiquidityRemoved),
531        _ => None,
532    }
533}
534
535// ============================================================================
536// SIMD utilities for log detection
537// ============================================================================
538#[inline]
539pub fn detect_pumpfun_create(logs: &[String]) -> bool {
540    logs.iter().any(|log| PUMPFUN_CREATE_FINDER.find(log.as_bytes()).is_some())
541}
542
543/// SIMD 优化的 "invoke [" 查找器
544static INVOKE_FINDER: Lazy<memmem::Finder> = Lazy::new(|| memmem::Finder::new(b"invoke ["));
545
546/// 从日志中解析指令调用信息 (SIMD 优化版本)
547/// 返回 (program_id, depth)
548#[inline]
549pub fn parse_invoke_info(log: &str) -> Option<(&str, usize)> {
550    let log_bytes = log.as_bytes();
551
552    // SIMD 快速查找 "invoke ["
553    let invoke_start = INVOKE_FINDER.find(log_bytes)?;
554    let bracket_start = invoke_start + 8; // "invoke [" 长度
555
556    // 边界检查
557    if bracket_start >= log_bytes.len() {
558        return None;
559    }
560
561    // 解析深度数字，直到遇到 ']'
562    let mut depth = 0usize;
563    for &byte in &log_bytes[bracket_start..] {
564        match byte {
565            b'0'..=b'9' => {
566                depth = depth * 10 + (byte - b'0') as usize;
567            }
568            b']' => break,
569            _ => return None, // 遇到非数字非']'字符，解析失败
570        }
571    }
572
573    // 提取程序ID：从 "Program " 开始到 " invoke" 结束
574    if invoke_start < 8 {
575        return None; // 没有足够空间放 "Program "
576    }
577
578    let program_start = 8; // "Program " 的长度
579    let program_end = invoke_start - 1; // " invoke" 前面的空格位置
580
581    if program_end <= program_start {
582        return None;
583    }
584
585    let program_id = std::str::from_utf8(&log_bytes[program_start..program_end]).ok()?;
586
587    Some((program_id, depth))
588}
sol_parser_sdk/logs/optimized_matcher.rs

sol_parser_sdk/logs/
optimized_matcher.rs
