sol-parser-sdk 0.3.1

//! 统一解析器 - 简化的单一入口解析器
//!
//! 提供完整的交易解析能力，支持指令和日志数据处理
//!
//! ## 零延迟优化
//! - 使用 SmallVec 避免小数组堆分配
//! - 内联热路径函数
//! - 流式处理，立即回调
//! - 分支预测提示

use crate::core::events::*;
use smallvec::{smallvec, SmallVec};
use solana_sdk::{pubkey::Pubkey, signature::Signature};

/// 主要解析函数 - 解析完整交易并返回所有 DEX 事件
///
/// 参数：
/// - instruction_data: 交易指令数据
/// - accounts: 账户列表
/// - logs: 交易日志
/// - signature: 交易签名
/// - slot: 区块高度
/// - block_time_us: 区块时间
/// - program_id: 程序 ID
///
/// ## 零延迟优化
/// - 使用 SmallVec<[DexEvent; 4]> 栈分配，大多数交易 ≤ 4 个事件
/// - 预分配容量，避免动态扩容
#[inline] // 零延迟优化：内联
pub fn parse_transaction_events(
    _instruction_data: &[u8],
    _accounts: &[Pubkey],
    logs: &[String],
    signature: Signature,
    slot: u64,
    _tx_index: u64,
    block_time_us: Option<i64>,
    _program_id: &Pubkey,
) -> SmallVec<[DexEvent; 4]> {
    // 零延迟优化：SmallVec 栈分配
    let mut events = smallvec![]; // 栈分配，容量 4

    // 2. 解析日志事件 - 大多数日志会成功解析
    for log in logs {
        if let Some(log_event) = crate::logs::parse_log_unified(log, signature, slot, block_time_us)
        {
            events.push(log_event); // 热路径：成功解析
        }
        // 冷路径：解析失败，继续下一个
    }

    events
}

/// 简化版本 - 仅解析日志事件
#[inline] // 零延迟优化：内联
pub fn parse_logs_only(
    logs: &[String],
    signature: Signature,
    slot: u64,
    block_time_us: Option<i64>,
) -> SmallVec<[DexEvent; 4]> {
    // 零延迟优化：SmallVec 栈分配
    let mut events = SmallVec::with_capacity(logs.len().min(4)); // 预分配容量

    for log in logs {
        if let Some(event) = crate::logs::parse_log_unified(log, signature, slot, block_time_us) {
            events.push(event);
        }
    }

    events
}

/// 事件监听器 trait - 用户可以实现此 trait 来处理解析出的事件
pub trait EventListener {
    fn on_dex_event(&self, event: &DexEvent);
}

/// 使用监听器解析交易的便捷函数
pub fn parse_transaction_with_listener<T: EventListener>(
    instruction_data: &[u8],
    accounts: &[Pubkey],
    logs: &[String],
    signature: Signature,
    slot: u64,
    tx_index: u64,
    block_time_us: Option<i64>,
    program_id: &Pubkey,
    listener: &T,
) {
    let events = parse_transaction_events(
        instruction_data,
        accounts,
        logs,
        signature,
        slot,
        tx_index,
        block_time_us,
        program_id,
    );

    for event in &events {
        listener.on_dex_event(event);
    }
}

/// 流式解析交易事件 - 每解析出一个事件就立即回调
///
/// 这个版本不做事件合并，确保每个事件都能立即被处理
/// 适用于需要实时响应的场景
pub fn parse_transaction_events_streaming<F>(
    _instruction_data: &[u8],
    _accounts: &[Pubkey],
    logs: &[String],
    signature: Signature,
    slot: u64,
    _tx_index: u64,
    block_time_us: Option<i64>,
    _program_id: &Pubkey,
    mut callback: F,
) where
    F: FnMut(DexEvent),
{
    // 1. 先解析指令事件（如果有） - 立即回调
    // if let Some(instr_event) = crate::instr::parse_instruction_unified(
    //     instruction_data, accounts, signature, slot, tx_index, block_time_us, program_id
    // ) {
    //     callback(instr_event);  // 立即回调指令事件
    // }

    // 2. 逐个解析日志事件 - 每个事件立即回调
    for log in logs {
        if let Some(log_event) = crate::logs::parse_log_unified(log, signature, slot, block_time_us)
        {
            callback(log_event); // 立即回调日志事件，不等待其他日志
        }
    }

    // 注意：这里完全不做事件合并和缓存，确保每个事件都是立即回调
    // 回调顺序：先指令事件，然后按日志顺序回调日志事件
}

/// 流式解析日志事件 - 每解析出一个事件就立即回调
pub fn parse_logs_streaming<F>(
    logs: &[String],
    signature: Signature,
    slot: u64,
    block_time_us: Option<i64>,
    mut callback: F,
) where
    F: FnMut(DexEvent),
{
    for log in logs {
        if let Some(event) = crate::logs::parse_log_unified(log, signature, slot, block_time_us) {
            callback(event);
        }
    }
}

/// 流式事件监听器 trait - 适用于流式处理
pub trait StreamingEventListener {
    fn on_dex_event_streaming(&mut self, event: DexEvent);
}

/// 使用流式监听器解析交易的便捷函数
pub fn parse_transaction_with_streaming_listener<T: StreamingEventListener>(
    instruction_data: &[u8],
    accounts: &[Pubkey],
    logs: &[String],
    signature: Signature,
    slot: u64,
    tx_index: u64,
    block_time_us: Option<i64>,
    program_id: &Pubkey,
    listener: &mut T,
) {
    parse_transaction_events_streaming(
        instruction_data,
        accounts,
        logs,
        signature,
        slot,
        tx_index,
        block_time_us,
        program_id,
        |event| listener.on_dex_event_streaming(event),
    );
}