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)]
247/// `recent_blockhash`: pass as `Option<&[u8]>`; only cloned when an event is built (low latency).
248pub fn parse_log_optimized(
249    log: &str,
250    signature: Signature,
251    slot: u64,
252    tx_index: u64,
253    block_time_us: Option<i64>,
254    grpc_recv_us: i64,
255    event_type_filter: Option<&EventTypeFilter>,
256    is_created_buy: bool,
257    recent_blockhash: Option<&[u8]>,
258) -> Option<DexEvent> {
259    // Step 1: Find "Program data: " prefix using SIMD
260    let log_bytes = log.as_bytes();
261    let pos = PROGRAM_DATA_FINDER.find(log_bytes)?;
262    let data_start = pos + 14; // "Program data: " length
263    
264    if log_bytes.len() <= data_start {
265        return None;
266    }
267    
268    // Step 2: Decode base64 ONCE to stack buffer (compiler auto-vectorized, zero heap allocation)
269    let mut buf = [0u8; 2048];  // Increased back to 2048 to prevent buffer overflow panics
270    let data_part = &log[data_start..];
271    let trimmed = data_part.trim();
272
273    // Validate input size before decoding (base64: 4 chars -> 3 bytes, so max input = (2048/3)*4 = ~2730 chars)
274    // Add safety margin to prevent base64-simd assertion failures
275    if trimmed.len() > 2700 {
276        return None;
277    }
278
279    // SIMD-accelerated base64 decoding (AVX2/SSE4/NEON)
280    use base64_simd::AsOut;
281    let decoded_slice = base64_simd::STANDARD
282        .decode(trimmed.as_bytes(), buf.as_mut().as_out())
283        .ok()?;
284    let decoded_len = decoded_slice.len();
285    
286    if decoded_len < 8 {
287        return None;
288    }
289    
290    let program_data = &buf[..decoded_len];
291    
292    // Step 3: Extract discriminator (~5ns, just read 8 bytes)
293    let discriminator = unsafe {
294        let ptr = program_data.as_ptr() as *const u64;
295        ptr.read_unaligned()
296    };
297    
298    // Step 4: Map discriminator to EventType for early filtering
299    let event_type = discriminator_to_event_type(discriminator);
300    
301    // Step 5: Early filter check - BEFORE parsing any fields!
302    if let Some(filter) = event_type_filter {
303        if let Some(et) = event_type {
304            if !filter.should_include(et) {
305                return None; // Skip ALL parsing - saves ~200-500ns
306            }
307        } else {
308            // Unknown discriminator - check if any supported protocol is wanted
309            if let Some(ref include_only) = filter.include_only {
310                let wants_supported = include_only.iter().any(|t| matches!(t,
311                    EventType::PumpFunTrade | EventType::PumpFunCreate | EventType::PumpFunMigrate |
312                    EventType::PumpFunBuy | EventType::PumpFunSell | EventType::PumpFunBuyExactSolIn |
313                    EventType::PumpSwapBuy | EventType::PumpSwapSell | EventType::PumpSwapCreatePool |
314                    EventType::PumpSwapLiquidityAdded | EventType::PumpSwapLiquidityRemoved
315                ));
316                if !wants_supported {
317                    return None;
318                }
319            }
320        }
321    }
322    
323    // Step 6: Parse the specific event type (data already decoded!)
324    let data = &program_data[8..]; // Skip discriminator
325
326    use crate::core::events::*;
327
328    let metadata = EventMetadata {
329        signature,
330        slot,
331        tx_index,
332        block_time_us: block_time_us.unwrap_or(0),
333        grpc_recv_us,
334        recent_blockhash: recent_blockhash.map(|s| bs58::encode(s).into_string()),
335    };
336
337    // ========================================================================
338    // Hot-path optimization: Fast check for top 5 most common discriminators
339    // This avoids the large match statement for ~80% of events
340    // Expected savings: 5-20ns per hot event
341    // ========================================================================
342
343    // Check hot-path discriminators first (ordered by frequency)
344    if likely(discriminator == discriminators::PUMPFUN_TRADE) {
345        // PumpFun Trade - Most common (~40% of all events)
346        let event = crate::logs::pump::parse_trade_from_data(data, metadata, is_created_buy)?;
347        // Secondary filter check
348        if let Some(filter) = event_type_filter {
349            if let Some(ref include_only) = filter.include_only {
350                let has_specific_filter = include_only.iter().any(|t| matches!(t,
351                    EventType::PumpFunBuy | EventType::PumpFunSell | EventType::PumpFunBuyExactSolIn
352                        | EventType::PumpFunCreate | EventType::PumpFunCreateV2
353                ));
354                if has_specific_filter {
355                    let event_type_matches = match &event {
356                        DexEvent::PumpFunBuy(_) => include_only.contains(&EventType::PumpFunBuy),
357                        DexEvent::PumpFunSell(_) => include_only.contains(&EventType::PumpFunSell),
358                        DexEvent::PumpFunBuyExactSolIn(_) => include_only.contains(&EventType::PumpFunBuyExactSolIn),
359                        DexEvent::PumpFunTrade(_) => include_only.contains(&EventType::PumpFunTrade),
360                        DexEvent::PumpFunCreate(_) => include_only.contains(&EventType::PumpFunCreate),
361                        DexEvent::PumpFunCreateV2(_) => include_only.contains(&EventType::PumpFunCreateV2),
362                        _ => false,
363                    };
364                    if !event_type_matches {
365                        return None;
366                    }
367                }
368            }
369        }
370        return Some(event);
371    }
372
373    if likely(discriminator == discriminators::RAYDIUM_CLMM_SWAP) {
374        // Raydium CLMM Swap - High frequency (~20% of events)
375        return crate::logs::raydium_clmm::parse_swap_from_data(data, metadata);
376    }
377
378    if likely(discriminator == discriminators::RAYDIUM_AMM_SWAP_BASE_IN) {
379        // Raydium AMM Swap Base In - High frequency (~15% of events)
380        return crate::logs::raydium_amm::parse_swap_base_in_from_data(data, metadata);
381    }
382
383    if likely(discriminator == discriminators::PUMPSWAP_BUY) {
384        // PumpSwap Buy - Medium frequency (~10% of events)
385        return crate::logs::pump_amm::parse_buy_from_data(data, metadata);
386    }
387
388    if discriminator == discriminators::PUMPSWAP_SELL {
389        // PumpSwap Sell - Medium frequency (~5% of events)
390        return crate::logs::pump_amm::parse_sell_from_data(data, metadata);
391    }
392
393    // ========================================================================
394    // Cold path: Handle remaining ~10% of events via match statement
395    // ========================================================================
396
397    match discriminator {
398        // Note: Hot-path discriminators (PUMPFUN_TRADE, RAYDIUM_CLMM_SWAP, RAYDIUM_AMM_SWAP_BASE_IN,
399        // PUMPSWAP_BUY, PUMPSWAP_SELL) are handled above and never reach this match statement
400
401        // PumpFun events (cold path)
402        discriminators::PUMPFUN_CREATE => {
403            crate::logs::pump::parse_create_from_data(data, metadata)
404        }
405        discriminators::PUMPFUN_MIGRATE => {
406            crate::logs::pump::parse_migrate_from_data(data, metadata)
407        }
408
409        // PumpSwap events (cold path)
410        discriminators::PUMPSWAP_CREATE_POOL => {
411            crate::logs::pump_amm::parse_create_pool_from_data(data, metadata)
412        }
413        discriminators::PUMPSWAP_ADD_LIQUIDITY => {
414            crate::logs::pump_amm::parse_add_liquidity_from_data(data, metadata)
415        }
416        discriminators::PUMPSWAP_REMOVE_LIQUIDITY => {
417            crate::logs::pump_amm::parse_remove_liquidity_from_data(data, metadata)
418        }
419
420        // ========== Other protocols - route by discriminator ==========
421        // Raydium CLMM - use from_data functions (cold path)
422        discriminators::RAYDIUM_CLMM_INCREASE_LIQUIDITY => {
423            crate::logs::raydium_clmm::parse_increase_liquidity_from_data(data, metadata)
424        }
425        discriminators::RAYDIUM_CLMM_DECREASE_LIQUIDITY => {
426            crate::logs::raydium_clmm::parse_decrease_liquidity_from_data(data, metadata)
427        }
428        discriminators::RAYDIUM_CLMM_CREATE_POOL => {
429            crate::logs::raydium_clmm::parse_create_pool_from_data(data, metadata)
430        }
431        discriminators::RAYDIUM_CLMM_COLLECT_FEE => {
432            crate::logs::raydium_clmm::parse_collect_fee_from_data(data, metadata)
433        }
434        
435        // Raydium CPMM - use from_data functions (single decode)
436        discriminators::RAYDIUM_CPMM_SWAP_BASE_IN => {
437            crate::logs::raydium_cpmm::parse_swap_base_in_from_data(data, metadata)
438        }
439        discriminators::RAYDIUM_CPMM_SWAP_BASE_OUT => {
440            crate::logs::raydium_cpmm::parse_swap_base_out_from_data(data, metadata)
441        }
442        // Note: RAYDIUM_CPMM_CREATE_POOL discriminator conflicts with RAYDIUM_CLMM_CREATE_POOL
443        // CPMM create pool is rare, handled via log content detection if needed
444        discriminators::RAYDIUM_CPMM_DEPOSIT => {
445            crate::logs::raydium_cpmm::parse_deposit_from_data(data, metadata)
446        }
447        discriminators::RAYDIUM_CPMM_WITHDRAW => {
448            crate::logs::raydium_cpmm::parse_withdraw_from_data(data, metadata)
449        }
450        
451        // Raydium AMM V4 - use from_data functions (single decode)
452        discriminators::RAYDIUM_AMM_SWAP_BASE_IN => {
453            crate::logs::raydium_amm::parse_swap_base_in_from_data(data, metadata)
454        }
455        discriminators::RAYDIUM_AMM_SWAP_BASE_OUT => {
456            crate::logs::raydium_amm::parse_swap_base_out_from_data(data, metadata)
457        }
458        discriminators::RAYDIUM_AMM_DEPOSIT => {
459            crate::logs::raydium_amm::parse_deposit_from_data(data, metadata)
460        }
461        discriminators::RAYDIUM_AMM_WITHDRAW => {
462            crate::logs::raydium_amm::parse_withdraw_from_data(data, metadata)
463        }
464        discriminators::RAYDIUM_AMM_INITIALIZE2 => {
465            crate::logs::raydium_amm::parse_initialize2_from_data(data, metadata)
466        }
467        
468        // Orca Whirlpool - use from_data functions (single decode)
469        discriminators::ORCA_TRADED => {
470            crate::logs::orca_whirlpool::parse_traded_from_data(data, metadata)
471        }
472        discriminators::ORCA_LIQUIDITY_INCREASED => {
473            crate::logs::orca_whirlpool::parse_liquidity_increased_from_data(data, metadata)
474        }
475        discriminators::ORCA_LIQUIDITY_DECREASED => {
476            crate::logs::orca_whirlpool::parse_liquidity_decreased_from_data(data, metadata)
477        }
478        discriminators::ORCA_POOL_INITIALIZED => {
479            crate::logs::orca_whirlpool::parse_pool_initialized_from_data(data, metadata)
480        }
481        
482        // Meteora AMM - use from_data functions (single decode)
483        discriminators::METEORA_AMM_SWAP => {
484            crate::logs::meteora_amm::parse_swap_from_data(data, metadata)
485        }
486        discriminators::METEORA_AMM_ADD_LIQUIDITY => {
487            crate::logs::meteora_amm::parse_add_liquidity_from_data(data, metadata)
488        }
489        discriminators::METEORA_AMM_REMOVE_LIQUIDITY => {
490            crate::logs::meteora_amm::parse_remove_liquidity_from_data(data, metadata)
491        }
492        discriminators::METEORA_AMM_BOOTSTRAP_LIQUIDITY => {
493            crate::logs::meteora_amm::parse_bootstrap_liquidity_from_data(data, metadata)
494        }
495        discriminators::METEORA_AMM_POOL_CREATED => {
496            crate::logs::meteora_amm::parse_pool_created_from_data(data, metadata)
497        }
498        
499        // Meteora DAMM V2
500        discriminators::METEORA_DAMM_SWAP |
501        discriminators::METEORA_DAMM_SWAP2 |
502        discriminators::METEORA_DAMM_ADD_LIQUIDITY |
503        discriminators::METEORA_DAMM_REMOVE_LIQUIDITY |
504        discriminators::METEORA_DAMM_INITIALIZE_POOL |
505        discriminators::METEORA_DAMM_CREATE_POSITION |
506        discriminators::METEORA_DAMM_CLOSE_POSITION => {
507            crate::logs::parse_meteora_damm_log(log, signature, slot, tx_index, block_time_us, grpc_recv_us)
508        }
509        
510        // NOTE: Meteora DLMM discriminators conflict with Raydium CPMM!
511        // METEORA_DLMM_SWAP == RAYDIUM_CPMM_SWAP_BASE_IN
512        // Handle DLMM in fallback using log content detection
513        
514        // Unknown discriminator - try fallback protocols
515        _ => {
516            // Try Meteora DLMM (has discriminator conflict with Raydium CPMM)
517            if let Some(event) = crate::logs::parse_meteora_dlmm_log(log, signature, slot, tx_index, block_time_us, grpc_recv_us) {
518                return Some(event);
519            }
520            None
521        }
522    }
523}
524
525/// Map discriminator to EventType (compile-time optimized match)
526#[inline(always)]
527fn discriminator_to_event_type(discriminator: u64) -> Option<EventType> {
528    match discriminator {
529        discriminators::PUMPFUN_CREATE => Some(EventType::PumpFunCreate),
530        discriminators::PUMPFUN_TRADE => Some(EventType::PumpFunTrade),
531        discriminators::PUMPFUN_MIGRATE => Some(EventType::PumpFunMigrate),
532        discriminators::PUMPSWAP_BUY => Some(EventType::PumpSwapBuy),
533        discriminators::PUMPSWAP_SELL => Some(EventType::PumpSwapSell),
534        discriminators::PUMPSWAP_CREATE_POOL => Some(EventType::PumpSwapCreatePool),
535        discriminators::PUMPSWAP_ADD_LIQUIDITY => Some(EventType::PumpSwapLiquidityAdded),
536        discriminators::PUMPSWAP_REMOVE_LIQUIDITY => Some(EventType::PumpSwapLiquidityRemoved),
537        _ => None,
538    }
539}
540
541// ============================================================================
542// SIMD utilities for log detection
543// ============================================================================
544#[inline]
545pub fn detect_pumpfun_create(logs: &[String]) -> bool {
546    logs.iter().any(|log| PUMPFUN_CREATE_FINDER.find(log.as_bytes()).is_some())
547}
548
549/// SIMD 优化的 "invoke [" 查找器
550static INVOKE_FINDER: Lazy<memmem::Finder> = Lazy::new(|| memmem::Finder::new(b"invoke ["));
551
552/// 从日志中解析指令调用信息 (SIMD 优化版本)
553/// 返回 (program_id, depth)
554#[inline]
555pub fn parse_invoke_info(log: &str) -> Option<(&str, usize)> {
556    let log_bytes = log.as_bytes();
557
558    // SIMD 快速查找 "invoke ["
559    let invoke_start = INVOKE_FINDER.find(log_bytes)?;
560    let bracket_start = invoke_start + 8; // "invoke [" 长度
561
562    // 边界检查
563    if bracket_start >= log_bytes.len() {
564        return None;
565    }
566
567    // 解析深度数字，直到遇到 ']'
568    let mut depth = 0usize;
569    for &byte in &log_bytes[bracket_start..] {
570        match byte {
571            b'0'..=b'9' => {
572                depth = depth * 10 + (byte - b'0') as usize;
573            }
574            b']' => break,
575            _ => return None, // 遇到非数字非']'字符，解析失败
576        }
577    }
578
579    // 提取程序ID：从 "Program " 开始到 " invoke" 结束
580    if invoke_start < 8 {
581        return None; // 没有足够空间放 "Program "
582    }
583
584    let program_start = 8; // "Program " 的长度
585    let program_end = invoke_start - 1; // " invoke" 前面的空格位置
586
587    if program_end <= program_start {
588        return None;
589    }
590
591    let program_id = std::str::from_utf8(&log_bytes[program_start..program_end]).ok()?;
592
593    Some((program_id, depth))
594}
sol_parser_sdk/logs/optimized_matcher.rs

sol_parser_sdk/logs/
optimized_matcher.rs
