Compare commits

...
24 Commits
Author SHA1 Message Date
Wood 7d2ecd57e9 Release v3.6.4: bump version, update README (EN/CN), ensure all examples build
- Bump version to 3.6.4 in Cargo.toml
- Update version references in README.md and README_CN.md
- Add RELEASE_v3.6.4.md with English release notes
- All workspace examples verified to compile (cargo build --workspace)
- Remove RELEASE_v3.6.3.md and docs/ASYNC_EXECUTOR_REVIEW.md
- Minor updates in types, lib, transaction_builder, async_executor, executor, params

Made-with: Cursor
2026-03-17 04:52:27 +08:00
Wood 667d6d2c5b Release v3.6.3: bump version, update README and add release notes
Made-with: Cursor
2026-03-17 03:44:42 +08:00
Wood e957bc4bee Parallel multi-SWQoS submit: builder pool and executor tweaks
- transaction_pool: add PARALLEL_SENDER_COUNT (18), ensure prefill >= 18 so
  multi-channel build never serializes; prefill 64 with max(PREFILL, 18)
- async_executor: document that pool prefill must match sender thread count
- executor: do not sort submit_timings (avoids any extra work); comment that
  log order is completion order
- docs: add ASYNC_EXECUTOR_REVIEW.md
- remove RELEASE_NOTES_v3.6.2.md; minor example/types/lib/params updates

Made-with: Cursor
2026-03-17 03:32:59 +08:00
WoodandGitHub 274636abc5 Merge pull request #91 from hookenful/style/rust-fmt
style/run rustfmt
2026-03-15 22:25:00 +08:00
hookenful 2abcf3839e style: run rustfmt 2026-03-14 18:16:55 +02:00
WoodandGitHub 0ec826fcbd Merge pull request #90 from HelvetiCrypt/fix/swqos-worker-busy-spin
Fix SWQOS worker pool busy-spin causing high CPU at idle
2026-03-13 13:10:48 +08:00
vibesandClaude Opus 4.6 6607d276db Fix SWQOS worker pool busy-spin causing 300%+ CPU at idle
The 32 worker tasks in swqos_worker_loop use yield_now().await when
the queue is empty, which busy-spins across all tokio threads. Replace
with tokio::sync::Notify so workers sleep until jobs are enqueued.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
2026-03-13 00:37:14 +00:00
Wood d7b0985844 Release v3.6.2: Node1 QUIC, Speedlanding reliability, version bump
- Add Node1 QUIC transport (UUID auth, bincode tx, region endpoints)
- Speedlanding: ensure_connected, connect/send timeouts, lock on reconnect, SNI from endpoint host
- Bump version to 3.6.2; update README(s) and add release notes

Made-with: Cursor
2026-03-12 15:40:56 +08:00
Wood 2e39649ebc chore: release v3.6.1
Made-with: Cursor
2026-03-10 01:44:03 +08:00
Wood 0201d3443a chore: remove obsolete release notes (v3.4.1, v3.5.0)
Made-with: Cursor
2026-03-10 01:42:43 +08:00
Wood 9872b1b4a7 feat(swqos): Astralane QUIC region-based endpoints and endpoint selection
- Add SWQOS_ENDPOINTS_ASTRALANE_QUIC array (8 regions: ny, fr, ams, lim, sg, ams, la, lim)
- Select QUIC endpoint by region and optional custom URL in get_swqos_client (was hardcoded)
- Use primary endpoints only; avoid ams2/fr2 for lower latency
- Align with Astralane QUIC docs (port 7000, gateway.astralane.io)

Made-with: Cursor
2026-03-10 01:40:25 +08:00
Wood e16cd6620f chore: release v3.6.0
Made-with: Cursor
2026-03-09 00:40:59 +08:00
Wood 9f79e865ab perf: P0/P1/P2 low-latency and maintainability improvements
P0:
- blockhash/nonce: get_transaction_blockhash returns Result; buy/sell entry validation
- Bonk sell: remove RPC get_token_balance from build path; require caller to pass input_amount
- Hot path: drop redundant dex_type/protocol_params clone; tip via sol_f64_to_lamports (no String)
- compute_budget cache uses Arc; extend_compute_budget_instructions avoids SmallVec clone

P1:
- middleware protocol_name takes &str; executor destructures result once to avoid signatures/submit_timings clone
- Error messages include dex context; params use Copy for Pubkey; pumpswap utils clone only Pool for pools[0]

P2:
- transaction_pool capacity constants; TIP_ACCOUNT_CACHE marked allow(dead_code)
- Fix error copy in raydium_amm_v4 / meteora_damm_v2 to correct protocol names

Made-with: Cursor
2026-03-08 01:05:27 +08:00
Wood e32e7ee6ab docs: creator_vault/coin_creator from events; CODE_REVIEW three-repo check
Made-with: Cursor
2026-03-07 11:45:33 +08:00
Wood 30c91a74af docs: wrap Astralane QUIC line in README for readability
Made-with: Cursor
2026-03-07 10:54:35 +08:00
Wood 6c41e1cfe6 docs: README Astralane HTTP+QUIC example, sync README_CN
Made-with: Cursor
2026-03-07 10:52:42 +08:00
Wood a003fd4d2f feat: Astralane QUIC, code review fixes, README & example updates
- Add Astralane QUIC client (astralane_quic.rs) and SwqosTransport::Quic
- README: Astralane QUIC usage, remove third-party doc links, add QUIC to examples
- Code review: API key not in logs, PumpSwap no clone, PDA Result, SELL_DISCRIMINATOR, ensure_wsol_ata refactor, tracing in astralane, only supports fix
- instruction/utils: pumpfun/pumpswap PDA & discriminator unit tests
- trading_client example: SwqosTransport, Astralane QUIC in config
- cli_trading: fix all unused variable warnings (_prefix)
- Add docs/CODE_REVIEW_REPORT.md

Made-with: Cursor
2026-03-07 10:47:31 +08:00
Wood fd5af3ab61 feat: track_volume only when cashback; PumpSwap sell cashback use quote_mint ATA; sync IDL
- Pump/PumpSwap buy: track_volume OptionBool = Some(true) only when is_cashback_coin, else Some(false)
- PumpSwap sell cashback: use get_user_volume_accumulator_quote_ata(quote_mint) instead of WSOL-only ATA
- Sync idl/pump_amm.json, idl/pump_fees.json from pump-public-docs

Made-with: Cursor
2026-03-06 16:41:29 +08:00
Wood 7e2860d8de feat: PumpSwap/PumpFun improvements, align with pump-public-docs, sync IDL
- Re-apply PumpSwap pool lookup, lib/utils/seed changes (no Fastlane)
- pump-public-docs: Mayhem fee recipient at index 11 use WSOL ATA; random mayhem fee recipients (PumpSwap + PumpFun); Pool decode 244 bytes
- Sync idl (pump.json, pump_amm.json, pump_fees.json) from pump-fun/pump-public-docs

Made-with: Cursor
2026-03-06 15:18:04 +08:00
Wood 07487c06cb Revert "feat: Fastlane SWQoS, PumpSwap pool lookup and init robustness"
This reverts commit 1142829394.
2026-03-06 11:03:49 +08:00
Wood 1142829394 feat: Fastlane SWQoS, PumpSwap pool lookup and init robustness
- swqos: add Fastlane client and endpoint/tip constants
- pumpswap: pool 244-byte DataSize, canonical PDA lookup, find_by_mint diagnostics
- lib: find_pool_by_mint generic entry; rent/SWQoS init timeouts and default rent fallback
- lib: create WSOL ATA only when SOL balance is sufficient
- utils: get_token_balance_with_options and get_payer_token_balance_with_program (seed ATA aligned)

Made-with: Cursor
2026-03-05 23:45:42 +08:00
WoodandGitHub d2ce193e2c Merge pull request #85 from hookenful/perf/serializer-coldstart-main
Perf/serializer coldstart main
2026-03-01 14:15:35 +08:00
Hookie 46564f1bb5 test(swqos): add serializer cold-start perf tests
Add manual ignored performance tests for SWQOS serializer cold start:\n- compare legacy eager zero-fill preallocation vs bounded prewarm\n- measure lazy growth amortization (first vs reused serialize)\n\nAlso fix PumpFun test fixture to include newly required SwapParams\nfields so lib tests compile when running targeted perf tests.\n\nValidation:\n- cargo test --release perf_serializer_ -- --ignored --nocapture
2026-02-28 22:58:27 +02:00
Hookie 63afac7aea perf(swqos): reduce serializer cold-start overhead
Replace full eager preallocation in the SWQOS serializer with a bounded\nprewarm and lazy growth strategy.\n\n- Prewarm only a small hot set of buffers instead of allocating all\n  10,000 upfront\n- Keep the same max pool capacity and buffer size to preserve the\n  original buffer-pool design in steady state\n- Stop zero-filling each buffer during initialization; reserve capacity\n  and grow data on demand\n- Return temporary buffers in Base64Encoder::serialize_and_encode to\n  avoid pool depletion\n- Add tests for bounded prewarm and lazy allocation/return behavior\n\nWhy:\nThe previous Lazy initialization allocated and touched ~2.44 GiB\n(10,000 x 256 KiB) on first use, which can block the first buy submit\npath and produce a large cold-start latency spike.\n\nValidation:\n- cargo fmt --all\n- cargo check --lib\n- cargo check (SolBot workspace with path dependency)
2026-02-28 22:58:05 +02:00
90 changed files with 4075 additions and 2137 deletions
+4 -2
View File
@@ -1,6 +1,6 @@
[package] [package]
name = "sol-trade-sdk" name = "sol-trade-sdk"
version = "3.5.7" version = "3.6.4"
edition = "2021" edition = "2021"
authors = [ authors = [
"William <byteblock6@gmail.com>", "William <byteblock6@gmail.com>",
@@ -107,7 +107,9 @@ parking_lot = "0.12"
arc-swap = "1.7" arc-swap = "1.7"
sha2 = "0.10" sha2 = "0.10"
tonic-prost = "0.14.2" tonic-prost = "0.14.2"
quinn = {version = "0.11", default-features = false, features = ["rustls"]} quinn = { version = "0.11", default-features = false, features = ["rustls"] }
rcgen = "0.13"
uuid = "1.11"
# Performance optimization dependencies # Performance optimization dependencies
crossbeam-queue = "0.3" crossbeam-queue = "0.3"
+56 -6
View File
@@ -48,6 +48,7 @@
- [⚡ Trading Parameters](#-trading-parameters) - [⚡ Trading Parameters](#-trading-parameters)
- [📊 Usage Examples Summary Table](#-usage-examples-summary-table) - [📊 Usage Examples Summary Table](#-usage-examples-summary-table)
- [⚙️ SWQoS Service Configuration](#-swqos-service-configuration) - [⚙️ SWQoS Service Configuration](#-swqos-service-configuration)
- [Astralane QUIC (Low-Latency)](#astralane-quic-low-latency)
- [🔧 Middleware System](#-middleware-system) - [🔧 Middleware System](#-middleware-system)
- [🔍 Address Lookup Tables](#-address-lookup-tables) - [🔍 Address Lookup Tables](#-address-lookup-tables)
- [🔍 Nonce Cache](#-nonce-cache) - [🔍 Nonce Cache](#-nonce-cache)
@@ -89,14 +90,14 @@ Add the dependency to your `Cargo.toml`:
```toml ```toml
# Add to your Cargo.toml # Add to your Cargo.toml
sol-trade-sdk = { path = "./sol-trade-sdk", version = "3.5.7" } sol-trade-sdk = { path = "./sol-trade-sdk", version = "3.6.4" }
``` ```
### Use crates.io ### Use crates.io
```toml ```toml
# Add to your Cargo.toml # Add to your Cargo.toml
sol-trade-sdk = "3.5.7" sol-trade-sdk = "3.6.4"
``` ```
## 🛠️ Usage Examples ## 🛠️ Usage Examples
@@ -119,6 +120,14 @@ let swqos_configs: Vec<SwqosConfig> = vec![
SwqosConfig::Default(rpc_url.clone()), SwqosConfig::Default(rpc_url.clone()),
SwqosConfig::Jito("your uuid".to_string(), SwqosRegion::Frankfurt, None), SwqosConfig::Jito("your uuid".to_string(), SwqosRegion::Frankfurt, None),
SwqosConfig::Bloxroute("your api_token".to_string(), SwqosRegion::Frankfurt, None), SwqosConfig::Bloxroute("your api_token".to_string(), SwqosRegion::Frankfurt, None),
// Astralane: HTTP (4th param None) or QUIC (Some(SwqosTransport::Quic)); same API key
SwqosConfig::Astralane("your_astralane_api_key".to_string(), SwqosRegion::Frankfurt, None, None), // HTTP
SwqosConfig::Astralane(
"your_astralane_api_key".to_string(),
SwqosRegion::Frankfurt,
None,
Some(SwqosTransport::Quic),
), // QUIC
]; ];
// Create TradeConfig instance // Create TradeConfig instance
let trade_config = TradeConfig::new(rpc_url, swqos_configs, commitment); let trade_config = TradeConfig::new(rpc_url, swqos_configs, commitment);
@@ -257,6 +266,29 @@ let bloxroute_config = SwqosConfig::Bloxroute(
When using multiple MEV services, you need to use `Durable Nonce`. You need to use the `fetch_nonce_info` function to get the latest `nonce` value, and use it as the `durable_nonce` when trading. When using multiple MEV services, you need to use `Durable Nonce`. You need to use the `fetch_nonce_info` function to get the latest `nonce` value, and use it as the `durable_nonce` when trading.
#### Astralane QUIC (Low-Latency)
Astralane supports both HTTP and **QUIC** transport. QUIC reduces connection overhead and can lower submission latency. To use the QUIC channel, pass `Some(SwqosTransport::Quic)` as the fourth parameter of `SwqosConfig::Astralane`. Astralanes QUIC service uses a **single endpoint** (no per-region endpoints); the SDK ignores the `region` (and optional custom URL) when QUIC is selected. You can pass the same region as your other SWQoS configs for consistency.
```rust
use sol_trade_sdk::{SwqosConfig, SwqosRegion, SwqosTransport};
// Astralane over QUIC (low-latency); region is ignored (single QUIC endpoint)
let swqos_configs: Vec<SwqosConfig> = vec![
SwqosConfig::Default(rpc_url.clone()),
SwqosConfig::Astralane(
"your_astralane_api_key".to_string(),
SwqosRegion::Frankfurt, // same as other services; ignored for QUIC
None,
Some(SwqosTransport::Quic),
),
];
// Then create TradeConfig / TradingClient as usual with swqos_configs
```
- **HTTP** (default): use `None` or `Some(SwqosTransport::Http)`; region and optional custom URL apply.
- **QUIC**: use `Some(SwqosTransport::Quic)`; the SDK uses a single QUIC endpoint and ignores region. Same API key as HTTP.
--- ---
### 🔧 Middleware System ### 🔧 Middleware System
@@ -289,6 +321,24 @@ PumpFun and PumpSwap support **cashback** for eligible tokens: part of the tradi
- The **pumpfun_copy_trading** and **pumpfun_sniper_trading** examples use sol-parser-sdk for gRPC subscription and pass `e.is_cashback_coin` when building params. - The **pumpfun_copy_trading** and **pumpfun_sniper_trading** examples use sol-parser-sdk for gRPC subscription and pass `e.is_cashback_coin` when building params.
- **Claim**: Use `client.claim_cashback_pumpfun()` and `client.claim_cashback_pumpswap(...)` to claim accumulated cashback. - **Claim**: Use `client.claim_cashback_pumpfun()` and `client.claim_cashback_pumpswap(...)` to claim accumulated cashback.
#### PumpFun: Creator Rewards Sharing (creator_vault)
Some PumpFun coins use **Creator Rewards Sharing**, so the on-chain `creator_vault` can differ from the default derivation. If you reuse cached params from a **buy** when **selling**, you may see program error **2006 (seeds constraint violated)**. To avoid this:
- **From gRPC/events (no RPC needed)**: You can get both `creator` and `creator_vault` from parsed transaction events:
- **sol-parser-sdk**: Before pushing events, the pipeline calls `fill_trade_accounts`, which fills `creator_vault` from the buy/sell instruction accounts (buy index 9, sell index 8). `creator` comes from the TradeEvent log. Use `PumpFunParams::from_trade(..., e.creator, e.creator_vault, ...)` or `from_dev_trade(..., e.creator, e.creator_vault, ...)` with the event `e`.
- **solana-streamer**: Instruction parsers set `creator_vault` from accounts[9] (buy) or accounts[8] (sell); `creator` comes from the merged CPI TradeEvent log. Use the same `from_trade` / `from_dev_trade` with `e.creator` and `e.creator_vault`.
- **Override after RPC**: If you get params via `PumpFunParams::from_mint_by_rpc` but later receive a newer `creator_vault` from gRPC, call `.with_creator_vault(latest_creator_vault)` on the params before selling.
The SDK does not fetch creator_vault from RPC on every sell (to avoid latency); pass the up-to-date vault from gRPC/events when available.
#### PumpSwap: coin_creator_vault from events (no RPC)
For **PumpSwap** (Pump AMM), `coin_creator_vault_ata` and `coin_creator_vault_authority` are required in buy/sell instructions. Both are available from parsed events without RPC:
- **sol-parser-sdk**: Instruction parser sets them from accounts 17 and 18; the account filler also fills them when the event comes from logs. Use `PumpSwapParams::from_trade(..., e.coin_creator_vault_ata, e.coin_creator_vault_authority, ...)` with the buy/sell event `e`.
- **solana-streamer**: Instruction parser sets them from `accounts.get(17)` and `accounts.get(18)`. Use the same `from_trade` with the events `coin_creator_vault_ata` and `coin_creator_vault_authority`.
## 🛡️ MEV Protection Services ## 🛡️ MEV Protection Services
You can apply for a key through the official website: [Community Website](https://fnzero.dev/swqos) You can apply for a key through the official website: [Community Website](https://fnzero.dev/swqos)
@@ -297,10 +347,10 @@ You can apply for a key through the official website: [Community Website](https:
- **ZeroSlot**: Zero-latency transactions - **ZeroSlot**: Zero-latency transactions
- **Temporal**: Time-sensitive transactions - **Temporal**: Time-sensitive transactions
- **Bloxroute**: Blockchain network acceleration - **Bloxroute**: Blockchain network acceleration
- **FlashBlock**: High-speed transaction execution with API key authentication - [Official Documentation](https://doc.flashblock.trade/) - **FlashBlock**: High-speed transaction execution with API key authentication
- **BlockRazor**: High-speed transaction execution with API key authentication - [Official Documentation](https://blockrazor.gitbook.io/blockrazor/) - **BlockRazor**: High-speed transaction execution with API key authentication
- **Node1**: High-speed transaction execution with API key authentication - [Official Documentation](https://node1.me/docs.html) - **Node1**: High-speed transaction execution with API key authentication
- **Astralane**: Blockchain network acceleration - **Astralane**: Blockchain network acceleration (supports HTTP and QUIC; see [Astralane QUIC](#astralane-quic-low-latency) above)
## 📁 Project Structure ## 📁 Project Structure
+56 -6
View File
@@ -48,6 +48,7 @@
- [⚡ 交易参数](#-交易参数) - [⚡ 交易参数](#-交易参数)
- [📊 使用示例汇总表格](#-使用示例汇总表格) - [📊 使用示例汇总表格](#-使用示例汇总表格)
- [⚙️ SWQoS 服务配置说明](#-swqos-服务配置说明) - [⚙️ SWQoS 服务配置说明](#-swqos-服务配置说明)
- [Astralane QUIC(低延迟)](#astralane-quic低延迟)
- [🔧 中间件系统说明](#-中间件系统说明) - [🔧 中间件系统说明](#-中间件系统说明)
- [🔍 地址查找表](#-地址查找表) - [🔍 地址查找表](#-地址查找表)
- [🔍 Nonce 缓存](#-nonce-缓存) - [🔍 Nonce 缓存](#-nonce-缓存)
@@ -89,14 +90,14 @@ git clone https://github.com/0xfnzero/sol-trade-sdk
```toml ```toml
# 添加到您的 Cargo.toml # 添加到您的 Cargo.toml
sol-trade-sdk = { path = "./sol-trade-sdk", version = "3.5.7" } sol-trade-sdk = { path = "./sol-trade-sdk", version = "3.6.4" }
``` ```
### 使用 crates.io ### 使用 crates.io
```toml ```toml
# 添加到您的 Cargo.toml # 添加到您的 Cargo.toml
sol-trade-sdk = "3.5.7" sol-trade-sdk = "3.6.4"
``` ```
## 🛠️ 使用示例 ## 🛠️ 使用示例
@@ -119,6 +120,14 @@ let swqos_configs: Vec<SwqosConfig> = vec![
SwqosConfig::Default(rpc_url.clone()), SwqosConfig::Default(rpc_url.clone()),
SwqosConfig::Jito("your uuid".to_string(), SwqosRegion::Frankfurt, None), SwqosConfig::Jito("your uuid".to_string(), SwqosRegion::Frankfurt, None),
SwqosConfig::Bloxroute("your api_token".to_string(), SwqosRegion::Frankfurt, None), SwqosConfig::Bloxroute("your api_token".to_string(), SwqosRegion::Frankfurt, None),
// Astralane:第4个参数 None 为 HTTPSome(SwqosTransport::Quic) 为 QUIC;同一 API key
SwqosConfig::Astralane("your_astralane_api_key".to_string(), SwqosRegion::Frankfurt, None, None), // HTTP
SwqosConfig::Astralane(
"your_astralane_api_key".to_string(),
SwqosRegion::Frankfurt,
None,
Some(SwqosTransport::Quic),
), // QUIC
]; ];
// 创建 TradeConfig 实例 // 创建 TradeConfig 实例
let trade_config = TradeConfig::new(rpc_url, swqos_configs, commitment); let trade_config = TradeConfig::new(rpc_url, swqos_configs, commitment);
@@ -256,6 +265,29 @@ let bloxroute_config = SwqosConfig::Bloxroute(
当使用多个MEV服务时,需要使用`Durable Nonce`。你需要使用`fetch_nonce_info`函数获取最新的`nonce`值,并在交易的时候将`durable_nonce`填入交易参数。 当使用多个MEV服务时,需要使用`Durable Nonce`。你需要使用`fetch_nonce_info`函数获取最新的`nonce`值,并在交易的时候将`durable_nonce`填入交易参数。
#### Astralane QUIC(低延迟)
Astralane 支持 **HTTP****QUIC** 两种传输方式。QUIC 可减少连接开销,降低提交延迟。使用 QUIC 时,将 `SwqosConfig::Astralane` 的第四个参数设为 `Some(SwqosTransport::Quic)`。Astralane 的 QUIC 服务使用**单一端点**(无分区域端点),选 QUIC 时 SDK 会忽略 `region` 与可选自定义 URL;为与其他 SWQoS 配置一致,可传入相同 region。
```rust
use sol_trade_sdk::{SwqosConfig, SwqosRegion, SwqosTransport};
// Astralane 使用 QUIC(低延迟);region 会被忽略(QUIC 单一端点)
let swqos_configs: Vec<SwqosConfig> = vec![
SwqosConfig::Default(rpc_url.clone()),
SwqosConfig::Astralane(
"your_astralane_api_key".to_string(),
SwqosRegion::Frankfurt, // 与其他服务一致即可;QUIC 时会被忽略
None,
Some(SwqosTransport::Quic),
),
];
// 然后照常使用 swqos_configs 创建 TradeConfig / TradingClient
```
- **HTTP**(默认):第四个参数为 `None``Some(SwqosTransport::Http)`region 与可选自定义 URL 生效。
- **QUIC**:第四个参数为 `Some(SwqosTransport::Quic)`;SDK 使用单一 QUIC 端点并忽略 region。与 HTTP 使用同一 API key。
--- ---
### 🔧 中间件系统说明 ### 🔧 中间件系统说明
@@ -288,6 +320,24 @@ PumpFun 与 PumpSwap 支持**返现(Cashback**:部分手续费可返还
- **pumpfun_copy_trading**、**pumpfun_sniper_trading** 示例使用 sol-parser-sdk 订阅 gRPC 事件,并在构造参数时传入 `e.is_cashback_coin` - **pumpfun_copy_trading**、**pumpfun_sniper_trading** 示例使用 sol-parser-sdk 订阅 gRPC 事件,并在构造参数时传入 `e.is_cashback_coin`
- **领取返现**:使用 `client.claim_cashback_pumpfun()``client.claim_cashback_pumpswap(...)` 领取累计的返现。 - **领取返现**:使用 `client.claim_cashback_pumpfun()``client.claim_cashback_pumpswap(...)` 领取累计的返现。
#### PumpFunCreator Rewards Sharingcreator_vault
部分 PumpFun 代币启用了 **Creator Rewards Sharing**,链上 `creator_vault` 可能与默认推导结果不同。若在**卖出**时复用**买入**时缓存的 params,可能触发程序错误 **2006seeds constraint violated**。建议:
- **来自 gRPC/事件(无需 RPC**`creator``creator_vault` 均可从解析后的事件中直接拿到:
- **sol-parser-sdk**:推送前会调用 `fill_trade_accounts`,从 buy/sell 指令账户补全 `creator_vault`buy 索引 9sell 索引 8);`creator` 来自 TradeEvent 日志。用 `PumpFunParams::from_trade(..., e.creator, e.creator_vault, ...)``from_dev_trade(..., e.creator, e.creator_vault, ...)` 即可。
- **solana-streamer**:指令解析时从 accounts[9]buy/ accounts[8]sell)写入 `creator_vault``creator` 来自合并后的 CPI TradeEvent 日志。同样用事件的 `e.creator``e.creator_vault` 调用 `from_trade` / `from_dev_trade`
- **RPC 后覆盖**:若通过 `PumpFunParams::from_mint_by_rpc` 得到 params,之后又从 gRPC 拿到更新的 `creator_vault`,在卖出前对 params 调用 `.with_creator_vault(latest_creator_vault)`
SDK 不会在每次卖出时通过 RPC 拉取 creator_vault(以避免延迟);请从 gRPC/事件中传入最新 vault。
#### PumpSwap:从事件拿 coin_creator_vault(无需 RPC
**PumpSwap**Pump AMM)的 buy/sell 指令需要 `coin_creator_vault_ata``coin_creator_vault_authority`,二者均可从解析事件中拿到,无需 RPC:
- **sol-parser-sdk**:指令解析从账户 17、18 写入;若事件来自日志,账户填充器也会从指令补全。用 `PumpSwapParams::from_trade(..., e.coin_creator_vault_ata, e.coin_creator_vault_authority, ...)` 即可。
- **solana-streamer**:指令解析从 `accounts.get(17)``accounts.get(18)` 写入。同样用事件的 `coin_creator_vault_ata``coin_creator_vault_authority` 调用 `from_trade`
## 🛡️ MEV 保护服务 ## 🛡️ MEV 保护服务
可以通过官网申请密钥:[社区官网](https://fnzero.dev/swqos) 可以通过官网申请密钥:[社区官网](https://fnzero.dev/swqos)
@@ -296,10 +346,10 @@ PumpFun 与 PumpSwap 支持**返现(Cashback**:部分手续费可返还
- **ZeroSlot**: 零延迟交易 - **ZeroSlot**: 零延迟交易
- **Temporal**: 时间敏感交易 - **Temporal**: 时间敏感交易
- **Bloxroute**: 区块链网络加速 - **Bloxroute**: 区块链网络加速
- **FlashBlock**: 高速交易执行,支持 API 密钥认证 - [官方文档](https://doc.flashblock.trade/) - **FlashBlock**: 高速交易执行,支持 API 密钥认证
- **BlockRazor**: 高速交易执行,支持 API 密钥认证 - [官方文档](https://blockrazor.gitbook.io/blockrazor/) - **BlockRazor**: 高速交易执行,支持 API 密钥认证
- **Node1**: 高速交易执行,支持 API 密钥认证 - [官方文档](https://node1.me/docs.html) - **Node1**: 高速交易执行,支持 API 密钥认证
- **Astralane**: 高速交易执行,支持 API 密钥认证 - **Astralane**: 区块链网络加速(支持 HTTP 与 QUIC,见上方 [Astralane QUIC](#astralane-quic低延迟)
## 📁 项目结构 ## 📁 项目结构
+8
View File
@@ -0,0 +1,8 @@
# Release v3.6.4
## Changes from v3.6.3
- **Examples**: All workspace examples are verified to build successfully (`cargo build --workspace`).
- **Docs**: Version references in README.md and README_CN.md updated to 3.6.4.
No API or behavior changes in this release.
+170
View File
@@ -0,0 +1,170 @@
# sol-trade-sdk 代码审查报告
审查维度:**逻辑准确性**、**可读性**、**模块化**、**超低延迟**、**代码质量**、**安全性**。
---
## 1. 代码逻辑准确性
### 1.1 Instruction 与 IDL / 官方行为
| 模块 | 结论 | 说明 |
|------|------|------|
| PumpFun buy/sell | ✅ 一致 | 账户顺序、discriminator、track_volume 与 `idl/pump.json` 一致;cashback 时 remainingAccounts 顺序正确 |
| PumpSwap buy/sell | ✅ 一致 | 与 `idl/pump_amm.json` 一致;sell cashback 使用 quote_mint ATA |
| PDA 推导 | ✅ 一致 | bonding_curve_v2、pool_v2、user_volume_accumulator、creator_vault 等 seeds 与官方一致 |
### 1.2 需修正的逻辑/风格
- **`src/instruction/utils/pumpswap.rs` 约 258、291 行**`let program_id: &Pubkey = &&accounts::AMM_PROGRAM` 为双重引用,易误导且多余。建议改为 `&accounts::AMM_PROGRAM`
---
## 2. 代码可读性
### 2.1 命名与注释
- 多数模块有中英文注释,instruction 与 IDL 的对应关系有标注。
- **建议**`src/instruction/pumpfun.rs` 约 259 行 sell 的 discriminator 使用魔法数组 `[51, 230, 133, ...]`,建议改为 `SELL_DISCRIMINATOR` 常量(与 buy 路径一致)。
### 2.2 错误信息与文案
- **建议**`src/lib.rs` 中 “Current version only support” 应为 “only supports”;类似拼写/语法可统一检查。
### 2.3 过长函数
- **建议**`src/lib.rs``ensure_wsol_ata` 可拆为「入口 + 重试循环」与「单次尝试 + 结果判断」,便于单测和阅读。
---
## 3. 模块化
### 3.1 职责与分层
- **instruction**:按协议分(pumpfun / pumpswap / bonk / raydium_* 等),实现 `InstructionBuilder`,边界清晰。
- **instruction/utils**PDA、常量、类型、池子解析与上层「组指令」分工明确。
- **swqos**:按提供商分模块,common 放序列化、确认轮询、HTTP 客户端,无循环依赖。
- **结论**:分层合理,模块化良好。
---
## 4. 超低延迟
### 4.1 必须改:避免多余 clone
| 位置 | 问题 | 建议 |
|------|------|------|
| `src/instruction/pumpswap.rs` 约 232、429 行 | `Instruction { accounts: accounts.clone(), data }` 对已拥有的 `Vec<AccountMeta>` 做完整 clone | 改为直接移动:`Instruction { program_id, accounts, data }`,不再 clone |
### 4.2 建议
- 若多路 SWQOS 并发发**同一笔**交易,可在调用方序列化一次,再传 `&[u8]` 给各 client,减少重复 bincode 序列化。
- 热点路径未见不必要的 `Mutex`/`RwLock` 竞争,当前设计可接受。
---
## 5. 代码质量
### 5.1 必须改:避免 panic 的 unwrap
以下 PDA 或关键 `Option` 使用 `.unwrap()`,在异常输入下会直接 panic,建议改为 `Result` 并向上传播错误:
| 文件 | 行号(约) | 说明 |
|------|------------|------|
| `src/instruction/pumpfun.rs` | 67, 101, 149, 221, 291, 295 | `get_bonding_curve_pda``get_user_volume_accumulator_pda``get_bonding_curve_v2_pda` |
| `src/instruction/pumpswap.rs` | 184, 198, 385, 407 | `get_user_volume_accumulator_pda``get_pool_v2_pda` |
| `src/instruction/utils/pumpfun.rs` | 229 | `DEFAULT_CREATOR_VAULT.unwrap()`LazyLock 未初始化时可能 panic |
| `src/instruction/bonk.rs` | 多处 | `get_pool_pda``get_vault_pda``params.rpc.as_ref().unwrap()` |
| `src/instruction/utils/bonk.rs` | 110116, 148152 | `checked_*` 链后 `.unwrap()`,数学假设不成立会 panic |
| `src/instruction/raydium_cpmm.rs` | 46, 106, 189, 250 | PDA / 状态相关 unwrap |
| `src/instruction/utils/raydium_cpmm.rs` | 83, 85, 141 | `get_vault_pda(...).unwrap()` |
**建议**:统一改为 `.ok_or_else(|| anyhow!("..."))?` 或返回 `Result`,在调用链顶层处理错误,避免进程退出。
### 5.2 建议
- **测试**:为 instruction 构建(或至少 PDA + discriminator/data 布局)增加单元测试,固定输入与预期 bytes/accounts 比对,便于 IDL 升级时回归。
- **错误类型**`claim_cashback_*` 等返回 `Option<Instruction>`;可考虑统一为 `Result<Instruction>` 并带“无法构建”原因,或在文档中明确 None 的语义。
---
## 6. 安全性
### 6.1 必须改:API key 不得写入日志
| 位置 | 问题 | 建议 |
|------|------|------|
| `src/swqos/astralane_quic.rs` 约 61 行 | `info!(..., "api_key as CN: {}", api_key)` | 移除 api_key 或改为占位(如 `***` / 仅长度) |
| `src/swqos/astralane_quic.rs` 约 74 行 | `info!(..., "Connected at {} (api_key: {})", addr, api_key)` | 同上 |
### 6.2 必须改:SkipServerVerification 风险
| 位置 | 问题 | 建议 |
|------|------|------|
| `src/swqos/astralane_quic.rs` 约 179181 行 | `with_custom_certificate_verifier(SkipServerVerification)` 完全跳过服务端证书校验 | 1)若服务端提供证书:用 `RootCertStore` 或固定证书做校验;2)若仅 dev/内网:用 feature 或配置限制,并在文档/日志中明确“仅受控环境使用”;3)默认/生产构建建议不跳过校验 |
### 6.3 建议
- **敏感配置**:确保生产环境从环境变量或安全配置读取 API key,并在文档中说明。
- **依赖**:定期执行 `cargo audit` 与依赖升级。
- **unsafe**`perf/hardware_optimizations.rs``realtime_tuning.rs` 中的 `unsafe` 使用范围可控,需保持注释中的安全约定。
---
## 7. 三库联动与超低延迟检查(sol-trade-sdk / sol-parser-sdk / solana-streamer
### 7.1 逻辑一致性(已确认)
| 检查项 | sol-parser-sdk | solana-streamer | 说明 |
|--------|----------------|-----------------|------|
| Pump buy 账户数 | 16fill 用 get(9) 填 creator_vault | 16accounts[9]=creator_vault | 与 idl/pumpfun.json 一致 |
| Pump sell 账户数 | 14get(8)=creator_vault | 14accounts[8]=creator_vault | 一致 |
| PumpSwap buy 17/18 | 指令解析 + fill_buy_accounts get(17)/get(18) | 指令解析 accounts.get(17)/get(18) | coin_creator_vault_ata/authority 正确 |
| PumpSwap sell 17/18 | fill_sell_accounts 同左 | 同左 | 一致 |
| 填充顺序 | 先 parselog 或 instruction)→ fill_accounts → push | 指令解析直接写 17/18;无单独 fill 步骤 | 两者均保证事件带齐 creator_vault / coin_creator_vault |
| find_instruction_invoke | 选「账户数最多」的 invoke,保证取到 outer buy/sell | N/A(按当前 instruction 解析) | 正确 |
### 7.2 版本化交易账户解析
- **sol-parser-sdk**`get_instruction_account_getter` 正确支持 versioned tx:先 `account_keys`,再 `loaded_writable_addresses`,再 `loaded_readonly_addresses`,与 Solana 约定一致。
- **solana-streamer**:指令的 `accounts` 为索引,通过 `accounts.get(idx as usize).copied()` 从完整 `accounts: &[Pubkey]` 解析;调用方需传入已包含 loaded 的完整账户列表,否则高索引会得到 `default()`
### 7.3 超低延迟相关
| 项目 | 状态 / 建议 |
|------|-------------|
| solana-streamer 热路径 | 已改为**顺序执行** inner 解析与 swap_data 提取,去掉 `thread::scope` + 双 spawn/join,减少 μs 级开销。 |
| sol-parser-sdk | log 与 instruction 并行(rayon::join);fill 仅在有 invoke 时做;`find_instruction_invoke` 为 O(invokes),单程序单 tx 下可接受。 |
| sol-trade-sdk | 见上文第 4 节;PumpSwap instruction 构建避免 `accounts.clone()` 已列为必须改。 |
### 7.4 建议
- **solana-streamer**:若 gRPC 上游已提供完整 `accounts`(含 loaded),可避免对每笔 tx 做 `accounts.to_vec()`,仅在需要 resize 时克隆,进一步降低分配。
- **三库**:保持 IDL 与账户索引注释同步(pumpfun.json / pump_amm.json),避免后续扩展时索引错位。
---
## 8. 汇总:必须改 vs 建议改
### 必须改(优先处理)
| 序号 | 项 | 位置 |
|------|----|------|
| 1 | 移除 astralane_quic 中 API key 的日志输出 | `src/swqos/astralane_quic.rs` 61、74 行 |
| 2 | SkipServerVerification:改为证书校验或仅限 dev 并文档化 | `src/swqos/astralane_quic.rs` 179181 行 |
| 3 | instruction 中 PDA 等 `.unwrap()` 改为 `Result` 并传播错误 | pumpfun.rs、pumpswap.rs、pumpfun/utils、bonk、raydium_cpmm 等 |
| 4 | PumpSwap 构建 instruction 时避免 `accounts.clone()`,改为移动 | `src/instruction/pumpswap.rs` 232、429 行 |
### 建议改(可分批)
| 序号 | 项 | 位置 |
|------|----|------|
| 5 | PDA 的 `program_id``&&AMM_PROGRAM` 改为 `&AMM_PROGRAM` | `src/instruction/utils/pumpswap.rs` 258、291 行 |
| 6 | PumpFun sell discriminator 改为命名常量 | `src/instruction/pumpfun.rs` 约 259 行 |
| 7 | `ensure_wsol_ata` 拆分;修正 “only support” 等文案 | `src/lib.rs` |
| 8 | 为 instruction 构建与 PDA 增加单元测试 | 新建 tests 或模块下 |
| 9 | 生产日志用 tracing 替代 println!/eprintln! | `src/swqos/astralane.rs` 等 |
---
*报告基于当前仓库与 IDL 的静态阅读;若官方 SDK 或链上程序有未公开变更,建议再与官方实现或链上行为做一次对照验证。*
+10 -3
View File
@@ -16,7 +16,10 @@ use sol_trade_sdk::common::{GasFeeStrategy, TradeConfig};
use sol_trade_sdk::{ use sol_trade_sdk::{
common::AnyResult, common::AnyResult,
swqos::SwqosConfig, swqos::SwqosConfig,
trading::{core::params::{PumpFunParams, DexParamEnum}, factory::DexType}, trading::{
core::params::{DexParamEnum, PumpFunParams},
factory::DexType,
},
SolanaTrade, SolanaTrade,
}; };
use solana_commitment_config::CommitmentConfig; use solana_commitment_config::CommitmentConfig;
@@ -63,7 +66,9 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
loop { loop {
if let Some(event) = queue.pop() { if let Some(event) = queue.pop() {
let run = match &event { let run = match &event {
DexEvent::PumpFunBuy(e) | DexEvent::PumpFunSell(e) | DexEvent::PumpFunBuyExactSolIn(e) => { DexEvent::PumpFunBuy(e)
| DexEvent::PumpFunSell(e)
| DexEvent::PumpFunBuyExactSolIn(e) => {
if !ALREADY_EXECUTED.swap(true, Ordering::SeqCst) { if !ALREADY_EXECUTED.swap(true, Ordering::SeqCst) {
Some(e.clone()) Some(e.clone())
} else { } else {
@@ -123,7 +128,9 @@ async fn pumpfun_copy_trade_with_grpc(
let lookup_table_key = Pubkey::from_str("use_your_lookup_table_key_here").unwrap(); let lookup_table_key = Pubkey::from_str("use_your_lookup_table_key_here").unwrap();
let address_lookup_table_account = let address_lookup_table_account =
fetch_address_lookup_table_account(&client.infrastructure.rpc, &lookup_table_key).await.ok(); fetch_address_lookup_table_account(&client.infrastructure.rpc, &lookup_table_key)
.await
.ok();
let gas_fee_strategy = GasFeeStrategy::new(); let gas_fee_strategy = GasFeeStrategy::new();
gas_fee_strategy.set_global_fee_strategy(150000, 150000, 500000, 500000, 0.001, 0.001); gas_fee_strategy.set_global_fee_strategy(150000, 150000, 500000, 500000, 0.001, 0.001);
+5 -9
View File
@@ -10,7 +10,10 @@ use sol_trade_sdk::common::{
use sol_trade_sdk::{ use sol_trade_sdk::{
common::AnyResult, common::AnyResult,
swqos::SwqosConfig, swqos::SwqosConfig,
trading::{core::params::{BonkParams, DexParamEnum}, factory::DexType}, trading::{
core::params::{BonkParams, DexParamEnum},
factory::DexType,
},
SolanaTrade, SolanaTrade,
}; };
use solana_commitment_config::CommitmentConfig; use solana_commitment_config::CommitmentConfig;
@@ -127,14 +130,7 @@ async fn bonk_copy_trade_with_grpc(trade_info: BonkTradeEvent) -> AnyResult<()>
let recent_blockhash = client.infrastructure.rpc.get_latest_blockhash().await?; let recent_blockhash = client.infrastructure.rpc.get_latest_blockhash().await?;
let gas_fee_strategy = GasFeeStrategy::new(); let gas_fee_strategy = GasFeeStrategy::new();
gas_fee_strategy.set_global_fee_strategy( gas_fee_strategy.set_global_fee_strategy(150000, 150000, 500000, 500000, 0.001, 0.001);
150000,
150000,
500000,
500000,
0.001,
0.001,
);
// Buy tokens // Buy tokens
println!("Buying tokens from Bonk..."); println!("Buying tokens from Bonk...");
+5 -9
View File
@@ -3,7 +3,10 @@ use sol_trade_sdk::common::TradeConfig;
use sol_trade_sdk::{ use sol_trade_sdk::{
common::AnyResult, common::AnyResult,
swqos::SwqosConfig, swqos::SwqosConfig,
trading::{core::params::{BonkParams, DexParamEnum}, factory::DexType}, trading::{
core::params::{BonkParams, DexParamEnum},
factory::DexType,
},
SolanaTrade, SolanaTrade,
}; };
use solana_commitment_config::CommitmentConfig; use solana_commitment_config::CommitmentConfig;
@@ -95,14 +98,7 @@ async fn bonk_sniper_trade_with_shreds(trade_info: BonkTradeEvent) -> AnyResult<
let recent_blockhash = client.infrastructure.rpc.get_latest_blockhash().await?; let recent_blockhash = client.infrastructure.rpc.get_latest_blockhash().await?;
let gas_fee_strategy = sol_trade_sdk::common::GasFeeStrategy::new(); let gas_fee_strategy = sol_trade_sdk::common::GasFeeStrategy::new();
gas_fee_strategy.set_global_fee_strategy( gas_fee_strategy.set_global_fee_strategy(150000, 150000, 500000, 500000, 0.001, 0.001);
150000,
150000,
500000,
500000,
0.001,
0.001,
);
let token_type = if trade_info.quote_token_mint == sol_trade_sdk::constants::USD1_TOKEN_ACCOUNT let token_type = if trade_info.quote_token_mint == sol_trade_sdk::constants::USD1_TOKEN_ACCOUNT
{ {
+49 -40
View File
@@ -12,7 +12,8 @@ use sol_trade_sdk::{
swqos::SwqosConfig, swqos::SwqosConfig,
trading::{ trading::{
core::params::{ core::params::{
BonkParams, PumpFunParams, PumpSwapParams, RaydiumAmmV4Params, RaydiumCpmmParams, DexParamEnum, BonkParams, DexParamEnum, PumpFunParams, PumpSwapParams, RaydiumAmmV4Params,
RaydiumCpmmParams,
}, },
factory::DexType, factory::DexType,
}, },
@@ -531,7 +532,7 @@ async fn handle_buy(
let client = initialize_real_client().await?; let client = initialize_real_client().await?;
let (create_mint_ata, use_seed, owner_pubkey, amount_f64, decimals) = let (create_mint_ata, use_seed, owner_pubkey, _amount_f64, _decimals) =
check_mint_ata(&client, mint).await?; check_mint_ata(&client, mint).await?;
match dex { match dex {
@@ -565,7 +566,7 @@ async fn handle_buy_rv4(
slippage: Option<u64>, slippage: Option<u64>,
) -> Result<(), Box<dyn std::error::Error>> { ) -> Result<(), Box<dyn std::error::Error>> {
let client = initialize_real_client().await?; let client = initialize_real_client().await?;
let (create_mint_ata, use_seed, owner_pubkey, amount_f64, decimals) = let (create_mint_ata, use_seed, owner_pubkey, _amount_f64, _decimals) =
check_mint_ata(&client, mint).await?; check_mint_ata(&client, mint).await?;
handle_buy_raydium_v4(mint, amm, sol_amount, slippage, create_mint_ata, use_seed, owner_pubkey) handle_buy_raydium_v4(mint, amm, sol_amount, slippage, create_mint_ata, use_seed, owner_pubkey)
.await?; .await?;
@@ -579,7 +580,7 @@ async fn handle_buy_rcpmm(
slippage: Option<u64>, slippage: Option<u64>,
) -> Result<(), Box<dyn std::error::Error>> { ) -> Result<(), Box<dyn std::error::Error>> {
let client = initialize_real_client().await?; let client = initialize_real_client().await?;
let (create_mint_ata, use_seed, owner_pubkey, amount_f64, decimals) = let (create_mint_ata, use_seed, owner_pubkey, _amount_f64, _decimals) =
check_mint_ata(&client, mint).await?; check_mint_ata(&client, mint).await?;
handle_buy_raydium_cpmm( handle_buy_raydium_cpmm(
mint, mint,
@@ -599,8 +600,8 @@ async fn handle_buy_pumpfun(
sol_amount: f64, sol_amount: f64,
slippage: Option<u64>, slippage: Option<u64>,
create_mint_ata: bool, create_mint_ata: bool,
use_seed: bool, _use_seed: bool,
owner_pubkey: Pubkey, _owner_pubkey: Pubkey,
) -> Result<(), Box<dyn std::error::Error>> { ) -> Result<(), Box<dyn std::error::Error>> {
println!("🔥 BUY PUMPFUN COMMAND"); println!("🔥 BUY PUMPFUN COMMAND");
println!(" Token Mint: {}", mint); println!(" Token Mint: {}", mint);
@@ -655,7 +656,7 @@ async fn handle_buy_pumpswap(
sol_amount: f64, sol_amount: f64,
slippage: Option<u64>, slippage: Option<u64>,
create_mint_ata: bool, create_mint_ata: bool,
use_seed: bool, _use_seed: bool,
_owner_pubkey: Pubkey, _owner_pubkey: Pubkey,
) -> Result<(), Box<dyn std::error::Error>> { ) -> Result<(), Box<dyn std::error::Error>> {
let client = initialize_real_client().await?; let client = initialize_real_client().await?;
@@ -710,8 +711,8 @@ async fn handle_buy_bonk(
sol_amount: f64, sol_amount: f64,
slippage: Option<u64>, slippage: Option<u64>,
create_mint_ata: bool, create_mint_ata: bool,
use_seed: bool, _use_seed: bool,
owner_pubkey: Pubkey, _owner_pubkey: Pubkey,
) -> Result<(), Box<dyn std::error::Error>> { ) -> Result<(), Box<dyn std::error::Error>> {
let client = initialize_real_client().await?; let client = initialize_real_client().await?;
println!("🔥 BUY BONK COMMAND"); println!("🔥 BUY BONK COMMAND");
@@ -721,7 +722,8 @@ async fn handle_buy_bonk(
println!(" Slippage: {}%", slippage.unwrap()); println!(" Slippage: {}%", slippage.unwrap());
} }
let mint_pubkey = Pubkey::from_str(mint)?; let mint_pubkey = Pubkey::from_str(mint)?;
let param = BonkParams::from_mint_by_rpc(&client.infrastructure.rpc, &mint_pubkey, false).await?; let param =
BonkParams::from_mint_by_rpc(&client.infrastructure.rpc, &mint_pubkey, false).await?;
let recent_blockhash = client.infrastructure.rpc.get_latest_blockhash().await?; let recent_blockhash = client.infrastructure.rpc.get_latest_blockhash().await?;
let sol_lamports = sol_str_to_lamports(sol_amount.to_string().as_str()).unwrap(); let sol_lamports = sol_str_to_lamports(sol_amount.to_string().as_str()).unwrap();
@@ -766,8 +768,8 @@ async fn handle_buy_raydium_v4(
sol_amount: f64, sol_amount: f64,
slippage: Option<u64>, slippage: Option<u64>,
create_mint_ata: bool, create_mint_ata: bool,
use_seed: bool, _use_seed: bool,
owner_pubkey: Pubkey, _owner_pubkey: Pubkey,
) -> Result<(), Box<dyn std::error::Error>> { ) -> Result<(), Box<dyn std::error::Error>> {
let client = initialize_real_client().await?; let client = initialize_real_client().await?;
println!("🔥 BUY RAYDIUM V4 COMMAND"); println!("🔥 BUY RAYDIUM V4 COMMAND");
@@ -780,7 +782,8 @@ async fn handle_buy_raydium_v4(
let mint_pubkey = Pubkey::from_str(mint)?; let mint_pubkey = Pubkey::from_str(mint)?;
let amm_pubkey = Pubkey::from_str(amm)?; let amm_pubkey = Pubkey::from_str(amm)?;
let param = RaydiumAmmV4Params::from_amm_address_by_rpc(&client.infrastructure.rpc, amm_pubkey).await?; let param =
RaydiumAmmV4Params::from_amm_address_by_rpc(&client.infrastructure.rpc, amm_pubkey).await?;
let recent_blockhash = client.infrastructure.rpc.get_latest_blockhash().await?; let recent_blockhash = client.infrastructure.rpc.get_latest_blockhash().await?;
let sol_lamports = sol_str_to_lamports(sol_amount.to_string().as_str()).unwrap(); let sol_lamports = sol_str_to_lamports(sol_amount.to_string().as_str()).unwrap();
@@ -825,8 +828,8 @@ async fn handle_buy_raydium_cpmm(
sol_amount: f64, sol_amount: f64,
slippage: Option<u64>, slippage: Option<u64>,
create_mint_ata: bool, create_mint_ata: bool,
use_seed: bool, _use_seed: bool,
owner_pubkey: Pubkey, _owner_pubkey: Pubkey,
) -> Result<(), Box<dyn std::error::Error>> { ) -> Result<(), Box<dyn std::error::Error>> {
let client = initialize_real_client().await?; let client = initialize_real_client().await?;
println!("🔥 BUY RAYDIUM CPMM COMMAND"); println!("🔥 BUY RAYDIUM CPMM COMMAND");
@@ -839,7 +842,9 @@ async fn handle_buy_raydium_cpmm(
let mint_pubkey = Pubkey::from_str(mint)?; let mint_pubkey = Pubkey::from_str(mint)?;
let pool_pubkey = Pubkey::from_str(pool_address)?; let pool_pubkey = Pubkey::from_str(pool_address)?;
let param = RaydiumCpmmParams::from_pool_address_by_rpc(&client.infrastructure.rpc, &pool_pubkey).await?; let param =
RaydiumCpmmParams::from_pool_address_by_rpc(&client.infrastructure.rpc, &pool_pubkey)
.await?;
let recent_blockhash = client.infrastructure.rpc.get_latest_blockhash().await?; let recent_blockhash = client.infrastructure.rpc.get_latest_blockhash().await?;
let sol_lamports = sol_str_to_lamports(sol_amount.to_string().as_str()).unwrap(); let sol_lamports = sol_str_to_lamports(sol_amount.to_string().as_str()).unwrap();
@@ -992,11 +997,11 @@ async fn handle_sell_pumpfun(
mint: &str, mint: &str,
token_amount: Option<f64>, token_amount: Option<f64>,
slippage: Option<u64>, slippage: Option<u64>,
create_mint_ata: bool, _create_mint_ata: bool,
use_seed: bool, _use_seed: bool,
owner_pubkey: Pubkey, _owner_pubkey: Pubkey,
amount_f64: f64, amount_f64: f64,
decimals: u8, _decimals: u8,
) -> Result<(), Box<dyn std::error::Error>> { ) -> Result<(), Box<dyn std::error::Error>> {
let amount = if token_amount.is_some() { token_amount.unwrap() } else { amount_f64 }; let amount = if token_amount.is_some() { token_amount.unwrap() } else { amount_f64 };
@@ -1053,11 +1058,11 @@ async fn handle_sell_pumpswap(
mint: &str, mint: &str,
token_amount: Option<f64>, token_amount: Option<f64>,
slippage: Option<u64>, slippage: Option<u64>,
create_mint_ata: bool, _create_mint_ata: bool,
use_seed: bool, _use_seed: bool,
owner_pubkey: Pubkey, _owner_pubkey: Pubkey,
amount_f64: f64, amount_f64: f64,
decimals: u8, _decimals: u8,
) -> Result<(), Box<dyn std::error::Error>> { ) -> Result<(), Box<dyn std::error::Error>> {
let amount = if token_amount.is_some() { token_amount.unwrap() } else { amount_f64 }; let amount = if token_amount.is_some() { token_amount.unwrap() } else { amount_f64 };
println!("🔥 SELL PUMPSWAP COMMAND"); println!("🔥 SELL PUMPSWAP COMMAND");
@@ -1097,7 +1102,7 @@ async fn handle_sell_pumpswap(
match client.sell(sell_params).await { match client.sell(sell_params).await {
Ok((_, signature, _)) => { Ok((_, signature, _)) => {
println!(" ✅ Successfully sold tokens from PumpSwap!"); println!(" ✅ Successfully sold tokens from PumpSwap!");
println!(" ✅ Transaction Signature: {:?}", signature); println!(" ✅ Transaction Signature: {:?}", signature);
} }
Err(e) => { Err(e) => {
println!(" ❌ Failed to sell tokens from PumpSwap: {}", e); println!(" ❌ Failed to sell tokens from PumpSwap: {}", e);
@@ -1111,11 +1116,11 @@ async fn handle_sell_bonk(
mint: &str, mint: &str,
token_amount: Option<f64>, token_amount: Option<f64>,
slippage: Option<u64>, slippage: Option<u64>,
create_mint_ata: bool, _create_mint_ata: bool,
use_seed: bool, _use_seed: bool,
owner_pubkey: Pubkey, _owner_pubkey: Pubkey,
amount_f64: f64, amount_f64: f64,
decimals: u8, _decimals: u8,
) -> Result<(), Box<dyn std::error::Error>> { ) -> Result<(), Box<dyn std::error::Error>> {
let amount = if token_amount.is_some() { token_amount.unwrap() } else { amount_f64 }; let amount = if token_amount.is_some() { token_amount.unwrap() } else { amount_f64 };
println!("🔥 SELL PUMPSWAP COMMAND"); println!("🔥 SELL PUMPSWAP COMMAND");
@@ -1126,7 +1131,8 @@ async fn handle_sell_bonk(
} }
let client = initialize_real_client().await?; let client = initialize_real_client().await?;
let mint_pubkey = Pubkey::from_str(mint)?; let mint_pubkey = Pubkey::from_str(mint)?;
let param = BonkParams::from_mint_by_rpc(&client.infrastructure.rpc, &mint_pubkey, false).await?; let param =
BonkParams::from_mint_by_rpc(&client.infrastructure.rpc, &mint_pubkey, false).await?;
let recent_blockhash = client.infrastructure.rpc.get_latest_blockhash().await?; let recent_blockhash = client.infrastructure.rpc.get_latest_blockhash().await?;
let gas_fee_strategy = sol_trade_sdk::common::GasFeeStrategy::new(); let gas_fee_strategy = sol_trade_sdk::common::GasFeeStrategy::new();
@@ -1170,11 +1176,11 @@ async fn handle_sell_raydium_v4(
mint: &str, mint: &str,
token_amount: Option<f64>, token_amount: Option<f64>,
slippage: Option<u64>, slippage: Option<u64>,
create_mint_ata: bool, _create_mint_ata: bool,
use_seed: bool, _use_seed: bool,
owner_pubkey: Pubkey, _owner_pubkey: Pubkey,
amount_f64: f64, amount_f64: f64,
decimals: u8, _decimals: u8,
) -> Result<(), Box<dyn std::error::Error>> { ) -> Result<(), Box<dyn std::error::Error>> {
let amount = if token_amount.is_some() { token_amount.unwrap() } else { amount_f64 }; let amount = if token_amount.is_some() { token_amount.unwrap() } else { amount_f64 };
println!("🔥 SELL RAYDIUM V4 COMMAND"); println!("🔥 SELL RAYDIUM V4 COMMAND");
@@ -1187,7 +1193,8 @@ async fn handle_sell_raydium_v4(
let client = initialize_real_client().await?; let client = initialize_real_client().await?;
let amm_pubkey = Pubkey::from_str(amm)?; let amm_pubkey = Pubkey::from_str(amm)?;
let mint_pubkey = Pubkey::from_str(mint)?; let mint_pubkey = Pubkey::from_str(mint)?;
let param = RaydiumAmmV4Params::from_amm_address_by_rpc(&client.infrastructure.rpc, amm_pubkey).await?; let param =
RaydiumAmmV4Params::from_amm_address_by_rpc(&client.infrastructure.rpc, amm_pubkey).await?;
let recent_blockhash = client.infrastructure.rpc.get_latest_blockhash().await?; let recent_blockhash = client.infrastructure.rpc.get_latest_blockhash().await?;
let gas_fee_strategy = sol_trade_sdk::common::GasFeeStrategy::new(); let gas_fee_strategy = sol_trade_sdk::common::GasFeeStrategy::new();
@@ -1231,11 +1238,11 @@ async fn handle_sell_raydium_cpmm(
pool_address: &str, pool_address: &str,
token_amount: Option<f64>, token_amount: Option<f64>,
slippage: Option<u64>, slippage: Option<u64>,
create_mint_ata: bool, _create_mint_ata: bool,
use_seed: bool, _use_seed: bool,
owner_pubkey: Pubkey, _owner_pubkey: Pubkey,
amount_f64: f64, amount_f64: f64,
decimals: u8, _decimals: u8,
) -> Result<(), Box<dyn std::error::Error>> { ) -> Result<(), Box<dyn std::error::Error>> {
let amount = if token_amount.is_some() { token_amount.unwrap() } else { amount_f64 }; let amount = if token_amount.is_some() { token_amount.unwrap() } else { amount_f64 };
println!("🔥 SELL RAYDIUM CPMM COMMAND"); println!("🔥 SELL RAYDIUM CPMM COMMAND");
@@ -1248,7 +1255,9 @@ async fn handle_sell_raydium_cpmm(
let client = initialize_real_client().await?; let client = initialize_real_client().await?;
let pool_pubkey = Pubkey::from_str(pool_address)?; let pool_pubkey = Pubkey::from_str(pool_address)?;
let mint_pubkey = Pubkey::from_str(mint)?; let mint_pubkey = Pubkey::from_str(mint)?;
let param = RaydiumCpmmParams::from_pool_address_by_rpc(&client.infrastructure.rpc, &pool_pubkey).await?; let param =
RaydiumCpmmParams::from_pool_address_by_rpc(&client.infrastructure.rpc, &pool_pubkey)
.await?;
let recent_blockhash = client.infrastructure.rpc.get_latest_blockhash().await?; let recent_blockhash = client.infrastructure.rpc.get_latest_blockhash().await?;
let gas_fee_strategy = sol_trade_sdk::common::GasFeeStrategy::new(); let gas_fee_strategy = sol_trade_sdk::common::GasFeeStrategy::new();
@@ -1,7 +1,13 @@
use sol_trade_sdk::{ use sol_trade_sdk::{
SolanaTrade, TradeTokenType, common::{ common::{
AnyResult, TradeConfig, fast_fn::get_associated_token_address_with_program_id_fast_use_seed fast_fn::get_associated_token_address_with_program_id_fast_use_seed, AnyResult, TradeConfig,
}, swqos::SwqosConfig, trading::{core::params::{MeteoraDammV2Params, DexParamEnum}, factory::DexType} },
swqos::SwqosConfig,
trading::{
core::params::{DexParamEnum, MeteoraDammV2Params},
factory::DexType,
},
SolanaTrade, TradeTokenType,
}; };
use solana_commitment_config::CommitmentConfig; use solana_commitment_config::CommitmentConfig;
use solana_sdk::signature::Keypair; use solana_sdk::signature::Keypair;
@@ -32,7 +38,8 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
slippage_basis_points: slippage_basis_points, slippage_basis_points: slippage_basis_points,
recent_blockhash: Some(recent_blockhash), recent_blockhash: Some(recent_blockhash),
extension_params: DexParamEnum::MeteoraDammV2( extension_params: DexParamEnum::MeteoraDammV2(
MeteoraDammV2Params::from_pool_address_by_rpc(&client.infrastructure.rpc, &pool).await?, MeteoraDammV2Params::from_pool_address_by_rpc(&client.infrastructure.rpc, &pool)
.await?,
), ),
address_lookup_table_account: None, address_lookup_table_account: None,
wait_transaction_confirmed: true, wait_transaction_confirmed: true,
@@ -54,7 +61,12 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
let rpc = client.infrastructure.rpc.clone(); let rpc = client.infrastructure.rpc.clone();
let payer = client.payer.pubkey(); let payer = client.payer.pubkey();
let program_id = sol_trade_sdk::constants::TOKEN_PROGRAM; let program_id = sol_trade_sdk::constants::TOKEN_PROGRAM;
let account = get_associated_token_address_with_program_id_fast_use_seed(&payer, &mint_pubkey, &program_id, client.use_seed_optimize); let account = get_associated_token_address_with_program_id_fast_use_seed(
&payer,
&mint_pubkey,
&program_id,
client.use_seed_optimize,
);
let balance = rpc.get_token_account_balance(&account).await?; let balance = rpc.get_token_account_balance(&account).await?;
let amount_token = balance.amount.parse::<u64>().unwrap(); let amount_token = balance.amount.parse::<u64>().unwrap();
println!("Token balance: {}", amount_token); println!("Token balance: {}", amount_token);
@@ -67,7 +79,8 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
recent_blockhash: Some(recent_blockhash), recent_blockhash: Some(recent_blockhash),
with_tip: false, with_tip: false,
extension_params: DexParamEnum::MeteoraDammV2( extension_params: DexParamEnum::MeteoraDammV2(
MeteoraDammV2Params::from_pool_address_by_rpc(&client.infrastructure.rpc, &pool).await?, MeteoraDammV2Params::from_pool_address_by_rpc(&client.infrastructure.rpc, &pool)
.await?,
), ),
address_lookup_table_account: None, address_lookup_table_account: None,
wait_transaction_confirmed: true, wait_transaction_confirmed: true,
+6 -4
View File
@@ -3,7 +3,8 @@ use sol_trade_sdk::{
common::{AnyResult, TradeConfig}, common::{AnyResult, TradeConfig},
swqos::SwqosConfig, swqos::SwqosConfig,
trading::{ trading::{
core::params::{PumpSwapParams, DexParamEnum}, factory::DexType, core::params::{DexParamEnum, PumpSwapParams},
factory::DexType,
InstructionMiddleware, MiddlewareManager, InstructionMiddleware, MiddlewareManager,
}, },
SolanaTrade, TradeTokenType, SolanaTrade, TradeTokenType,
@@ -30,7 +31,7 @@ impl InstructionMiddleware for CustomMiddleware {
fn process_protocol_instructions( fn process_protocol_instructions(
&self, &self,
protocol_instructions: Vec<Instruction>, protocol_instructions: Vec<Instruction>,
_protocol_name: String, _protocol_name: &str,
_is_buy: bool, _is_buy: bool,
) -> Result<Vec<Instruction>> { ) -> Result<Vec<Instruction>> {
// do anything you want here // do anything you want here
@@ -41,7 +42,7 @@ impl InstructionMiddleware for CustomMiddleware {
fn process_full_instructions( fn process_full_instructions(
&self, &self,
full_instructions: Vec<Instruction>, full_instructions: Vec<Instruction>,
_protocol_name: String, _protocol_name: &str,
_is_buy: bool, _is_buy: bool,
) -> Result<Vec<Instruction>> { ) -> Result<Vec<Instruction>> {
// do anything you want here // do anything you want here
@@ -91,7 +92,8 @@ async fn test_middleware() -> AnyResult<()> {
slippage_basis_points: slippage_basis_points, slippage_basis_points: slippage_basis_points,
recent_blockhash: Some(recent_blockhash), recent_blockhash: Some(recent_blockhash),
extension_params: DexParamEnum::PumpSwap( extension_params: DexParamEnum::PumpSwap(
PumpSwapParams::from_pool_address_by_rpc(&client.infrastructure.rpc, &pool_address).await?, PumpSwapParams::from_pool_address_by_rpc(&client.infrastructure.rpc, &pool_address)
.await?,
), ),
address_lookup_table_account: None, address_lookup_table_account: None,
wait_transaction_confirmed: true, wait_transaction_confirmed: true,
+7 -2
View File
@@ -16,7 +16,10 @@ use sol_trade_sdk::TradeTokenType;
use sol_trade_sdk::{ use sol_trade_sdk::{
common::AnyResult, common::AnyResult,
swqos::SwqosConfig, swqos::SwqosConfig,
trading::{core::params::{PumpFunParams, DexParamEnum}, factory::DexType}, trading::{
core::params::{DexParamEnum, PumpFunParams},
factory::DexType,
},
SolanaTrade, SolanaTrade,
}; };
use solana_commitment_config::CommitmentConfig; use solana_commitment_config::CommitmentConfig;
@@ -62,7 +65,9 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
loop { loop {
if let Some(event) = queue.pop() { if let Some(event) = queue.pop() {
let run = match &event { let run = match &event {
DexEvent::PumpFunBuy(e) | DexEvent::PumpFunSell(e) | DexEvent::PumpFunBuyExactSolIn(e) => { DexEvent::PumpFunBuy(e)
| DexEvent::PumpFunSell(e)
| DexEvent::PumpFunBuyExactSolIn(e) => {
if !ALREADY_EXECUTED.swap(true, Ordering::SeqCst) { if !ALREADY_EXECUTED.swap(true, Ordering::SeqCst) {
Some(e.clone()) Some(e.clone())
} else { } else {
+14 -7
View File
@@ -2,7 +2,10 @@
//! //!
//! 收到 PumpFun 买卖事件后,用事件中的参数(含 is_cashback_coin)构造交易并执行一次买+卖。 //! 收到 PumpFun 买卖事件后,用事件中的参数(含 is_cashback_coin)构造交易并执行一次买+卖。
use std::sync::{atomic::{AtomicBool, Ordering}, Arc}; use std::sync::{
atomic::{AtomicBool, Ordering},
Arc,
};
use sol_parser_sdk::grpc::{ use sol_parser_sdk::grpc::{
AccountFilter, ClientConfig, EventType, EventTypeFilter, OrderMode, Protocol, AccountFilter, ClientConfig, EventType, EventTypeFilter, OrderMode, Protocol,
@@ -16,7 +19,10 @@ use sol_trade_sdk::TradeTokenType;
use sol_trade_sdk::{ use sol_trade_sdk::{
common::AnyResult, common::AnyResult,
swqos::SwqosConfig, swqos::SwqosConfig,
trading::{core::params::{PumpFunParams, DexParamEnum}, factory::DexType}, trading::{
core::params::{DexParamEnum, PumpFunParams},
factory::DexType,
},
SolanaTrade, SolanaTrade,
}; };
use solana_commitment_config::CommitmentConfig; use solana_commitment_config::CommitmentConfig;
@@ -65,7 +71,9 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
loop { loop {
if let Some(event) = queue.pop() { if let Some(event) = queue.pop() {
let run = match &event { let run = match &event {
DexEvent::PumpFunBuy(e) | DexEvent::PumpFunSell(e) | DexEvent::PumpFunBuyExactSolIn(e) => { DexEvent::PumpFunBuy(e)
| DexEvent::PumpFunSell(e)
| DexEvent::PumpFunBuyExactSolIn(e) => {
if !ALREADY_EXECUTED.swap(true, Ordering::SeqCst) { if !ALREADY_EXECUTED.swap(true, Ordering::SeqCst) {
Some(e.clone()) Some(e.clone())
} else { } else {
@@ -102,16 +110,15 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
async fn create_solana_trade_client() -> AnyResult<SolanaTrade> { async fn create_solana_trade_client() -> AnyResult<SolanaTrade> {
let payer = Keypair::from_base58_string("use_your_payer_keypair_here"); let payer = Keypair::from_base58_string("use_your_payer_keypair_here");
let rpc_url = std::env::var("RPC_URL").unwrap_or_else(|_| "https://api.mainnet-beta.solana.com".to_string()); let rpc_url = std::env::var("RPC_URL")
.unwrap_or_else(|_| "https://api.mainnet-beta.solana.com".to_string());
let commitment = CommitmentConfig::confirmed(); let commitment = CommitmentConfig::confirmed();
let swqos_configs: Vec<SwqosConfig> = vec![SwqosConfig::Default(rpc_url.clone())]; let swqos_configs: Vec<SwqosConfig> = vec![SwqosConfig::Default(rpc_url.clone())];
let trade_config = TradeConfig::new(rpc_url, swqos_configs, commitment); let trade_config = TradeConfig::new(rpc_url, swqos_configs, commitment);
Ok(SolanaTrade::new(Arc::new(payer), trade_config).await) Ok(SolanaTrade::new(Arc::new(payer), trade_config).await)
} }
async fn pumpfun_copy_trade( async fn pumpfun_copy_trade(e: sol_parser_sdk::core::events::PumpFunTradeEvent) -> AnyResult<()> {
e: sol_parser_sdk::core::events::PumpFunTradeEvent,
) -> AnyResult<()> {
let client = create_solana_trade_client().await?; let client = create_solana_trade_client().await?;
let mint_pubkey = e.mint; let mint_pubkey = e.mint;
let slippage_basis_points = Some(100u64); let slippage_basis_points = Some(100u64);
+11 -6
View File
@@ -3,7 +3,10 @@
//! 监听创建者首次买入(Create 后同笔/首笔 Buyis_created_buy == true), //! 监听创建者首次买入(Create 后同笔/首笔 Buyis_created_buy == true),
//! 用事件参数(含 is_cashback_coin)构造 from_dev_trade 并执行一次买+卖。 //! 用事件参数(含 is_cashback_coin)构造 from_dev_trade 并执行一次买+卖。
use std::sync::{atomic::{AtomicBool, Ordering}, Arc}; use std::sync::{
atomic::{AtomicBool, Ordering},
Arc,
};
use sol_parser_sdk::grpc::{ use sol_parser_sdk::grpc::{
AccountFilter, ClientConfig, EventType, EventTypeFilter, OrderMode, Protocol, AccountFilter, ClientConfig, EventType, EventTypeFilter, OrderMode, Protocol,
@@ -16,7 +19,10 @@ use sol_trade_sdk::TradeTokenType;
use sol_trade_sdk::{ use sol_trade_sdk::{
common::AnyResult, common::AnyResult,
swqos::SwqosConfig, swqos::SwqosConfig,
trading::{core::params::{PumpFunParams, DexParamEnum}, factory::DexType}, trading::{
core::params::{DexParamEnum, PumpFunParams},
factory::DexType,
},
SolanaTrade, SolanaTrade,
}; };
use solana_commitment_config::CommitmentConfig; use solana_commitment_config::CommitmentConfig;
@@ -94,16 +100,15 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
async fn create_solana_trade_client() -> AnyResult<SolanaTrade> { async fn create_solana_trade_client() -> AnyResult<SolanaTrade> {
let payer = Keypair::from_base58_string("use_your_payer_keypair_here"); let payer = Keypair::from_base58_string("use_your_payer_keypair_here");
let rpc_url = std::env::var("RPC_URL").unwrap_or_else(|_| "https://api.mainnet-beta.solana.com".to_string()); let rpc_url = std::env::var("RPC_URL")
.unwrap_or_else(|_| "https://api.mainnet-beta.solana.com".to_string());
let commitment = CommitmentConfig::confirmed(); let commitment = CommitmentConfig::confirmed();
let swqos_configs: Vec<SwqosConfig> = vec![SwqosConfig::Default(rpc_url.clone())]; let swqos_configs: Vec<SwqosConfig> = vec![SwqosConfig::Default(rpc_url.clone())];
let trade_config = TradeConfig::new(rpc_url, swqos_configs, commitment); let trade_config = TradeConfig::new(rpc_url, swqos_configs, commitment);
Ok(SolanaTrade::new(Arc::new(payer), trade_config).await) Ok(SolanaTrade::new(Arc::new(payer), trade_config).await)
} }
async fn pumpfun_sniper_trade( async fn pumpfun_sniper_trade(e: sol_parser_sdk::core::events::PumpFunTradeEvent) -> AnyResult<()> {
e: sol_parser_sdk::core::events::PumpFunTradeEvent,
) -> AnyResult<()> {
let client = create_solana_trade_client().await?; let client = create_solana_trade_client().await?;
let mint_pubkey = e.mint; let mint_pubkey = e.mint;
let slippage_basis_points = Some(300u64); let slippage_basis_points = Some(300u64);
+15 -4
View File
@@ -1,7 +1,13 @@
use sol_trade_sdk::{ use sol_trade_sdk::{
SolanaTrade, TradeTokenType, common::{ common::{
AnyResult, TradeConfig, fast_fn::get_associated_token_address_with_program_id_fast_use_seed fast_fn::get_associated_token_address_with_program_id_fast_use_seed, AnyResult, TradeConfig,
}, swqos::SwqosConfig, trading::{core::params::{PumpSwapParams, DexParamEnum}, factory::DexType} },
swqos::SwqosConfig,
trading::{
core::params::{DexParamEnum, PumpSwapParams},
factory::DexType,
},
SolanaTrade, TradeTokenType,
}; };
use solana_commitment_config::CommitmentConfig; use solana_commitment_config::CommitmentConfig;
use solana_sdk::signature::Keypair; use solana_sdk::signature::Keypair;
@@ -54,7 +60,12 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
let rpc = client.infrastructure.rpc.clone(); let rpc = client.infrastructure.rpc.clone();
let payer = client.payer.pubkey(); let payer = client.payer.pubkey();
let program_id = sol_trade_sdk::constants::TOKEN_PROGRAM_2022; let program_id = sol_trade_sdk::constants::TOKEN_PROGRAM_2022;
let account = get_associated_token_address_with_program_id_fast_use_seed(&payer, &mint_pubkey, &program_id, client.use_seed_optimize); let account = get_associated_token_address_with_program_id_fast_use_seed(
&payer,
&mint_pubkey,
&program_id,
client.use_seed_optimize,
);
let balance = rpc.get_token_account_balance(&account).await?; let balance = rpc.get_token_account_balance(&account).await?;
let amount_token = balance.amount.parse::<u64>().unwrap(); let amount_token = balance.amount.parse::<u64>().unwrap();
let sell_params = sol_trade_sdk::TradeSellParams { let sell_params = sol_trade_sdk::TradeSellParams {
+11 -3
View File
@@ -1,11 +1,14 @@
use sol_trade_sdk::common::fast_fn::get_associated_token_address_with_program_id_fast_use_seed; use sol_trade_sdk::common::fast_fn::get_associated_token_address_with_program_id_fast_use_seed;
use sol_trade_sdk::common::TradeConfig; use sol_trade_sdk::common::TradeConfig;
use sol_trade_sdk::TradeTokenType;
use sol_trade_sdk::instruction::utils::pumpswap::fetch_pool; use sol_trade_sdk::instruction::utils::pumpswap::fetch_pool;
use sol_trade_sdk::TradeTokenType;
use sol_trade_sdk::{ use sol_trade_sdk::{
common::AnyResult, common::AnyResult,
swqos::SwqosConfig, swqos::SwqosConfig,
trading::{core::params::{PumpSwapParams, DexParamEnum}, factory::DexType}, trading::{
core::params::{DexParamEnum, PumpSwapParams},
factory::DexType,
},
SolanaTrade, SolanaTrade,
}; };
use solana_commitment_config::CommitmentConfig; use solana_commitment_config::CommitmentConfig;
@@ -244,7 +247,12 @@ async fn pumpswap_trade_with_grpc(
} else { } else {
params.quote_token_program params.quote_token_program
}; };
let account = get_associated_token_address_with_program_id_fast_use_seed(&payer, &mint_pubkey, &program_id, client.use_seed_optimize); let account = get_associated_token_address_with_program_id_fast_use_seed(
&payer,
&mint_pubkey,
&program_id,
client.use_seed_optimize,
);
let balance = rpc.get_token_account_balance(&account).await?; let balance = rpc.get_token_account_balance(&account).await?;
let amount_token = balance.amount.parse::<u64>().unwrap(); let amount_token = balance.amount.parse::<u64>().unwrap();
let sell_params = sol_trade_sdk::TradeSellParams { let sell_params = sol_trade_sdk::TradeSellParams {
+14 -12
View File
@@ -2,7 +2,10 @@ use sol_trade_sdk::common::fast_fn::get_associated_token_address_with_program_id
use sol_trade_sdk::{ use sol_trade_sdk::{
common::AnyResult, common::AnyResult,
swqos::SwqosConfig, swqos::SwqosConfig,
trading::{core::params::{RaydiumAmmV4Params, DexParamEnum}, factory::DexType}, trading::{
core::params::{DexParamEnum, RaydiumAmmV4Params},
factory::DexType,
},
SolanaTrade, SolanaTrade,
}; };
use sol_trade_sdk::{ use sol_trade_sdk::{
@@ -122,8 +125,12 @@ async fn raydium_amm_v4_copy_trade_with_grpc(trade_info: RaydiumAmmV4SwapEvent)
let recent_blockhash = client.infrastructure.rpc.get_latest_blockhash().await?; let recent_blockhash = client.infrastructure.rpc.get_latest_blockhash().await?;
let amm_info = fetch_amm_info(&client.infrastructure.rpc, trade_info.amm).await?; let amm_info = fetch_amm_info(&client.infrastructure.rpc, trade_info.amm).await?;
let (coin_reserve, pc_reserve) = let (coin_reserve, pc_reserve) = get_multi_token_balances(
get_multi_token_balances(&client.infrastructure.rpc, &amm_info.token_coin, &amm_info.token_pc).await?; &client.infrastructure.rpc,
&amm_info.token_coin,
&amm_info.token_pc,
)
.await?;
let mint_pubkey = if amm_info.pc_mint == sol_trade_sdk::constants::WSOL_TOKEN_ACCOUNT let mint_pubkey = if amm_info.pc_mint == sol_trade_sdk::constants::WSOL_TOKEN_ACCOUNT
|| amm_info.pc_mint == sol_trade_sdk::constants::USDC_TOKEN_ACCOUNT || amm_info.pc_mint == sol_trade_sdk::constants::USDC_TOKEN_ACCOUNT
{ {
@@ -142,14 +149,7 @@ async fn raydium_amm_v4_copy_trade_with_grpc(trade_info: RaydiumAmmV4SwapEvent)
); );
let gas_fee_strategy = sol_trade_sdk::common::GasFeeStrategy::new(); let gas_fee_strategy = sol_trade_sdk::common::GasFeeStrategy::new();
gas_fee_strategy.set_global_fee_strategy( gas_fee_strategy.set_global_fee_strategy(150000, 150000, 500000, 500000, 0.001, 0.001);
150000,
150000,
500000,
500000,
0.001,
0.001,
);
// Buy tokens // Buy tokens
println!("Buying tokens from Raydium_amm_v4..."); println!("Buying tokens from Raydium_amm_v4...");
@@ -194,7 +194,9 @@ async fn raydium_amm_v4_copy_trade_with_grpc(trade_info: RaydiumAmmV4SwapEvent)
let amount_token = balance.amount.parse::<u64>().unwrap(); let amount_token = balance.amount.parse::<u64>().unwrap();
println!("Selling {} tokens", amount_token); println!("Selling {} tokens", amount_token);
let params = RaydiumAmmV4Params::from_amm_address_by_rpc(&client.infrastructure.rpc, trade_info.amm).await?; let params =
RaydiumAmmV4Params::from_amm_address_by_rpc(&client.infrastructure.rpc, trade_info.amm)
.await?;
let sell_params = sol_trade_sdk::TradeSellParams { let sell_params = sol_trade_sdk::TradeSellParams {
dex_type: DexType::RaydiumAmmV4, dex_type: DexType::RaydiumAmmV4,
output_token_type: if is_wsol { TradeTokenType::WSOL } else { TradeTokenType::USDC }, output_token_type: if is_wsol { TradeTokenType::WSOL } else { TradeTokenType::USDC },
+13 -7
View File
@@ -1,7 +1,7 @@
use sol_trade_sdk::common::spl_associated_token_account::get_associated_token_address; use sol_trade_sdk::common::spl_associated_token_account::get_associated_token_address;
use sol_trade_sdk::common::TradeConfig; use sol_trade_sdk::common::TradeConfig;
use sol_trade_sdk::constants::{WSOL_TOKEN_ACCOUNT, USDC_TOKEN_ACCOUNT}; use sol_trade_sdk::constants::{USDC_TOKEN_ACCOUNT, WSOL_TOKEN_ACCOUNT};
use sol_trade_sdk::trading::core::params::{RaydiumCpmmParams, DexParamEnum}; use sol_trade_sdk::trading::core::params::{DexParamEnum, RaydiumCpmmParams};
use sol_trade_sdk::trading::factory::DexType; use sol_trade_sdk::trading::factory::DexType;
use sol_trade_sdk::TradeTokenType; use sol_trade_sdk::TradeTokenType;
use sol_trade_sdk::{common::AnyResult, swqos::SwqosConfig, SolanaTrade}; use sol_trade_sdk::{common::AnyResult, swqos::SwqosConfig, SolanaTrade};
@@ -132,9 +132,12 @@ async fn raydium_cpmm_copy_trade_with_grpc(trade_info: RaydiumCpmmSwapEvent) ->
let gas_fee_strategy = sol_trade_sdk::common::GasFeeStrategy::new(); let gas_fee_strategy = sol_trade_sdk::common::GasFeeStrategy::new();
gas_fee_strategy.set_global_fee_strategy(150000, 150000, 500000, 500000, 0.001, 0.001); gas_fee_strategy.set_global_fee_strategy(150000, 150000, 500000, 500000, 0.001, 0.001);
let buy_params = let buy_params = RaydiumCpmmParams::from_pool_address_by_rpc(
RaydiumCpmmParams::from_pool_address_by_rpc(&client.infrastructure.rpc, &trade_info.pool_state).await?; &client.infrastructure.rpc,
&trade_info.pool_state,
)
.await?;
let is_wsol = trade_info.input_token_mint == sol_trade_sdk::constants::WSOL_TOKEN_ACCOUNT let is_wsol = trade_info.input_token_mint == sol_trade_sdk::constants::WSOL_TOKEN_ACCOUNT
|| trade_info.output_token_mint == sol_trade_sdk::constants::WSOL_TOKEN_ACCOUNT; || trade_info.output_token_mint == sol_trade_sdk::constants::WSOL_TOKEN_ACCOUNT;
@@ -173,8 +176,11 @@ async fn raydium_cpmm_copy_trade_with_grpc(trade_info: RaydiumCpmmSwapEvent) ->
println!("Balance: {:?}", balance); println!("Balance: {:?}", balance);
let amount_token = balance.amount.parse::<u64>().unwrap(); let amount_token = balance.amount.parse::<u64>().unwrap();
let sell_params = let sell_params = RaydiumCpmmParams::from_pool_address_by_rpc(
RaydiumCpmmParams::from_pool_address_by_rpc(&client.infrastructure.rpc, &trade_info.pool_state).await?; &client.infrastructure.rpc,
&trade_info.pool_state,
)
.await?;
println!("Selling {} tokens", amount_token); println!("Selling {} tokens", amount_token);
let sell_params = sol_trade_sdk::TradeSellParams { let sell_params = sol_trade_sdk::TradeSellParams {
+4 -1
View File
@@ -3,7 +3,10 @@ use sol_trade_sdk::{
fast_fn::get_associated_token_address_with_program_id_fast_use_seed, AnyResult, TradeConfig, fast_fn::get_associated_token_address_with_program_id_fast_use_seed, AnyResult, TradeConfig,
}, },
swqos::SwqosConfig, swqos::SwqosConfig,
trading::{core::params::{PumpSwapParams, DexParamEnum}, factory::DexType}, trading::{
core::params::{DexParamEnum, PumpSwapParams},
factory::DexType,
},
SolanaTrade, TradeTokenType, SolanaTrade, TradeTokenType,
}; };
use solana_commitment_config::CommitmentConfig; use solana_commitment_config::CommitmentConfig;
+12 -8
View File
@@ -10,7 +10,7 @@
use sol_trade_sdk::{ use sol_trade_sdk::{
common::{AnyResult, InfrastructureConfig, TradeConfig}, common::{AnyResult, InfrastructureConfig, TradeConfig},
swqos::{SwqosConfig, SwqosRegion}, swqos::{SwqosConfig, SwqosRegion},
TradingClient, TradingInfrastructure, SwqosTransport, TradingClient, TradingInfrastructure,
}; };
use solana_commitment_config::CommitmentConfig; use solana_commitment_config::CommitmentConfig;
use solana_sdk::signature::Keypair; use solana_sdk::signature::Keypair;
@@ -46,19 +46,23 @@ async fn create_trading_client_simple() -> AnyResult<TradingClient> {
SwqosConfig::ZeroSlot("your_api_token".to_string(), SwqosRegion::Frankfurt, None), SwqosConfig::ZeroSlot("your_api_token".to_string(), SwqosRegion::Frankfurt, None),
SwqosConfig::Temporal("your_api_token".to_string(), SwqosRegion::Frankfurt, None), SwqosConfig::Temporal("your_api_token".to_string(), SwqosRegion::Frankfurt, None),
SwqosConfig::FlashBlock("your_api_token".to_string(), SwqosRegion::Frankfurt, None), SwqosConfig::FlashBlock("your_api_token".to_string(), SwqosRegion::Frankfurt, None),
SwqosConfig::Node1("your_api_token".to_string(), SwqosRegion::Frankfurt, None), SwqosConfig::Node1("your_api_token".to_string(), SwqosRegion::Frankfurt, None, None),
SwqosConfig::BlockRazor("your_api_token".to_string(), SwqosRegion::Frankfurt, None), SwqosConfig::BlockRazor("your_api_token".to_string(), SwqosRegion::Frankfurt, None),
SwqosConfig::Astralane("your_api_token".to_string(), SwqosRegion::Frankfurt, None), SwqosConfig::Astralane(
"your_api_token".to_string(),
SwqosRegion::Frankfurt,
None,
Some(SwqosTransport::Quic),
), // QUIC; use None for HTTP
// Helius Sender: 4th param swqos_only Some(true) => min tip 0.000005 SOL; None => 0.0002 SOL // Helius Sender: 4th param swqos_only Some(true) => min tip 0.000005 SOL; None => 0.0002 SOL
SwqosConfig::Helius("".to_string(), SwqosRegion::Default, None, Some(true)), SwqosConfig::Helius("".to_string(), SwqosRegion::Default, None, Some(true)),
]; ];
// Optional: Customize WSOL ATA and Seed optimization settings // Optional: Customize WSOL ATA and Seed optimization settings
let trade_config = TradeConfig::new(rpc_url, swqos_configs, commitment) let trade_config = TradeConfig::new(rpc_url, swqos_configs, commitment).with_wsol_ata_config(
.with_wsol_ata_config( true, // create_wsol_ata_on_startup: Check and create WSOL ATA on startup
true, // create_wsol_ata_on_startup: Check and create WSOL ATA on startup true, // use_seed_optimize: Enable seed optimization for all ATA operations
true, // use_seed_optimize: Enable seed optimization for all ATA operations );
);
// Creates new infrastructure internally // Creates new infrastructure internally
let client = TradingClient::new(Arc::new(payer), trade_config).await; let client = TradingClient::new(Arc::new(payer), trade_config).await;
+4 -1
View File
@@ -38,7 +38,10 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
// Example 2: Unwrap half of the WSOL back to SOL using seed account // Example 2: Unwrap half of the WSOL back to SOL using seed account
println!("\n🔄 Example 2: Unwrapping half of WSOL back to SOL using seed account"); println!("\n🔄 Example 2: Unwrapping half of WSOL back to SOL using seed account");
let unwrap_amount = wrap_amount / 2; // Half of the wrapped amount let unwrap_amount = wrap_amount / 2; // Half of the wrapped amount
println!("Unwrapping {} lamports (0.0005 SOL) back to SOL using seed account...", unwrap_amount); println!(
"Unwrapping {} lamports (0.0005 SOL) back to SOL using seed account...",
unwrap_amount
);
match solana_trade.wrap_wsol_to_sol(unwrap_amount).await { match solana_trade.wrap_wsol_to_sol(unwrap_amount).await {
Ok(signature) => { Ok(signature) => {
-45
View File
@@ -768,21 +768,6 @@
{ {
"name": "fee_program", "name": "fee_program",
"address": "pfeeUxB6jkeY1Hxd7CsFCAjcbHA9rWtchMGdZ6VojVZ" "address": "pfeeUxB6jkeY1Hxd7CsFCAjcbHA9rWtchMGdZ6VojVZ"
},
{
"name": "bonding_curve_v2",
"pda": {
"seeds": [
{
"kind": "const",
"value": [98, 111, 110, 100, 105, 110, 103, 45, 99, 117, 114, 118, 101, 45, 118, 50]
},
{
"kind": "account",
"path": "mint"
}
]
}
} }
], ],
"args": [ "args": [
@@ -1179,21 +1164,6 @@
{ {
"name": "fee_program", "name": "fee_program",
"address": "pfeeUxB6jkeY1Hxd7CsFCAjcbHA9rWtchMGdZ6VojVZ" "address": "pfeeUxB6jkeY1Hxd7CsFCAjcbHA9rWtchMGdZ6VojVZ"
},
{
"name": "bonding_curve_v2",
"pda": {
"seeds": [
{
"kind": "const",
"value": [98, 111, 110, 100, 105, 110, 103, 45, 99, 117, 114, 118, 101, 45, 118, 50]
},
{
"kind": "account",
"path": "mint"
}
]
}
} }
], ],
"args": [ "args": [
@@ -3987,21 +3957,6 @@
{ {
"name": "fee_program", "name": "fee_program",
"address": "pfeeUxB6jkeY1Hxd7CsFCAjcbHA9rWtchMGdZ6VojVZ" "address": "pfeeUxB6jkeY1Hxd7CsFCAjcbHA9rWtchMGdZ6VojVZ"
},
{
"name": "bonding_curve_v2",
"pda": {
"seeds": [
{
"kind": "const",
"value": [98, 111, 110, 100, 105, 110, 103, 45, 99, 117, 114, 118, 101, 45, 118, 50]
},
{
"kind": "account",
"path": "mint"
}
]
}
} }
], ],
"args": [ "args": [
-45
View File
@@ -688,21 +688,6 @@
{ {
"name": "fee_program", "name": "fee_program",
"address": "pfeeUxB6jkeY1Hxd7CsFCAjcbHA9rWtchMGdZ6VojVZ" "address": "pfeeUxB6jkeY1Hxd7CsFCAjcbHA9rWtchMGdZ6VojVZ"
},
{
"name": "pool_v2",
"pda": {
"seeds": [
{
"kind": "const",
"value": [112, 111, 111, 108, 45, 118, 50]
},
{
"kind": "account",
"path": "base_mint"
}
]
}
} }
], ],
"args": [ "args": [
@@ -1134,21 +1119,6 @@
{ {
"name": "fee_program", "name": "fee_program",
"address": "pfeeUxB6jkeY1Hxd7CsFCAjcbHA9rWtchMGdZ6VojVZ" "address": "pfeeUxB6jkeY1Hxd7CsFCAjcbHA9rWtchMGdZ6VojVZ"
},
{
"name": "pool_v2",
"pda": {
"seeds": [
{
"kind": "const",
"value": [112, 111, 111, 108, 45, 118, 50]
},
{
"kind": "account",
"path": "base_mint"
}
]
}
} }
], ],
"args": [ "args": [
@@ -3192,21 +3162,6 @@
{ {
"name": "fee_program", "name": "fee_program",
"address": "pfeeUxB6jkeY1Hxd7CsFCAjcbHA9rWtchMGdZ6VojVZ" "address": "pfeeUxB6jkeY1Hxd7CsFCAjcbHA9rWtchMGdZ6VojVZ"
},
{
"name": "pool_v2",
"pda": {
"seeds": [
{
"kind": "const",
"value": [112, 111, 111, 108, 45, 118, 50]
},
{
"kind": "account",
"path": "base_mint"
}
]
}
} }
], ],
"args": [ "args": [
-40
View File
@@ -1,40 +0,0 @@
# sol-trade-sdk v3.4.1
Rust SDK for Solana DEX trading (Pump.fun, PumpSwap, Raydium, Bonk, Meteora, etc.).
## What's Changed
### New Features
- **PumpFun & PumpSwap Cashback** (#77): Support for cashback in PumpFun and PumpSwap trading flows. See [Cashback documentation](docs/PUMP_CASHBACK_README.md).
- **Events**: `is_cashback_coin` is now passed from events; PumpFun examples use `sol-parser-sdk` only for event parsing.
### Performance
- **SWQoS**: Reduced submit latency; fixed high latency after ~5 minutes idle.
### Bug Fixes
- **WSOL ATA**: WSOL Associated Token Account creation now runs in background with retry and timeout for more reliable setup.
- Silenced unused and deprecated compiler warnings.
### Documentation
- README (EN/中文): Added Cashback section, outline, examples and tables.
- Updated crates.io / docs references in README.
---
## Cargo
**From Git (this release):**
```toml
sol-trade-sdk = { git = "https://github.com/0xfnzero/sol-trade-sdk", tag = "v3.4.1" }
```
**From crates.io** (when published):
```toml
sol-trade-sdk = "3.4.1"
```
**Full Changelog**: https://github.com/0xfnzero/sol-trade-sdk/compare/v3.4.0...v3.4.1
-37
View File
@@ -1,37 +0,0 @@
# sol-trade-sdk v3.5.0
Rust SDK for Solana DEX trading (Pump.fun, PumpSwap, Raydium, Bonk, Meteora, etc.).
## What's Changed
### Performance
- **Executor hot path**: Sample `Instant::now()` only when `log_enabled` or `simulate` (total, build, submit, confirm) to reduce cold-path syscalls.
- **SWQOS**: `execute_parallel` now takes `&[Arc<SwqosClient>]` to avoid cloning the client list on each swap.
- **Constants**: Named constants for instruction/account sizes and HTTP client timeouts; no magic numbers in hot paths.
### Code Quality
- **Protocol params**: Single `validate_protocol_params()` used for both buy and sell; removed duplicate match blocks in `lib.rs`.
- **Comments**: Bilingual (English + 中文) doc and inline comments across execution, executor, perf (hardware_optimizations, syscall_bypass), and swqos/common.
- **Prefetch/safety**: Clearer docs for cache prefetch and `unsafe` usage (valid read-only ref, no concurrent write).
### Documentation
- README (EN/CN): Version references updated to 3.5.0 for path and crates.io usage.
---
## Cargo
**From Git (this release):**
```toml
sol-trade-sdk = { git = "https://github.com/0xfnzero/sol-trade-sdk", tag = "v3.5.0" }
```
**From crates.io:**
```toml
sol-trade-sdk = "3.5.0"
```
**Full Changelog**: https://github.com/0xfnzero/sol-trade-sdk/compare/v3.4.1...v3.5.0
+6 -6
View File
@@ -46,7 +46,10 @@ static INSTRUCTION_CACHE: Lazy<DashMap<InstructionCacheKey, Arc<Vec<Instruction>
/// Get cached instruction, compute and cache if not exists (lock-free) /// Get cached instruction, compute and cache if not exists (lock-free)
/// 🚀 返回 Arc 避免每次调用克隆整个 Vec /// 🚀 返回 Arc 避免每次调用克隆整个 Vec
#[inline] #[inline]
pub fn get_cached_instructions<F>(cache_key: InstructionCacheKey, compute_fn: F) -> Arc<Vec<Instruction>> pub fn get_cached_instructions<F>(
cache_key: InstructionCacheKey,
compute_fn: F,
) -> Arc<Vec<Instruction>>
where where
F: FnOnce() -> Vec<Instruction>, F: FnOnce() -> Vec<Instruction>,
{ {
@@ -63,10 +66,7 @@ where
}; };
// Lock-free cache lookup with entry API // Lock-free cache lookup with entry API
INSTRUCTION_CACHE INSTRUCTION_CACHE.entry(cache_key).or_insert_with(|| Arc::new(compute_fn())).clone()
.entry(cache_key)
.or_insert_with(|| Arc::new(compute_fn()))
.clone()
} }
// --------------------- Associated Token Account --------------------- // --------------------- Associated Token Account ---------------------
@@ -141,7 +141,7 @@ pub fn _create_associated_token_account_idempotent_fast(
}] }]
}) })
}; };
// 🚀 性能优化:尝试零开销解包 Arc,如果引用计数=1则直接移出,否则克隆 // 🚀 性能优化:尝试零开销解包 Arc,如果引用计数=1则直接移出,否则克隆
Arc::try_unwrap(arc_instructions).unwrap_or_else(|arc| (*arc).clone()) Arc::try_unwrap(arc_instructions).unwrap_or_else(|arc| (*arc).clone())
} }
+4 -11
View File
@@ -2,9 +2,9 @@
//! //!
//! 使用 syscall_bypass 提供的快速时间戳避免频繁的系统调用 //! 使用 syscall_bypass 提供的快速时间戳避免频繁的系统调用
use std::time::{Duration, Instant};
use once_cell::sync::Lazy;
use crate::perf::syscall_bypass::SystemCallBypassManager; use crate::perf::syscall_bypass::SystemCallBypassManager;
use once_cell::sync::Lazy;
use std::time::{Duration, Instant};
/// 全局快速时间提供器 /// 全局快速时间提供器
static FAST_TIMER: Lazy<FastTimer> = Lazy::new(|| FastTimer::new()); static FAST_TIMER: Lazy<FastTimer> = Lazy::new(|| FastTimer::new());
@@ -26,11 +26,7 @@ impl FastTimer {
let base_instant = Instant::now(); let base_instant = Instant::now();
let base_nanos = bypass_manager.fast_timestamp_nanos(); let base_nanos = bypass_manager.fast_timestamp_nanos();
Self { Self { bypass_manager, _base_instant: base_instant, _base_nanos: base_nanos }
bypass_manager,
_base_instant: base_instant,
_base_nanos: base_nanos,
}
} }
/// 🚀 获取当前时间戳(纳秒) - 使用快速系统调用绕过 /// 🚀 获取当前时间戳(纳秒) - 使用快速系统调用绕过
@@ -107,10 +103,7 @@ impl FastStopwatch {
/// 创建并启动计时器 /// 创建并启动计时器
#[inline(always)] #[inline(always)]
pub fn start(label: &'static str) -> Self { pub fn start(label: &'static str) -> Self {
Self { Self { start_nanos: fast_now_nanos(), label }
start_nanos: fast_now_nanos(),
label,
}
} }
/// 获取已耗时(纳秒) /// 获取已耗时(纳秒)
+1 -1
View File
@@ -2,11 +2,11 @@ pub mod address_lookup;
pub mod bonding_curve; pub mod bonding_curve;
pub mod clock; pub mod clock;
pub mod fast_fn; pub mod fast_fn;
pub mod sdk_log;
pub mod fast_timing; pub mod fast_timing;
pub mod gas_fee_strategy; pub mod gas_fee_strategy;
pub mod global; pub mod global;
pub mod nonce_cache; pub mod nonce_cache;
pub mod sdk_log;
pub mod seed; pub mod seed;
pub mod spl_associated_token_account; pub mod spl_associated_token_account;
pub mod spl_token; pub mod spl_token;
+18 -5
View File
@@ -1,13 +1,13 @@
use crate::common::SolanaRpcClient; use crate::common::SolanaRpcClient;
use anyhow::anyhow; use anyhow::anyhow;
use fnv::FnvHasher; use fnv::FnvHasher;
use once_cell::sync::Lazy;
use solana_sdk::{instruction::Instruction, pubkey::Pubkey}; use solana_sdk::{instruction::Instruction, pubkey::Pubkey};
use solana_system_interface::instruction::create_account_with_seed; use solana_system_interface::instruction::create_account_with_seed;
use std::hash::Hasher; use std::hash::Hasher;
use std::sync::Arc;
use std::sync::atomic::{AtomicU64, Ordering}; use std::sync::atomic::{AtomicU64, Ordering};
use std::sync::Arc;
use tokio::time::{sleep, Duration}; use tokio::time::{sleep, Duration};
use once_cell::sync::Lazy;
// 🚀 优化:使用 AtomicU64 替代 RwLock,性能提升 5-10x // 🚀 优化:使用 AtomicU64 替代 RwLock,性能提升 5-10x
// u64::MAX 表示未初始化状态 // u64::MAX 表示未初始化状态
@@ -17,7 +17,7 @@ static SPL_TOKEN_2022_RENT: Lazy<AtomicU64> = Lazy::new(|| AtomicU64::new(u64::M
/// 更新租金缓存(后台任务调用) /// 更新租金缓存(后台任务调用)
pub async fn update_rents(client: &SolanaRpcClient) -> Result<(), anyhow::Error> { pub async fn update_rents(client: &SolanaRpcClient) -> Result<(), anyhow::Error> {
let rent = fetch_rent_for_token_account(client, false).await?; let rent = fetch_rent_for_token_account(client, false).await?;
SPL_TOKEN_RENT.store(rent, Ordering::Release); // Release 确保其他线程可见 SPL_TOKEN_RENT.store(rent, Ordering::Release); // Release 确保其他线程可见
let rent = fetch_rent_for_token_account(client, true).await?; let rent = fetch_rent_for_token_account(client, true).await?;
SPL_TOKEN_2022_RENT.store(rent, Ordering::Release); SPL_TOKEN_2022_RENT.store(rent, Ordering::Release);
@@ -25,6 +25,15 @@ pub async fn update_rents(client: &SolanaRpcClient) -> Result<(), anyhow::Error>
Ok(()) Ok(())
} }
/// 165 字节 Token 账户的典型租金(lamports),RPC 超时时用作回退
const DEFAULT_TOKEN_ACCOUNT_RENT: u64 = 2_039_280;
/// 当 RPC 超时或不可用时设置默认租金,避免客户端创建卡死
pub fn set_default_rents() {
SPL_TOKEN_RENT.store(DEFAULT_TOKEN_ACCOUNT_RENT, Ordering::Release);
SPL_TOKEN_2022_RENT.store(DEFAULT_TOKEN_ACCOUNT_RENT, Ordering::Release);
}
pub fn start_rent_updater(client: Arc<SolanaRpcClient>) { pub fn start_rent_updater(client: Arc<SolanaRpcClient>) {
tokio::spawn(async move { tokio::spawn(async move {
loop { loop {
@@ -53,11 +62,15 @@ pub fn create_associated_token_account_use_seed(
// Relaxed: 租金值不变,无需同步;Release/Acquire 在 update_rents 保证初始化可见性 // Relaxed: 租金值不变,无需同步;Release/Acquire 在 update_rents 保证初始化可见性
let rent = if is_2022_token { let rent = if is_2022_token {
let v = SPL_TOKEN_2022_RENT.load(Ordering::Relaxed); let v = SPL_TOKEN_2022_RENT.load(Ordering::Relaxed);
if v == u64::MAX { return Err(anyhow!("Rent not initialized")); } if v == u64::MAX {
return Err(anyhow!("Rent not initialized"));
}
v v
} else { } else {
let v = SPL_TOKEN_RENT.load(Ordering::Relaxed); let v = SPL_TOKEN_RENT.load(Ordering::Relaxed);
if v == u64::MAX { return Err(anyhow!("Rent not initialized")); } if v == u64::MAX {
return Err(anyhow!("Rent not initialized"));
}
v v
}; };
+1 -5
View File
@@ -52,11 +52,7 @@ pub fn transfer(
accounts.push(AccountMeta::new_readonly(**signer, true)); accounts.push(AccountMeta::new_readonly(**signer, true));
} }
Ok(Instruction { Ok(Instruction { program_id: *token_program_id, accounts, data })
program_id: *token_program_id,
accounts,
data,
})
} }
pub fn initialize_account3( pub fn initialize_account3(
+5 -12
View File
@@ -17,11 +17,7 @@ impl InfrastructureConfig {
swqos_configs: Vec<SwqosConfig>, swqos_configs: Vec<SwqosConfig>,
commitment: CommitmentConfig, commitment: CommitmentConfig,
) -> Self { ) -> Self {
Self { Self { rpc_url, swqos_configs, commitment }
rpc_url,
swqos_configs,
commitment,
}
} }
/// Create from TradeConfig (extract infrastructure-only settings) /// Create from TradeConfig (extract infrastructure-only settings)
@@ -72,8 +68,6 @@ pub struct TradeConfig {
pub create_wsol_ata_on_startup: bool, pub create_wsol_ata_on_startup: bool,
/// Whether to use seed optimization for all ATA operations (default: true) /// Whether to use seed optimization for all ATA operations (default: true)
pub use_seed_optimize: bool, pub use_seed_optimize: bool,
/// Whether to pin parallel submit tasks to CPU cores (can reduce latency; set false in containers). Default true.
pub use_core_affinity: bool,
/// Whether to output all SDK logs (timing, SWQOS submit/confirm, WSOL, blacklist, etc.). Default true. /// Whether to output all SDK logs (timing, SWQOS submit/confirm, WSOL, blacklist, etc.). Default true.
pub log_enabled: bool, pub log_enabled: bool,
/// Whether to check minimum tip per SWQOS provider (filter out configs below min). Default false to save latency. /// Whether to check minimum tip per SWQOS provider (filter out configs below min). Default false to save latency.
@@ -94,11 +88,10 @@ impl TradeConfig {
rpc_url, rpc_url,
swqos_configs, swqos_configs,
commitment, commitment,
create_wsol_ata_on_startup: true, // default: check and create on startup create_wsol_ata_on_startup: true, // default: check and create on startup
use_seed_optimize: true, // default: use seed optimization use_seed_optimize: true, // default: use seed optimization
use_core_affinity: true, // default: pin parallel submit tasks to cores log_enabled: true, // default: enable all SDK logs
log_enabled: true, // default: enable all SDK logs check_min_tip: false, // default: skip min tip check to reduce latency
check_min_tip: false, // default: skip min tip check to reduce latency
} }
} }
+1 -2
View File
@@ -43,8 +43,7 @@ pub const USD1_TOKEN_ACCOUNT_META: solana_sdk::instruction::AccountMeta =
}; };
// USDC (mainnet) mint and meta // USDC (mainnet) mint and meta
pub const USDC_TOKEN_ACCOUNT: Pubkey = pub const USDC_TOKEN_ACCOUNT: Pubkey = pubkey!("EPjFWdd5AufqSSqeM2qN1xzybapC8G4wEGGkZwyTDt1v");
pubkey!("EPjFWdd5AufqSSqeM2qN1xzybapC8G4wEGGkZwyTDt1v");
pub const USDC_TOKEN_ACCOUNT_META: solana_sdk::instruction::AccountMeta = pub const USDC_TOKEN_ACCOUNT_META: solana_sdk::instruction::AccountMeta =
solana_sdk::instruction::AccountMeta { solana_sdk::instruction::AccountMeta {
pubkey: USDC_TOKEN_ACCOUNT, pubkey: USDC_TOKEN_ACCOUNT,
+31 -5
View File
@@ -1,7 +1,6 @@
use solana_program::pubkey; use solana_program::pubkey;
use solana_sdk::pubkey::Pubkey; use solana_sdk::pubkey::Pubkey;
pub const JITO_TIP_ACCOUNTS: &[Pubkey] = &[ pub const JITO_TIP_ACCOUNTS: &[Pubkey] = &[
pubkey!("96gYZGLnJYVFmbjzopPSU6QiEV5fGqZNyN9nmNhvrZU5"), pubkey!("96gYZGLnJYVFmbjzopPSU6QiEV5fGqZNyN9nmNhvrZU5"),
pubkey!("HFqU5x63VTqvQss8hp11i4wVV8bD44PvwucfZ2bU7gRe"), pubkey!("HFqU5x63VTqvQss8hp11i4wVV8bD44PvwucfZ2bU7gRe"),
@@ -165,13 +164,13 @@ pub const SPEEDLANDING_TIP_ACCOUNTS: &[Pubkey] = &[
// Default, // Default,
pub const SWQOS_ENDPOINTS_JITO: [&str; 8] = [ pub const SWQOS_ENDPOINTS_JITO: [&str; 8] = [
"https://ny.mainnet.block-engine.jito.wtf", "https://ny.mainnet.block-engine.jito.wtf",
"https://frankfurt.mainnet.block-engine.jito.wtf", "https://frankfurt.mainnet.block-engine.jito.wtf",
"https://amsterdam.mainnet.block-engine.jito.wtf", "https://amsterdam.mainnet.block-engine.jito.wtf",
"https://slc.mainnet.block-engine.jito.wtf", "https://slc.mainnet.block-engine.jito.wtf",
"https://tokyo.mainnet.block-engine.jito.wtf", "https://tokyo.mainnet.block-engine.jito.wtf",
"https://london.mainnet.block-engine.jito.wtf", "https://london.mainnet.block-engine.jito.wtf",
"https://ny.mainnet.block-engine.jito.wtf", "https://ny.mainnet.block-engine.jito.wtf",
"https://mainnet.block-engine.jito.wtf", "https://mainnet.block-engine.jito.wtf",
]; ];
@@ -180,8 +179,8 @@ pub const SWQOS_ENDPOINTS_NEXTBLOCK: [&str; 8] = [
"http://frankfurt.nextblock.io", "http://frankfurt.nextblock.io",
"http://amsterdam.nextblock.io", "http://amsterdam.nextblock.io",
"http://slc.nextblock.io", "http://slc.nextblock.io",
"http://tokyo.nextblock.io", "http://tokyo.nextblock.io",
"http://london.nextblock.io", "http://london.nextblock.io",
"http://singapore.nextblock.io", "http://singapore.nextblock.io",
"http://frankfurt.nextblock.io", "http://frankfurt.nextblock.io",
]; ];
@@ -230,6 +229,19 @@ pub const SWQOS_ENDPOINTS_NODE1: [&str; 8] = [
"http://fra.node1.me", "http://fra.node1.me",
]; ];
/// Node1 QUIC: port 16666. Region order: NewYork, Frankfurt, Amsterdam, SLC→ny, Tokyo, London, LosAngeles→ny, Default→ny.
/// server_name = host part (e.g. ny.node1.me). Auth: first bi stream = 16-byte UUID; each tx = new bi stream, bincode body.
pub const SWQOS_ENDPOINTS_NODE1_QUIC: [&str; 8] = [
"ny.node1.me:16666",
"fra.node1.me:16666",
"ams.node1.me:16666",
"ny.node1.me:16666", // SLC → ny
"tk.node1.me:16666",
"lon.node1.me:16666",
"ny.node1.me:16666", // LA → ny
"ny.node1.me:16666", // Default → ny
];
pub const SWQOS_ENDPOINTS_FLASHBLOCK: [&str; 8] = [ pub const SWQOS_ENDPOINTS_FLASHBLOCK: [&str; 8] = [
"http://ny.flashblock.trade", "http://ny.flashblock.trade",
"http://fra.flashblock.trade", "http://fra.flashblock.trade",
@@ -242,6 +254,7 @@ pub const SWQOS_ENDPOINTS_FLASHBLOCK: [&str; 8] = [
]; ];
/// BlockRazor Send Transaction v2: plain-text Base64 body, auth in URI, Content-Type: text/plain. Keep-alive: POST /v2/health. /// BlockRazor Send Transaction v2: plain-text Base64 body, auth in URI, Content-Type: text/plain. Keep-alive: POST /v2/health.
/// 若 HTTP 返回 500,可尝试 HTTPShttps://<region>.solana.blockrazor.io/v2/sendTransactionFrankfurt/NewYork/Tokyo),通过 custom_url 覆盖。
pub const SWQOS_ENDPOINTS_BLOCKRAZOR: [&str; 8] = [ pub const SWQOS_ENDPOINTS_BLOCKRAZOR: [&str; 8] = [
"http://newyork.solana.blockrazor.xyz:443/v2/sendTransaction", "http://newyork.solana.blockrazor.xyz:443/v2/sendTransaction",
"http://frankfurt.solana.blockrazor.xyz:443/v2/sendTransaction", "http://frankfurt.solana.blockrazor.xyz:443/v2/sendTransaction",
@@ -267,6 +280,19 @@ pub const SWQOS_ENDPOINTS_ASTRALANE: [&str; 8] = [
"http://lim.gateway.astralane.io/irisb", "http://lim.gateway.astralane.io/irisb",
]; ];
/// Astralane QUIC endpoints (port 7000). Region order: NewYork, Frankfurt, Amsterdam, SLC, Tokyo, London, LosAngeles, Default.
/// See: https://github.com/Astralane/astralane-quic-client. We use fr, ams, la, ny, lim, sg only (avoid ams2/fr2 for lower latency).
pub const SWQOS_ENDPOINTS_ASTRALANE_QUIC: [&str; 8] = [
"ny.gateway.astralane.io:7000", // NewYork
"fr.gateway.astralane.io:7000", // Frankfurt
"ams.gateway.astralane.io:7000", // Amsterdam
"lim.gateway.astralane.io:7000", // SLC (no slc, use lim)
"sg.gateway.astralane.io:7000", // Tokyo (Asia)
"ams.gateway.astralane.io:7000", // London (Europe, avoid ams2)
"la.gateway.astralane.io:7000", // LosAngeles
"lim.gateway.astralane.io:7000", // Default
];
pub const SWQOS_ENDPOINTS_STELLIUM: [&str; 8] = [ pub const SWQOS_ENDPOINTS_STELLIUM: [&str; 8] = [
"http://ewr1.flashrpc.com", "http://ewr1.flashrpc.com",
"http://fra1.flashrpc.com", "http://fra1.flashrpc.com",
+1 -1
View File
@@ -6,4 +6,4 @@ pub mod trade {
pub const DEFAULT_SELL_TIP_FEE: f64 = 0.0001; pub const DEFAULT_SELL_TIP_FEE: f64 = 0.0001;
pub const DEFAULT_RPC_UNIT_LIMIT: u32 = 150000; pub const DEFAULT_RPC_UNIT_LIMIT: u32 = 150000;
pub const DEFAULT_RPC_UNIT_PRICE: u64 = 500000; pub const DEFAULT_RPC_UNIT_PRICE: u64 = 500000;
} }
+7 -23
View File
@@ -4,12 +4,9 @@ use crate::{
accounts, get_pool_pda, get_vault_pda, BUY_EXECT_IN_DISCRIMINATOR, accounts, get_pool_pda, get_vault_pda, BUY_EXECT_IN_DISCRIMINATOR,
SELL_EXECT_IN_DISCRIMINATOR, SELL_EXECT_IN_DISCRIMINATOR,
}, },
trading::{ trading::core::{
common::utils::get_token_balance, params::{BonkParams, SwapParams},
core::{ traits::InstructionBuilder,
params::{BonkParams, SwapParams},
traits::InstructionBuilder,
},
}, },
utils::calc::bonk::{ utils::calc::bonk::{
get_buy_token_amount_from_sol_amount, get_sell_sol_amount_from_token_amount, get_buy_token_amount_from_sol_amount, get_sell_sol_amount_from_token_amount,
@@ -177,9 +174,10 @@ impl InstructionBuilder for BonkInstructionBuilder {
// ======================================== // ========================================
// Parameter validation and basic data preparation // Parameter validation and basic data preparation
// ======================================== // ========================================
if params.rpc.is_none() { let amount = params
return Err(anyhow!("RPC is not set")); .input_amount
} .filter(|&a| a > 0)
.ok_or_else(|| anyhow!("Bonk sell requires input_amount (token amount to sell); fetch balance via RPC before calling build_sell"))?;
let protocol_params = params let protocol_params = params
.protocol_params .protocol_params
@@ -189,20 +187,6 @@ impl InstructionBuilder for BonkInstructionBuilder {
let usd1_pool = protocol_params.global_config == accounts::USD1_GLOBAL_CONFIG; let usd1_pool = protocol_params.global_config == accounts::USD1_GLOBAL_CONFIG;
let rpc = params.rpc.as_ref().unwrap().clone();
let mut amount = params.input_amount;
if params.input_amount.is_none() || params.input_amount.unwrap_or(0) == 0 {
let balance_u64 =
get_token_balance(rpc.as_ref(), &params.payer.pubkey(), &params.input_mint).await?;
amount = Some(balance_u64);
}
let amount = amount.unwrap_or(0);
if amount == 0 {
return Err(anyhow!("Amount cannot be zero"));
}
let pool_state = if protocol_params.pool_state == Pubkey::default() { let pool_state = if protocol_params.pool_state == Pubkey::default() {
if usd1_pool { if usd1_pool {
get_pool_pda(&params.input_mint, &crate::constants::USD1_TOKEN_ACCOUNT).unwrap() get_pool_pda(&params.input_mint, &crate::constants::USD1_TOKEN_ACCOUNT).unwrap()
+14 -8
View File
@@ -27,10 +27,12 @@ impl InstructionBuilder for MeteoraDammV2InstructionBuilder {
.protocol_params .protocol_params
.as_any() .as_any()
.downcast_ref::<MeteoraDammV2Params>() .downcast_ref::<MeteoraDammV2Params>()
.ok_or_else(|| anyhow!("Invalid protocol params for RaydiumCpmm"))?; .ok_or_else(|| anyhow!("Invalid protocol params for MeteoraDammV2"))?;
let is_wsol = protocol_params.token_a_mint == crate::constants::WSOL_TOKEN_ACCOUNT || protocol_params.token_b_mint == crate::constants::WSOL_TOKEN_ACCOUNT; let is_wsol = protocol_params.token_a_mint == crate::constants::WSOL_TOKEN_ACCOUNT
let is_usdc = protocol_params.token_a_mint == crate::constants::USDC_TOKEN_ACCOUNT || protocol_params.token_b_mint == crate::constants::USDC_TOKEN_ACCOUNT; || protocol_params.token_b_mint == crate::constants::WSOL_TOKEN_ACCOUNT;
let is_usdc = protocol_params.token_a_mint == crate::constants::USDC_TOKEN_ACCOUNT
|| protocol_params.token_b_mint == crate::constants::USDC_TOKEN_ACCOUNT;
if !is_wsol && !is_usdc { if !is_wsol && !is_usdc {
return Err(anyhow!("Pool must contain WSOL or USDC")); return Err(anyhow!("Pool must contain WSOL or USDC"));
} }
@@ -38,7 +40,8 @@ impl InstructionBuilder for MeteoraDammV2InstructionBuilder {
// ======================================== // ========================================
// Trade calculation and account address preparation // Trade calculation and account address preparation
// ======================================== // ========================================
let is_a_in = protocol_params.token_a_mint == crate::constants::WSOL_TOKEN_ACCOUNT || protocol_params.token_a_mint == crate::constants::USDC_TOKEN_ACCOUNT; let is_a_in = protocol_params.token_a_mint == crate::constants::WSOL_TOKEN_ACCOUNT
|| protocol_params.token_a_mint == crate::constants::USDC_TOKEN_ACCOUNT;
let amount_in: u64 = params.input_amount.unwrap_or(0); let amount_in: u64 = params.input_amount.unwrap_or(0);
let minimum_amount_out: u64 = match params.fixed_output_amount { let minimum_amount_out: u64 = match params.fixed_output_amount {
Some(fixed) => fixed, Some(fixed) => fixed,
@@ -135,14 +138,16 @@ impl InstructionBuilder for MeteoraDammV2InstructionBuilder {
.protocol_params .protocol_params
.as_any() .as_any()
.downcast_ref::<MeteoraDammV2Params>() .downcast_ref::<MeteoraDammV2Params>()
.ok_or_else(|| anyhow!("Invalid protocol params for RaydiumCpmm"))?; .ok_or_else(|| anyhow!("Invalid protocol params for MeteoraDammV2"))?;
if params.input_amount.is_none() || params.input_amount.unwrap_or(0) == 0 { if params.input_amount.is_none() || params.input_amount.unwrap_or(0) == 0 {
return Err(anyhow!("Token amount is not set")); return Err(anyhow!("Token amount is not set"));
} }
let is_wsol = protocol_params.token_b_mint == crate::constants::WSOL_TOKEN_ACCOUNT || protocol_params.token_a_mint == crate::constants::WSOL_TOKEN_ACCOUNT; let is_wsol = protocol_params.token_b_mint == crate::constants::WSOL_TOKEN_ACCOUNT
let is_usdc = protocol_params.token_b_mint == crate::constants::USDC_TOKEN_ACCOUNT || protocol_params.token_a_mint == crate::constants::USDC_TOKEN_ACCOUNT; || protocol_params.token_a_mint == crate::constants::WSOL_TOKEN_ACCOUNT;
let is_usdc = protocol_params.token_b_mint == crate::constants::USDC_TOKEN_ACCOUNT
|| protocol_params.token_a_mint == crate::constants::USDC_TOKEN_ACCOUNT;
if !is_wsol && !is_usdc { if !is_wsol && !is_usdc {
return Err(anyhow!("Pool must contain WSOL or USDC")); return Err(anyhow!("Pool must contain WSOL or USDC"));
} }
@@ -150,7 +155,8 @@ impl InstructionBuilder for MeteoraDammV2InstructionBuilder {
// ======================================== // ========================================
// Trade calculation and account address preparation // Trade calculation and account address preparation
// ======================================== // ========================================
let is_a_in = protocol_params.token_b_mint == crate::constants::WSOL_TOKEN_ACCOUNT || protocol_params.token_b_mint == crate::constants::USDC_TOKEN_ACCOUNT; let is_a_in = protocol_params.token_b_mint == crate::constants::WSOL_TOKEN_ACCOUNT
|| protocol_params.token_b_mint == crate::constants::USDC_TOKEN_ACCOUNT;
let minimum_amount_out: u64 = match params.fixed_output_amount { let minimum_amount_out: u64 = match params.fixed_output_amount {
Some(fixed) => fixed, Some(fixed) => fixed,
None => return Err(anyhow!("fixed_output_amount must be set for MeteoraDammV2 swap")), None => return Err(anyhow!("fixed_output_amount must be set for MeteoraDammV2 swap")),
+4 -4
View File
@@ -1,7 +1,7 @@
pub mod bonk;
pub mod meteora_damm_v2;
pub mod pumpfun; pub mod pumpfun;
pub mod pumpswap; pub mod pumpswap;
pub mod bonk;
pub mod raydium_cpmm;
pub mod raydium_amm_v4; pub mod raydium_amm_v4;
pub mod meteora_damm_v2; pub mod raydium_cpmm;
pub mod utils; pub mod utils;
+37 -37
View File
@@ -9,8 +9,9 @@ use crate::{
use crate::{ use crate::{
instruction::utils::pumpfun::{ instruction::utils::pumpfun::{
accounts, get_bonding_curve_pda, get_bonding_curve_v2_pda, get_creator, accounts, get_bonding_curve_pda, get_bonding_curve_v2_pda, get_creator,
get_user_volume_accumulator_pda, global_constants::{self}, BUY_DISCRIMINATOR, get_mayhem_fee_recipient_meta_random, get_user_volume_accumulator_pda,
BUY_EXACT_SOL_IN_DISCRIMINATOR, global_constants::{self},
BUY_DISCRIMINATOR, BUY_EXACT_SOL_IN_DISCRIMINATOR, SELL_DISCRIMINATOR,
}, },
utils::calc::{ utils::calc::{
common::{calculate_with_slippage_buy, calculate_with_slippage_sell}, common::{calculate_with_slippage_buy, calculate_with_slippage_sell},
@@ -64,7 +65,9 @@ impl InstructionBuilder for PumpFunInstructionBuilder {
); );
let bonding_curve_addr = if bonding_curve.account == Pubkey::default() { let bonding_curve_addr = if bonding_curve.account == Pubkey::default() {
get_bonding_curve_pda(&params.output_mint).unwrap() get_bonding_curve_pda(&params.output_mint).ok_or_else(|| {
anyhow!("bonding_curve PDA derivation failed for mint {}", params.output_mint)
})?
} else { } else {
bonding_curve.account bonding_curve.account
}; };
@@ -97,8 +100,8 @@ impl InstructionBuilder for PumpFunInstructionBuilder {
params.open_seed_optimize, params.open_seed_optimize,
); );
let user_volume_accumulator = let user_volume_accumulator = get_user_volume_accumulator_pda(&params.payer.pubkey())
get_user_volume_accumulator_pda(&params.payer.pubkey()).unwrap(); .ok_or_else(|| anyhow!("user_volume_accumulator PDA derivation failed"))?;
// ======================================== // ========================================
// Build instructions // Build instructions
@@ -118,10 +121,11 @@ impl InstructionBuilder for PumpFunInstructionBuilder {
); );
} }
let mut buy_data = [0u8; 24]; // IDL: buy/buy_exact_sol_in 第三参数 track_volume: OptionBool,仅代币支持返现时传 Some(true)
let track_volume = if bonding_curve.is_cashback_coin { [1u8, 1u8] } else { [1u8, 0u8] }; // Some(true) / Some(false)
let mut buy_data = [0u8; 26];
if params.use_exact_sol_amount.unwrap_or(true) { if params.use_exact_sol_amount.unwrap_or(true) {
// buy_exact_sol_in(spendable_sol_in: u64, min_tokens_out: u64) // buy_exact_sol_in(spendable_sol_in: u64, min_tokens_out: u64, track_volume)
// Spend exactly the input SOL amount, get at least min_tokens_out
let min_tokens_out = calculate_with_slippage_sell( let min_tokens_out = calculate_with_slippage_sell(
buy_token_amount, buy_token_amount,
params.slippage_basis_points.unwrap_or(DEFAULT_SLIPPAGE), params.slippage_basis_points.unwrap_or(DEFAULT_SLIPPAGE),
@@ -129,22 +133,25 @@ impl InstructionBuilder for PumpFunInstructionBuilder {
buy_data[..8].copy_from_slice(&BUY_EXACT_SOL_IN_DISCRIMINATOR); buy_data[..8].copy_from_slice(&BUY_EXACT_SOL_IN_DISCRIMINATOR);
buy_data[8..16].copy_from_slice(&params.input_amount.unwrap_or(0).to_le_bytes()); buy_data[8..16].copy_from_slice(&params.input_amount.unwrap_or(0).to_le_bytes());
buy_data[16..24].copy_from_slice(&min_tokens_out.to_le_bytes()); buy_data[16..24].copy_from_slice(&min_tokens_out.to_le_bytes());
buy_data[24..26].copy_from_slice(&track_volume);
} else { } else {
// buy(token_amount: u64, max_sol_cost: u64) // buy(token_amount: u64, max_sol_cost: u64, track_volume)
// Buy exactly token_amount tokens, pay up to max_sol_cost
buy_data[..8].copy_from_slice(&BUY_DISCRIMINATOR); buy_data[..8].copy_from_slice(&BUY_DISCRIMINATOR);
buy_data[8..16].copy_from_slice(&buy_token_amount.to_le_bytes()); buy_data[8..16].copy_from_slice(&buy_token_amount.to_le_bytes());
buy_data[16..24].copy_from_slice(&max_sol_cost.to_le_bytes()); buy_data[16..24].copy_from_slice(&max_sol_cost.to_le_bytes());
buy_data[24..26].copy_from_slice(&track_volume);
} }
// Determine fee recipient based on mayhem mode // Determine fee recipient based on mayhem mode (pump-public-docs: 2nd account = Mayhem fee recipient; use any one randomly)
let fee_recipient_meta = if is_mayhem_mode { let fee_recipient_meta = if is_mayhem_mode {
global_constants::MAYHEM_FEE_RECIPIENT_META get_mayhem_fee_recipient_meta_random()
} else { } else {
global_constants::FEE_RECIPIENT_META global_constants::FEE_RECIPIENT_META
}; };
let bonding_curve_v2 = get_bonding_curve_v2_pda(&params.output_mint).unwrap(); let bonding_curve_v2 = get_bonding_curve_v2_pda(&params.output_mint).ok_or_else(|| {
anyhow!("bonding_curve_v2 PDA derivation failed for mint {}", params.output_mint)
})?;
let mut accounts: Vec<AccountMeta> = vec![ let mut accounts: Vec<AccountMeta> = vec![
global_constants::GLOBAL_ACCOUNT_META, global_constants::GLOBAL_ACCOUNT_META,
fee_recipient_meta, fee_recipient_meta,
@@ -163,13 +170,9 @@ impl InstructionBuilder for PumpFunInstructionBuilder {
accounts::FEE_CONFIG_META, accounts::FEE_CONFIG_META,
accounts::FEE_PROGRAM_META, accounts::FEE_PROGRAM_META,
]; ];
accounts.push(AccountMeta::new_readonly(bonding_curve_v2, false)); // bonding_curve_v2 (readonly) at end accounts.push(AccountMeta::new_readonly(bonding_curve_v2, false)); // remainingAccounts: @pump-fun/pump-sdk 要求末尾传 bondingCurveV2Pda(mint),勿删
instructions.push(Instruction::new_with_bytes( instructions.push(Instruction::new_with_bytes(accounts::PUMPFUN, &buy_data, accounts));
accounts::PUMPFUN,
&buy_data,
accounts,
));
Ok(instructions) Ok(instructions)
} }
@@ -216,7 +219,9 @@ impl InstructionBuilder for PumpFunInstructionBuilder {
}; };
let bonding_curve_addr = if bonding_curve.account == Pubkey::default() { let bonding_curve_addr = if bonding_curve.account == Pubkey::default() {
get_bonding_curve_pda(&params.input_mint).unwrap() get_bonding_curve_pda(&params.input_mint).ok_or_else(|| {
anyhow!("bonding_curve PDA derivation failed for mint {}", params.input_mint)
})?
} else { } else {
bonding_curve.account bonding_curve.account
}; };
@@ -255,13 +260,13 @@ impl InstructionBuilder for PumpFunInstructionBuilder {
let mut instructions = Vec::with_capacity(2); let mut instructions = Vec::with_capacity(2);
let mut sell_data = [0u8; 24]; let mut sell_data = [0u8; 24];
sell_data[..8].copy_from_slice(&[51, 230, 133, 164, 1, 127, 131, 173]); // Method ID sell_data[..8].copy_from_slice(&SELL_DISCRIMINATOR);
sell_data[8..16].copy_from_slice(&token_amount.to_le_bytes()); sell_data[8..16].copy_from_slice(&token_amount.to_le_bytes());
sell_data[16..24].copy_from_slice(&min_sol_output.to_le_bytes()); sell_data[16..24].copy_from_slice(&min_sol_output.to_le_bytes());
// Determine fee recipient based on mayhem mode // Determine fee recipient based on mayhem mode (pump-public-docs: 2nd account = Mayhem fee recipient; use any one randomly)
let fee_recipient_meta = if is_mayhem_mode { let fee_recipient_meta = if is_mayhem_mode {
global_constants::MAYHEM_FEE_RECIPIENT_META get_mayhem_fee_recipient_meta_random()
} else { } else {
global_constants::FEE_RECIPIENT_META global_constants::FEE_RECIPIENT_META
}; };
@@ -286,18 +291,17 @@ impl InstructionBuilder for PumpFunInstructionBuilder {
// Cashback: Bonding Curve Sell expects UserVolumeAccumulator PDA at 0th remaining account (writable) // Cashback: Bonding Curve Sell expects UserVolumeAccumulator PDA at 0th remaining account (writable)
if bonding_curve.is_cashback_coin { if bonding_curve.is_cashback_coin {
let user_volume_accumulator = let user_volume_accumulator =
get_user_volume_accumulator_pda(&params.payer.pubkey()).unwrap(); get_user_volume_accumulator_pda(&params.payer.pubkey())
.ok_or_else(|| anyhow!("user_volume_accumulator PDA derivation failed"))?;
accounts.push(AccountMeta::new(user_volume_accumulator, false)); accounts.push(AccountMeta::new(user_volume_accumulator, false));
} }
// Program upgrade: bonding_curve_v2 (readonly) at end of account list // remainingAccounts: @pump-fun/pump-sdk sell 要求末尾传 bondingCurveV2Pda(mint)cashback 时在 user_volume_accumulator 之后),勿删
let bonding_curve_v2 = get_bonding_curve_v2_pda(&params.input_mint).unwrap(); let bonding_curve_v2 = get_bonding_curve_v2_pda(&params.input_mint).ok_or_else(|| {
anyhow!("bonding_curve_v2 PDA derivation failed for mint {}", params.input_mint)
})?;
accounts.push(AccountMeta::new_readonly(bonding_curve_v2, false)); accounts.push(AccountMeta::new_readonly(bonding_curve_v2, false));
instructions.push(Instruction::new_with_bytes( instructions.push(Instruction::new_with_bytes(accounts::PUMPFUN, &sell_data, accounts));
accounts::PUMPFUN,
&sell_data,
accounts,
));
// Optional: Close token account // Optional: Close token account
if protocol_params.close_token_account_when_sell.unwrap_or(false) if protocol_params.close_token_account_when_sell.unwrap_or(false)
@@ -321,15 +325,11 @@ pub fn claim_cashback_pumpfun_instruction(payer: &Pubkey) -> Option<Instruction>
const CLAIM_CASHBACK_DISCRIMINATOR: [u8; 8] = [37, 58, 35, 126, 190, 53, 228, 197]; const CLAIM_CASHBACK_DISCRIMINATOR: [u8; 8] = [37, 58, 35, 126, 190, 53, 228, 197];
let user_volume_accumulator = get_user_volume_accumulator_pda(payer)?; let user_volume_accumulator = get_user_volume_accumulator_pda(payer)?;
let accounts = vec![ let accounts = vec![
AccountMeta::new(*payer, true), // user (signer, writable) AccountMeta::new(*payer, true), // user (signer, writable)
AccountMeta::new(user_volume_accumulator, false), // user_volume_accumulator (writable, not signer) AccountMeta::new(user_volume_accumulator, false), // user_volume_accumulator (writable, not signer)
crate::constants::SYSTEM_PROGRAM_META, crate::constants::SYSTEM_PROGRAM_META,
accounts::EVENT_AUTHORITY_META, accounts::EVENT_AUTHORITY_META,
accounts::PUMPFUN_META, accounts::PUMPFUN_META,
]; ];
Some(Instruction::new_with_bytes( Some(Instruction::new_with_bytes(accounts::PUMPFUN, &CLAIM_CASHBACK_DISCRIMINATOR, accounts))
accounts::PUMPFUN,
&CLAIM_CASHBACK_DISCRIMINATOR,
accounts,
))
} }
+70 -77
View File
@@ -1,9 +1,10 @@
use crate::{ use crate::{
constants::trade::trade::DEFAULT_SLIPPAGE, constants::trade::trade::DEFAULT_SLIPPAGE,
instruction::utils::pumpswap::{ instruction::utils::pumpswap::{
accounts, fee_recipient_ata, get_pool_v2_pda, get_user_volume_accumulator_pda, accounts, fee_recipient_ata, get_mayhem_fee_recipient_random, get_pool_v2_pda,
get_user_volume_accumulator_wsol_ata, BUY_DISCRIMINATOR, get_user_volume_accumulator_pda, get_user_volume_accumulator_quote_ata,
BUY_EXACT_QUOTE_IN_DISCRIMINATOR, SELL_DISCRIMINATOR, get_user_volume_accumulator_wsol_ata, BUY_DISCRIMINATOR, BUY_EXACT_QUOTE_IN_DISCRIMINATOR,
SELL_DISCRIMINATOR,
}, },
trading::{ trading::{
common::wsol_manager, common::wsol_manager,
@@ -119,16 +120,18 @@ impl InstructionBuilder for PumpSwapInstructionBuilder {
params.open_seed_optimize, params.open_seed_optimize,
); );
// Determine fee recipient based on mayhem mode // Determine fee recipient based on mayhem mode (pump-public-docs: 10th = Mayhem fee recipient, 11th = WSOL ATA of Mayhem; use any one randomly)
let is_mayhem_mode = protocol_params.is_mayhem_mode; let is_mayhem_mode = protocol_params.is_mayhem_mode;
let fee_recipient = let (fee_recipient, fee_recipient_meta) = if is_mayhem_mode {
if is_mayhem_mode { accounts::MAYHEM_FEE_RECIPIENT } else { accounts::FEE_RECIPIENT }; get_mayhem_fee_recipient_random()
let fee_recipient_meta = if is_mayhem_mode {
accounts::MAYHEM_FEE_RECIPIENT_META
} else { } else {
accounts::FEE_RECIPIENT_META (accounts::FEE_RECIPIENT, accounts::FEE_RECIPIENT_META)
};
let fee_recipient_ata = if is_mayhem_mode {
fee_recipient_ata(fee_recipient, crate::constants::WSOL_TOKEN_ACCOUNT)
} else {
fee_recipient_ata(fee_recipient, quote_mint)
}; };
let fee_recipient_ata = fee_recipient_ata(fee_recipient, quote_mint);
// ======================================== // ========================================
// Build instructions // Build instructions
@@ -177,10 +180,9 @@ impl InstructionBuilder for PumpSwapInstructionBuilder {
]); ]);
if quote_is_wsol_or_usdc { if quote_is_wsol_or_usdc {
accounts.push(accounts::GLOBAL_VOLUME_ACCUMULATOR_META); accounts.push(accounts::GLOBAL_VOLUME_ACCUMULATOR_META);
accounts.push(AccountMeta::new( let uva = get_user_volume_accumulator_pda(&params.payer.pubkey())
get_user_volume_accumulator_pda(&params.payer.pubkey()).unwrap(), .ok_or_else(|| anyhow!("user_volume_accumulator PDA derivation failed"))?;
false, accounts.push(AccountMeta::new(uva, false));
));
} }
accounts.push(accounts::FEE_CONFIG_META); accounts.push(accounts::FEE_CONFIG_META);
accounts.push(accounts::FEE_PROGRAM_META); accounts.push(accounts::FEE_PROGRAM_META);
@@ -190,51 +192,40 @@ impl InstructionBuilder for PumpSwapInstructionBuilder {
accounts.push(AccountMeta::new(wsol_ata, false)); accounts.push(AccountMeta::new(wsol_ata, false));
} }
} }
// Program upgrade: pool_v2 (readonly) at end of account list // remainingAccounts: @pump-fun/pump-swap-sdk 要求末尾传 poolV2Pda(baseMint),勿删
accounts.push(AccountMeta::new_readonly( let pool_v2 = get_pool_v2_pda(&base_mint)
get_pool_v2_pda(&base_mint).unwrap(), .ok_or_else(|| anyhow!("pool_v2 PDA derivation failed for base_mint {}", base_mint))?;
false, accounts.push(AccountMeta::new_readonly(pool_v2, false));
));
// Create instruction data // Create instruction databuy/buy_exact_quote_in 第三参数 track_volume: OptionBool,仅代币支持返现时传 Some(true);sell 仅两参数)
let mut data = [0u8; 24]; let track_volume = if protocol_params.is_cashback_coin { [1u8, 1u8] } else { [1u8, 0u8] }; // Some(true) / Some(false)
if quote_is_wsol_or_usdc { let data: Vec<u8> = if quote_is_wsol_or_usdc {
let mut buf = [0u8; 26];
if params.use_exact_sol_amount.unwrap_or(true) { if params.use_exact_sol_amount.unwrap_or(true) {
// buy_exact_quote_in(spendable_quote_in: u64, min_base_amount_out: u64)
// Spend exactly the input SOL/quote amount, get at least min_base_amount_out
let min_base_amount_out = crate::utils::calc::common::calculate_with_slippage_sell( let min_base_amount_out = crate::utils::calc::common::calculate_with_slippage_sell(
token_amount, token_amount,
params.slippage_basis_points.unwrap_or(DEFAULT_SLIPPAGE), params.slippage_basis_points.unwrap_or(DEFAULT_SLIPPAGE),
); );
data[..8].copy_from_slice(&BUY_EXACT_QUOTE_IN_DISCRIMINATOR); buf[..8].copy_from_slice(&BUY_EXACT_QUOTE_IN_DISCRIMINATOR);
// spendable_quote_in (exact SOL amount to spend) buf[8..16].copy_from_slice(&params.input_amount.unwrap_or(0).to_le_bytes());
data[8..16].copy_from_slice(&params.input_amount.unwrap_or(0).to_le_bytes()); buf[16..24].copy_from_slice(&min_base_amount_out.to_le_bytes());
// min_base_amount_out (minimum tokens to receive) buf[24..26].copy_from_slice(&track_volume);
data[16..24].copy_from_slice(&min_base_amount_out.to_le_bytes());
} else { } else {
// buy(base_amount_out: u64, max_quote_amount_in: u64) buf[..8].copy_from_slice(&BUY_DISCRIMINATOR);
// Buy exactly base_amount_out tokens, pay up to max_quote_amount_in buf[8..16].copy_from_slice(&token_amount.to_le_bytes());
data[..8].copy_from_slice(&BUY_DISCRIMINATOR); buf[16..24].copy_from_slice(&sol_amount.to_le_bytes());
// base_amount_out buf[24..26].copy_from_slice(&track_volume);
data[8..16].copy_from_slice(&token_amount.to_le_bytes());
// max_quote_amount_in
data[16..24].copy_from_slice(&sol_amount.to_le_bytes());
} }
buf.to_vec()
} else { } else {
data[..8].copy_from_slice(&SELL_DISCRIMINATOR); let mut buf = [0u8; 24];
// base_amount_in buf[..8].copy_from_slice(&SELL_DISCRIMINATOR);
data[8..16].copy_from_slice(&sol_amount.to_le_bytes()); buf[8..16].copy_from_slice(&sol_amount.to_le_bytes());
// min_quote_amount_out buf[16..24].copy_from_slice(&token_amount.to_le_bytes());
data[16..24].copy_from_slice(&token_amount.to_le_bytes()); buf.to_vec()
}
let buy_instruction = Instruction {
program_id: accounts::AMM_PROGRAM,
accounts: accounts.clone(),
data: data.to_vec(),
}; };
instructions.push(buy_instruction); instructions.push(Instruction { program_id: accounts::AMM_PROGRAM, accounts, data });
if close_wsol_ata { if close_wsol_ata {
// Close wSOL ATA account, reclaim rent // Close wSOL ATA account, reclaim rent
instructions.extend(crate::trading::common::close_wsol(&params.payer.pubkey())); instructions.extend(crate::trading::common::close_wsol(&params.payer.pubkey()));
@@ -320,16 +311,18 @@ impl InstructionBuilder for PumpSwapInstructionBuilder {
sol_amount = params.fixed_output_amount.unwrap(); sol_amount = params.fixed_output_amount.unwrap();
} }
// Determine fee recipient based on mayhem mode // Determine fee recipient based on mayhem mode (pump-public-docs: 10th = Mayhem fee recipient, 11th = WSOL ATA of Mayhem; use any one randomly)
let is_mayhem_mode = protocol_params.is_mayhem_mode; let is_mayhem_mode = protocol_params.is_mayhem_mode;
let fee_recipient = let (fee_recipient, fee_recipient_meta) = if is_mayhem_mode {
if is_mayhem_mode { accounts::MAYHEM_FEE_RECIPIENT } else { accounts::FEE_RECIPIENT }; get_mayhem_fee_recipient_random()
let fee_recipient_meta = if is_mayhem_mode {
accounts::MAYHEM_FEE_RECIPIENT_META
} else { } else {
accounts::FEE_RECIPIENT_META (accounts::FEE_RECIPIENT, accounts::FEE_RECIPIENT_META)
};
let fee_recipient_ata = if is_mayhem_mode {
fee_recipient_ata(fee_recipient, crate::constants::WSOL_TOKEN_ACCOUNT)
} else {
fee_recipient_ata(fee_recipient, quote_mint)
}; };
let fee_recipient_ata = fee_recipient_ata(fee_recipient, quote_mint);
let user_base_token_account = let user_base_token_account =
crate::common::fast_fn::get_associated_token_address_with_program_id_fast_use_seed( crate::common::fast_fn::get_associated_token_address_with_program_id_fast_use_seed(
@@ -380,28 +373,30 @@ impl InstructionBuilder for PumpSwapInstructionBuilder {
]); ]);
if !quote_is_wsol_or_usdc { if !quote_is_wsol_or_usdc {
accounts.push(accounts::GLOBAL_VOLUME_ACCUMULATOR_META); accounts.push(accounts::GLOBAL_VOLUME_ACCUMULATOR_META);
accounts.push(AccountMeta::new( let uva = get_user_volume_accumulator_pda(&params.payer.pubkey())
get_user_volume_accumulator_pda(&params.payer.pubkey()).unwrap(), .ok_or_else(|| anyhow!("user_volume_accumulator PDA derivation failed"))?;
false, accounts.push(AccountMeta::new(uva, false));
));
} }
accounts.push(accounts::FEE_CONFIG_META); accounts.push(accounts::FEE_CONFIG_META);
accounts.push(accounts::FEE_PROGRAM_META); accounts.push(accounts::FEE_PROGRAM_META);
// Cashback: remaining_accounts[0] = WSOL ATA of UserVolumeAccumulator, remaining_accounts[1] = UserVolumeAccumulator PDA // Cashback sell: 官方 remainingAccounts = [accumulator 的 quote_mint ATA, accumulator PDA, poolV2](用 quote_mint 非固定 WSOL
if protocol_params.is_cashback_coin { if protocol_params.is_cashback_coin {
if let (Some(wsol_ata), Some(accumulator)) = ( if let (Some(quote_ata), Some(accumulator)) = (
get_user_volume_accumulator_wsol_ata(&params.payer.pubkey()), get_user_volume_accumulator_quote_ata(
&params.payer.pubkey(),
&quote_mint,
&quote_token_program,
),
get_user_volume_accumulator_pda(&params.payer.pubkey()), get_user_volume_accumulator_pda(&params.payer.pubkey()),
) { ) {
accounts.push(AccountMeta::new(wsol_ata, false)); accounts.push(AccountMeta::new(quote_ata, false));
accounts.push(AccountMeta::new(accumulator, false)); accounts.push(AccountMeta::new(accumulator, false));
} }
} }
// Program upgrade: pool_v2 (readonly) at end of account list // remainingAccounts: @pump-fun/pump-swap-sdk sell 要求末尾传 poolV2Pda(baseMint),勿删
accounts.push(AccountMeta::new_readonly( let pool_v2 = get_pool_v2_pda(&base_mint)
get_pool_v2_pda(&base_mint).unwrap(), .ok_or_else(|| anyhow!("pool_v2 PDA derivation failed for base_mint {}", base_mint))?;
false, accounts.push(AccountMeta::new_readonly(pool_v2, false));
));
// Create instruction data // Create instruction data
let mut data = [0u8; 24]; let mut data = [0u8; 24];
@@ -419,13 +414,11 @@ impl InstructionBuilder for PumpSwapInstructionBuilder {
data[16..24].copy_from_slice(&token_amount.to_le_bytes()); data[16..24].copy_from_slice(&token_amount.to_le_bytes());
} }
let sell_instruction = Instruction { instructions.push(Instruction {
program_id: accounts::AMM_PROGRAM, program_id: accounts::AMM_PROGRAM,
accounts: accounts.clone(), accounts,
data: data.to_vec(), data: data.to_vec(),
}; });
instructions.push(sell_instruction);
if close_wsol_ata { if close_wsol_ata {
instructions.extend(crate::trading::common::close_wsol(&params.payer.pubkey())); instructions.extend(crate::trading::common::close_wsol(&params.payer.pubkey()));
@@ -465,12 +458,12 @@ pub fn claim_cashback_pumpswap_instruction(
// IDL order: user, user_volume_accumulator, quote_mint, quote_token_program, // IDL order: user, user_volume_accumulator, quote_mint, quote_token_program,
// user_volume_accumulator_wsol_token_account, user_wsol_token_account, system_program, event_authority, program // user_volume_accumulator_wsol_token_account, user_wsol_token_account, system_program, event_authority, program
let accounts = vec![ let accounts = vec![
AccountMeta::new(*payer, true), // user (signer, writable) AccountMeta::new(*payer, true), // user (signer, writable)
AccountMeta::new(user_volume_accumulator, false), // user_volume_accumulator (writable) AccountMeta::new(user_volume_accumulator, false), // user_volume_accumulator (writable)
AccountMeta::new_readonly(quote_mint, false), AccountMeta::new_readonly(quote_mint, false),
AccountMeta::new_readonly(quote_token_program, false), AccountMeta::new_readonly(quote_token_program, false),
AccountMeta::new(user_volume_accumulator_wsol_ata, false), // writable AccountMeta::new(user_volume_accumulator_wsol_ata, false), // writable
AccountMeta::new(user_wsol_ata, false), // writable AccountMeta::new(user_wsol_ata, false), // writable
crate::constants::SYSTEM_PROGRAM_META, crate::constants::SYSTEM_PROGRAM_META,
accounts::EVENT_AUTHORITY_META, accounts::EVENT_AUTHORITY_META,
accounts::AMM_PROGRAM_META, accounts::AMM_PROGRAM_META,
+13 -5
View File
@@ -29,7 +29,7 @@ impl InstructionBuilder for RaydiumAmmV4InstructionBuilder {
.protocol_params .protocol_params
.as_any() .as_any()
.downcast_ref::<RaydiumAmmV4Params>() .downcast_ref::<RaydiumAmmV4Params>()
.ok_or_else(|| anyhow!("Invalid protocol params for RaydiumCpmm"))?; .ok_or_else(|| anyhow!("Invalid protocol params for RaydiumAmmV4"))?;
let is_wsol = protocol_params.coin_mint == crate::constants::WSOL_TOKEN_ACCOUNT let is_wsol = protocol_params.coin_mint == crate::constants::WSOL_TOKEN_ACCOUNT
|| protocol_params.pc_mint == crate::constants::WSOL_TOKEN_ACCOUNT; || protocol_params.pc_mint == crate::constants::WSOL_TOKEN_ACCOUNT;
@@ -44,7 +44,7 @@ impl InstructionBuilder for RaydiumAmmV4InstructionBuilder {
// ======================================== // ========================================
// Trade calculation and account address preparation // Trade calculation and account address preparation
// ======================================== // ========================================
let is_base_in = protocol_params.coin_mint == crate::constants::WSOL_TOKEN_ACCOUNT let is_base_in = protocol_params.coin_mint == crate::constants::WSOL_TOKEN_ACCOUNT
|| protocol_params.coin_mint == crate::constants::USDC_TOKEN_ACCOUNT; || protocol_params.coin_mint == crate::constants::USDC_TOKEN_ACCOUNT;
let amount_in: u64 = params.input_amount.unwrap_or(0); let amount_in: u64 = params.input_amount.unwrap_or(0);
let swap_result = compute_swap_amount( let swap_result = compute_swap_amount(
@@ -62,7 +62,11 @@ impl InstructionBuilder for RaydiumAmmV4InstructionBuilder {
let user_source_token_account = let user_source_token_account =
crate::common::fast_fn::get_associated_token_address_with_program_id_fast_use_seed( crate::common::fast_fn::get_associated_token_address_with_program_id_fast_use_seed(
&params.payer.pubkey(), &params.payer.pubkey(),
if is_wsol { &crate::constants::WSOL_TOKEN_ACCOUNT } else { &crate::constants::USDC_TOKEN_ACCOUNT }, if is_wsol {
&crate::constants::WSOL_TOKEN_ACCOUNT
} else {
&crate::constants::USDC_TOKEN_ACCOUNT
},
&crate::constants::TOKEN_PROGRAM, &crate::constants::TOKEN_PROGRAM,
params.open_seed_optimize, params.open_seed_optimize,
); );
@@ -144,7 +148,7 @@ impl InstructionBuilder for RaydiumAmmV4InstructionBuilder {
.protocol_params .protocol_params
.as_any() .as_any()
.downcast_ref::<RaydiumAmmV4Params>() .downcast_ref::<RaydiumAmmV4Params>()
.ok_or_else(|| anyhow!("Invalid protocol params for RaydiumCpmm"))?; .ok_or_else(|| anyhow!("Invalid protocol params for RaydiumAmmV4"))?;
if params.input_amount.is_none() || params.input_amount.unwrap_or(0) == 0 { if params.input_amount.is_none() || params.input_amount.unwrap_or(0) == 0 {
return Err(anyhow!("Token amount is not set")); return Err(anyhow!("Token amount is not set"));
@@ -187,7 +191,11 @@ impl InstructionBuilder for RaydiumAmmV4InstructionBuilder {
let user_destination_token_account = let user_destination_token_account =
crate::common::fast_fn::get_associated_token_address_with_program_id_fast_use_seed( crate::common::fast_fn::get_associated_token_address_with_program_id_fast_use_seed(
&params.payer.pubkey(), &params.payer.pubkey(),
if is_wsol { &crate::constants::WSOL_TOKEN_ACCOUNT } else { &crate::constants::USDC_TOKEN_ACCOUNT }, if is_wsol {
&crate::constants::WSOL_TOKEN_ACCOUNT
} else {
&crate::constants::USDC_TOKEN_ACCOUNT
},
&crate::constants::TOKEN_PROGRAM, &crate::constants::TOKEN_PROGRAM,
params.open_seed_optimize, params.open_seed_optimize,
); );
+44 -18
View File
@@ -61,7 +61,7 @@ impl InstructionBuilder for RaydiumCpmmInstructionBuilder {
// ======================================== // ========================================
// Trade calculation and account address preparation // Trade calculation and account address preparation
// ======================================== // ========================================
let is_base_in = protocol_params.base_mint == crate::constants::WSOL_TOKEN_ACCOUNT let is_base_in = protocol_params.base_mint == crate::constants::WSOL_TOKEN_ACCOUNT
|| protocol_params.base_mint == crate::constants::USDC_TOKEN_ACCOUNT; || protocol_params.base_mint == crate::constants::USDC_TOKEN_ACCOUNT;
let mint_token_program = if is_base_in { let mint_token_program = if is_base_in {
protocol_params.quote_token_program protocol_params.quote_token_program
@@ -84,7 +84,11 @@ impl InstructionBuilder for RaydiumCpmmInstructionBuilder {
let input_token_account = get_associated_token_address_with_program_id_fast_use_seed( let input_token_account = get_associated_token_address_with_program_id_fast_use_seed(
&params.payer.pubkey(), &params.payer.pubkey(),
if is_wsol { &crate::constants::WSOL_TOKEN_ACCOUNT } else { &crate::constants::USDC_TOKEN_ACCOUNT }, if is_wsol {
&crate::constants::WSOL_TOKEN_ACCOUNT
} else {
&crate::constants::USDC_TOKEN_ACCOUNT
},
&crate::constants::TOKEN_PROGRAM, &crate::constants::TOKEN_PROGRAM,
params.open_seed_optimize, params.open_seed_optimize,
); );
@@ -97,10 +101,15 @@ impl InstructionBuilder for RaydiumCpmmInstructionBuilder {
let input_vault_account = get_vault_account( let input_vault_account = get_vault_account(
&pool_state, &pool_state,
if is_wsol { &crate::constants::WSOL_TOKEN_ACCOUNT } else { &crate::constants::USDC_TOKEN_ACCOUNT }, if is_wsol {
&crate::constants::WSOL_TOKEN_ACCOUNT
} else {
&crate::constants::USDC_TOKEN_ACCOUNT
},
protocol_params, protocol_params,
); );
let output_vault_account = get_vault_account(&pool_state, &params.output_mint, protocol_params); let output_vault_account =
get_vault_account(&pool_state, &params.output_mint, protocol_params);
let observation_state_account = if protocol_params.observation_state == Pubkey::default() { let observation_state_account = if protocol_params.observation_state == Pubkey::default() {
get_observation_state_pda(&pool_state).unwrap() get_observation_state_pda(&pool_state).unwrap()
@@ -136,13 +145,17 @@ impl InstructionBuilder for RaydiumCpmmInstructionBuilder {
accounts::AUTHORITY_META, // Authority (readonly) accounts::AUTHORITY_META, // Authority (readonly)
AccountMeta::new(protocol_params.amm_config, false), // Amm Config (readonly) AccountMeta::new(protocol_params.amm_config, false), // Amm Config (readonly)
AccountMeta::new(pool_state, false), // Pool State AccountMeta::new(pool_state, false), // Pool State
AccountMeta::new(input_token_account, false), // Input Token Account AccountMeta::new(input_token_account, false), // Input Token Account
AccountMeta::new(output_token_account, false), // Output Token Account AccountMeta::new(output_token_account, false), // Output Token Account
AccountMeta::new(input_vault_account, false), // Input Vault Account AccountMeta::new(input_vault_account, false), // Input Vault Account
AccountMeta::new(output_vault_account, false), // Output Vault Account AccountMeta::new(output_vault_account, false), // Output Vault Account
crate::constants::TOKEN_PROGRAM_META, // Input Token Program (readonly) crate::constants::TOKEN_PROGRAM_META, // Input Token Program (readonly)
AccountMeta::new_readonly(mint_token_program, false), // Output Token Program (readonly) AccountMeta::new_readonly(mint_token_program, false), // Output Token Program (readonly)
if is_wsol { crate::constants::WSOL_TOKEN_ACCOUNT_META } else { crate::constants::USDC_TOKEN_ACCOUNT_META }, // Input token mint (readonly) if is_wsol {
crate::constants::WSOL_TOKEN_ACCOUNT_META
} else {
crate::constants::USDC_TOKEN_ACCOUNT_META
}, // Input token mint (readonly)
AccountMeta::new_readonly(params.output_mint, false), // Output token mint (readonly) AccountMeta::new_readonly(params.output_mint, false), // Output token mint (readonly)
AccountMeta::new(observation_state_account, false), // Observation State Account AccountMeta::new(observation_state_account, false), // Observation State Account
]; ];
@@ -196,7 +209,7 @@ impl InstructionBuilder for RaydiumCpmmInstructionBuilder {
let is_usdc = protocol_params.base_mint == crate::constants::USDC_TOKEN_ACCOUNT let is_usdc = protocol_params.base_mint == crate::constants::USDC_TOKEN_ACCOUNT
|| protocol_params.quote_mint == crate::constants::USDC_TOKEN_ACCOUNT; || protocol_params.quote_mint == crate::constants::USDC_TOKEN_ACCOUNT;
if !is_wsol && !is_usdc { if !is_wsol && !is_usdc {
return Err(anyhow!("Pool must contain WSOL or USDC")); return Err(anyhow!("Pool must contain WSOL or USDC"));
} }
@@ -228,7 +241,11 @@ impl InstructionBuilder for RaydiumCpmmInstructionBuilder {
let output_token_account = get_associated_token_address_with_program_id_fast_use_seed( let output_token_account = get_associated_token_address_with_program_id_fast_use_seed(
&params.payer.pubkey(), &params.payer.pubkey(),
if is_wsol { &crate::constants::WSOL_TOKEN_ACCOUNT } else { &crate::constants::USDC_TOKEN_ACCOUNT }, if is_wsol {
&crate::constants::WSOL_TOKEN_ACCOUNT
} else {
&crate::constants::USDC_TOKEN_ACCOUNT
},
&crate::constants::TOKEN_PROGRAM, &crate::constants::TOKEN_PROGRAM,
params.open_seed_optimize, params.open_seed_optimize,
); );
@@ -241,10 +258,15 @@ impl InstructionBuilder for RaydiumCpmmInstructionBuilder {
let output_vault_account = get_vault_account( let output_vault_account = get_vault_account(
&pool_state, &pool_state,
if is_wsol { &crate::constants::WSOL_TOKEN_ACCOUNT } else { &crate::constants::USDC_TOKEN_ACCOUNT }, if is_wsol {
&crate::constants::WSOL_TOKEN_ACCOUNT
} else {
&crate::constants::USDC_TOKEN_ACCOUNT
},
protocol_params, protocol_params,
); );
let input_vault_account = get_vault_account(&pool_state, &params.input_mint, protocol_params); let input_vault_account =
get_vault_account(&pool_state, &params.input_mint, protocol_params);
let observation_state_account = if protocol_params.observation_state == Pubkey::default() { let observation_state_account = if protocol_params.observation_state == Pubkey::default() {
get_observation_state_pda(&pool_state).unwrap() get_observation_state_pda(&pool_state).unwrap()
@@ -267,14 +289,18 @@ impl InstructionBuilder for RaydiumCpmmInstructionBuilder {
accounts::AUTHORITY_META, // Authority (readonly) accounts::AUTHORITY_META, // Authority (readonly)
AccountMeta::new(protocol_params.amm_config, false), // Amm Config (readonly) AccountMeta::new(protocol_params.amm_config, false), // Amm Config (readonly)
AccountMeta::new(pool_state, false), // Pool State AccountMeta::new(pool_state, false), // Pool State
AccountMeta::new(input_token_account, false), // Input Token Account AccountMeta::new(input_token_account, false), // Input Token Account
AccountMeta::new(output_token_account, false), // Output Token Account AccountMeta::new(output_token_account, false), // Output Token Account
AccountMeta::new(input_vault_account, false), // Input Vault Account AccountMeta::new(input_vault_account, false), // Input Vault Account
AccountMeta::new(output_vault_account, false), // Output Vault Account AccountMeta::new(output_vault_account, false), // Output Vault Account
AccountMeta::new_readonly(mint_token_program, false), // Input Token Program (readonly) AccountMeta::new_readonly(mint_token_program, false), // Input Token Program (readonly)
crate::constants::TOKEN_PROGRAM_META, // Output Token Program (readonly) crate::constants::TOKEN_PROGRAM_META, // Output Token Program (readonly)
AccountMeta::new_readonly(params.input_mint, false), // Input token mint (readonly) AccountMeta::new_readonly(params.input_mint, false), // Input token mint (readonly)
if is_wsol { crate::constants::WSOL_TOKEN_ACCOUNT_META } else { crate::constants::USDC_TOKEN_ACCOUNT_META }, // Output token mint (readonly) if is_wsol {
crate::constants::WSOL_TOKEN_ACCOUNT_META
} else {
crate::constants::USDC_TOKEN_ACCOUNT_META
}, // Output token mint (readonly)
AccountMeta::new(observation_state_account, false), // Observation State Account AccountMeta::new(observation_state_account, false), // Observation State Account
]; ];
// Create instruction data // Create instruction data
+2 -2
View File
@@ -1,13 +1,13 @@
pub mod bonk; pub mod bonk;
pub mod meteora_damm_v2;
pub mod pumpfun; pub mod pumpfun;
pub mod pumpswap; pub mod pumpswap;
pub mod raydium_amm_v4; pub mod raydium_amm_v4;
pub mod raydium_cpmm; pub mod raydium_cpmm;
pub mod meteora_damm_v2;
// types // types
pub mod bonk_types; pub mod bonk_types;
pub mod meteora_damm_v2_types;
pub mod pumpswap_types; pub mod pumpswap_types;
pub mod raydium_amm_v4_types; pub mod raydium_amm_v4_types;
pub mod raydium_cpmm_types; pub mod raydium_cpmm_types;
pub mod meteora_damm_v2_types;
+55 -7
View File
@@ -1,6 +1,7 @@
use crate::common::{bonding_curve::BondingCurveAccount, SolanaRpcClient}; use crate::common::{bonding_curve::BondingCurveAccount, SolanaRpcClient};
use anyhow::anyhow; use anyhow::anyhow;
use solana_sdk::pubkey::Pubkey; use rand::seq::IndexedRandom;
use solana_sdk::{instruction::AccountMeta, pubkey::Pubkey};
use std::sync::Arc; use std::sync::Arc;
/// Constants used as seeds for deriving PDAs (Program Derived Addresses) /// Constants used as seeds for deriving PDAs (Program Derived Addresses)
@@ -59,8 +60,18 @@ pub mod global_constants {
is_writable: true, is_writable: true,
}; };
pub const MAYHEM_FEE_RECIPIENT: Pubkey = /// Mayhem fee recipients (pump-public-docs: use any one randomly)
pubkey!("GesfTA3X2arioaHp8bbKdjG9vJtskViWACZoYvxp4twS"); pub const MAYHEM_FEE_RECIPIENTS: [Pubkey; 8] = [
pubkey!("GesfTA3X2arioaHp8bbKdjG9vJtskViWACZoYvxp4twS"),
pubkey!("4budycTjhs9fD6xw62VBducVTNgMgJJ5BgtKq7mAZwn6"),
pubkey!("8SBKzEQU4nLSzcwF4a74F2iaUDQyTfjGndn6qUWBnrpR"),
pubkey!("4UQeTP1T39KZ9Sfxzo3WR5skgsaP6NZa87BAkuazLEKH"),
pubkey!("8sNeir4QsLsJdYpc9RZacohhK1Y5FLU3nC5LXgYB4aa6"),
pubkey!("Fh9HmeLNUMVCvejxCtCL2DbYaRyBFVJ5xrWkLnMH6fdk"),
pubkey!("463MEnMeGyJekNZFQSTUABBEbLnvMTALbT6ZmsxAbAdq"),
pubkey!("6AUH3WEHucYZyC61hqpqYUWVto5qA5hjHuNQ32GNnNxA"),
];
pub const MAYHEM_FEE_RECIPIENT: Pubkey = MAYHEM_FEE_RECIPIENTS[0];
pub const MAYHEM_FEE_RECIPIENT_META: solana_sdk::instruction::AccountMeta = pub const MAYHEM_FEE_RECIPIENT_META: solana_sdk::instruction::AccountMeta =
solana_sdk::instruction::AccountMeta { solana_sdk::instruction::AccountMeta {
pubkey: MAYHEM_FEE_RECIPIENT, pubkey: MAYHEM_FEE_RECIPIENT,
@@ -161,6 +172,15 @@ pub const BUY_DISCRIMINATOR: [u8; 8] = [102, 6, 61, 18, 1, 218, 235, 234];
pub const BUY_EXACT_SOL_IN_DISCRIMINATOR: [u8; 8] = [56, 252, 116, 8, 158, 223, 205, 95]; pub const BUY_EXACT_SOL_IN_DISCRIMINATOR: [u8; 8] = [56, 252, 116, 8, 158, 223, 205, 95];
pub const SELL_DISCRIMINATOR: [u8; 8] = [51, 230, 133, 164, 1, 127, 131, 173]; pub const SELL_DISCRIMINATOR: [u8; 8] = [51, 230, 133, 164, 1, 127, 131, 173];
/// Returns a random Mayhem fee recipient AccountMeta (pump-public-docs: Bonding Curve 2nd account = Mayhem fee recipient; use any one randomly).
#[inline]
pub fn get_mayhem_fee_recipient_meta_random() -> AccountMeta {
let recipient = *global_constants::MAYHEM_FEE_RECIPIENTS
.choose(&mut rand::rng())
.unwrap_or(&global_constants::MAYHEM_FEE_RECIPIENTS[0]);
AccountMeta { pubkey: recipient, is_signer: false, is_writable: true }
}
pub struct Symbol; pub struct Symbol;
impl Symbol { impl Symbol {
@@ -202,10 +222,9 @@ pub fn get_creator(creator_vault_pda: &Pubkey) -> Pubkey {
// Fast check against cached default creator vault // Fast check against cached default creator vault
static DEFAULT_CREATOR_VAULT: std::sync::LazyLock<Option<Pubkey>> = static DEFAULT_CREATOR_VAULT: std::sync::LazyLock<Option<Pubkey>> =
std::sync::LazyLock::new(|| get_creator_vault_pda(&Pubkey::default())); std::sync::LazyLock::new(|| get_creator_vault_pda(&Pubkey::default()));
if creator_vault_pda.eq(&DEFAULT_CREATOR_VAULT.unwrap()) { match DEFAULT_CREATOR_VAULT.as_ref() {
Pubkey::default() Some(default) if creator_vault_pda.eq(default) => Pubkey::default(),
} else { _ => *creator_vault_pda,
*creator_vault_pda
} }
} }
} }
@@ -275,3 +294,32 @@ pub fn get_buy_price(
s_u64.min(real_token_reserves) s_u64.min(real_token_reserves)
} }
#[cfg(test)]
mod tests {
use super::*;
use solana_sdk::pubkey::Pubkey;
#[test]
fn pumpfun_discriminators_are_8_bytes() {
assert_eq!(BUY_DISCRIMINATOR.len(), 8);
assert_eq!(BUY_EXACT_SOL_IN_DISCRIMINATOR.len(), 8);
assert_eq!(SELL_DISCRIMINATOR.len(), 8);
}
#[test]
fn pumpfun_bonding_curve_and_v2_pda_differ_for_same_mint() {
let mint = Pubkey::new_unique();
let pda = get_bonding_curve_pda(&mint).unwrap();
let pda_v2 = get_bonding_curve_v2_pda(&mint).unwrap();
assert_ne!(pda, pda_v2, "bonding_curve and bonding_curve_v2 PDAs must differ");
}
#[test]
fn pumpfun_creator_vault_pda_deterministic() {
let creator = Pubkey::new_unique();
let a = get_creator_vault_pda(&creator).unwrap();
let b = get_creator_vault_pda(&creator).unwrap();
assert_eq!(a, b);
}
}
+174 -27
View File
@@ -2,12 +2,17 @@ use crate::{
common::{ common::{
spl_associated_token_account::get_associated_token_address_with_program_id, SolanaRpcClient, spl_associated_token_account::get_associated_token_address_with_program_id, SolanaRpcClient,
}, },
constants::TOKEN_PROGRAM, constants::{TOKEN_PROGRAM, WSOL_TOKEN_ACCOUNT},
instruction::utils::pumpswap_types::{pool_decode, Pool}, instruction::utils::pumpswap_types::{pool_decode, Pool},
}; };
use anyhow::anyhow; use anyhow::anyhow;
use rand::seq::IndexedRandom;
use solana_account_decoder::UiAccountEncoding; use solana_account_decoder::UiAccountEncoding;
use solana_sdk::pubkey::Pubkey; use solana_sdk::{instruction::AccountMeta, pubkey::Pubkey};
/// PumpSwap 池账户总长度(见 pump-public-docs Breaking Change):8 字节 discriminator + 244 字节 Pool。
/// 官方文档:pool structure needs to be 244 bytes (was 243),含 is_mayhem_mode。DataSize 必须与此一致,否则 getProgramAccounts 会返回 0。
const POOL_ACCOUNT_DATA_LEN: u64 = 8 + 244;
/// Constants used as seeds for deriving PDAs (Program Derived Addresses) /// Constants used as seeds for deriving PDAs (Program Derived Addresses)
pub mod seeds { pub mod seeds {
@@ -29,6 +34,10 @@ pub mod seeds {
/// Seed for pool v2 PDA (required by program upgrade, readonly at end of account list) /// Seed for pool v2 PDA (required by program upgrade, readonly at end of account list)
pub const POOL_V2_SEED: &[u8] = b"pool-v2"; pub const POOL_V2_SEED: &[u8] = b"pool-v2";
/// Legacy pool PDA seed (used with index, creator, base_mint, quote_mint)
pub const POOL_SEED: &[u8] = b"pool";
/// Pump program: pool-authority PDA seed (creator for canonical pool)
pub const POOL_AUTHORITY_SEED: &[u8] = b"pool-authority";
} }
/// Constants related to program accounts and authorities /// Constants related to program accounts and authorities
@@ -53,6 +62,8 @@ pub mod accounts {
pubkey!("62qc2CNXwrYqQScmEdiZFFAnJR262PxWEuNQtxfafNgV"); pubkey!("62qc2CNXwrYqQScmEdiZFFAnJR262PxWEuNQtxfafNgV");
pub const AMM_PROGRAM: Pubkey = pubkey!("pAMMBay6oceH9fJKBRHGP5D4bD4sWpmSwMn52FMfXEA"); pub const AMM_PROGRAM: Pubkey = pubkey!("pAMMBay6oceH9fJKBRHGP5D4bD4sWpmSwMn52FMfXEA");
/// Pump Bonding Curve programcanonical pool 的 creator 来自此程序的 pool-authority PDA
pub const PUMP_PROGRAM_ID: Pubkey = pubkey!("6EF8rrecthR5Dkzon8Nwu78hRvfCKubJ14M5uBEwF6P");
pub const LP_FEE_BASIS_POINTS: u64 = 25; pub const LP_FEE_BASIS_POINTS: u64 = 25;
pub const PROTOCOL_FEE_BASIS_POINTS: u64 = 5; pub const PROTOCOL_FEE_BASIS_POINTS: u64 = 5;
@@ -68,9 +79,19 @@ pub mod accounts {
pub const DEFAULT_COIN_CREATOR_VAULT_AUTHORITY: Pubkey = pub const DEFAULT_COIN_CREATOR_VAULT_AUTHORITY: Pubkey =
pubkey!("8N3GDaZ2iwN65oxVatKTLPNooAVUJTbfiVJ1ahyqwjSk"); pubkey!("8N3GDaZ2iwN65oxVatKTLPNooAVUJTbfiVJ1ahyqwjSk");
/// Mayhem fee recipient (for mayhem mode coins) /// Mayhem fee recipients (pump-public-docs: use any one randomly for throughput)
pub const MAYHEM_FEE_RECIPIENT: Pubkey = pub const MAYHEM_FEE_RECIPIENTS: [Pubkey; 8] = [
pubkey!("GesfTA3X2arioaHp8bbKdjG9vJtskViWACZoYvxp4twS"); pubkey!("GesfTA3X2arioaHp8bbKdjG9vJtskViWACZoYvxp4twS"),
pubkey!("4budycTjhs9fD6xw62VBducVTNgMgJJ5BgtKq7mAZwn6"),
pubkey!("8SBKzEQU4nLSzcwF4a74F2iaUDQyTfjGndn6qUWBnrpR"),
pubkey!("4UQeTP1T39KZ9Sfxzo3WR5skgsaP6NZa87BAkuazLEKH"),
pubkey!("8sNeir4QsLsJdYpc9RZacohhK1Y5FLU3nC5LXgYB4aa6"),
pubkey!("Fh9HmeLNUMVCvejxCtCL2DbYaRyBFVJ5xrWkLnMH6fdk"),
pubkey!("463MEnMeGyJekNZFQSTUABBEbLnvMTALbT6ZmsxAbAdq"),
pubkey!("6AUH3WEHucYZyC61hqpqYUWVto5qA5hjHuNQ32GNnNxA"),
];
/// Default Mayhem fee recipient (first of MAYHEM_FEE_RECIPIENTS)
pub const MAYHEM_FEE_RECIPIENT: Pubkey = MAYHEM_FEE_RECIPIENTS[0];
// META // META
@@ -142,6 +163,16 @@ pub const BUY_DISCRIMINATOR: [u8; 8] = [102, 6, 61, 18, 1, 218, 235, 234];
pub const BUY_EXACT_QUOTE_IN_DISCRIMINATOR: [u8; 8] = [198, 46, 21, 82, 180, 217, 232, 112]; pub const BUY_EXACT_QUOTE_IN_DISCRIMINATOR: [u8; 8] = [198, 46, 21, 82, 180, 217, 232, 112];
pub const SELL_DISCRIMINATOR: [u8; 8] = [51, 230, 133, 164, 1, 127, 131, 173]; pub const SELL_DISCRIMINATOR: [u8; 8] = [51, 230, 133, 164, 1, 127, 131, 173];
/// Returns a random Mayhem fee recipient and its AccountMeta (pump-public-docs: use any one randomly).
#[inline]
pub fn get_mayhem_fee_recipient_random() -> (Pubkey, AccountMeta) {
let recipient = *accounts::MAYHEM_FEE_RECIPIENTS
.choose(&mut rand::rng())
.unwrap_or(&accounts::MAYHEM_FEE_RECIPIENTS[0]);
let meta = AccountMeta { pubkey: recipient, is_signer: false, is_writable: false };
(recipient, meta)
}
/// Pool v2 PDA (seeds: ["pool-v2", base_mint]). Required at end of buy/sell/buy_exact_quote_in accounts. /// Pool v2 PDA (seeds: ["pool-v2", base_mint]). Required at end of buy/sell/buy_exact_quote_in accounts.
#[inline] #[inline]
pub fn get_pool_v2_pda(base_mint: &Pubkey) -> Option<Pubkey> { pub fn get_pool_v2_pda(base_mint: &Pubkey) -> Option<Pubkey> {
@@ -152,6 +183,35 @@ pub fn get_pool_v2_pda(base_mint: &Pubkey) -> Option<Pubkey> {
Some(pda) Some(pda)
} }
/// Pump 程序上的 pool-authority PDAcanonical pool 的 creator),与 @pump-fun/pump-swap-sdk 一致。
#[inline]
pub fn get_pump_pool_authority_pda(mint: &Pubkey) -> Pubkey {
Pubkey::find_program_address(
&[seeds::POOL_AUTHORITY_SEED, mint.as_ref()],
&accounts::PUMP_PROGRAM_ID,
)
.0
}
/// Canonical Pump 池 PDAindex=0creator=pumpPoolAuthorityPda(mint)base_mint=mintquote_mint=WSOL。
/// 与 @pump-fun/pump-swap-sdk 的 canonicalPumpPoolPda(mint) 一致,用于从 bonding curve 迁移后的标准池查找。
#[inline]
pub fn get_canonical_pool_pda(mint: &Pubkey) -> Pubkey {
const CANONICAL_POOL_INDEX: u16 = 0;
let authority = get_pump_pool_authority_pda(mint);
let (pda, _) = Pubkey::find_program_address(
&[
seeds::POOL_SEED,
&CANONICAL_POOL_INDEX.to_le_bytes(),
authority.as_ref(),
mint.as_ref(),
WSOL_TOKEN_ACCOUNT.as_ref(),
],
&accounts::AMM_PROGRAM,
);
pda
}
// Find a pool for a specific mint // Find a pool for a specific mint
pub async fn find_pool(rpc: &SolanaRpcClient, mint: &Pubkey) -> Result<Pubkey, anyhow::Error> { pub async fn find_pool(rpc: &SolanaRpcClient, mint: &Pubkey) -> Result<Pubkey, anyhow::Error> {
let (pool_address, _) = find_by_mint(rpc, mint).await?; let (pool_address, _) = find_by_mint(rpc, mint).await?;
@@ -191,14 +251,14 @@ pub fn get_user_volume_accumulator_pda(user: &Pubkey) -> Option<Pubkey> {
crate::common::fast_fn::PdaCacheKey::PumpSwapUserVolume(*user), crate::common::fast_fn::PdaCacheKey::PumpSwapUserVolume(*user),
|| { || {
let seeds: &[&[u8]; 2] = &[&seeds::USER_VOLUME_ACCUMULATOR_SEED, user.as_ref()]; let seeds: &[&[u8]; 2] = &[&seeds::USER_VOLUME_ACCUMULATOR_SEED, user.as_ref()];
let program_id: &Pubkey = &&accounts::AMM_PROGRAM; let program_id: &Pubkey = &accounts::AMM_PROGRAM;
let pda: Option<(Pubkey, u8)> = Pubkey::try_find_program_address(seeds, program_id); let pda: Option<(Pubkey, u8)> = Pubkey::try_find_program_address(seeds, program_id);
pda.map(|pubkey| pubkey.0) pda.map(|pubkey| pubkey.0)
}, },
) )
} }
/// WSOL ATA of UserVolumeAccumulator for Pump AMM (used for cashback remaining accounts). /// WSOL ATA of UserVolumeAccumulator for Pump AMM (buy cashback: remaining_accounts[0] 官方用 NATIVE_MINT).
pub fn get_user_volume_accumulator_wsol_ata(user: &Pubkey) -> Option<Pubkey> { pub fn get_user_volume_accumulator_wsol_ata(user: &Pubkey) -> Option<Pubkey> {
let accumulator = get_user_volume_accumulator_pda(user)?; let accumulator = get_user_volume_accumulator_pda(user)?;
Some(crate::common::fast_fn::get_associated_token_address_with_program_id_fast( Some(crate::common::fast_fn::get_associated_token_address_with_program_id_fast(
@@ -208,9 +268,23 @@ pub fn get_user_volume_accumulator_wsol_ata(user: &Pubkey) -> Option<Pubkey> {
)) ))
} }
/// Quote-mint ATA of UserVolumeAccumulatorsell cashback 时官方用 quoteMint,非固定 WSOL.
pub fn get_user_volume_accumulator_quote_ata(
user: &Pubkey,
quote_mint: &Pubkey,
quote_token_program: &Pubkey,
) -> Option<Pubkey> {
let accumulator = get_user_volume_accumulator_pda(user)?;
Some(crate::common::fast_fn::get_associated_token_address_with_program_id_fast(
&accumulator,
quote_mint,
quote_token_program,
))
}
pub fn get_global_volume_accumulator_pda() -> Option<Pubkey> { pub fn get_global_volume_accumulator_pda() -> Option<Pubkey> {
let seeds: &[&[u8]; 1] = &[&seeds::GLOBAL_VOLUME_ACCUMULATOR_SEED]; let seeds: &[&[u8]; 1] = &[&seeds::GLOBAL_VOLUME_ACCUMULATOR_SEED];
let program_id: &Pubkey = &&accounts::AMM_PROGRAM; let program_id: &Pubkey = &accounts::AMM_PROGRAM;
let pda: Option<(Pubkey, u8)> = Pubkey::try_find_program_address(seeds, program_id); let pda: Option<(Pubkey, u8)> = Pubkey::try_find_program_address(seeds, program_id);
pda.map(|pubkey| pubkey.0) pda.map(|pubkey| pubkey.0)
} }
@@ -231,11 +305,12 @@ pub async fn find_by_base_mint(
rpc: &SolanaRpcClient, rpc: &SolanaRpcClient,
base_mint: &Pubkey, base_mint: &Pubkey,
) -> Result<(Pubkey, Pool), anyhow::Error> { ) -> Result<(Pubkey, Pool), anyhow::Error> {
// Use getProgramAccounts to find pools for the given mint // Use getProgramAccounts to find pools for the given mint.
// base_mint 在账户布局中的偏移:8(discriminator) + 1(bump) + 2(index) + 32(creator) = 43
let filters = vec![ let filters = vec![
// solana_rpc_client_api::filter::RpcFilterType::DataSize(211), // Pool account size solana_rpc_client_api::filter::RpcFilterType::DataSize(POOL_ACCOUNT_DATA_LEN),
solana_rpc_client_api::filter::RpcFilterType::Memcmp( solana_rpc_client_api::filter::RpcFilterType::Memcmp(
solana_client::rpc_filter::Memcmp::new_base58_encoded(43, &base_mint.to_bytes()), solana_client::rpc_filter::Memcmp::new_base58_encoded(43, base_mint.as_ref()),
), ),
]; ];
let config = solana_rpc_client_api::config::RpcProgramAccountsConfig { let config = solana_rpc_client_api::config::RpcProgramAccountsConfig {
@@ -255,7 +330,7 @@ pub async fn find_by_base_mint(
if accounts.is_empty() { if accounts.is_empty() {
return Err(anyhow!("No pool found for mint {}", base_mint)); return Err(anyhow!("No pool found for mint {}", base_mint));
} }
let accounts_count = accounts.len(); // 🔧 保存长度,因为 into_iter() 会消耗 accounts let accounts_count = accounts.len(); // 🔧 保存长度,因为 into_iter() 会消耗 accounts
let mut pools: Vec<_> = accounts let mut pools: Vec<_> = accounts
.into_iter() .into_iter()
.filter_map(|(addr, acc)| { .filter_map(|(addr, acc)| {
@@ -270,23 +345,28 @@ pub async fn find_by_base_mint(
// 🔧 修复:检查过滤后的 pools 是否为空(accounts 可能不为空但解码全部失败) // 🔧 修复:检查过滤后的 pools 是否为空(accounts 可能不为空但解码全部失败)
if pools.is_empty() { if pools.is_empty() {
return Err(anyhow!("No valid pool decoded for mint {} (found {} accounts but all decode failed)", base_mint, accounts_count)); return Err(anyhow!(
"No valid pool decoded for mint {} (found {} accounts but all decode failed)",
base_mint,
accounts_count
));
} }
pools.sort_by(|a, b| b.1.lp_supply.cmp(&a.1.lp_supply)); pools.sort_by(|a, b| b.1.lp_supply.cmp(&a.1.lp_supply));
let (address, pool) = pools[0].clone(); let first = &pools[0];
Ok((address, pool)) Ok((first.0, first.1.clone()))
} }
pub async fn find_by_quote_mint( pub async fn find_by_quote_mint(
rpc: &SolanaRpcClient, rpc: &SolanaRpcClient,
quote_mint: &Pubkey, quote_mint: &Pubkey,
) -> Result<(Pubkey, Pool), anyhow::Error> { ) -> Result<(Pubkey, Pool), anyhow::Error> {
// Use getProgramAccounts to find pools for the given mint // Use getProgramAccounts to find pools for the given mint.
// quote_mint 在账户布局中的偏移:8 + 1 + 2 + 32 + 32 = 75
let filters = vec![ let filters = vec![
// solana_rpc_client_api::filter::RpcFilterType::DataSize(211), // Pool account size solana_rpc_client_api::filter::RpcFilterType::DataSize(POOL_ACCOUNT_DATA_LEN),
solana_rpc_client_api::filter::RpcFilterType::Memcmp( solana_rpc_client_api::filter::RpcFilterType::Memcmp(
solana_client::rpc_filter::Memcmp::new_base58_encoded(75, &quote_mint.to_bytes()), solana_client::rpc_filter::Memcmp::new_base58_encoded(75, quote_mint.as_ref()),
), ),
]; ];
let config = solana_rpc_client_api::config::RpcProgramAccountsConfig { let config = solana_rpc_client_api::config::RpcProgramAccountsConfig {
@@ -306,7 +386,7 @@ pub async fn find_by_quote_mint(
if accounts.is_empty() { if accounts.is_empty() {
return Err(anyhow!("No pool found for mint {}", quote_mint)); return Err(anyhow!("No pool found for mint {}", quote_mint));
} }
let accounts_count = accounts.len(); // 🔧 保存长度,因为 into_iter() 会消耗 accounts let accounts_count = accounts.len(); // 🔧 保存长度,因为 into_iter() 会消耗 accounts
let mut pools: Vec<_> = accounts let mut pools: Vec<_> = accounts
.into_iter() .into_iter()
.filter_map(|(addr, acc)| { .filter_map(|(addr, acc)| {
@@ -321,25 +401,64 @@ pub async fn find_by_quote_mint(
// 🔧 修复:检查过滤后的 pools 是否为空(accounts 可能不为空但解码全部失败) // 🔧 修复:检查过滤后的 pools 是否为空(accounts 可能不为空但解码全部失败)
if pools.is_empty() { if pools.is_empty() {
return Err(anyhow!("No valid pool decoded for quote_mint {} (found {} accounts but all decode failed)", quote_mint, accounts_count)); return Err(anyhow!(
"No valid pool decoded for quote_mint {} (found {} accounts but all decode failed)",
quote_mint,
accounts_count
));
} }
pools.sort_by(|a, b| b.1.lp_supply.cmp(&a.1.lp_supply)); pools.sort_by(|a, b| b.1.lp_supply.cmp(&a.1.lp_supply));
let (address, pool) = pools[0].clone(); let first = &pools[0];
Ok((address, pool)) Ok((first.0, first.1.clone()))
} }
/// 按 mint 查找 PumpSwap 池(本函数仅用于 PumpSwap,其他 DEX 勿用)。
///
/// 查找顺序(与 @pump-fun/pump-swap-sdk 一致):
/// 1. Pool v2 PDA ["pool-v2", base_mint] — 一次 getAccount
/// 2. Canonical pool PDA ["pool", 0, pumpPoolAuthority(mint), mint, WSOL] — 迁移后的标准池
/// 3. getProgramAccounts 按 base_mint / quote_mint 过滤
pub async fn find_by_mint( pub async fn find_by_mint(
rpc: &SolanaRpcClient, rpc: &SolanaRpcClient,
mint: &Pubkey, mint: &Pubkey,
) -> Result<(Pubkey, Pool), anyhow::Error> { ) -> Result<(Pubkey, Pool), anyhow::Error> {
if let Ok((address, pool)) = find_by_base_mint(rpc, mint).await { let mut diag = Vec::<String>::new();
return Ok((address, pool));
// 1. PumpSwap v2 PDAseeds: ["pool-v2", base_mint]
if let Some(pool_address) = get_pool_v2_pda(mint) {
diag.push(format!("PDA(v2)={}", pool_address));
match fetch_pool(rpc, &pool_address).await {
Ok(pool) if pool.base_mint == *mint => return Ok((pool_address, pool)),
Ok(_) => diag.push("PDA(v2) 账户存在但 base_mint 不匹配".into()),
Err(e) => diag.push(format!("PDA(v2) get_account/decode 失败: {}", e)),
}
} }
if let Ok((address, pool)) = find_by_quote_mint(rpc, mint).await {
return Ok((address, pool)); // 2. Canonical pool PDA(与 pump-swap-sdk canonicalPumpPoolPda(mint) 一致)
let canonical_address = get_canonical_pool_pda(mint);
diag.push(format!("canonical={}", canonical_address));
match fetch_pool(rpc, &canonical_address).await {
Ok(pool) if pool.base_mint == *mint => return Ok((canonical_address, pool)),
Ok(_) => diag.push("canonical 账户存在但 base_mint 不匹配".into()),
Err(e) => diag.push(format!("canonical get_account/decode 失败: {}", e)),
} }
Err(anyhow!("No pool found for mint {}", mint))
// 3. 回退:getProgramAccounts 按 base_mint / quote_mint
match find_by_base_mint(rpc, mint).await {
Ok((address, pool)) => return Ok((address, pool)),
Err(e) => diag.push(format!("getProgramAccounts(base_mint): {}", e)),
}
match find_by_quote_mint(rpc, mint).await {
Ok((address, pool)) => return Ok((address, pool)),
Err(e) => diag.push(format!("getProgramAccounts(quote_mint): {}", e)),
}
Err(anyhow!(
"No pool found for mint {}. 诊断: {}。若使用自建 RPC 请确认已开启 getProgramAccounts 或换用公共 RPC 重试;若代币未在 PumpSwap 建池请先在 pump.fun/DEX 上确认",
mint,
diag.join("; ")
))
} }
pub async fn get_token_balances( pub async fn get_token_balances(
@@ -362,3 +481,31 @@ pub fn get_fee_config_pda() -> Option<Pubkey> {
let pda: Option<(Pubkey, u8)> = Pubkey::try_find_program_address(seeds, program_id); let pda: Option<(Pubkey, u8)> = Pubkey::try_find_program_address(seeds, program_id);
pda.map(|pubkey| pubkey.0) pda.map(|pubkey| pubkey.0)
} }
#[cfg(test)]
mod tests {
use super::*;
use solana_sdk::pubkey::Pubkey;
#[test]
fn pumpswap_user_volume_accumulator_pda_deterministic() {
let user = Pubkey::new_unique();
let a = get_user_volume_accumulator_pda(&user).unwrap();
let b = get_user_volume_accumulator_pda(&user).unwrap();
assert_eq!(a, b);
}
#[test]
fn pumpswap_global_volume_accumulator_matches_constant() {
let pda = get_global_volume_accumulator_pda().unwrap();
assert_eq!(pda, accounts::GLOBAL_VOLUME_ACCUMULATOR);
}
#[test]
fn pumpswap_pool_v2_pda_deterministic() {
let base_mint = Pubkey::new_unique();
let a = get_pool_v2_pda(&base_mint).unwrap();
let b = get_pool_v2_pda(&base_mint).unwrap();
assert_eq!(a, b);
}
}
+4 -1
View File
@@ -17,9 +17,12 @@ pub struct Pool {
pub is_mayhem_mode: bool, pub is_mayhem_mode: bool,
/// Whether this pool's coin has cashback enabled /// Whether this pool's coin has cashback enabled
pub is_cashback_coin: bool, pub is_cashback_coin: bool,
/// Reserved for future fields (pump-public-docs: pool structure = 244 bytes total)
pub _reserved: [u8; 7],
} }
pub const POOL_SIZE: usize = 1 + 2 + 32 * 6 + 8 + 32 + 1 + 1; /// Borsh 解码用的 Pool 长度。链上池为 244 字节(pump-public-docs Breaking Change),与 POOL_SIZE 一致。
pub const POOL_SIZE: usize = 244;
pub fn pool_decode(data: &[u8]) -> Option<Pool> { pub fn pool_decode(data: &[u8]) -> Option<Pool> {
if data.len() < POOL_SIZE { if data.len() < POOL_SIZE {
+4 -2
View File
@@ -135,9 +135,11 @@ pub fn get_vault_account(
) -> Pubkey { ) -> Pubkey {
if protocol_params.base_mint == *token_mint && protocol_params.base_vault != Pubkey::default() { if protocol_params.base_mint == *token_mint && protocol_params.base_vault != Pubkey::default() {
protocol_params.base_vault protocol_params.base_vault
} else if protocol_params.quote_mint == *token_mint && protocol_params.quote_vault != Pubkey::default() { } else if protocol_params.quote_mint == *token_mint
&& protocol_params.quote_vault != Pubkey::default()
{
protocol_params.quote_vault protocol_params.quote_vault
} else { } else {
get_vault_pda(pool_state, token_mint).unwrap() get_vault_pda(pool_state, token_mint).unwrap()
} }
} }
+319 -168
View File
@@ -8,7 +8,7 @@ pub mod utils;
use crate::common::nonce_cache::DurableNonceInfo; use crate::common::nonce_cache::DurableNonceInfo;
use crate::common::sdk_log; use crate::common::sdk_log;
use crate::common::GasFeeStrategy; use crate::common::GasFeeStrategy;
use crate::common::{TradeConfig, InfrastructureConfig}; use crate::common::{InfrastructureConfig, TradeConfig};
#[cfg(feature = "perf-trace")] #[cfg(feature = "perf-trace")]
use crate::constants::trade::trade::DEFAULT_SLIPPAGE; use crate::constants::trade::trade::DEFAULT_SLIPPAGE;
use crate::constants::SOL_TOKEN_ACCOUNT; use crate::constants::SOL_TOKEN_ACCOUNT;
@@ -19,13 +19,15 @@ use crate::swqos::common::TradeError;
use crate::swqos::SwqosClient; use crate::swqos::SwqosClient;
use crate::swqos::SwqosConfig; use crate::swqos::SwqosConfig;
use crate::swqos::TradeType; use crate::swqos::TradeType;
// Re-export for SWQOS HTTP/QUIC choice in SwqosConfig (e.g. Astralane)
pub use crate::swqos::SwqosTransport;
use crate::trading::core::params::BonkParams; use crate::trading::core::params::BonkParams;
use crate::trading::core::params::DexParamEnum;
use crate::trading::core::params::MeteoraDammV2Params; use crate::trading::core::params::MeteoraDammV2Params;
use crate::trading::core::params::PumpFunParams; use crate::trading::core::params::PumpFunParams;
use crate::trading::core::params::PumpSwapParams; use crate::trading::core::params::PumpSwapParams;
use crate::trading::core::params::RaydiumAmmV4Params; use crate::trading::core::params::RaydiumAmmV4Params;
use crate::trading::core::params::RaydiumCpmmParams; use crate::trading::core::params::RaydiumCpmmParams;
use crate::trading::core::params::DexParamEnum;
use crate::trading::factory::DexType; use crate::trading::factory::DexType;
use crate::trading::MiddlewareManager; use crate::trading::MiddlewareManager;
use crate::trading::SwapParams; use crate::trading::SwapParams;
@@ -54,6 +56,20 @@ fn validate_protocol_params(dex_type: DexType, params: &DexParamEnum) -> bool {
} }
} }
/// 按 mint 查找池地址(通用入口,根据 DEX 类型分发,仅 PumpSwap 等已实现的类型会走优化路径)。
///
/// * `dex_type`PumpSwap 时先走 PDA 再回退 getProgramAccounts,其他类型返回未实现错误。
pub async fn find_pool_by_mint(
rpc: &SolanaRpcClient,
mint: &Pubkey,
dex_type: DexType,
) -> Result<Pubkey, anyhow::Error> {
match dex_type {
DexType::PumpSwap => crate::instruction::utils::pumpswap::find_pool(rpc, mint).await,
_ => Err(anyhow::anyhow!("find_pool_by_mint not implemented for {:?}", dex_type)),
}
}
/// Type of the token to buy /// Type of the token to buy
#[derive(Clone, PartialEq)] #[derive(Clone, PartialEq)]
pub enum TradeTokenType { pub enum TradeTokenType {
@@ -70,10 +86,14 @@ pub enum TradeTokenType {
pub struct TradingInfrastructure { pub struct TradingInfrastructure {
/// Shared RPC client for blockchain interactions /// Shared RPC client for blockchain interactions
pub rpc: Arc<SolanaRpcClient>, pub rpc: Arc<SolanaRpcClient>,
/// Shared SWQOS clients for transaction priority and routing /// Shared SWQOS clients for transaction priority and routing. Arc<Vec<..>> so cloning into SwapParams is a single Arc clone.
pub swqos_clients: Vec<Arc<SwqosClient>>, pub swqos_clients: Arc<Vec<Arc<SwqosClient>>>,
/// Configuration used to create this infrastructure /// Configuration used to create this infrastructure
pub config: InfrastructureConfig, pub config: InfrastructureConfig,
/// Precomputed at init: min(swqos_clients.len(), 2/3 * num_cores). Not computed on trade hot path.
pub max_sender_concurrency: usize,
/// Precomputed at init: first max_sender_concurrency CoreIds for job affinity. Empty if no cores. Not computed on trade hot path.
pub effective_core_ids: Arc<Vec<core_affinity::CoreId>>,
} }
impl TradingInfrastructure { impl TradingInfrastructure {
@@ -97,27 +117,47 @@ impl TradingInfrastructure {
config.commitment.clone(), config.commitment.clone(),
)); ));
// Initialize rent cache and start background updater // Initialize rent cache (with timeout so slow RPC doesn't block forever)
common::seed::update_rents(&rpc).await.unwrap(); const RENT_UPDATE_TIMEOUT: std::time::Duration = std::time::Duration::from_secs(15);
match tokio::time::timeout(RENT_UPDATE_TIMEOUT, common::seed::update_rents(&rpc)).await {
Ok(Ok(())) => {}
Ok(Err(e)) => {
if sdk_log::sdk_log_enabled() {
warn!(target: "sol_trade_sdk", "rent update failed: {}, using defaults", e);
}
common::seed::set_default_rents();
}
Err(_) => {
if sdk_log::sdk_log_enabled() {
warn!(target: "sol_trade_sdk", "rent update timed out ({}s), using defaults; check RPC", RENT_UPDATE_TIMEOUT.as_secs());
}
common::seed::set_default_rents();
}
}
common::seed::start_rent_updater(rpc.clone()); common::seed::start_rent_updater(rpc.clone());
// Create SWQOS clients with blacklist checking // Create SWQOS clients with blacklist checking(单节点超时 5s,避免某一家卡死整段初始化)
const SWQOS_CLIENT_TIMEOUT: std::time::Duration = std::time::Duration::from_secs(5);
let mut swqos_clients: Vec<Arc<SwqosClient>> = vec![]; let mut swqos_clients: Vec<Arc<SwqosClient>> = vec![];
for swqos in &config.swqos_configs { for swqos in &config.swqos_configs {
// Check blacklist, skip disabled providers
if swqos.is_blacklisted() { if swqos.is_blacklisted() {
if sdk_log::sdk_log_enabled() { if sdk_log::sdk_log_enabled() {
warn!(target: "sol_trade_sdk", "⚠️ SWQOS {:?} is blacklisted, skipping", swqos.swqos_type()); warn!(target: "sol_trade_sdk", "⚠️ SWQOS {:?} is blacklisted, skipping", swqos.swqos_type());
} }
continue; continue;
} }
match SwqosConfig::get_swqos_client( match tokio::time::timeout(
config.rpc_url.clone(), SWQOS_CLIENT_TIMEOUT,
config.commitment.clone(), SwqosConfig::get_swqos_client(
swqos.clone(), config.rpc_url.clone(),
).await { config.commitment.clone(),
Ok(swqos_client) => swqos_clients.push(swqos_client), swqos.clone(),
Err(err) => { ),
)
.await
{
Ok(Ok(swqos_client)) => swqos_clients.push(swqos_client),
Ok(Err(err)) => {
if sdk_log::sdk_log_enabled() { if sdk_log::sdk_log_enabled() {
warn!( warn!(
target: "sol_trade_sdk", target: "sol_trade_sdk",
@@ -126,13 +166,36 @@ impl TradingInfrastructure {
); );
} }
} }
Err(_) => {
if sdk_log::sdk_log_enabled() {
warn!(
target: "sol_trade_sdk",
"swqos {:?} init timed out ({}s), skipping",
swqos.swqos_type(),
SWQOS_CLIENT_TIMEOUT.as_secs()
);
}
}
} }
} }
let swqos_count = swqos_clients.len();
let (max_sender_concurrency, effective_core_ids) = {
let num_cores = core_affinity::get_core_ids().map(|c| c.len()).unwrap_or(0);
let max_by_cores = (num_cores * 2 / 3).max(1);
let cap = swqos_count.min(max_by_cores).max(1);
let ids = core_affinity::get_core_ids()
.map(|all| all.into_iter().take(cap).collect::<Vec<_>>())
.unwrap_or_default();
(cap, Arc::new(ids))
};
Self { Self {
rpc, rpc,
swqos_clients, swqos_clients: Arc::new(swqos_clients),
config, config,
max_sender_concurrency,
effective_core_ids,
} }
} }
} }
@@ -153,8 +216,14 @@ pub struct TradingClient {
/// Whether to use seed optimization for all ATA operations (default: true) /// Whether to use seed optimization for all ATA operations (default: true)
/// Applies to all token account creations across buy and sell operations /// Applies to all token account creations across buy and sell operations
pub use_seed_optimize: bool, pub use_seed_optimize: bool,
/// Whether to pin parallel submit tasks to CPU cores (from TradeConfig.use_core_affinity). Default true. /// Internal: use dedicated sender threads (default false). Set via with_dedicated_sender_threads() for advanced use.
pub use_core_affinity: bool, pub use_dedicated_sender_threads: bool,
/// Internal: core indices for dedicated sender threads. Trimmed to ≤ max_sender_concurrency at set.
pub sender_thread_cores: Option<Arc<Vec<usize>>>,
/// Internal: precomputed at infra init (min(swqos_count, 2/3*cores)). Not user-configurable.
pub max_sender_concurrency: usize,
/// Internal: precomputed at infra init for job affinity. Not user-configurable.
pub effective_core_ids: Arc<Vec<core_affinity::CoreId>>,
/// Whether to output all SDK logs (from TradeConfig.log_enabled). /// Whether to output all SDK logs (from TradeConfig.log_enabled).
pub log_enabled: bool, pub log_enabled: bool,
/// Whether to check minimum tip per SWQOS (from TradeConfig.check_min_tip). Default false for lower latency. /// Whether to check minimum tip per SWQOS (from TradeConfig.check_min_tip). Default false for lower latency.
@@ -173,7 +242,10 @@ impl Clone for TradingClient {
infrastructure: self.infrastructure.clone(), infrastructure: self.infrastructure.clone(),
middleware_manager: self.middleware_manager.clone(), middleware_manager: self.middleware_manager.clone(),
use_seed_optimize: self.use_seed_optimize, use_seed_optimize: self.use_seed_optimize,
use_core_affinity: self.use_core_affinity, use_dedicated_sender_threads: self.use_dedicated_sender_threads,
sender_thread_cores: self.sender_thread_cores.clone(),
max_sender_concurrency: self.max_sender_concurrency,
effective_core_ids: self.effective_core_ids.clone(),
log_enabled: self.log_enabled, log_enabled: self.log_enabled,
check_min_tip: self.check_min_tip, check_min_tip: self.check_min_tip,
} }
@@ -296,13 +368,18 @@ impl TradingClient {
) -> Self { ) -> Self {
// Initialize wallet-specific caches (fast, synchronous) // Initialize wallet-specific caches (fast, synchronous)
crate::common::fast_fn::fast_init(&payer.pubkey()); crate::common::fast_fn::fast_init(&payer.pubkey());
let max_sender_concurrency = infrastructure.max_sender_concurrency;
let effective_core_ids = infrastructure.effective_core_ids.clone();
Self { Self {
payer, payer,
infrastructure, infrastructure,
middleware_manager: None, middleware_manager: None,
use_seed_optimize, use_seed_optimize,
use_core_affinity: true, use_dedicated_sender_threads: false,
sender_thread_cores: None,
max_sender_concurrency,
effective_core_ids,
log_enabled: true, log_enabled: true,
check_min_tip: false, check_min_tip: false,
} }
@@ -337,139 +414,133 @@ impl TradingClient {
} }
} }
let max_sender_concurrency = infrastructure.max_sender_concurrency;
let effective_core_ids = infrastructure.effective_core_ids.clone();
Self { Self {
payer, payer,
infrastructure, infrastructure,
middleware_manager: None, middleware_manager: None,
use_seed_optimize, use_seed_optimize,
use_core_affinity: true, use_dedicated_sender_threads: false,
sender_thread_cores: None,
max_sender_concurrency,
effective_core_ids,
log_enabled: true, log_enabled: true,
check_min_tip: false, check_min_tip: false,
} }
} }
/// Helper to ensure WSOL ATA exists for a wallet /// 单次尝试创建 WSOL ATA:获取 blockhash、组交易、发送并确认。成功或账户已存在返回 Ok(()),否则返回 Err(错误信息)。
async fn try_create_wsol_ata_once(
rpc: &SolanaRpcClient,
payer: &Arc<Keypair>,
wsol_ata: &solana_sdk::pubkey::Pubkey,
create_ata_ixs: &[solana_sdk::instruction::Instruction],
timeout_secs: u64,
) -> Result<(), String> {
use solana_sdk::transaction::Transaction;
let recent_blockhash = rpc
.get_latest_blockhash()
.await
.map_err(|e| format!("Failed to get blockhash: {}", e))?;
let tx = Transaction::new_signed_with_payer(
create_ata_ixs,
Some(&payer.pubkey()),
&[payer.as_ref()],
recent_blockhash,
);
let send_result = tokio::time::timeout(
tokio::time::Duration::from_secs(timeout_secs),
rpc.send_and_confirm_transaction(&tx),
)
.await;
match send_result {
Ok(Ok(_signature)) => Ok(()),
Ok(Err(e)) => {
if rpc.get_account(wsol_ata).await.is_ok() {
return Ok(());
}
Err(format!("{}", e))
}
Err(_) => Err(format!("Transaction confirmation timeout ({}s)", timeout_secs)),
}
}
/// 确保钱包存在 WSOL ATA;不存在则发交易创建(会花费租金 + 手续费,初始化阶段唯一会扣钱的逻辑)
async fn ensure_wsol_ata(payer: &Arc<Keypair>, rpc: &Arc<SolanaRpcClient>) { async fn ensure_wsol_ata(payer: &Arc<Keypair>, rpc: &Arc<SolanaRpcClient>) {
let wsol_ata = const MAX_RETRIES: usize = 3;
crate::common::fast_fn::get_associated_token_address_with_program_id_fast( const TIMEOUT_SECS: u64 = 10;
&payer.pubkey(),
&WSOL_TOKEN_ACCOUNT,
&crate::constants::TOKEN_PROGRAM,
);
match rpc.get_account(&wsol_ata).await { let wsol_ata = crate::common::fast_fn::get_associated_token_address_with_program_id_fast(
Ok(_) => { &payer.pubkey(),
if sdk_log::sdk_log_enabled() { &WSOL_TOKEN_ACCOUNT,
info!(target: "sol_trade_sdk", "✅ WSOL ATA already exists: {}", wsol_ata); &crate::constants::TOKEN_PROGRAM,
} );
return;
if rpc.get_account(&wsol_ata).await.is_ok() {
if sdk_log::sdk_log_enabled() {
info!(target: "sol_trade_sdk", "✅ WSOL ATA already exists: {}", wsol_ata);
} }
Err(_) => { return;
}
let create_ata_ixs = crate::trading::common::wsol_manager::create_wsol_ata(&payer.pubkey());
if create_ata_ixs.is_empty() {
if sdk_log::sdk_log_enabled() {
info!(target: "sol_trade_sdk", "️ WSOL ATA already exists (no need to create)");
}
return;
}
if sdk_log::sdk_log_enabled() {
info!(target: "sol_trade_sdk", "🔨 Creating WSOL ATA: {}", wsol_ata);
}
let mut last_error = None;
for attempt in 1..=MAX_RETRIES {
if attempt > 1 {
if sdk_log::sdk_log_enabled() { if sdk_log::sdk_log_enabled() {
info!(target: "sol_trade_sdk", "🔨 Creating WSOL ATA: {}", wsol_ata); info!(target: "sol_trade_sdk", "🔄 Retrying WSOL ATA creation (attempt {}/{})...", attempt, MAX_RETRIES);
} }
let create_ata_ixs = tokio::time::sleep(tokio::time::Duration::from_secs(2)).await;
crate::trading::common::wsol_manager::create_wsol_ata(&payer.pubkey()); }
match Self::try_create_wsol_ata_once(
if !create_ata_ixs.is_empty() { rpc.as_ref(),
use solana_sdk::transaction::Transaction; payer,
&wsol_ata,
// 重试逻辑:最多尝试3次,每次超时10秒 &create_ata_ixs,
const MAX_RETRIES: usize = 3; TIMEOUT_SECS,
const TIMEOUT_SECS: u64 = 10; )
let mut last_error = None; .await
{
for attempt in 1..=MAX_RETRIES { Ok(()) => {
if attempt > 1 {
if sdk_log::sdk_log_enabled() {
info!(target: "sol_trade_sdk", "🔄 Retrying WSOL ATA creation (attempt {}/{})...", attempt, MAX_RETRIES);
}
tokio::time::sleep(tokio::time::Duration::from_secs(2)).await;
}
let recent_blockhash = match rpc.get_latest_blockhash().await {
Ok(hash) => hash,
Err(e) => {
if sdk_log::sdk_log_enabled() {
warn!(target: "sol_trade_sdk", "⚠️ Failed to get latest blockhash: {}", e);
}
last_error = Some(format!("Failed to get blockhash: {}", e));
continue;
}
};
let tx = Transaction::new_signed_with_payer(
&create_ata_ixs,
Some(&payer.pubkey()),
&[payer.as_ref()],
recent_blockhash,
);
// 使用超时包装 send_and_confirm_transaction
let send_result = tokio::time::timeout(
tokio::time::Duration::from_secs(TIMEOUT_SECS),
rpc.send_and_confirm_transaction(&tx)
).await;
match send_result {
Ok(Ok(signature)) => {
if sdk_log::sdk_log_enabled() {
info!(target: "sol_trade_sdk", "✅ WSOL ATA created successfully: {}", signature);
}
return;
}
Ok(Err(e)) => {
last_error = Some(format!("{}", e));
// 检查账户是否实际已存在
if let Ok(_) = rpc.get_account(&wsol_ata).await {
if sdk_log::sdk_log_enabled() {
info!(target: "sol_trade_sdk", "✅ WSOL ATA already exists (tx failed but account exists): {}", wsol_ata);
}
return;
}
if attempt < MAX_RETRIES && sdk_log::sdk_log_enabled() {
warn!(target: "sol_trade_sdk", "⚠️ Attempt {} failed: {}", attempt, e);
}
}
Err(_) => {
last_error = Some(format!("Transaction confirmation timeout ({}s)", TIMEOUT_SECS));
if sdk_log::sdk_log_enabled() {
warn!(target: "sol_trade_sdk", "⚠️ Attempt {} timed out", attempt);
}
}
}
}
// 所有重试都失败了
if let Some(err) = last_error {
if sdk_log::sdk_log_enabled() {
error!(target: "sol_trade_sdk", "❌ WSOL ATA creation failed after {} retries: {}", MAX_RETRIES, wsol_ata);
error!(target: "sol_trade_sdk", " Error: {}", err);
error!(target: "sol_trade_sdk", " 💡 Possible causes:");
error!(target: "sol_trade_sdk", " 1. Insufficient SOL balance (need ~0.002 SOL for rent exemption)");
error!(target: "sol_trade_sdk", " 2. RPC timeout or network congestion");
error!(target: "sol_trade_sdk", " 3. Insufficient transaction fee");
error!(target: "sol_trade_sdk", " 🔧 Solutions:");
error!(target: "sol_trade_sdk", " 1. Fund wallet with at least 0.1 SOL");
error!(target: "sol_trade_sdk", " 2. Retry after a few seconds");
error!(target: "sol_trade_sdk", " 3. Check RPC connection");
error!(target: "sol_trade_sdk", " ⚠️ Process will exit in 5 seconds, restart after fixing the above");
}
std::thread::sleep(std::time::Duration::from_secs(5));
panic!(
"❌ WSOL ATA creation failed and account does not exist: {}. Error: {}",
wsol_ata, err
);
}
} else {
if sdk_log::sdk_log_enabled() { if sdk_log::sdk_log_enabled() {
info!(target: "sol_trade_sdk", " WSOL ATA already exists (no need to create)"); info!(target: "sol_trade_sdk", " WSOL ATA created or already exists");
}
return;
}
Err(e) => {
last_error = Some(e.clone());
if attempt < MAX_RETRIES && sdk_log::sdk_log_enabled() {
warn!(target: "sol_trade_sdk", "⚠️ Attempt {} failed: {}", attempt, e);
} }
} }
} }
} }
if let Some(err) = last_error {
if sdk_log::sdk_log_enabled() {
error!(target: "sol_trade_sdk", "❌ WSOL ATA creation failed after {} retries: {}", MAX_RETRIES, wsol_ata);
error!(target: "sol_trade_sdk", " Error: {}", err);
error!(target: "sol_trade_sdk", " 💡 Possible causes: insufficient SOL, RPC timeout, or fee");
error!(target: "sol_trade_sdk", " 🔧 Solutions: fund wallet (e.g. 0.1 SOL), retry, check RPC");
}
std::thread::sleep(std::time::Duration::from_secs(5));
panic!(
"❌ WSOL ATA creation failed and account does not exist: {}. Error: {}",
wsol_ata, err
);
}
} }
/// Creates a new SolTradingSDK instance with the specified configuration /// Creates a new SolTradingSDK instance with the specified configuration
@@ -496,17 +567,44 @@ impl TradingClient {
// Initialize wallet-specific caches // Initialize wallet-specific caches
crate::common::fast_fn::fast_init(&payer.pubkey()); crate::common::fast_fn::fast_init(&payer.pubkey());
// Handle WSOL ATA creation if configured // ═══════════════════════════════════════════════════════════════════════════════
// 初始化阶段会花费租金/手续费的唯一路径:创建 WSOL ATAensure_wsol_ata
// - 触发条件:create_wsol_ata_on_startup == true 且钱包 SOL >= MIN_SOL_FOR_WSOL_ATA_LAMPORTS
// - 花费:ATA 租金(约 0.00203928 SOL)+ 交易手续费;钱包不足时已跳过
// - 其它初始化(TradingInfrastructure::new、update_rents、get_swqos_client)仅 RPC/HTTP,不发送交易
// ═══════════════════════════════════════════════════════════════════════════════
if trade_config.create_wsol_ata_on_startup { if trade_config.create_wsol_ata_on_startup {
Self::ensure_wsol_ata(&payer, &infrastructure.rpc).await; const MIN_SOL_FOR_WSOL_ATA_LAMPORTS: u64 = 500_000; // 约 0.0005 SOL,用于 ATA 租金 + 手续费
const BALANCE_CHECK_TIMEOUT: std::time::Duration = std::time::Duration::from_secs(5);
let balance = tokio::time::timeout(
BALANCE_CHECK_TIMEOUT,
infrastructure.rpc.get_balance(&payer.pubkey()),
)
.await
.unwrap_or(Ok(0))
.unwrap_or(0);
if balance >= MIN_SOL_FOR_WSOL_ATA_LAMPORTS {
Self::ensure_wsol_ata(&payer, &infrastructure.rpc).await;
} else if sdk_log::sdk_log_enabled() {
info!(
target: "sol_trade_sdk",
"⏭️ 跳过创建 WSOL ATA:钱包 SOL 不足(当前 {} lamports,需要至少 {}",
balance,
MIN_SOL_FOR_WSOL_ATA_LAMPORTS
);
}
} }
// 并发/核心相关由 infrastructure 预计算,用户无需配置
let instance = Self { let instance = Self {
payer, payer,
infrastructure, infrastructure: infrastructure.clone(),
middleware_manager: None, middleware_manager: None,
use_seed_optimize: trade_config.use_seed_optimize, use_seed_optimize: trade_config.use_seed_optimize,
use_core_affinity: trade_config.use_core_affinity, use_dedicated_sender_threads: false,
sender_thread_cores: None,
max_sender_concurrency: infrastructure.max_sender_concurrency,
effective_core_ids: infrastructure.effective_core_ids.clone(),
log_enabled: trade_config.log_enabled, log_enabled: trade_config.log_enabled,
check_min_tip: trade_config.check_min_tip, check_min_tip: trade_config.check_min_tip,
}; };
@@ -532,6 +630,31 @@ impl TradingClient {
self self
} }
/// **Advanced.** Use dedicated OS threads for sender pool (and optionally pin to cores).
/// By default the SDK uses a shared tokio pool; this can reduce scheduling contention when sending many txs.
/// Concurrency and core count are capped internally (≤ swqos count, ≤ 2/3 of CPU cores).
/// - `None`: keep default (shared tokio pool).
/// - `Some(vec![])`: dedicated threads with default count, no core pinning.
/// - `Some(indices)`: dedicated threads pinned to those core indices (trimmed to cap).
pub fn with_dedicated_sender_threads(mut self, core_indices: Option<Vec<usize>>) -> Self {
match core_indices {
None => {
self.use_dedicated_sender_threads = false;
self.sender_thread_cores = None;
}
Some(v) if v.is_empty() => {
self.use_dedicated_sender_threads = true;
self.sender_thread_cores = None;
}
Some(v) => {
self.use_dedicated_sender_threads = true;
let cap = v.len().min(self.max_sender_concurrency);
self.sender_thread_cores = Some(Arc::new(if cap < v.len() { v[..cap].to_vec() } else { v }));
}
}
self
}
/// Gets the RPC client instance for direct Solana blockchain interactions /// Gets the RPC client instance for direct Solana blockchain interactions
/// ///
/// This provides access to the underlying Solana RPC client that can be used /// This provides access to the underlying Solana RPC client that can be used
@@ -590,6 +713,11 @@ impl TradingClient {
&self, &self,
params: TradeBuyParams, params: TradeBuyParams,
) -> Result<(bool, Vec<Signature>, Option<TradeError>), anyhow::Error> { ) -> Result<(bool, Vec<Signature>, Option<TradeError>), anyhow::Error> {
if params.recent_blockhash.is_none() && params.durable_nonce.is_none() {
return Err(anyhow::anyhow!(
"Must provide either recent_blockhash or durable_nonce for buy (required for transaction validity)"
));
}
#[cfg(feature = "perf-trace")] #[cfg(feature = "perf-trace")]
if sdk_log::sdk_log_enabled() && params.slippage_basis_points.is_none() { if sdk_log::sdk_log_enabled() && params.slippage_basis_points.is_none() {
debug!( debug!(
@@ -600,7 +728,14 @@ impl TradingClient {
} }
if params.input_token_type == TradeTokenType::USD1 && params.dex_type != DexType::Bonk { if params.input_token_type == TradeTokenType::USD1 && params.dex_type != DexType::Bonk {
return Err(anyhow::anyhow!( return Err(anyhow::anyhow!(
" Current version only support USD1 trading on Bonk protocols" " Current version only supports USD1 trading on Bonk protocols"
));
}
let protocol_params = params.extension_params;
if !validate_protocol_params(params.dex_type, &protocol_params) {
return Err(anyhow::anyhow!(
"Invalid protocol params for Trade (dex={:?})",
params.dex_type
)); ));
} }
let input_token_mint = if params.input_token_type == TradeTokenType::SOL { let input_token_mint = if params.input_token_type == TradeTokenType::SOL {
@@ -612,8 +747,7 @@ impl TradingClient {
} else { } else {
USD1_TOKEN_ACCOUNT USD1_TOKEN_ACCOUNT
}; };
let executor = TradeFactory::create_executor(params.dex_type.clone()); let executor = TradeFactory::create_executor(params.dex_type);
let protocol_params = params.extension_params;
let buy_params = SwapParams { let buy_params = SwapParams {
rpc: Some(self.infrastructure.rpc.clone()), rpc: Some(self.infrastructure.rpc.clone()),
payer: self.payer.clone(), payer: self.payer.clone(),
@@ -627,7 +761,7 @@ impl TradingClient {
address_lookup_table_account: params.address_lookup_table_account, address_lookup_table_account: params.address_lookup_table_account,
recent_blockhash: params.recent_blockhash, recent_blockhash: params.recent_blockhash,
wait_transaction_confirmed: params.wait_transaction_confirmed, wait_transaction_confirmed: params.wait_transaction_confirmed,
protocol_params: protocol_params.clone(), protocol_params,
open_seed_optimize: self.use_seed_optimize, // 使用全局seed优化配置 open_seed_optimize: self.use_seed_optimize, // 使用全局seed优化配置
swqos_clients: self.infrastructure.swqos_clients.clone(), swqos_clients: self.infrastructure.swqos_clients.clone(),
middleware_manager: self.middleware_manager.clone(), middleware_manager: self.middleware_manager.clone(),
@@ -641,16 +775,15 @@ impl TradingClient {
gas_fee_strategy: params.gas_fee_strategy, gas_fee_strategy: params.gas_fee_strategy,
simulate: params.simulate, simulate: params.simulate,
log_enabled: self.log_enabled, log_enabled: self.log_enabled,
use_core_affinity: self.use_core_affinity, use_dedicated_sender_threads: self.use_dedicated_sender_threads,
sender_thread_cores: self.sender_thread_cores.clone(),
max_sender_concurrency: self.max_sender_concurrency,
effective_core_ids: self.effective_core_ids.clone(),
check_min_tip: self.check_min_tip, check_min_tip: self.check_min_tip,
grpc_recv_us: params.grpc_recv_us, grpc_recv_us: params.grpc_recv_us,
use_exact_sol_amount: params.use_exact_sol_amount, use_exact_sol_amount: params.use_exact_sol_amount,
}; };
if !validate_protocol_params(params.dex_type, &protocol_params) {
return Err(anyhow::anyhow!("Invalid protocol params for Trade"));
}
let swap_result = executor.swap(buy_params).await; let swap_result = executor.swap(buy_params).await;
let result = let result =
swap_result.map(|(success, sigs, err)| (success, sigs, err.map(TradeError::from))); swap_result.map(|(success, sigs, err)| (success, sigs, err.map(TradeError::from)));
@@ -695,13 +828,24 @@ impl TradingClient {
DEFAULT_SLIPPAGE DEFAULT_SLIPPAGE
); );
} }
if params.output_token_type == TradeTokenType::USD1 && params.dex_type != DexType::Bonk { if params.recent_blockhash.is_none() && params.durable_nonce.is_none() {
return Err(anyhow::anyhow!( return Err(anyhow::anyhow!(
" Current version only support USD1 trading on Bonk protocols" "Must provide either recent_blockhash or durable_nonce for sell (required for transaction validity)"
));
}
if params.output_token_type == TradeTokenType::USD1 && params.dex_type != DexType::Bonk {
return Err(anyhow::anyhow!(
" Current version only supports USD1 trading on Bonk protocols"
)); ));
} }
let executor = TradeFactory::create_executor(params.dex_type.clone());
let protocol_params = params.extension_params; let protocol_params = params.extension_params;
if !validate_protocol_params(params.dex_type, &protocol_params) {
return Err(anyhow::anyhow!(
"Invalid protocol params for Trade (dex={:?})",
params.dex_type
));
}
let executor = TradeFactory::create_executor(params.dex_type);
let output_token_mint = if params.output_token_type == TradeTokenType::SOL { let output_token_mint = if params.output_token_type == TradeTokenType::SOL {
SOL_TOKEN_ACCOUNT SOL_TOKEN_ACCOUNT
} else if params.output_token_type == TradeTokenType::WSOL { } else if params.output_token_type == TradeTokenType::WSOL {
@@ -724,7 +868,7 @@ impl TradingClient {
address_lookup_table_account: params.address_lookup_table_account, address_lookup_table_account: params.address_lookup_table_account,
recent_blockhash: params.recent_blockhash, recent_blockhash: params.recent_blockhash,
wait_transaction_confirmed: params.wait_transaction_confirmed, wait_transaction_confirmed: params.wait_transaction_confirmed,
protocol_params: protocol_params.clone(), protocol_params,
with_tip: params.with_tip, with_tip: params.with_tip,
open_seed_optimize: self.use_seed_optimize, // 使用全局seed优化配置 open_seed_optimize: self.use_seed_optimize, // 使用全局seed优化配置
swqos_clients: self.infrastructure.swqos_clients.clone(), swqos_clients: self.infrastructure.swqos_clients.clone(),
@@ -738,16 +882,15 @@ impl TradingClient {
gas_fee_strategy: params.gas_fee_strategy, gas_fee_strategy: params.gas_fee_strategy,
simulate: params.simulate, simulate: params.simulate,
log_enabled: self.log_enabled, log_enabled: self.log_enabled,
use_core_affinity: self.use_core_affinity, use_dedicated_sender_threads: self.use_dedicated_sender_threads,
sender_thread_cores: self.sender_thread_cores.clone(),
max_sender_concurrency: self.max_sender_concurrency,
effective_core_ids: self.effective_core_ids.clone(),
check_min_tip: self.check_min_tip, check_min_tip: self.check_min_tip,
grpc_recv_us: params.grpc_recv_us, grpc_recv_us: params.grpc_recv_us,
use_exact_sol_amount: None, use_exact_sol_amount: None,
}; };
if !validate_protocol_params(params.dex_type, &protocol_params) {
return Err(anyhow::anyhow!("Invalid protocol params for Trade"));
}
let swap_result = executor.swap(sell_params).await; let swap_result = executor.swap(sell_params).await;
let result = let result =
swap_result.map(|(success, sigs, err)| (success, sigs, err.map(TradeError::from))); swap_result.map(|(success, sigs, err)| (success, sigs, err.map(TradeError::from)));
@@ -914,8 +1057,10 @@ impl TradingClient {
/// - 交易执行或确认失败 /// - 交易执行或确认失败
/// - 网络或 RPC 错误 /// - 网络或 RPC 错误
pub async fn wrap_wsol_to_sol(&self, amount: u64) -> Result<String, anyhow::Error> { pub async fn wrap_wsol_to_sol(&self, amount: u64) -> Result<String, anyhow::Error> {
use crate::trading::common::wsol_manager::{wrap_wsol_to_sol as wrap_wsol_to_sol_internal, wrap_wsol_to_sol_without_create};
use crate::common::seed::get_associated_token_address_with_program_id_use_seed; use crate::common::seed::get_associated_token_address_with_program_id_use_seed;
use crate::trading::common::wsol_manager::{
wrap_wsol_to_sol as wrap_wsol_to_sol_internal, wrap_wsol_to_sol_without_create,
};
use solana_sdk::transaction::Transaction; use solana_sdk::transaction::Transaction;
// 检查临时seed账户是否已存在 // 检查临时seed账户是否已存在
@@ -936,7 +1081,8 @@ impl TradingClient {
}; };
let recent_blockhash = self.infrastructure.rpc.get_latest_blockhash().await?; let recent_blockhash = self.infrastructure.rpc.get_latest_blockhash().await?;
let mut transaction = Transaction::new_with_payer(&instructions, Some(&self.payer.pubkey())); let mut transaction =
Transaction::new_with_payer(&instructions, Some(&self.payer.pubkey()));
transaction.sign(&[&*self.payer], recent_blockhash); transaction.sign(&[&*self.payer], recent_blockhash);
let signature = self.infrastructure.rpc.send_and_confirm_transaction(&transaction).await?; let signature = self.infrastructure.rpc.send_and_confirm_transaction(&transaction).await?;
Ok(signature.to_string()) Ok(signature.to_string())
@@ -952,8 +1098,10 @@ impl TradingClient {
/// * `Err(anyhow::Error)` - Build or send failure (e.g. invalid PDA) /// * `Err(anyhow::Error)` - Build or send failure (e.g. invalid PDA)
pub async fn claim_cashback_pumpfun(&self) -> Result<String, anyhow::Error> { pub async fn claim_cashback_pumpfun(&self) -> Result<String, anyhow::Error> {
use solana_sdk::transaction::Transaction; use solana_sdk::transaction::Transaction;
let ix = crate::instruction::pumpfun::claim_cashback_pumpfun_instruction(&self.payer.pubkey()) let ix = crate::instruction::pumpfun::claim_cashback_pumpfun_instruction(
.ok_or_else(|| anyhow::anyhow!("Failed to build PumpFun claim_cashback instruction"))?; &self.payer.pubkey(),
)
.ok_or_else(|| anyhow::anyhow!("Failed to build PumpFun claim_cashback instruction"))?;
let recent_blockhash = self.infrastructure.rpc.get_latest_blockhash().await?; let recent_blockhash = self.infrastructure.rpc.get_latest_blockhash().await?;
let mut transaction = Transaction::new_with_payer(&[ix], Some(&self.payer.pubkey())); let mut transaction = Transaction::new_with_payer(&[ix], Some(&self.payer.pubkey()));
transaction.sign(&[&*self.payer], recent_blockhash); transaction.sign(&[&*self.payer], recent_blockhash);
@@ -971,21 +1119,24 @@ impl TradingClient {
/// * `Err(anyhow::Error)` - Build or send failure /// * `Err(anyhow::Error)` - Build or send failure
pub async fn claim_cashback_pumpswap(&self) -> Result<String, anyhow::Error> { pub async fn claim_cashback_pumpswap(&self) -> Result<String, anyhow::Error> {
use solana_sdk::transaction::Transaction; use solana_sdk::transaction::Transaction;
let mut instructions = crate::common::fast_fn::create_associated_token_account_idempotent_fast_use_seed( let mut instructions =
&self.payer.pubkey(), crate::common::fast_fn::create_associated_token_account_idempotent_fast_use_seed(
&self.payer.pubkey(), &self.payer.pubkey(),
&WSOL_TOKEN_ACCOUNT, &self.payer.pubkey(),
&crate::constants::TOKEN_PROGRAM, &WSOL_TOKEN_ACCOUNT,
self.use_seed_optimize, &crate::constants::TOKEN_PROGRAM,
); self.use_seed_optimize,
);
let ix = crate::instruction::pumpswap::claim_cashback_pumpswap_instruction( let ix = crate::instruction::pumpswap::claim_cashback_pumpswap_instruction(
&self.payer.pubkey(), &self.payer.pubkey(),
WSOL_TOKEN_ACCOUNT, WSOL_TOKEN_ACCOUNT,
crate::constants::TOKEN_PROGRAM, crate::constants::TOKEN_PROGRAM,
).ok_or_else(|| anyhow::anyhow!("Failed to build PumpSwap claim_cashback instruction"))?; )
.ok_or_else(|| anyhow::anyhow!("Failed to build PumpSwap claim_cashback instruction"))?;
instructions.push(ix); instructions.push(ix);
let recent_blockhash = self.infrastructure.rpc.get_latest_blockhash().await?; let recent_blockhash = self.infrastructure.rpc.get_latest_blockhash().await?;
let mut transaction = Transaction::new_with_payer(&instructions, Some(&self.payer.pubkey())); let mut transaction =
Transaction::new_with_payer(&instructions, Some(&self.payer.pubkey()));
transaction.sign(&[&*self.payer], recent_blockhash); transaction.sign(&[&*self.payer], recent_blockhash);
let signature = self.infrastructure.rpc.send_and_confirm_transaction(&transaction).await?; let signature = self.infrastructure.rpc.send_and_confirm_transaction(&transaction).await?;
Ok(signature.to_string()) Ok(signature.to_string())
+90 -75
View File
@@ -1,5 +1,5 @@
//! 🚀 编译器级性能优化 - 极致编译时优化 //! 🚀 编译器级性能优化 - 极致编译时优化
//! //!
//! 实现编译时的极致性能优化,包括: //! 实现编译时的极致性能优化,包括:
//! - 编译器标志优化配置 //! - 编译器标志优化配置
//! - 编译时代码生成 //! - 编译时代码生成
@@ -137,100 +137,110 @@ impl CompilerOptimizer {
stats: CompilerOptimizationStats::default(), stats: CompilerOptimizationStats::default(),
} }
} }
/// 🚀 生成超高性能编译配置 /// 🚀 生成超高性能编译配置
pub fn generate_ultra_performance_config(&self) -> Result<CompilerConfig> { pub fn generate_ultra_performance_config(&self) -> Result<CompilerConfig> {
tracing::info!(target: "sol_trade_sdk","🚀 Generating ultra-performance compiler configuration..."); tracing::info!(target: "sol_trade_sdk","🚀 Generating ultra-performance compiler configuration...");
let mut rustflags = Vec::new(); let mut rustflags = Vec::new();
// 基础优化标志 // 基础优化标志
rustflags.push("-C".to_string()); rustflags.push("-C".to_string());
rustflags.push("opt-level=3".to_string()); // 最高优化级别 rustflags.push("opt-level=3".to_string()); // 最高优化级别
// 链接时优化 // 链接时优化
if self.optimization_flags.enable_lto { if self.optimization_flags.enable_lto {
rustflags.push("-C".to_string()); rustflags.push("-C".to_string());
rustflags.push("lto=fat".to_string()); // 胖LTO获得最佳优化 rustflags.push("lto=fat".to_string()); // 胖LTO获得最佳优化
} }
// 目标CPU优化 // 目标CPU优化
if !self.optimization_flags.target_cpu.is_empty() { if !self.optimization_flags.target_cpu.is_empty() {
rustflags.push("-C".to_string()); rustflags.push("-C".to_string());
rustflags.push(format!("target-cpu={}", self.optimization_flags.target_cpu)); rustflags.push(format!("target-cpu={}", self.optimization_flags.target_cpu));
} }
// 目标特性 // 目标特性
if !self.optimization_flags.target_features.is_empty() { if !self.optimization_flags.target_features.is_empty() {
rustflags.push("-C".to_string()); rustflags.push("-C".to_string());
rustflags.push(format!("target-feature={}", self.optimization_flags.target_features.join(","))); rustflags.push(format!(
"target-feature={}",
self.optimization_flags.target_features.join(",")
));
} }
// 代码模型 // 代码模型
rustflags.push("-C".to_string()); rustflags.push("-C".to_string());
rustflags.push(format!("code-model={:?}", self.optimization_flags.code_model).to_lowercase()); rustflags
.push(format!("code-model={:?}", self.optimization_flags.code_model).to_lowercase());
// 恐慌处理 // 恐慌处理
if self.codegen_config.panic_abort { if self.codegen_config.panic_abort {
rustflags.push("-C".to_string()); rustflags.push("-C".to_string());
rustflags.push("panic=abort".to_string()); rustflags.push("panic=abort".to_string());
} }
// 溢出检查 // 溢出检查
if !self.codegen_config.overflow_checks { if !self.codegen_config.overflow_checks {
rustflags.push("-C".to_string()); rustflags.push("-C".to_string());
rustflags.push("overflow-checks=no".to_string()); rustflags.push("overflow-checks=no".to_string());
} }
// 代码生成单元 // 代码生成单元
if let Some(units) = self.optimization_flags.codegen_units { if let Some(units) = self.optimization_flags.codegen_units {
rustflags.push("-C".to_string()); rustflags.push("-C".to_string());
rustflags.push(format!("codegen-units={}", units)); rustflags.push(format!("codegen-units={}", units));
} }
// 内联阈值 // 内联阈值
rustflags.push("-C".to_string()); rustflags.push("-C".to_string());
rustflags.push(format!("inline-threshold={}", self.inline_strategy.inline_threshold)); rustflags.push(format!("inline-threshold={}", self.inline_strategy.inline_threshold));
// 额外的性能优化标志 // 额外的性能优化标志
rustflags.extend([ rustflags.extend([
"-C".to_string(), "embed-bitcode=no".to_string(), // 不嵌入位码以减少体积 "-C".to_string(),
"-C".to_string(), "debuginfo=0".to_string(), // 禁用调试信息 "embed-bitcode=no".to_string(), // 不嵌入位码以减少体积
"-C".to_string(), "rpath=no".to_string(), // 禁用rpath "-C".to_string(),
"-C".to_string(), "force-frame-pointers=no".to_string(), // 禁用帧指针 "debuginfo=0".to_string(), // 禁用调试信息
"-C".to_string(),
"rpath=no".to_string(), // 禁用rpath
"-C".to_string(),
"force-frame-pointers=no".to_string(), // 禁用帧指针
]); ]);
let config = CompilerConfig { let config = CompilerConfig {
rustflags, rustflags,
env_vars: self.generate_env_vars(), env_vars: self.generate_env_vars(),
cargo_config: self.generate_cargo_config(), cargo_config: self.generate_cargo_config(),
}; };
tracing::info!(target: "sol_trade_sdk","✅ Ultra-performance compiler configuration generated"); tracing::info!(target: "sol_trade_sdk","✅ Ultra-performance compiler configuration generated");
Ok(config) Ok(config)
} }
/// 生成环境变量配置 /// 生成环境变量配置
fn generate_env_vars(&self) -> HashMap<String, String> { fn generate_env_vars(&self) -> HashMap<String, String> {
let mut env_vars = HashMap::new(); let mut env_vars = HashMap::new();
// CPU特定优化 // CPU特定优化
env_vars.insert("CARGO_CFG_TARGET_FEATURE".to_string(), env_vars.insert(
self.optimization_flags.target_features.join(",")); "CARGO_CFG_TARGET_FEATURE".to_string(),
self.optimization_flags.target_features.join(","),
);
// 启用不稳定特性 // 启用不稳定特性
env_vars.insert("RUSTC_BOOTSTRAP".to_string(), "1".to_string()); env_vars.insert("RUSTC_BOOTSTRAP".to_string(), "1".to_string());
// 编译缓存设置 // 编译缓存设置
if self.optimization_flags.incremental { if self.optimization_flags.incremental {
env_vars.insert("CARGO_INCREMENTAL".to_string(), "1".to_string()); env_vars.insert("CARGO_INCREMENTAL".to_string(), "1".to_string());
} else { } else {
env_vars.insert("CARGO_INCREMENTAL".to_string(), "0".to_string()); env_vars.insert("CARGO_INCREMENTAL".to_string(), "0".to_string());
} }
env_vars env_vars
} }
/// 生成Cargo配置 /// 生成Cargo配置
fn generate_cargo_config(&self) -> CargoConfig { fn generate_cargo_config(&self) -> CargoConfig {
CargoConfig { CargoConfig {
@@ -244,17 +254,21 @@ impl CompilerOptimizer {
debug_assertions: false, debug_assertions: false,
rpath: false, rpath: false,
strip: true, // 去除符号表 strip: true, // 去除符号表
} },
} }
} }
/// 获取统计信息 /// 获取统计信息
pub fn get_stats(&self) -> CompilerOptimizationStats { pub fn get_stats(&self) -> CompilerOptimizationStats {
CompilerOptimizationStats { CompilerOptimizationStats {
inlined_functions: AtomicU64::new(self.stats.inlined_functions.load(Ordering::Relaxed)), inlined_functions: AtomicU64::new(self.stats.inlined_functions.load(Ordering::Relaxed)),
constant_folding: AtomicU64::new(self.stats.constant_folding.load(Ordering::Relaxed)), constant_folding: AtomicU64::new(self.stats.constant_folding.load(Ordering::Relaxed)),
dead_code_elimination: AtomicU64::new(self.stats.dead_code_elimination.load(Ordering::Relaxed)), dead_code_elimination: AtomicU64::new(
loop_optimizations: AtomicU64::new(self.stats.loop_optimizations.load(Ordering::Relaxed)), self.stats.dead_code_elimination.load(Ordering::Relaxed),
),
loop_optimizations: AtomicU64::new(
self.stats.loop_optimizations.load(Ordering::Relaxed),
),
} }
} }
} }
@@ -279,12 +293,12 @@ impl OptimizationFlags {
Self { Self {
opt_level: OptLevel::Aggressive, opt_level: OptLevel::Aggressive,
enable_lto: true, enable_lto: true,
enable_pgo: false, // PGO需要多阶段构建 enable_pgo: false, // PGO需要多阶段构建
target_cpu: "native".to_string(), // 使用本机CPU特性 target_cpu: "native".to_string(), // 使用本机CPU特性
target_features, target_features,
code_model: CodeModel::Small, code_model: CodeModel::Small,
debug_info: false, debug_info: false,
incremental: false, // 发布版本禁用增量编译 incremental: false, // 发布版本禁用增量编译
codegen_units: Some(1), // 单个代码生成单元获得最佳优化 codegen_units: Some(1), // 单个代码生成单元获得最佳优化
} }
} }
@@ -294,7 +308,7 @@ impl CodegenConfig {
/// 超高性能配置 /// 超高性能配置
pub fn ultra_performance() -> Self { pub fn ultra_performance() -> Self {
Self { Self {
panic_abort: true, // 恐慌即中止,避免展开开销 panic_abort: true, // 恐慌即中止,避免展开开销
overflow_checks: false, // 生产环境禁用溢出检查 overflow_checks: false, // 生产环境禁用溢出检查
fat_lto: true, fat_lto: true,
enable_simd: true, enable_simd: true,
@@ -353,14 +367,14 @@ macro_rules! compile_time_optimize {
(const $expr:expr) => { (const $expr:expr) => {
const { $expr } const { $expr }
}; };
// 强制内联热路径 // 强制内联热路径
(inline_hot $fn_name:ident) => { (inline_hot $fn_name:ident) => {
#[inline(always)] #[inline(always)]
#[hot] #[hot]
$fn_name $fn_name
}; };
// 标记冷路径 // 标记冷路径
(cold $fn_name:ident) => { (cold $fn_name:ident) => {
#[inline(never)] #[inline(never)]
@@ -372,10 +386,10 @@ macro_rules! compile_time_optimize {
/// 🚀 零成本抽象特征 /// 🚀 零成本抽象特征
pub trait ZeroCostAbstraction { pub trait ZeroCostAbstraction {
type Output; type Output;
/// 编译时计算 /// 编译时计算
fn compute_at_compile_time(&self) -> Self::Output; fn compute_at_compile_time(&self) -> Self::Output;
/// 内联操作 /// 内联操作
#[inline(always)] #[inline(always)]
fn inline_operation(&self) -> Self::Output { fn inline_operation(&self) -> Self::Output {
@@ -399,35 +413,35 @@ impl CompileTimeOptimizedEventProcessor {
route_table: Self::precompute_route_table(), route_table: Self::precompute_route_table(),
} }
} }
/// 编译时预计算哈希表 /// 编译时预计算哈希表
const fn precompute_hash_table() -> [u64; 256] { const fn precompute_hash_table() -> [u64; 256] {
let mut table = [0u64; 256]; let mut table = [0u64; 256];
let mut i = 0; let mut i = 0;
while i < 256 { while i < 256 {
// 使用编译时常量计算哈希值 // 使用编译时常量计算哈希值
table[i] = Self::const_hash(i as u8); table[i] = Self::const_hash(i as u8);
i += 1; i += 1;
} }
table table
} }
/// 编译时预计算路由表 /// 编译时预计算路由表
const fn precompute_route_table() -> [u32; 1024] { const fn precompute_route_table() -> [u32; 1024] {
let mut table = [0u32; 1024]; let mut table = [0u32; 1024];
let mut i = 0; let mut i = 0;
while i < 1024 { while i < 1024 {
// 预计算路由信息 // 预计算路由信息
table[i] = (i as u32) % 16; // 16个工作线程 table[i] = (i as u32) % 16; // 16个工作线程
i += 1; i += 1;
} }
table table
} }
/// 编译时常量哈希函数 /// 编译时常量哈希函数
const fn const_hash(input: u8) -> u64 { const fn const_hash(input: u8) -> u64 {
// 使用简单的编译时常量哈希 // 使用简单的编译时常量哈希
@@ -437,16 +451,14 @@ impl CompileTimeOptimizedEventProcessor {
hash ^= hash << 17; hash ^= hash << 17;
hash hash
} }
/// 🚀 零开销事件路由 /// 🚀 零开销事件路由
#[inline(always)] #[inline(always)]
pub fn route_event_zero_cost(&self, event_id: u8) -> u32 { pub fn route_event_zero_cost(&self, event_id: u8) -> u32 {
// 编译时优化:直接数组访问,无边界检查 // 编译时优化:直接数组访问,无边界检查
unsafe { unsafe { *self.route_table.get_unchecked((event_id as usize) & 1023) }
*self.route_table.get_unchecked((event_id as usize) & 1023)
}
} }
/// 🚀 编译时优化的哈希查找 /// 🚀 编译时优化的哈希查找
#[inline(always)] #[inline(always)]
pub fn hash_lookup_optimized(&self, key: u8) -> u64 { pub fn hash_lookup_optimized(&self, key: u8) -> u64 {
@@ -464,28 +476,28 @@ impl SIMDCompileTimeOptimizer {
#[target_feature(enable = "avx2")] #[target_feature(enable = "avx2")]
pub unsafe fn vectorized_sum_compile_time(data: &[u64]) -> u64 { pub unsafe fn vectorized_sum_compile_time(data: &[u64]) -> u64 {
use std::arch::x86_64::*; use std::arch::x86_64::*;
if data.len() < 4 { if data.len() < 4 {
return data.iter().sum(); return data.iter().sum();
} }
let chunks = data.len() / 4; let chunks = data.len() / 4;
let mut sum_vec = _mm256_setzero_si256(); let mut sum_vec = _mm256_setzero_si256();
for i in 0..chunks { for i in 0..chunks {
let ptr = data.as_ptr().add(i * 4) as *const __m256i; let ptr = data.as_ptr().add(i * 4) as *const __m256i;
let vec = _mm256_loadu_si256(ptr); let vec = _mm256_loadu_si256(ptr);
sum_vec = _mm256_add_epi64(sum_vec, vec); sum_vec = _mm256_add_epi64(sum_vec, vec);
} }
// 水平求和 // 水平求和
let mut result = [0u64; 4]; let mut result = [0u64; 4];
_mm256_storeu_si256(result.as_mut_ptr() as *mut __m256i, sum_vec); _mm256_storeu_si256(result.as_mut_ptr() as *mut __m256i, sum_vec);
let partial_sum: u64 = result.iter().sum(); let partial_sum: u64 = result.iter().sum();
// 处理剩余元素 // 处理剩余元素
let remaining: u64 = data[chunks * 4..].iter().sum(); let remaining: u64 = data[chunks * 4..].iter().sum();
partial_sum + remaining partial_sum + remaining
} }
@@ -523,7 +535,8 @@ fn main() {
println!("cargo:rustc-link-arg=-fprofile-use"); println!("cargo:rustc-link-arg=-fprofile-use");
} }
} }
"#.to_string() "#
.to_string()
} }
/// 🚀 生成.cargo/config.toml /// 🚀 生成.cargo/config.toml
@@ -576,56 +589,58 @@ rustflags = [
rustflags = [ rustflags = [
"-C", "target-feature=+sse4.2,+avx,+avx2,+fma,+bmi1,+bmi2,+lzcnt,+popcnt", "-C", "target-feature=+sse4.2,+avx,+avx2,+fma,+bmi1,+bmi2,+lzcnt,+popcnt",
] ]
"#.to_string() "#
.to_string()
} }
#[cfg(test)] #[cfg(test)]
mod tests { mod tests {
use super::*; use super::*;
#[test] #[test]
fn test_compiler_optimizer_creation() { fn test_compiler_optimizer_creation() {
let optimizer = CompilerOptimizer::new(); let optimizer = CompilerOptimizer::new();
assert!(optimizer.optimization_flags.enable_lto); assert!(optimizer.optimization_flags.enable_lto);
assert_eq!(optimizer.optimization_flags.opt_level as u8, OptLevel::Aggressive as u8); assert_eq!(optimizer.optimization_flags.opt_level as u8, OptLevel::Aggressive as u8);
} }
#[test] #[test]
fn test_compile_time_processor() { fn test_compile_time_processor() {
const PROCESSOR: CompileTimeOptimizedEventProcessor = CompileTimeOptimizedEventProcessor::new(); const PROCESSOR: CompileTimeOptimizedEventProcessor =
CompileTimeOptimizedEventProcessor::new();
let route = PROCESSOR.route_event_zero_cost(42); let route = PROCESSOR.route_event_zero_cost(42);
assert!(route < 16); // 应该路由到16个工作线程之一 assert!(route < 16); // 应该路由到16个工作线程之一
let hash = PROCESSOR.hash_lookup_optimized(100); let hash = PROCESSOR.hash_lookup_optimized(100);
assert!(hash > 0); // 哈希值应该非零 assert!(hash > 0); // 哈希值应该非零
} }
#[test] #[test]
fn test_ultra_performance_config() { fn test_ultra_performance_config() {
let flags = OptimizationFlags::ultra_performance(); let flags = OptimizationFlags::ultra_performance();
assert!(flags.enable_lto); assert!(flags.enable_lto);
assert_eq!(flags.target_cpu, "native"); assert_eq!(flags.target_cpu, "native");
assert!(!flags.target_features.is_empty()); assert!(!flags.target_features.is_empty());
let codegen = CodegenConfig::ultra_performance(); let codegen = CodegenConfig::ultra_performance();
assert!(codegen.panic_abort); assert!(codegen.panic_abort);
assert!(!codegen.overflow_checks); assert!(!codegen.overflow_checks);
assert!(codegen.enable_simd); assert!(codegen.enable_simd);
} }
#[test] #[test]
fn test_compiler_config_generation() { fn test_compiler_config_generation() {
let optimizer = CompilerOptimizer::new(); let optimizer = CompilerOptimizer::new();
let config = optimizer.generate_ultra_performance_config().unwrap(); let config = optimizer.generate_ultra_performance_config().unwrap();
assert!(!config.rustflags.is_empty()); assert!(!config.rustflags.is_empty());
assert!(config.rustflags.contains(&"-C".to_string())); assert!(config.rustflags.contains(&"-C".to_string()));
assert!(config.rustflags.contains(&"opt-level=3".to_string())); assert!(config.rustflags.contains(&"opt-level=3".to_string()));
assert!(config.env_vars.contains_key("CARGO_INCREMENTAL")); assert!(config.env_vars.contains_key("CARGO_INCREMENTAL"));
} }
#[test] #[test]
fn test_simd_compile_time_optimization() { fn test_simd_compile_time_optimization() {
#[cfg(any(target_arch = "x86", target_arch = "x86_64"))] #[cfg(any(target_arch = "x86", target_arch = "x86_64"))]
@@ -642,7 +657,7 @@ mod tests {
assert_eq!(sum, 36); // 1+2+3+4+5+6+7+8 = 36 assert_eq!(sum, 36); // 1+2+3+4+5+6+7+8 = 36
} }
} }
#[test] #[test]
fn test_build_script_generation() { fn test_build_script_generation() {
let build_script = generate_build_script(); let build_script = generate_build_script();
@@ -650,7 +665,7 @@ mod tests {
assert!(build_script.contains("TARGET_FEATURE")); assert!(build_script.contains("TARGET_FEATURE"));
assert!(build_script.contains("lld")); assert!(build_script.contains("lld"));
} }
#[test] #[test]
fn test_cargo_config_generation() { fn test_cargo_config_generation() {
let config = generate_cargo_config_toml(); let config = generate_cargo_config_toml();
@@ -659,4 +674,4 @@ mod tests {
assert!(config.contains("target-cpu=native")); assert!(config.contains("target-cpu=native"));
assert!(config.contains("panic = \"abort\"")); assert!(config.contains("panic = \"abort\""));
} }
} }
+82 -93
View File
@@ -1,11 +1,11 @@
//! Hardware-oriented optimizations: cache-line alignment, prefetch, SIMD, branch hints, memory barriers. //! Hardware-oriented optimizations: cache-line alignment, prefetch, SIMD, branch hints, memory barriers.
//! 硬件级优化:缓存行对齐与预取、SIMD、分支提示、内存屏障。 //! 硬件级优化:缓存行对齐与预取、SIMD、分支提示、内存屏障。
use std::sync::atomic::{AtomicU64, Ordering}; use anyhow::Result;
use crossbeam_utils::CachePadded;
use std::mem::size_of; use std::mem::size_of;
use std::ptr; use std::ptr;
use crossbeam_utils::CachePadded; use std::sync::atomic::{AtomicU64, Ordering};
use anyhow::Result;
/// Typical CPU cache line size in bytes. 典型 CPU 缓存行大小(字节)。 /// Typical CPU cache line size in bytes. 典型 CPU 缓存行大小(字节)。
pub const CACHE_LINE_SIZE: usize = 64; pub const CACHE_LINE_SIZE: usize = 64;
@@ -32,7 +32,7 @@ impl SIMDMemoryOps {
_ => Self::memcpy_avx512_or_fallback(dst, src, len), _ => Self::memcpy_avx512_or_fallback(dst, src, len),
} }
} }
/// Copy 18 bytes (scalar / small word). 小数据拷贝(18 字节)。 /// Copy 18 bytes (scalar / small word). 小数据拷贝(18 字节)。
#[inline(always)] #[inline(always)]
unsafe fn memcpy_small(dst: *mut u8, src: *const u8, len: usize) { unsafe fn memcpy_small(dst: *mut u8, src: *const u8, len: usize) {
@@ -53,52 +53,52 @@ impl SIMDMemoryOps {
_ => unreachable!(), _ => unreachable!(),
} }
} }
/// Copy 916 bytes using SSE (128-bit). SSE 拷贝(916 字节)。 /// Copy 916 bytes using SSE (128-bit). SSE 拷贝(916 字节)。
#[inline(always)] #[inline(always)]
unsafe fn memcpy_sse(dst: *mut u8, src: *const u8, len: usize) { unsafe fn memcpy_sse(dst: *mut u8, src: *const u8, len: usize) {
#[cfg(target_arch = "x86_64")] #[cfg(target_arch = "x86_64")]
{ {
use std::arch::x86_64::{__m128i, _mm_loadu_si128, _mm_storeu_si128}; use std::arch::x86_64::{__m128i, _mm_loadu_si128, _mm_storeu_si128};
if len <= 16 { if len <= 16 {
let chunk = _mm_loadu_si128(src as *const __m128i); let chunk = _mm_loadu_si128(src as *const __m128i);
_mm_storeu_si128(dst as *mut __m128i, chunk); _mm_storeu_si128(dst as *mut __m128i, chunk);
} }
} }
#[cfg(not(target_arch = "x86_64"))] #[cfg(not(target_arch = "x86_64"))]
{ {
ptr::copy_nonoverlapping(src, dst, len); ptr::copy_nonoverlapping(src, dst, len);
} }
} }
/// Copy 1732 bytes using AVX (256-bit). AVX 拷贝(1732 字节)。 /// Copy 1732 bytes using AVX (256-bit). AVX 拷贝(1732 字节)。
#[inline(always)] #[inline(always)]
unsafe fn memcpy_avx(dst: *mut u8, src: *const u8, len: usize) { unsafe fn memcpy_avx(dst: *mut u8, src: *const u8, len: usize) {
#[cfg(target_arch = "x86_64")] #[cfg(target_arch = "x86_64")]
{ {
use std::arch::x86_64::{__m256i, _mm256_loadu_si256, _mm256_storeu_si256}; use std::arch::x86_64::{__m256i, _mm256_loadu_si256, _mm256_storeu_si256};
if len <= 32 { if len <= 32 {
let chunk = _mm256_loadu_si256(src as *const __m256i); let chunk = _mm256_loadu_si256(src as *const __m256i);
_mm256_storeu_si256(dst as *mut __m256i, chunk); _mm256_storeu_si256(dst as *mut __m256i, chunk);
} }
} }
#[cfg(not(target_arch = "x86_64"))] #[cfg(not(target_arch = "x86_64"))]
{ {
ptr::copy_nonoverlapping(src, dst, len); ptr::copy_nonoverlapping(src, dst, len);
} }
} }
/// Copy 3364 bytes using AVX2 (256-bit, two chunks). AVX2 拷贝(3364 字节,两段)。 /// Copy 3364 bytes using AVX2 (256-bit, two chunks). AVX2 拷贝(3364 字节,两段)。
#[inline(always)] #[inline(always)]
unsafe fn memcpy_avx2(dst: *mut u8, src: *const u8, len: usize) { unsafe fn memcpy_avx2(dst: *mut u8, src: *const u8, len: usize) {
#[cfg(target_arch = "x86_64")] #[cfg(target_arch = "x86_64")]
{ {
use std::arch::x86_64::{__m256i, _mm256_loadu_si256, _mm256_storeu_si256}; use std::arch::x86_64::{__m256i, _mm256_loadu_si256, _mm256_storeu_si256};
let chunk1 = _mm256_loadu_si256(src as *const __m256i); let chunk1 = _mm256_loadu_si256(src as *const __m256i);
_mm256_storeu_si256(dst as *mut __m256i, chunk1); _mm256_storeu_si256(dst as *mut __m256i, chunk1);
if len > 32 { if len > 32 {
@@ -109,52 +109,52 @@ impl SIMDMemoryOps {
} }
} }
} }
#[cfg(not(target_arch = "x86_64"))] #[cfg(not(target_arch = "x86_64"))]
{ {
ptr::copy_nonoverlapping(src, dst, len); ptr::copy_nonoverlapping(src, dst, len);
} }
} }
/// Copy >64 bytes: AVX-512 64-byte chunks when available, else AVX2 32-byte chunks. >64 字节:有 AVX512 用 64 字节块,否则 AVX2 32 字节块。 /// Copy >64 bytes: AVX-512 64-byte chunks when available, else AVX2 32-byte chunks. >64 字节:有 AVX512 用 64 字节块,否则 AVX2 32 字节块。
#[inline(always)] #[inline(always)]
unsafe fn memcpy_avx512_or_fallback(dst: *mut u8, src: *const u8, len: usize) { unsafe fn memcpy_avx512_or_fallback(dst: *mut u8, src: *const u8, len: usize) {
#[cfg(all(target_arch = "x86_64", target_feature = "avx512f"))] #[cfg(all(target_arch = "x86_64", target_feature = "avx512f"))]
{ {
use std::arch::x86_64::{__m512i, _mm512_loadu_si512, _mm512_storeu_si512}; use std::arch::x86_64::{__m512i, _mm512_loadu_si512, _mm512_storeu_si512};
let chunks = len / 64; let chunks = len / 64;
let mut offset = 0; let mut offset = 0;
for _ in 0..chunks { for _ in 0..chunks {
let chunk = _mm512_loadu_si512(src.add(offset) as *const __m512i); let chunk = _mm512_loadu_si512(src.add(offset) as *const __m512i);
_mm512_storeu_si512(dst.add(offset) as *mut __m512i, chunk); _mm512_storeu_si512(dst.add(offset) as *mut __m512i, chunk);
offset += 64; offset += 64;
} }
let remaining = len % 64; let remaining = len % 64;
if remaining > 0 { if remaining > 0 {
Self::memcpy_avx2(dst.add(offset), src.add(offset), remaining); Self::memcpy_avx2(dst.add(offset), src.add(offset), remaining);
} }
} }
#[cfg(not(all(target_arch = "x86_64", target_feature = "avx512f")))] #[cfg(not(all(target_arch = "x86_64", target_feature = "avx512f")))]
{ {
let chunks = len / 32; let chunks = len / 32;
let mut offset = 0; let mut offset = 0;
for _ in 0..chunks { for _ in 0..chunks {
Self::memcpy_avx2(dst.add(offset), src.add(offset), 32); Self::memcpy_avx2(dst.add(offset), src.add(offset), 32);
offset += 32; offset += 32;
} }
let remaining = len % 32; let remaining = len % 32;
if remaining > 0 { if remaining > 0 {
Self::memcpy_avx(dst.add(offset), src.add(offset), remaining); Self::memcpy_avx(dst.add(offset), src.add(offset), remaining);
} }
} }
} }
/// SIMD-optimized byte equality; dispatches by length (small / SSE / AVX2 / large). SIMD 加速的内存比较,按长度分派。 /// SIMD-optimized byte equality; dispatches by length (small / SSE / AVX2 / large). SIMD 加速的内存比较,按长度分派。
#[inline(always)] #[inline(always)]
pub unsafe fn memcmp_simd_optimized(a: *const u8, b: *const u8, len: usize) -> bool { pub unsafe fn memcmp_simd_optimized(a: *const u8, b: *const u8, len: usize) -> bool {
@@ -166,109 +166,108 @@ impl SIMDMemoryOps {
_ => Self::memcmp_large(a, b, len), _ => Self::memcmp_large(a, b, len),
} }
} }
/// Compare 18 bytes (scalar). 小数据比较(18 字节)。 /// Compare 18 bytes (scalar). 小数据比较(18 字节)。
#[inline(always)] #[inline(always)]
unsafe fn memcmp_small(a: *const u8, b: *const u8, len: usize) -> bool { unsafe fn memcmp_small(a: *const u8, b: *const u8, len: usize) -> bool {
match len { match len {
1 => *a == *b, 1 => *a == *b,
2 => *(a as *const u16) == *(b as *const u16), 2 => *(a as *const u16) == *(b as *const u16),
3 => { 3 => *(a as *const u16) == *(b as *const u16) && *a.add(2) == *b.add(2),
*(a as *const u16) == *(b as *const u16) &&
*a.add(2) == *b.add(2)
}
4 => *(a as *const u32) == *(b as *const u32), 4 => *(a as *const u32) == *(b as *const u32),
5..=8 => *(a as *const u64) == *(b as *const u64), 5..=8 => *(a as *const u64) == *(b as *const u64),
_ => unreachable!(), _ => unreachable!(),
} }
} }
/// Compare 916 bytes using SSE. SSE 比较(916 字节)。 /// Compare 916 bytes using SSE. SSE 比较(916 字节)。
#[inline(always)] #[inline(always)]
unsafe fn memcmp_sse(a: *const u8, b: *const u8, len: usize) -> bool { unsafe fn memcmp_sse(a: *const u8, b: *const u8, len: usize) -> bool {
#[cfg(target_arch = "x86_64")] #[cfg(target_arch = "x86_64")]
{ {
use std::arch::x86_64::{__m128i, _mm_loadu_si128, _mm_cmpeq_epi8, _mm_movemask_epi8}; use std::arch::x86_64::{__m128i, _mm_cmpeq_epi8, _mm_loadu_si128, _mm_movemask_epi8};
let chunk_a = _mm_loadu_si128(a as *const __m128i); let chunk_a = _mm_loadu_si128(a as *const __m128i);
let chunk_b = _mm_loadu_si128(b as *const __m128i); let chunk_b = _mm_loadu_si128(b as *const __m128i);
let cmp_result = _mm_cmpeq_epi8(chunk_a, chunk_b); let cmp_result = _mm_cmpeq_epi8(chunk_a, chunk_b);
let mask = _mm_movemask_epi8(cmp_result) as u32; let mask = _mm_movemask_epi8(cmp_result) as u32;
let valid_mask = if len >= 16 { 0xFFFF } else { (1u32 << len) - 1 }; let valid_mask = if len >= 16 { 0xFFFF } else { (1u32 << len) - 1 };
(mask & valid_mask) == valid_mask (mask & valid_mask) == valid_mask
} }
#[cfg(not(target_arch = "x86_64"))] #[cfg(not(target_arch = "x86_64"))]
{ {
(0..len).all(|i| *a.add(i) == *b.add(i)) (0..len).all(|i| *a.add(i) == *b.add(i))
} }
} }
/// Compare 1732 bytes using AVX2. AVX2 比较(1732 字节)。 /// Compare 1732 bytes using AVX2. AVX2 比较(1732 字节)。
#[inline(always)] #[inline(always)]
unsafe fn memcmp_avx2(a: *const u8, b: *const u8, len: usize) -> bool { unsafe fn memcmp_avx2(a: *const u8, b: *const u8, len: usize) -> bool {
#[cfg(target_arch = "x86_64")] #[cfg(target_arch = "x86_64")]
{ {
use std::arch::x86_64::{__m256i, _mm256_loadu_si256, _mm256_cmpeq_epi8, _mm256_movemask_epi8}; use std::arch::x86_64::{
__m256i, _mm256_cmpeq_epi8, _mm256_loadu_si256, _mm256_movemask_epi8,
};
let chunk_a = _mm256_loadu_si256(a as *const __m256i); let chunk_a = _mm256_loadu_si256(a as *const __m256i);
let chunk_b = _mm256_loadu_si256(b as *const __m256i); let chunk_b = _mm256_loadu_si256(b as *const __m256i);
let cmp_result = _mm256_cmpeq_epi8(chunk_a, chunk_b); let cmp_result = _mm256_cmpeq_epi8(chunk_a, chunk_b);
let mask = _mm256_movemask_epi8(cmp_result) as u32; let mask = _mm256_movemask_epi8(cmp_result) as u32;
let valid_mask = if len >= 32 { 0xFFFFFFFF } else { (1u32 << len) - 1 }; let valid_mask = if len >= 32 { 0xFFFFFFFF } else { (1u32 << len) - 1 };
(mask & valid_mask) == valid_mask (mask & valid_mask) == valid_mask
} }
#[cfg(not(target_arch = "x86_64"))] #[cfg(not(target_arch = "x86_64"))]
{ {
(0..len).all(|i| *a.add(i) == *b.add(i)) (0..len).all(|i| *a.add(i) == *b.add(i))
} }
} }
/// Compare >32 bytes in 32-byte AVX2 chunks. 大数据比较(32 字节 AVX2 分块)。 /// Compare >32 bytes in 32-byte AVX2 chunks. 大数据比较(32 字节 AVX2 分块)。
#[inline(always)] #[inline(always)]
unsafe fn memcmp_large(a: *const u8, b: *const u8, len: usize) -> bool { unsafe fn memcmp_large(a: *const u8, b: *const u8, len: usize) -> bool {
let chunks = len / 32; let chunks = len / 32;
for i in 0..chunks { for i in 0..chunks {
let offset = i * 32; let offset = i * 32;
if !Self::memcmp_avx2(a.add(offset), b.add(offset), 32) { if !Self::memcmp_avx2(a.add(offset), b.add(offset), 32) {
return false; return false;
} }
} }
let remaining = len % 32; let remaining = len % 32;
if remaining > 0 { if remaining > 0 {
return Self::memcmp_avx2(a.add(chunks * 32), b.add(chunks * 32), remaining); return Self::memcmp_avx2(a.add(chunks * 32), b.add(chunks * 32), remaining);
} }
true true
} }
/// SIMD-optimized zero memory. SIMD 加速的内存清零。 /// SIMD-optimized zero memory. SIMD 加速的内存清零。
#[inline(always)] #[inline(always)]
pub unsafe fn memzero_simd_optimized(ptr: *mut u8, len: usize) { pub unsafe fn memzero_simd_optimized(ptr: *mut u8, len: usize) {
#[cfg(target_arch = "x86_64")] #[cfg(target_arch = "x86_64")]
{ {
use std::arch::x86_64::{__m256i, _mm256_setzero_si256, _mm256_storeu_si256}; use std::arch::x86_64::{__m256i, _mm256_setzero_si256, _mm256_storeu_si256};
let zero = _mm256_setzero_si256(); let zero = _mm256_setzero_si256();
let chunks = len / 32; let chunks = len / 32;
let mut offset = 0; let mut offset = 0;
for _ in 0..chunks { for _ in 0..chunks {
_mm256_storeu_si256(ptr.add(offset) as *mut __m256i, zero); _mm256_storeu_si256(ptr.add(offset) as *mut __m256i, zero);
offset += 32; offset += 32;
} }
let remaining = len % 32; let remaining = len % 32;
for i in 0..remaining { for i in 0..remaining {
*ptr.add(offset + i) = 0; *ptr.add(offset + i) = 0;
} }
} }
#[cfg(not(target_arch = "x86_64"))] #[cfg(not(target_arch = "x86_64"))]
{ {
ptr::write_bytes(ptr, 0, len); ptr::write_bytes(ptr, 0, len);
@@ -291,17 +290,17 @@ impl CacheAlignedCounter {
_padding: [0; CACHE_LINE_SIZE - size_of::<AtomicU64>()], _padding: [0; CACHE_LINE_SIZE - size_of::<AtomicU64>()],
} }
} }
#[inline(always)] #[inline(always)]
pub fn increment(&self) -> u64 { pub fn increment(&self) -> u64 {
self.value.fetch_add(1, Ordering::Relaxed) self.value.fetch_add(1, Ordering::Relaxed)
} }
#[inline(always)] #[inline(always)]
pub fn load(&self) -> u64 { pub fn load(&self) -> u64 {
self.value.load(Ordering::Relaxed) self.value.load(Ordering::Relaxed)
} }
#[inline(always)] #[inline(always)]
pub fn store(&self, val: u64) { pub fn store(&self, val: u64) {
self.value.store(val, Ordering::Relaxed) self.value.store(val, Ordering::Relaxed)
@@ -312,7 +311,7 @@ impl CacheLineAligned for CacheAlignedCounter {
fn ensure_cache_aligned(&self) -> bool { fn ensure_cache_aligned(&self) -> bool {
(self as *const Self as usize) % CACHE_LINE_SIZE == 0 (self as *const Self as usize) % CACHE_LINE_SIZE == 0
} }
fn prefetch_data(&self) { fn prefetch_data(&self) {
#[cfg(target_arch = "x86_64")] #[cfg(target_arch = "x86_64")]
unsafe { unsafe {
@@ -339,10 +338,10 @@ impl<T: Copy + Default> CacheOptimizedRingBuffer<T> {
if !capacity.is_power_of_two() { if !capacity.is_power_of_two() {
return Err(anyhow::anyhow!("Capacity must be a power of 2")); return Err(anyhow::anyhow!("Capacity must be a power of 2"));
} }
let mut buffer = Vec::with_capacity(capacity); let mut buffer = Vec::with_capacity(capacity);
buffer.resize_with(capacity, Default::default); buffer.resize_with(capacity, Default::default);
Ok(Self { Ok(Self {
buffer, buffer,
producer_head: CachePadded::new(AtomicU64::new(0)), producer_head: CachePadded::new(AtomicU64::new(0)),
@@ -351,7 +350,7 @@ impl<T: Copy + Default> CacheOptimizedRingBuffer<T> {
mask: capacity - 1, mask: capacity - 1,
}) })
} }
/// Lock-free push; returns false if full. 无锁写入,满则返回 false。 /// Lock-free push; returns false if full. 无锁写入,满则返回 false。
#[inline(always)] #[inline(always)]
pub fn try_push(&self, item: T) -> bool { pub fn try_push(&self, item: T) -> bool {
@@ -368,7 +367,7 @@ impl<T: Copy + Default> CacheOptimizedRingBuffer<T> {
self.producer_head.store(current_head + 1, Ordering::Release); self.producer_head.store(current_head + 1, Ordering::Release);
true true
} }
/// Lock-free pop; returns None if empty. 无锁读取,空则返回 None。 /// Lock-free pop; returns None if empty. 无锁读取,空则返回 None。
#[inline(always)] #[inline(always)]
pub fn try_pop(&self) -> Option<T> { pub fn try_pop(&self) -> Option<T> {
@@ -385,7 +384,7 @@ impl<T: Copy + Default> CacheOptimizedRingBuffer<T> {
self.consumer_tail.store(current_tail + 1, Ordering::Release); self.consumer_tail.store(current_tail + 1, Ordering::Release);
Some(item) Some(item)
} }
/// Current number of elements. 当前元素个数。 /// Current number of elements. 当前元素个数。
#[inline(always)] #[inline(always)]
pub fn len(&self) -> usize { pub fn len(&self) -> usize {
@@ -393,12 +392,11 @@ impl<T: Copy + Default> CacheOptimizedRingBuffer<T> {
let tail = self.consumer_tail.load(Ordering::Relaxed); let tail = self.consumer_tail.load(Ordering::Relaxed);
((head + self.capacity as u64 - tail) & self.mask as u64) as usize ((head + self.capacity as u64 - tail) & self.mask as u64) as usize
} }
/// True if no elements. 是否为空。 /// True if no elements. 是否为空。
#[inline(always)] #[inline(always)]
pub fn is_empty(&self) -> bool { pub fn is_empty(&self) -> bool {
self.producer_head.load(Ordering::Relaxed) == self.producer_head.load(Ordering::Relaxed) == self.consumer_tail.load(Ordering::Relaxed)
self.consumer_tail.load(Ordering::Relaxed)
} }
} }
@@ -406,7 +404,7 @@ impl<T> CacheLineAligned for CacheOptimizedRingBuffer<T> {
fn ensure_cache_aligned(&self) -> bool { fn ensure_cache_aligned(&self) -> bool {
(self as *const Self as usize) % CACHE_LINE_SIZE == 0 (self as *const Self as usize) % CACHE_LINE_SIZE == 0
} }
fn prefetch_data(&self) { fn prefetch_data(&self) {
#[cfg(target_arch = "x86_64")] #[cfg(target_arch = "x86_64")]
unsafe { unsafe {
@@ -428,25 +426,25 @@ impl BranchOptimizer {
pub fn likely(condition: bool) -> bool { pub fn likely(condition: bool) -> bool {
#[cold] #[cold]
fn cold() {} fn cold() {}
if !condition { if !condition {
cold(); cold();
} }
condition condition
} }
/// Hint: condition is usually false. 提示编译器条件大概率为假。 /// Hint: condition is usually false. 提示编译器条件大概率为假。
#[inline(always)] #[inline(always)]
pub fn unlikely(condition: bool) -> bool { pub fn unlikely(condition: bool) -> bool {
#[cold] #[cold]
fn cold() {} fn cold() {}
if condition { if condition {
cold(); cold();
} }
condition condition
} }
/// Prefetch: load cache line at ptr into L1. Caller must ensure ptr is valid, read-only, no concurrent write. 预取:将 ptr 所在缓存行加载到 L1;调用方需保证有效、只读、无并发写。 /// Prefetch: load cache line at ptr into L1. Caller must ensure ptr is valid, read-only, no concurrent write. 预取:将 ptr 所在缓存行加载到 L1;调用方需保证有效、只读、无并发写。
#[inline(always)] #[inline(always)]
pub unsafe fn prefetch_read_data<T>(ptr: *const T) { pub unsafe fn prefetch_read_data<T>(ptr: *const T) {
@@ -457,7 +455,7 @@ impl BranchOptimizer {
_mm_prefetch(ptr as *const i8, _MM_HINT_T0); _mm_prefetch(ptr as *const i8, _MM_HINT_T0);
} }
} }
/// Prefetch for write (T1 hint). 写预取(T1 提示)。 /// Prefetch for write (T1 hint). 写预取(T1 提示)。
#[inline(always)] #[inline(always)]
pub unsafe fn prefetch_write_data<T>(ptr: *const T) { pub unsafe fn prefetch_write_data<T>(ptr: *const T) {
@@ -479,25 +477,25 @@ impl MemoryBarriers {
pub fn compiler_barrier() { pub fn compiler_barrier() {
std::sync::atomic::compiler_fence(Ordering::SeqCst); std::sync::atomic::compiler_fence(Ordering::SeqCst);
} }
/// Light barrier (Acquire). 轻量级屏障(Acquire)。 /// Light barrier (Acquire). 轻量级屏障(Acquire)。
#[inline(always)] #[inline(always)]
pub fn memory_barrier_light() { pub fn memory_barrier_light() {
std::sync::atomic::fence(Ordering::Acquire); std::sync::atomic::fence(Ordering::Acquire);
} }
/// Full sequential consistency barrier. 全序一致性屏障。 /// Full sequential consistency barrier. 全序一致性屏障。
#[inline(always)] #[inline(always)]
pub fn memory_barrier_heavy() { pub fn memory_barrier_heavy() {
std::sync::atomic::fence(Ordering::SeqCst); std::sync::atomic::fence(Ordering::SeqCst);
} }
/// Store/release barrier. 存储屏障,保证写入可见性。 /// Store/release barrier. 存储屏障,保证写入可见性。
#[inline(always)] #[inline(always)]
pub fn store_barrier() { pub fn store_barrier() {
std::sync::atomic::fence(Ordering::Release); std::sync::atomic::fence(Ordering::Release);
} }
/// Load/acquire barrier. 加载屏障,保证读取顺序。 /// Load/acquire barrier. 加载屏障,保证读取顺序。
#[inline(always)] #[inline(always)]
pub fn load_barrier() { pub fn load_barrier() {
@@ -508,63 +506,54 @@ impl MemoryBarriers {
#[cfg(test)] #[cfg(test)]
mod tests { mod tests {
use super::*; use super::*;
#[test] #[test]
fn test_cache_aligned_counter() { fn test_cache_aligned_counter() {
let counter = CacheAlignedCounter::new(0); let counter = CacheAlignedCounter::new(0);
assert!(counter.ensure_cache_aligned()); assert!(counter.ensure_cache_aligned());
assert_eq!(counter.load(), 0); assert_eq!(counter.load(), 0);
counter.increment(); counter.increment();
assert_eq!(counter.load(), 1); assert_eq!(counter.load(), 1);
} }
#[test] #[test]
fn test_simd_memcpy() { fn test_simd_memcpy() {
let src = [1u8, 2, 3, 4, 5, 6, 7, 8, 9, 10]; let src = [1u8, 2, 3, 4, 5, 6, 7, 8, 9, 10];
let mut dst = [0u8; 10]; let mut dst = [0u8; 10];
unsafe { unsafe {
SIMDMemoryOps::memcpy_simd_optimized( SIMDMemoryOps::memcpy_simd_optimized(dst.as_mut_ptr(), src.as_ptr(), src.len());
dst.as_mut_ptr(),
src.as_ptr(),
src.len()
);
} }
assert_eq!(src, dst); assert_eq!(src, dst);
} }
#[test] #[test]
fn test_cache_optimized_ring_buffer() { fn test_cache_optimized_ring_buffer() {
let buffer: CacheOptimizedRingBuffer<u64> = let buffer: CacheOptimizedRingBuffer<u64> = CacheOptimizedRingBuffer::new(16).unwrap();
CacheOptimizedRingBuffer::new(16).unwrap();
assert!(buffer.is_empty()); assert!(buffer.is_empty());
// 测试推入 // 测试推入
assert!(buffer.try_push(42)); assert!(buffer.try_push(42));
assert_eq!(buffer.len(), 1); assert_eq!(buffer.len(), 1);
// 测试弹出 // 测试弹出
assert_eq!(buffer.try_pop(), Some(42)); assert_eq!(buffer.try_pop(), Some(42));
assert!(buffer.is_empty()); assert!(buffer.is_empty());
} }
#[test] #[test]
fn test_simd_memcmp() { fn test_simd_memcmp() {
let a = [1u8, 2, 3, 4, 5]; let a = [1u8, 2, 3, 4, 5];
let b = [1u8, 2, 3, 4, 5]; let b = [1u8, 2, 3, 4, 5];
let c = [1u8, 2, 3, 4, 6]; let c = [1u8, 2, 3, 4, 6];
unsafe { unsafe {
assert!(SIMDMemoryOps::memcmp_simd_optimized( assert!(SIMDMemoryOps::memcmp_simd_optimized(a.as_ptr(), b.as_ptr(), a.len()));
a.as_ptr(), b.as_ptr(), a.len()
)); assert!(!SIMDMemoryOps::memcmp_simd_optimized(a.as_ptr(), c.as_ptr(), a.len()));
assert!(!SIMDMemoryOps::memcmp_simd_optimized(
a.as_ptr(), c.as_ptr(), a.len()
));
} }
} }
} }
+8 -8
View File
@@ -1,14 +1,14 @@
//! Performance: SIMD, cache prefetch, branch hints, zero-copy I/O, syscall bypass, compiler hints. //! Performance: SIMD, cache prefetch, branch hints, zero-copy I/O, syscall bypass, compiler hints.
//! 性能优化:SIMD、缓存预取、分支提示、零拷贝 I/O、系统调用绕过、编译器提示。 //! 性能优化:SIMD、缓存预取、分支提示、零拷贝 I/O、系统调用绕过、编译器提示。
pub mod simd;
pub mod hardware_optimizations;
pub mod zero_copy_io;
pub mod syscall_bypass;
pub mod compiler_optimization; pub mod compiler_optimization;
pub mod hardware_optimizations;
pub mod simd;
pub mod syscall_bypass;
pub mod zero_copy_io;
pub use simd::*;
pub use hardware_optimizations::*;
pub use zero_copy_io::*;
pub use syscall_bypass::*;
pub use compiler_optimization::*; pub use compiler_optimization::*;
pub use hardware_optimizations::*;
pub use simd::*;
pub use syscall_bypass::*;
pub use zero_copy_io::*;
+1 -1
View File
@@ -235,7 +235,7 @@ impl SIMDHash {
/// 批量计算 SHA256 哈希 /// 批量计算 SHA256 哈希
#[inline(always)] #[inline(always)]
pub fn hash_batch_sha256(data: &[&[u8]]) -> Vec<[u8; 32]> { pub fn hash_batch_sha256(data: &[&[u8]]) -> Vec<[u8; 32]> {
use sha2::{Sha256, Digest}; use sha2::{Digest, Sha256};
data.iter() data.iter()
.map(|item| { .map(|item| {
+154 -154
View File
@@ -1,11 +1,11 @@
//! Syscall bypass: batching, vDSO fast time, io_uring, mmap, userspace impl. //! Syscall bypass: batching, vDSO fast time, io_uring, mmap, userspace impl.
//! 系统调用绕过:批处理、vDSO 快速时间、io_uring、mmap、用户态实现。 //! 系统调用绕过:批处理、vDSO 快速时间、io_uring、mmap、用户态实现。
use std::sync::atomic::{AtomicU64, Ordering};
use std::sync::Arc;
use std::time::{SystemTime, UNIX_EPOCH, Duration, Instant};
#[allow(unused_imports)] #[allow(unused_imports)]
use std::fs::OpenOptions; use std::fs::OpenOptions;
use std::sync::atomic::{AtomicU64, Ordering};
use std::sync::Arc;
use std::time::{Duration, Instant, SystemTime, UNIX_EPOCH};
use anyhow::Result; use anyhow::Result;
use crossbeam_utils::CachePadded; use crossbeam_utils::CachePadded;
@@ -55,14 +55,30 @@ pub struct SyscallBatchProcessor {
#[derive(Debug, Clone)] #[derive(Debug, Clone)]
pub enum SyscallRequest { pub enum SyscallRequest {
Write { fd: i32, data: Vec<u8> }, Write {
Read { fd: i32, size: usize }, fd: i32,
Send { socket: i32, data: Vec<u8> }, data: Vec<u8>,
Recv { socket: i32, size: usize }, },
Read {
fd: i32,
size: usize,
},
Send {
socket: i32,
data: Vec<u8>,
},
Recv {
socket: i32,
size: usize,
},
GetTime, GetTime,
MemAlloc { size: usize }, MemAlloc {
size: usize,
},
/// 内存释放 /// 内存释放
MemFree { ptr: usize }, MemFree {
ptr: usize,
},
} }
/// 🚀 快速时间提供器 - 绕过系统调用获取时间 /// 🚀 快速时间提供器 - 绕过系统调用获取时间
@@ -86,24 +102,22 @@ impl FastTimeProvider {
pub fn new(enable_vdso: bool) -> Result<Self> { pub fn new(enable_vdso: bool) -> Result<Self> {
let now = SystemTime::now(); let now = SystemTime::now();
let instant_now = Instant::now(); let instant_now = Instant::now();
let provider = Self { let provider = Self {
_base_time: now, _base_time: now,
monotonic_start: instant_now, monotonic_start: instant_now,
time_cache: CachePadded::new(AtomicU64::new( time_cache: CachePadded::new(AtomicU64::new(
now.duration_since(UNIX_EPOCH)?.as_nanos() as u64 now.duration_since(UNIX_EPOCH)?.as_nanos() as u64,
)), )),
cache_update_interval_ns: 1_000_000, // 1ms cache_update_interval_ns: 1_000_000, // 1ms
last_update: CachePadded::new(AtomicU64::new( last_update: CachePadded::new(AtomicU64::new(instant_now.elapsed().as_nanos() as u64)),
instant_now.elapsed().as_nanos() as u64
)),
vdso_enabled: enable_vdso, vdso_enabled: enable_vdso,
}; };
tracing::info!(target: "sol_trade_sdk","🚀 Fast time provider initialized with vDSO: {}", enable_vdso); tracing::info!(target: "sol_trade_sdk","🚀 Fast time provider initialized with vDSO: {}", enable_vdso);
Ok(provider) Ok(provider)
} }
/// 🚀 超快速获取当前时间 - 绕过系统调用 /// 🚀 超快速获取当前时间 - 绕过系统调用
#[inline(always)] #[inline(always)]
pub fn fast_now_nanos(&self) -> u64 { pub fn fast_now_nanos(&self) -> u64 {
@@ -111,19 +125,19 @@ impl FastTimeProvider {
// 使用vDSO快速获取时间 // 使用vDSO快速获取时间
return self.vdso_time_nanos(); return self.vdso_time_nanos();
} }
// 使用缓存的时间 // 使用缓存的时间
let now_mono = self.monotonic_start.elapsed().as_nanos() as u64; let now_mono = self.monotonic_start.elapsed().as_nanos() as u64;
let last_update = self.last_update.load(Ordering::Relaxed); let last_update = self.last_update.load(Ordering::Relaxed);
if now_mono.saturating_sub(last_update) > self.cache_update_interval_ns { if now_mono.saturating_sub(last_update) > self.cache_update_interval_ns {
// 需要更新缓存 // 需要更新缓存
self.update_time_cache(); self.update_time_cache();
} }
self.time_cache.load(Ordering::Relaxed) self.time_cache.load(Ordering::Relaxed)
} }
/// vDSO时间获取 /// vDSO时间获取
#[inline(always)] #[inline(always)]
fn vdso_time_nanos(&self) -> u64 { fn vdso_time_nanos(&self) -> u64 {
@@ -132,36 +146,34 @@ impl FastTimeProvider {
// 在Linux上使用vDSO获取时间,避免系统调用 // 在Linux上使用vDSO获取时间,避免系统调用
unsafe { unsafe {
let mut ts = libc::timespec { tv_sec: 0, tv_nsec: 0 }; let mut ts = libc::timespec { tv_sec: 0, tv_nsec: 0 };
// CLOCK_MONOTONIC_RAW不受NTP调整影响,更适合性能测量 // CLOCK_MONOTONIC_RAW不受NTP调整影响,更适合性能测量
if libc::clock_gettime(libc::CLOCK_MONOTONIC_RAW, &mut ts) == 0 { if libc::clock_gettime(libc::CLOCK_MONOTONIC_RAW, &mut ts) == 0 {
return (ts.tv_sec as u64) * 1_000_000_000 + (ts.tv_nsec as u64); return (ts.tv_sec as u64) * 1_000_000_000 + (ts.tv_nsec as u64);
} }
} }
} }
// 回退到缓存时间 // 回退到缓存时间
self.time_cache.load(Ordering::Relaxed) self.time_cache.load(Ordering::Relaxed)
} }
/// 更新时间缓存 /// 更新时间缓存
fn update_time_cache(&self) { fn update_time_cache(&self) {
if let Ok(now) = SystemTime::now().duration_since(UNIX_EPOCH) { if let Ok(now) = SystemTime::now().duration_since(UNIX_EPOCH) {
let nanos = now.as_nanos() as u64; let nanos = now.as_nanos() as u64;
self.time_cache.store(nanos, Ordering::Relaxed); self.time_cache.store(nanos, Ordering::Relaxed);
self.last_update.store( self.last_update
self.monotonic_start.elapsed().as_nanos() as u64, .store(self.monotonic_start.elapsed().as_nanos() as u64, Ordering::Relaxed);
Ordering::Relaxed
);
} }
} }
/// 🚀 快速获取微秒时间戳 /// 🚀 快速获取微秒时间戳
#[inline(always)] #[inline(always)]
pub fn fast_now_micros(&self) -> u64 { pub fn fast_now_micros(&self) -> u64 {
self.fast_now_nanos() / 1000 self.fast_now_nanos() / 1000
} }
/// 🚀 快速获取毫秒时间戳 /// 🚀 快速获取毫秒时间戳
#[inline(always)] #[inline(always)]
pub fn fast_now_millis(&self) -> u64 { pub fn fast_now_millis(&self) -> u64 {
@@ -200,16 +212,16 @@ impl IOOptimizer {
/// 创建I/O优化器 /// 创建I/O优化器
pub fn new(_config: &SyscallBypassConfig) -> Result<Self> { pub fn new(_config: &SyscallBypassConfig) -> Result<Self> {
let io_uring_available = Self::check_io_uring_support(); let io_uring_available = Self::check_io_uring_support();
tracing::info!(target: "sol_trade_sdk","🚀 I/O Optimizer initialized - io_uring: {}", io_uring_available); tracing::info!(target: "sol_trade_sdk","🚀 I/O Optimizer initialized - io_uring: {}", io_uring_available);
Ok(Self { Ok(Self {
io_uring_available, io_uring_available,
async_io_stats: Arc::new(AsyncIOStats::default()), async_io_stats: Arc::new(AsyncIOStats::default()),
mmap_regions: Vec::new(), mmap_regions: Vec::new(),
}) })
} }
/// 检查io_uring支持 /// 检查io_uring支持
fn check_io_uring_support() -> bool { fn check_io_uring_support() -> bool {
#[cfg(target_os = "linux")] #[cfg(target_os = "linux")]
@@ -218,7 +230,7 @@ impl IOOptimizer {
if let Ok(uname) = std::process::Command::new("uname").arg("-r").output() { if let Ok(uname) = std::process::Command::new("uname").arg("-r").output() {
let kernel_version = String::from_utf8_lossy(&uname.stdout); let kernel_version = String::from_utf8_lossy(&uname.stdout);
tracing::info!(target: "sol_trade_sdk","Kernel version: {}", kernel_version.trim()); tracing::info!(target: "sol_trade_sdk","Kernel version: {}", kernel_version.trim());
// 简单检查:内核版本 >= 5.1 支持io_uring // 简单检查:内核版本 >= 5.1 支持io_uring
if let Some(version_str) = kernel_version.split('.').next() { if let Some(version_str) = kernel_version.split('.').next() {
if let Ok(major_version) = version_str.parse::<u32>() { if let Ok(major_version) = version_str.parse::<u32>() {
@@ -227,60 +239,62 @@ impl IOOptimizer {
} }
} }
} }
false false
} }
/// 🚀 批量异步写入 - 绕过多次系统调用 /// 🚀 批量异步写入 - 绕过多次系统调用
#[inline(always)] #[inline(always)]
pub async fn batch_async_write(&self, requests: &[(i32, &[u8])]) -> Result<Vec<usize>> { pub async fn batch_async_write(&self, requests: &[(i32, &[u8])]) -> Result<Vec<usize>> {
if self.io_uring_available && requests.len() > 1 { if self.io_uring_available && requests.len() > 1 {
return self.io_uring_batch_write(requests).await; return self.io_uring_batch_write(requests).await;
} }
// 回退到标准批量写入 // 回退到标准批量写入
self.standard_batch_write(requests).await self.standard_batch_write(requests).await
} }
/// 使用io_uring进行批量写入 /// 使用io_uring进行批量写入
async fn io_uring_batch_write(&self, requests: &[(i32, &[u8])]) -> Result<Vec<usize>> { async fn io_uring_batch_write(&self, requests: &[(i32, &[u8])]) -> Result<Vec<usize>> {
// 这里是伪代码 - 实际实现需要io_uring库 // 这里是伪代码 - 实际实现需要io_uring库
tracing::trace!(target: "sol_trade_sdk","Using io_uring for {} write operations", requests.len()); tracing::trace!(target: "sol_trade_sdk","Using io_uring for {} write operations", requests.len());
let mut results = Vec::with_capacity(requests.len()); let mut results = Vec::with_capacity(requests.len());
// 模拟批量提交到io_uring // 模拟批量提交到io_uring
for (_fd, data) in requests { for (_fd, data) in requests {
self.async_io_stats.operations_queued.fetch_add(1, Ordering::Relaxed); self.async_io_stats.operations_queued.fetch_add(1, Ordering::Relaxed);
// 实际的io_uring实现会在这里提交所有操作 // 实际的io_uring实现会在这里提交所有操作
// 然后等待完成,避免多次系统调用 // 然后等待完成,避免多次系统调用
results.push(data.len()); // 模拟写入成功 results.push(data.len()); // 模拟写入成功
self.async_io_stats.bytes_transferred.fetch_add(data.len() as u64, Ordering::Relaxed); self.async_io_stats.bytes_transferred.fetch_add(data.len() as u64, Ordering::Relaxed);
self.async_io_stats.operations_completed.fetch_add(1, Ordering::Relaxed); self.async_io_stats.operations_completed.fetch_add(1, Ordering::Relaxed);
} }
// 这是一个系统调用而不是N个 // 这是一个系统调用而不是N个
self.async_io_stats.syscalls_avoided.fetch_add(requests.len() as u64 - 1, Ordering::Relaxed); self.async_io_stats
.syscalls_avoided
.fetch_add(requests.len() as u64 - 1, Ordering::Relaxed);
Ok(results) Ok(results)
} }
/// 标准批量写入 /// 标准批量写入
async fn standard_batch_write(&self, requests: &[(i32, &[u8])]) -> Result<Vec<usize>> { async fn standard_batch_write(&self, requests: &[(i32, &[u8])]) -> Result<Vec<usize>> {
let mut results = Vec::with_capacity(requests.len()); let mut results = Vec::with_capacity(requests.len());
// 将所有写入打包成一个写操作 // 将所有写入打包成一个写操作
for (_fd, data) in requests { for (_fd, data) in requests {
// 模拟写入操作 // 模拟写入操作
results.push(data.len()); results.push(data.len());
self.async_io_stats.bytes_transferred.fetch_add(data.len() as u64, Ordering::Relaxed); self.async_io_stats.bytes_transferred.fetch_add(data.len() as u64, Ordering::Relaxed);
} }
Ok(results) Ok(results)
} }
/// 🚀 内存映射文件I/O - 避免read/write系统调用 /// 🚀 内存映射文件I/O - 避免read/write系统调用
pub fn create_memory_mapped_io(&mut self, file_path: &str, size: usize) -> Result<usize> { pub fn create_memory_mapped_io(&mut self, file_path: &str, size: usize) -> Result<usize> {
#[cfg(unix)] #[cfg(unix)]
@@ -298,16 +312,12 @@ impl IOOptimizer {
.custom_flags(libc::O_DIRECT) // 直接I/O,绕过页面缓存 .custom_flags(libc::O_DIRECT) // 直接I/O,绕过页面缓存
.open(file_path)? .open(file_path)?
}; };
#[cfg(not(target_os = "linux"))] #[cfg(not(target_os = "linux"))]
let file = OpenOptions::new() let file = OpenOptions::new().read(true).write(true).create(true).open(file_path)?;
.read(true)
.write(true)
.create(true)
.open(file_path)?;
let fd = file.as_raw_fd(); let fd = file.as_raw_fd();
unsafe { unsafe {
let addr = libc::mmap( let addr = libc::mmap(
std::ptr::null_mut(), std::ptr::null_mut(),
@@ -317,37 +327,42 @@ impl IOOptimizer {
fd, fd,
0, 0,
); );
if addr == libc::MAP_FAILED { if addr == libc::MAP_FAILED {
return Err(anyhow::anyhow!("Memory mapping failed")); return Err(anyhow::anyhow!("Memory mapping failed"));
} }
let region = MemoryMappedRegion { let region =
address: addr as usize, MemoryMappedRegion { address: addr as usize, size, file_descriptor: fd };
size,
file_descriptor: fd,
};
self.mmap_regions.push(region); self.mmap_regions.push(region);
tracing::info!(target: "sol_trade_sdk","✅ Memory mapped I/O created: {} bytes at {:p}", size, addr); tracing::info!(target: "sol_trade_sdk","✅ Memory mapped I/O created: {} bytes at {:p}", size, addr);
Ok(addr as usize) Ok(addr as usize)
} }
} }
#[cfg(not(unix))] #[cfg(not(unix))]
{ {
Err(anyhow::anyhow!("Memory mapped I/O not supported on this platform")) Err(anyhow::anyhow!("Memory mapped I/O not supported on this platform"))
} }
} }
/// 获取I/O统计 /// 获取I/O统计
pub fn get_stats(&self) -> AsyncIOStats { pub fn get_stats(&self) -> AsyncIOStats {
AsyncIOStats { AsyncIOStats {
operations_queued: AtomicU64::new(self.async_io_stats.operations_queued.load(Ordering::Relaxed)), operations_queued: AtomicU64::new(
operations_completed: AtomicU64::new(self.async_io_stats.operations_completed.load(Ordering::Relaxed)), self.async_io_stats.operations_queued.load(Ordering::Relaxed),
bytes_transferred: AtomicU64::new(self.async_io_stats.bytes_transferred.load(Ordering::Relaxed)), ),
syscalls_avoided: AtomicU64::new(self.async_io_stats.syscalls_avoided.load(Ordering::Relaxed)), operations_completed: AtomicU64::new(
self.async_io_stats.operations_completed.load(Ordering::Relaxed),
),
bytes_transferred: AtomicU64::new(
self.async_io_stats.bytes_transferred.load(Ordering::Relaxed),
),
syscalls_avoided: AtomicU64::new(
self.async_io_stats.syscalls_avoided.load(Ordering::Relaxed),
),
} }
} }
} }
@@ -357,47 +372,46 @@ impl SyscallBatchProcessor {
pub fn new(batch_size: usize) -> Result<Self> { pub fn new(batch_size: usize) -> Result<Self> {
let pending_calls = crossbeam_queue::ArrayQueue::new(batch_size * 10); let pending_calls = crossbeam_queue::ArrayQueue::new(batch_size * 10);
let executor = tokio::runtime::Handle::current(); let executor = tokio::runtime::Handle::current();
tracing::info!(target: "sol_trade_sdk","🚀 Syscall batch processor created with batch size: {}", batch_size); tracing::info!(target: "sol_trade_sdk","🚀 Syscall batch processor created with batch size: {}", batch_size);
Ok(Self { Ok(Self {
pending_calls, pending_calls,
_executor: executor, _executor: executor,
batch_stats: CachePadded::new(AtomicU64::new(0)), batch_stats: CachePadded::new(AtomicU64::new(0)),
}) })
} }
/// 🚀 提交系统调用请求到批处理队列 /// 🚀 提交系统调用请求到批处理队列
#[inline(always)] #[inline(always)]
pub fn submit_request(&self, request: SyscallRequest) -> Result<()> { pub fn submit_request(&self, request: SyscallRequest) -> Result<()> {
self.pending_calls.push(request) self.pending_calls.push(request).map_err(|_| anyhow::anyhow!("Batch queue full"))?;
.map_err(|_| anyhow::anyhow!("Batch queue full"))?;
Ok(()) Ok(())
} }
/// 🚀 执行批量系统调用 /// 🚀 执行批量系统调用
pub async fn execute_batch(&self) -> Result<usize> { pub async fn execute_batch(&self) -> Result<usize> {
let mut batch = Vec::new(); let mut batch = Vec::new();
// 收集批量请求 // 收集批量请求
while batch.len() < 100 && !self.pending_calls.is_empty() { while batch.len() < 100 && !self.pending_calls.is_empty() {
if let Some(request) = self.pending_calls.pop() { if let Some(request) = self.pending_calls.pop() {
batch.push(request); batch.push(request);
} }
} }
if batch.is_empty() { if batch.is_empty() {
return Ok(0); return Ok(0);
} }
let batch_size = batch.len(); let batch_size = batch.len();
// 按类型分组批量执行 // 按类型分组批量执行
let mut write_requests = Vec::new(); let mut write_requests = Vec::new();
let mut read_requests = Vec::new(); let mut read_requests = Vec::new();
let mut network_requests = Vec::new(); let mut network_requests = Vec::new();
for request in batch { for request in batch {
match request { match request {
SyscallRequest::Write { fd, data } => { SyscallRequest::Write { fd, data } => {
@@ -414,28 +428,28 @@ impl SyscallBatchProcessor {
} }
} }
} }
// 批量执行写入 // 批量执行写入
if !write_requests.is_empty() { if !write_requests.is_empty() {
self.batch_write_operations(write_requests).await?; self.batch_write_operations(write_requests).await?;
} }
// 批量执行读取 // 批量执行读取
if !read_requests.is_empty() { if !read_requests.is_empty() {
self.batch_read_operations(read_requests).await?; self.batch_read_operations(read_requests).await?;
} }
// 批量执行网络操作 // 批量执行网络操作
if !network_requests.is_empty() { if !network_requests.is_empty() {
self.batch_network_operations(network_requests).await?; self.batch_network_operations(network_requests).await?;
} }
self.batch_stats.fetch_add(1, Ordering::Relaxed); self.batch_stats.fetch_add(1, Ordering::Relaxed);
tracing::trace!(target: "sol_trade_sdk","Executed batch of {} syscalls", batch_size); tracing::trace!(target: "sol_trade_sdk","Executed batch of {} syscalls", batch_size);
Ok(batch_size) Ok(batch_size)
} }
/// 批量写入操作 /// 批量写入操作
async fn batch_write_operations(&self, requests: Vec<(i32, Vec<u8>)>) -> Result<()> { async fn batch_write_operations(&self, requests: Vec<(i32, Vec<u8>)>) -> Result<()> {
// 使用writev系统调用进行批量写入 // 使用writev系统调用进行批量写入
@@ -445,7 +459,7 @@ impl SyscallBatchProcessor {
} }
Ok(()) Ok(())
} }
/// 批量读取操作 /// 批量读取操作
async fn batch_read_operations(&self, requests: Vec<(i32, usize)>) -> Result<()> { async fn batch_read_operations(&self, requests: Vec<(i32, usize)>) -> Result<()> {
// 使用readv系统调用进行批量读取 // 使用readv系统调用进行批量读取
@@ -454,7 +468,7 @@ impl SyscallBatchProcessor {
} }
Ok(()) Ok(())
} }
/// 批量网络操作 /// 批量网络操作
async fn batch_network_operations(&self, requests: Vec<(i32, Vec<u8>)>) -> Result<()> { async fn batch_network_operations(&self, requests: Vec<(i32, Vec<u8>)>) -> Result<()> {
// 使用sendmsg/recvmsg进行批量网络操作 // 使用sendmsg/recvmsg进行批量网络操作
@@ -482,22 +496,16 @@ impl SystemCallBypassManager {
let fast_time_provider = Arc::new(FastTimeProvider::new(config.enable_vdso)?); let fast_time_provider = Arc::new(FastTimeProvider::new(config.enable_vdso)?);
let io_optimizer = Arc::new(IOOptimizer::new(&config)?); let io_optimizer = Arc::new(IOOptimizer::new(&config)?);
let stats = Arc::new(SyscallBypassStats::default()); let stats = Arc::new(SyscallBypassStats::default());
tracing::info!(target: "sol_trade_sdk","🚀 System Call Bypass Manager initialized"); tracing::info!(target: "sol_trade_sdk","🚀 System Call Bypass Manager initialized");
tracing::info!(target: "sol_trade_sdk"," 📦 Batch Processing: {}", config.enable_batch_processing); tracing::info!(target: "sol_trade_sdk"," 📦 Batch Processing: {}", config.enable_batch_processing);
tracing::info!(target: "sol_trade_sdk"," ⏰ Fast Time: {}", config.enable_fast_time); tracing::info!(target: "sol_trade_sdk"," ⏰ Fast Time: {}", config.enable_fast_time);
tracing::info!(target: "sol_trade_sdk"," 🚀 vDSO: {}", config.enable_vdso); tracing::info!(target: "sol_trade_sdk"," 🚀 vDSO: {}", config.enable_vdso);
tracing::info!(target: "sol_trade_sdk"," 📁 io_uring: {}", config.enable_io_uring); tracing::info!(target: "sol_trade_sdk"," 📁 io_uring: {}", config.enable_io_uring);
Ok(Self { Ok(Self { config, batch_processor, fast_time_provider, _io_optimizer: io_optimizer, stats })
config,
batch_processor,
fast_time_provider,
_io_optimizer: io_optimizer,
stats,
})
} }
/// 🚀 快速获取当前时间戳 - 绕过系统调用 /// 🚀 快速获取当前时间戳 - 绕过系统调用
#[inline(always)] #[inline(always)]
pub fn fast_timestamp_nanos(&self) -> u64 { pub fn fast_timestamp_nanos(&self) -> u64 {
@@ -505,28 +513,25 @@ impl SystemCallBypassManager {
self.stats.time_calls_cached.fetch_add(1, Ordering::Relaxed); self.stats.time_calls_cached.fetch_add(1, Ordering::Relaxed);
return self.fast_time_provider.fast_now_nanos(); return self.fast_time_provider.fast_now_nanos();
} }
// 回退到标准时间获取 // 回退到标准时间获取
SystemTime::now() SystemTime::now().duration_since(UNIX_EPOCH).unwrap_or_default().as_nanos() as u64
.duration_since(UNIX_EPOCH)
.unwrap_or_default()
.as_nanos() as u64
} }
/// 🚀 提交批量I/O操作 /// 🚀 提交批量I/O操作
pub async fn submit_batch_io(&self, operations: Vec<SyscallRequest>) -> Result<()> { pub async fn submit_batch_io(&self, operations: Vec<SyscallRequest>) -> Result<()> {
if !self.config.enable_batch_processing { if !self.config.enable_batch_processing {
return Err(anyhow::anyhow!("Batch processing disabled")); return Err(anyhow::anyhow!("Batch processing disabled"));
} }
for op in operations { for op in operations {
self.batch_processor.submit_request(op)?; self.batch_processor.submit_request(op)?;
} }
self.stats.syscalls_batched.fetch_add(1, Ordering::Relaxed); self.stats.syscalls_batched.fetch_add(1, Ordering::Relaxed);
Ok(()) Ok(())
} }
/// 🚀 执行优化的内存分配 - 绕过malloc系统调用 /// 🚀 执行优化的内存分配 - 绕过malloc系统调用
#[inline(always)] #[inline(always)]
pub fn fast_allocate(&self, size: usize) -> Result<*mut u8> { pub fn fast_allocate(&self, size: usize) -> Result<*mut u8> {
@@ -534,34 +539,30 @@ impl SystemCallBypassManager {
self.stats.memory_operations_avoided.fetch_add(1, Ordering::Relaxed); self.stats.memory_operations_avoided.fetch_add(1, Ordering::Relaxed);
return self.userspace_allocate(size); return self.userspace_allocate(size);
} }
// 回退到标准分配 // 回退到标准分配
let layout = std::alloc::Layout::from_size_align(size, 8)?; let layout = std::alloc::Layout::from_size_align(size, 8)?;
let ptr = unsafe { std::alloc::alloc(layout) }; let ptr = unsafe { std::alloc::alloc(layout) };
if ptr.is_null() { if ptr.is_null() {
Err(anyhow::anyhow!("Allocation failed")) Err(anyhow::anyhow!("Allocation failed"))
} else { } else {
Ok(ptr) Ok(ptr)
} }
} }
/// 用户空间内存分配 /// 用户空间内存分配
fn userspace_allocate(&self, size: usize) -> Result<*mut u8> { fn userspace_allocate(&self, size: usize) -> Result<*mut u8> {
use std::sync::Mutex;
use once_cell::sync::Lazy; use once_cell::sync::Lazy;
use std::sync::Mutex;
struct MemoryPool { struct MemoryPool {
pool: Box<[u8; 1024 * 1024]>, pool: Box<[u8; 1024 * 1024]>,
offset: usize, offset: usize,
} }
static MEMORY_POOL: Lazy<Mutex<MemoryPool>> = Lazy::new(|| { static MEMORY_POOL: Lazy<Mutex<MemoryPool>> =
Mutex::new(MemoryPool { Lazy::new(|| Mutex::new(MemoryPool { pool: Box::new([0; 1024 * 1024]), offset: 0 }));
pool: Box::new([0; 1024 * 1024]),
offset: 0,
})
});
let mut pool = MEMORY_POOL.lock().unwrap(); let mut pool = MEMORY_POOL.lock().unwrap();
@@ -574,18 +575,18 @@ impl SystemCallBypassManager {
Ok(ptr) Ok(ptr)
} }
/// 启动批处理工作线程 /// 启动批处理工作线程
pub async fn start_batch_processing(&self) -> Result<()> { pub async fn start_batch_processing(&self) -> Result<()> {
let processor = Arc::clone(&self.batch_processor); let processor = Arc::clone(&self.batch_processor);
let stats = Arc::clone(&self.stats); let stats = Arc::clone(&self.stats);
tokio::spawn(async move { tokio::spawn(async move {
let mut interval = tokio::time::interval(Duration::from_micros(100)); // 100μs间隔 let mut interval = tokio::time::interval(Duration::from_micros(100)); // 100μs间隔
loop { loop {
interval.tick().await; interval.tick().await;
if let Ok(processed) = processor.execute_batch().await { if let Ok(processed) = processor.execute_batch().await {
if processed > 0 { if processed > 0 {
stats.syscalls_bypassed.fetch_add(processed as u64, Ordering::Relaxed); stats.syscalls_bypassed.fetch_add(processed as u64, Ordering::Relaxed);
@@ -593,11 +594,11 @@ impl SystemCallBypassManager {
} }
} }
}); });
tracing::info!(target: "sol_trade_sdk","✅ Batch processing worker started"); tracing::info!(target: "sol_trade_sdk","✅ Batch processing worker started");
Ok(()) Ok(())
} }
/// 获取绕过统计 /// 获取绕过统计
pub fn get_bypass_stats(&self) -> SyscallBypassStatsSnapshot { pub fn get_bypass_stats(&self) -> SyscallBypassStatsSnapshot {
SyscallBypassStatsSnapshot { SyscallBypassStatsSnapshot {
@@ -608,7 +609,7 @@ impl SystemCallBypassManager {
memory_operations_avoided: self.stats.memory_operations_avoided.load(Ordering::Relaxed), memory_operations_avoided: self.stats.memory_operations_avoided.load(Ordering::Relaxed),
} }
} }
/// 🚀 极致优化配置 /// 🚀 极致优化配置
pub fn extreme_bypass_config() -> SyscallBypassConfig { pub fn extreme_bypass_config() -> SyscallBypassConfig {
SyscallBypassConfig { SyscallBypassConfig {
@@ -643,9 +644,11 @@ impl SyscallBypassStatsSnapshot {
tracing::info!(target: "sol_trade_sdk"," ⏰ Time Calls Cached: {}", self.time_calls_cached); tracing::info!(target: "sol_trade_sdk"," ⏰ Time Calls Cached: {}", self.time_calls_cached);
tracing::info!(target: "sol_trade_sdk"," 📁 I/O Operations Optimized: {}", self.io_operations_optimized); tracing::info!(target: "sol_trade_sdk"," 📁 I/O Operations Optimized: {}", self.io_operations_optimized);
tracing::info!(target: "sol_trade_sdk"," 💾 Memory Operations Avoided: {}", self.memory_operations_avoided); tracing::info!(target: "sol_trade_sdk"," 💾 Memory Operations Avoided: {}", self.memory_operations_avoided);
let total_optimizations = self.syscalls_bypassed + self.time_calls_cached + let total_optimizations = self.syscalls_bypassed
self.io_operations_optimized + self.memory_operations_avoided; + self.time_calls_cached
+ self.io_operations_optimized
+ self.memory_operations_avoided;
tracing::info!(target: "sol_trade_sdk"," 🏆 Total Optimizations: {}", total_optimizations); tracing::info!(target: "sol_trade_sdk"," 🏆 Total Optimizations: {}", total_optimizations);
} }
} }
@@ -657,7 +660,7 @@ macro_rules! bypass_syscall {
// 使用快速时间而不是系统调用 // 使用快速时间而不是系统调用
crate::performance::syscall_bypass::GLOBAL_TIME_PROVIDER.fast_now_nanos() crate::performance::syscall_bypass::GLOBAL_TIME_PROVIDER.fast_now_nanos()
}; };
(batch_io $ops:expr) => { (batch_io $ops:expr) => {
// 批量提交I/O操作 // 批量提交I/O操作
crate::performance::syscall_bypass::GLOBAL_BYPASS_MANAGER.submit_batch_io($ops).await crate::performance::syscall_bypass::GLOBAL_BYPASS_MANAGER.submit_batch_io($ops).await
@@ -667,60 +670,57 @@ macro_rules! bypass_syscall {
#[cfg(test)] #[cfg(test)]
mod tests { mod tests {
use super::*; use super::*;
#[tokio::test] #[tokio::test]
async fn test_fast_time_provider() { async fn test_fast_time_provider() {
let provider = FastTimeProvider::new(false).unwrap(); let provider = FastTimeProvider::new(false).unwrap();
let time1 = provider.fast_now_nanos(); let time1 = provider.fast_now_nanos();
tokio::time::sleep(Duration::from_millis(1)).await; tokio::time::sleep(Duration::from_millis(1)).await;
let time2 = provider.fast_now_nanos(); let time2 = provider.fast_now_nanos();
assert!(time2 > time1); assert!(time2 > time1);
assert!(time2 - time1 >= 1_000_000); // 至少1ms差异 assert!(time2 - time1 >= 1_000_000); // 至少1ms差异
} }
#[tokio::test] #[tokio::test]
async fn test_syscall_batch_processor() { async fn test_syscall_batch_processor() {
let processor = SyscallBatchProcessor::new(10).unwrap(); let processor = SyscallBatchProcessor::new(10).unwrap();
let request = SyscallRequest::Write { let request = SyscallRequest::Write { fd: 1, data: vec![1, 2, 3, 4, 5] };
fd: 1,
data: vec![1, 2, 3, 4, 5],
};
processor.submit_request(request).unwrap(); processor.submit_request(request).unwrap();
let processed = processor.execute_batch().await.unwrap(); let processed = processor.execute_batch().await.unwrap();
assert_eq!(processed, 1); assert_eq!(processed, 1);
} }
#[tokio::test] #[tokio::test]
async fn test_io_optimizer() { async fn test_io_optimizer() {
let config = SyscallBypassConfig::default(); let config = SyscallBypassConfig::default();
let optimizer = IOOptimizer::new(&config).unwrap(); let optimizer = IOOptimizer::new(&config).unwrap();
let requests = vec![(1, b"test data".as_ref())]; let requests = vec![(1, b"test data".as_ref())];
let results = optimizer.batch_async_write(&requests).await.unwrap(); let results = optimizer.batch_async_write(&requests).await.unwrap();
assert_eq!(results.len(), 1); assert_eq!(results.len(), 1);
assert_eq!(results[0], 9); // "test data".len() assert_eq!(results[0], 9); // "test data".len()
} }
#[tokio::test] #[tokio::test]
async fn test_system_call_bypass_manager() { async fn test_system_call_bypass_manager() {
let config = SyscallBypassConfig::default(); let config = SyscallBypassConfig::default();
let manager = SystemCallBypassManager::new(config).unwrap(); let manager = SystemCallBypassManager::new(config).unwrap();
// 测试快速时间戳 // 测试快速时间戳
let timestamp = manager.fast_timestamp_nanos(); let timestamp = manager.fast_timestamp_nanos();
assert!(timestamp > 0); assert!(timestamp > 0);
// 测试统计 // 测试统计
let stats = manager.get_bypass_stats(); let stats = manager.get_bypass_stats();
assert_eq!(stats.time_calls_cached, 1); assert_eq!(stats.time_calls_cached, 1);
} }
#[test] #[test]
fn test_extreme_bypass_config() { fn test_extreme_bypass_config() {
let config = SystemCallBypassManager::extreme_bypass_config(); let config = SystemCallBypassManager::extreme_bypass_config();
@@ -731,16 +731,16 @@ mod tests {
assert_eq!(config.batch_size, 1000); assert_eq!(config.batch_size, 1000);
assert_eq!(config.syscall_cache_size, 10000); assert_eq!(config.syscall_cache_size, 10000);
} }
#[test] #[test]
fn test_userspace_allocation() { fn test_userspace_allocation() {
let config = SyscallBypassConfig::default(); let config = SyscallBypassConfig::default();
let manager = SystemCallBypassManager::new(config).unwrap(); let manager = SystemCallBypassManager::new(config).unwrap();
let ptr = manager.fast_allocate(64).unwrap(); let ptr = manager.fast_allocate(64).unwrap();
assert!(!ptr.is_null()); assert!(!ptr.is_null());
let stats = manager.get_bypass_stats(); let stats = manager.get_bypass_stats();
assert_eq!(stats.memory_operations_avoided, 1); assert_eq!(stats.memory_operations_avoided, 1);
} }
} }
+133 -126
View File
@@ -1,5 +1,5 @@
//! 🚀 零拷贝内存映射IO - 完全消除数据拷贝开销 //! 🚀 零拷贝内存映射IO - 完全消除数据拷贝开销
//! //!
//! 实现极致的零拷贝策略,包括: //! 实现极致的零拷贝策略,包括:
//! - 内存映射文件IO //! - 内存映射文件IO
//! - 共享内存环形缓冲区 //! - 共享内存环形缓冲区
@@ -10,11 +10,11 @@
use std::sync::atomic::{AtomicU64, AtomicUsize, Ordering}; use std::sync::atomic::{AtomicU64, AtomicUsize, Ordering};
use std::sync::Arc; use std::sync::Arc;
// use std::mem::{size_of, MaybeUninit}; // use std::mem::{size_of, MaybeUninit};
use anyhow::{Context, Result};
use crossbeam_utils::CachePadded;
use memmap2::{MmapMut, MmapOptions};
use std::ptr::NonNull; use std::ptr::NonNull;
use std::slice; use std::slice;
use memmap2::{MmapMut, MmapOptions};
use anyhow::{Result, Context};
use crossbeam_utils::CachePadded;
/// 🚀 零拷贝内存管理器 /// 🚀 零拷贝内存管理器
pub struct ZeroCopyMemoryManager { pub struct ZeroCopyMemoryManager {
@@ -50,17 +50,17 @@ impl SharedMemoryPool {
// 确保块大小是64字节对齐(缓存行对齐) // 确保块大小是64字节对齐(缓存行对齐)
let aligned_block_size = (block_size + 63) & !63; let aligned_block_size = (block_size + 63) & !63;
let total_blocks = total_size / aligned_block_size; let total_blocks = total_size / aligned_block_size;
// 创建内存映射文件 // 创建内存映射文件
let memory_region = MmapOptions::new() let memory_region = MmapOptions::new()
.len(total_blocks * aligned_block_size) .len(total_blocks * aligned_block_size)
.map_anon() .map_anon()
.context("Failed to create memory mapped region")?; .context("Failed to create memory mapped region")?;
// 初始化空闲块位图 (每个u64可以管理64个块) // 初始化空闲块位图 (每个u64可以管理64个块)
let bitmap_size = (total_blocks + 63) / 64; let bitmap_size = (total_blocks + 63) / 64;
let mut free_blocks = Vec::with_capacity(bitmap_size); let mut free_blocks = Vec::with_capacity(bitmap_size);
// 将所有块标记为空闲(全1) // 将所有块标记为空闲(全1)
for i in 0..bitmap_size { for i in 0..bitmap_size {
let bits = if i == bitmap_size - 1 && total_blocks % 64 != 0 { let bits = if i == bitmap_size - 1 && total_blocks % 64 != 0 {
@@ -72,10 +72,10 @@ impl SharedMemoryPool {
}; };
free_blocks.push(AtomicU64::new(bits)); free_blocks.push(AtomicU64::new(bits));
} }
tracing::info!(target: "sol_trade_sdk","🚀 Created shared memory pool {} with {} blocks of {} bytes each", tracing::info!(target: "sol_trade_sdk","🚀 Created shared memory pool {} with {} blocks of {} bytes each",
pool_id, total_blocks, aligned_block_size); pool_id, total_blocks, aligned_block_size);
Ok(Self { Ok(Self {
memory_region, memory_region,
free_blocks, free_blocks,
@@ -85,31 +85,31 @@ impl SharedMemoryPool {
pool_id, pool_id,
}) })
} }
/// 🚀 零拷贝分配内存块 /// 🚀 零拷贝分配内存块
#[inline(always)] #[inline(always)]
pub fn allocate_block(&self) -> Option<ZeroCopyBlock> { pub fn allocate_block(&self) -> Option<ZeroCopyBlock> {
// 快速路径:尝试从预期位置分配 // 快速路径:尝试从预期位置分配
let start_index = self.allocator_head.load(Ordering::Relaxed) / 64; let start_index = self.allocator_head.load(Ordering::Relaxed) / 64;
// 遍历所有位图寻找空闲块 // 遍历所有位图寻找空闲块
for attempt in 0..self.free_blocks.len() { for attempt in 0..self.free_blocks.len() {
let bitmap_index = (start_index + attempt) % self.free_blocks.len(); let bitmap_index = (start_index + attempt) % self.free_blocks.len();
let bitmap = &self.free_blocks[bitmap_index]; let bitmap = &self.free_blocks[bitmap_index];
let mut current = bitmap.load(Ordering::Acquire); let mut current = bitmap.load(Ordering::Acquire);
while current != 0 { while current != 0 {
// 找到最低位的1(最小的空闲块) // 找到最低位的1(最小的空闲块)
let bit_pos = current.trailing_zeros() as usize; let bit_pos = current.trailing_zeros() as usize;
let mask = 1u64 << bit_pos; let mask = 1u64 << bit_pos;
// 尝试原子地清除这一位(标记为已分配) // 尝试原子地清除这一位(标记为已分配)
match bitmap.compare_exchange_weak( match bitmap.compare_exchange_weak(
current, current,
current & !mask, current & !mask,
Ordering::AcqRel, Ordering::AcqRel,
Ordering::Relaxed Ordering::Relaxed,
) { ) {
Ok(_) => { Ok(_) => {
// 成功分配 // 成功分配
@@ -119,20 +119,17 @@ impl SharedMemoryPool {
bitmap.fetch_or(mask, Ordering::Relaxed); bitmap.fetch_or(mask, Ordering::Relaxed);
break; break;
} }
let offset = block_index * self.block_size; let offset = block_index * self.block_size;
let ptr = unsafe { let ptr = unsafe {
NonNull::new_unchecked( NonNull::new_unchecked(
self.memory_region.as_ptr().add(offset) as *mut u8 self.memory_region.as_ptr().add(offset) as *mut u8
) )
}; };
// 更新分配器头指针 // 更新分配器头指针
self.allocator_head.store( self.allocator_head.store((block_index + 1) * 64, Ordering::Relaxed);
(block_index + 1) * 64,
Ordering::Relaxed
);
return Some(ZeroCopyBlock { return Some(ZeroCopyBlock {
ptr, ptr,
size: self.block_size, size: self.block_size,
@@ -147,10 +144,10 @@ impl SharedMemoryPool {
} }
} }
} }
None // 没有可用块 None // 没有可用块
} }
/// 🚀 零拷贝释放内存块 /// 🚀 零拷贝释放内存块
#[inline(always)] #[inline(always)]
pub fn deallocate_block(&self, block: ZeroCopyBlock) { pub fn deallocate_block(&self, block: ZeroCopyBlock) {
@@ -158,20 +155,21 @@ impl SharedMemoryPool {
tracing::error!(target: "sol_trade_sdk", "Attempting to deallocate block from wrong pool"); tracing::error!(target: "sol_trade_sdk", "Attempting to deallocate block from wrong pool");
return; return;
} }
let bitmap_index = block.block_index / 64; let bitmap_index = block.block_index / 64;
let bit_pos = block.block_index % 64; let bit_pos = block.block_index % 64;
let mask = 1u64 << bit_pos; let mask = 1u64 << bit_pos;
if bitmap_index < self.free_blocks.len() { if bitmap_index < self.free_blocks.len() {
// 原子地设置位为1(标记为空闲) // 原子地设置位为1(标记为空闲)
self.free_blocks[bitmap_index].fetch_or(mask, Ordering::Release); self.free_blocks[bitmap_index].fetch_or(mask, Ordering::Release);
} }
} }
/// 获取可用块数量 /// 获取可用块数量
pub fn available_blocks(&self) -> usize { pub fn available_blocks(&self) -> usize {
self.free_blocks.iter() self.free_blocks
.iter()
.map(|bitmap| bitmap.load(Ordering::Relaxed).count_ones() as usize) .map(|bitmap| bitmap.load(Ordering::Relaxed).count_ones() as usize)
.sum() .sum()
} }
@@ -195,49 +193,49 @@ impl ZeroCopyBlock {
pub fn as_ptr(&self) -> *mut u8 { pub fn as_ptr(&self) -> *mut u8 {
self.ptr.as_ptr() self.ptr.as_ptr()
} }
/// 获取只读切片 /// 获取只读切片
#[inline(always)] #[inline(always)]
pub unsafe fn as_slice(&self) -> &[u8] { pub unsafe fn as_slice(&self) -> &[u8] {
slice::from_raw_parts(self.ptr.as_ptr(), self.size) slice::from_raw_parts(self.ptr.as_ptr(), self.size)
} }
/// 获取可变切片 /// 获取可变切片
#[inline(always)] #[inline(always)]
pub unsafe fn as_mut_slice(&mut self) -> &mut [u8] { pub unsafe fn as_mut_slice(&mut self) -> &mut [u8] {
slice::from_raw_parts_mut(self.ptr.as_ptr(), self.size) slice::from_raw_parts_mut(self.ptr.as_ptr(), self.size)
} }
/// 获取块大小 /// 获取块大小
#[inline(always)] #[inline(always)]
pub fn size(&self) -> usize { pub fn size(&self) -> usize {
self.size self.size
} }
/// 零拷贝写入数据 /// 零拷贝写入数据
#[inline(always)] #[inline(always)]
pub unsafe fn write_bytes(&mut self, data: &[u8]) -> Result<()> { pub unsafe fn write_bytes(&mut self, data: &[u8]) -> Result<()> {
if data.len() > self.size { if data.len() > self.size {
return Err(anyhow::anyhow!("Data too large for block")); return Err(anyhow::anyhow!("Data too large for block"));
} }
// 使用硬件优化的内存拷贝 // 使用硬件优化的内存拷贝
super::hardware_optimizations::SIMDMemoryOps::memcpy_simd_optimized( super::hardware_optimizations::SIMDMemoryOps::memcpy_simd_optimized(
self.ptr.as_ptr(), self.ptr.as_ptr(),
data.as_ptr(), data.as_ptr(),
data.len() data.len(),
); );
Ok(()) Ok(())
} }
/// 零拷贝读取数据 /// 零拷贝读取数据
#[inline(always)] #[inline(always)]
pub unsafe fn read_bytes(&self, len: usize) -> Result<&[u8]> { pub unsafe fn read_bytes(&self, len: usize) -> Result<&[u8]> {
if len > self.size { if len > self.size {
return Err(anyhow::anyhow!("Read length exceeds block size")); return Err(anyhow::anyhow!("Read length exceeds block size"));
} }
Ok(slice::from_raw_parts(self.ptr.as_ptr(), len)) Ok(slice::from_raw_parts(self.ptr.as_ptr(), len))
} }
} }
@@ -266,9 +264,9 @@ impl MemoryMappedBuffer {
.len(size) .len(size)
.map_anon() .map_anon()
.context("Failed to create memory mapped buffer")?; .context("Failed to create memory mapped buffer")?;
tracing::info!(target: "sol_trade_sdk","🚀 Created memory mapped buffer {} with size {} bytes", buffer_id, size); tracing::info!(target: "sol_trade_sdk","🚀 Created memory mapped buffer {} with size {} bytes", buffer_id, size);
Ok(Self { Ok(Self {
mmap, mmap,
read_pos: CachePadded::new(AtomicUsize::new(0)), read_pos: CachePadded::new(AtomicUsize::new(0)),
@@ -277,128 +275,136 @@ impl MemoryMappedBuffer {
_buffer_id: buffer_id, _buffer_id: buffer_id,
}) })
} }
/// 🚀 零拷贝写入数据 /// 🚀 零拷贝写入数据
#[inline(always)] #[inline(always)]
pub fn write_data(&self, data: &[u8]) -> Result<usize> { pub fn write_data(&self, data: &[u8]) -> Result<usize> {
let data_len = data.len(); let data_len = data.len();
let current_write = self.write_pos.load(Ordering::Relaxed); let current_write = self.write_pos.load(Ordering::Relaxed);
let current_read = self.read_pos.load(Ordering::Acquire); let current_read = self.read_pos.load(Ordering::Acquire);
// 计算可用空间 // 计算可用空间
let available_space = if current_write >= current_read { let available_space = if current_write >= current_read {
self.size - (current_write - current_read) - 1 self.size - (current_write - current_read) - 1
} else { } else {
current_read - current_write - 1 current_read - current_write - 1
}; };
if data_len > available_space { if data_len > available_space {
return Err(anyhow::anyhow!("Insufficient buffer space")); return Err(anyhow::anyhow!("Insufficient buffer space"));
} }
// 零拷贝写入 // 零拷贝写入
unsafe { unsafe {
let write_ptr = self.mmap.as_ptr().add(current_write) as *mut u8; let write_ptr = self.mmap.as_ptr().add(current_write) as *mut u8;
if current_write + data_len <= self.size { if current_write + data_len <= self.size {
// 数据不跨越缓冲区边界 // 数据不跨越缓冲区边界
super::hardware_optimizations::SIMDMemoryOps::memcpy_simd_optimized( super::hardware_optimizations::SIMDMemoryOps::memcpy_simd_optimized(
write_ptr, data.as_ptr(), data_len write_ptr,
data.as_ptr(),
data_len,
); );
} else { } else {
// 数据跨越缓冲区边界,分两段写入 // 数据跨越缓冲区边界,分两段写入
let first_part = self.size - current_write; let first_part = self.size - current_write;
let second_part = data_len - first_part; let second_part = data_len - first_part;
// 写入第一部分 // 写入第一部分
super::hardware_optimizations::SIMDMemoryOps::memcpy_simd_optimized( super::hardware_optimizations::SIMDMemoryOps::memcpy_simd_optimized(
write_ptr, data.as_ptr(), first_part write_ptr,
data.as_ptr(),
first_part,
); );
// 写入第二部分(从缓冲区开头) // 写入第二部分(从缓冲区开头)
super::hardware_optimizations::SIMDMemoryOps::memcpy_simd_optimized( super::hardware_optimizations::SIMDMemoryOps::memcpy_simd_optimized(
self.mmap.as_ptr() as *mut u8, self.mmap.as_ptr() as *mut u8,
data.as_ptr().add(first_part), data.as_ptr().add(first_part),
second_part second_part,
); );
} }
} }
// 更新写指针 // 更新写指针
let new_write_pos = (current_write + data_len) % self.size; let new_write_pos = (current_write + data_len) % self.size;
self.write_pos.store(new_write_pos, Ordering::Release); self.write_pos.store(new_write_pos, Ordering::Release);
Ok(data_len) Ok(data_len)
} }
/// 🚀 零拷贝读取数据 /// 🚀 零拷贝读取数据
#[inline(always)] #[inline(always)]
pub fn read_data(&self, buffer: &mut [u8]) -> Result<usize> { pub fn read_data(&self, buffer: &mut [u8]) -> Result<usize> {
let buffer_len = buffer.len(); let buffer_len = buffer.len();
let current_read = self.read_pos.load(Ordering::Relaxed); let current_read = self.read_pos.load(Ordering::Relaxed);
let current_write = self.write_pos.load(Ordering::Acquire); let current_write = self.write_pos.load(Ordering::Acquire);
// 计算可读数据量 // 计算可读数据量
let available_data = if current_write >= current_read { let available_data = if current_write >= current_read {
current_write - current_read current_write - current_read
} else { } else {
self.size - (current_read - current_write) self.size - (current_read - current_write)
}; };
if available_data == 0 { if available_data == 0 {
return Ok(0); // 无数据可读 return Ok(0); // 无数据可读
} }
let read_len = buffer_len.min(available_data); let read_len = buffer_len.min(available_data);
// 零拷贝读取 // 零拷贝读取
unsafe { unsafe {
let read_ptr = self.mmap.as_ptr().add(current_read); let read_ptr = self.mmap.as_ptr().add(current_read);
if current_read + read_len <= self.size { if current_read + read_len <= self.size {
// 数据不跨越缓冲区边界 // 数据不跨越缓冲区边界
super::hardware_optimizations::SIMDMemoryOps::memcpy_simd_optimized( super::hardware_optimizations::SIMDMemoryOps::memcpy_simd_optimized(
buffer.as_mut_ptr(), read_ptr, read_len buffer.as_mut_ptr(),
read_ptr,
read_len,
); );
} else { } else {
// 数据跨越缓冲区边界,分两段读取 // 数据跨越缓冲区边界,分两段读取
let first_part = self.size - current_read; let first_part = self.size - current_read;
let second_part = read_len - first_part; let second_part = read_len - first_part;
// 读取第一部分 // 读取第一部分
super::hardware_optimizations::SIMDMemoryOps::memcpy_simd_optimized( super::hardware_optimizations::SIMDMemoryOps::memcpy_simd_optimized(
buffer.as_mut_ptr(), read_ptr, first_part buffer.as_mut_ptr(),
read_ptr,
first_part,
); );
// 读取第二部分(从缓冲区开头) // 读取第二部分(从缓冲区开头)
super::hardware_optimizations::SIMDMemoryOps::memcpy_simd_optimized( super::hardware_optimizations::SIMDMemoryOps::memcpy_simd_optimized(
buffer.as_mut_ptr().add(first_part), buffer.as_mut_ptr().add(first_part),
self.mmap.as_ptr(), self.mmap.as_ptr(),
second_part second_part,
); );
} }
} }
// 更新读指针 // 更新读指针
let new_read_pos = (current_read + read_len) % self.size; let new_read_pos = (current_read + read_len) % self.size;
self.read_pos.store(new_read_pos, Ordering::Release); self.read_pos.store(new_read_pos, Ordering::Release);
Ok(read_len) Ok(read_len)
} }
/// 获取可读数据量 /// 获取可读数据量
#[inline(always)] #[inline(always)]
pub fn available_data(&self) -> usize { pub fn available_data(&self) -> usize {
let current_read = self.read_pos.load(Ordering::Relaxed); let current_read = self.read_pos.load(Ordering::Relaxed);
let current_write = self.write_pos.load(Ordering::Relaxed); let current_write = self.write_pos.load(Ordering::Relaxed);
if current_write >= current_read { if current_write >= current_read {
current_write - current_read current_write - current_read
} else { } else {
self.size - (current_read - current_write) self.size - (current_read - current_write)
} }
} }
/// 获取可用空间 /// 获取可用空间
#[inline(always)] #[inline(always)]
pub fn available_space(&self) -> usize { pub fn available_space(&self) -> usize {
@@ -420,38 +426,39 @@ impl DirectMemoryAccessManager {
/// 创建DMA管理器 /// 创建DMA管理器
pub fn new(num_channels: usize) -> Result<Self> { pub fn new(num_channels: usize) -> Result<Self> {
let mut dma_channels = Vec::with_capacity(num_channels); let mut dma_channels = Vec::with_capacity(num_channels);
for i in 0..num_channels { for i in 0..num_channels {
dma_channels.push(Arc::new(DMAChannel::new(i)?)); dma_channels.push(Arc::new(DMAChannel::new(i)?));
} }
tracing::info!(target: "sol_trade_sdk","🚀 Created DMA manager with {} channels", num_channels); tracing::info!(target: "sol_trade_sdk","🚀 Created DMA manager with {} channels", num_channels);
Ok(Self { Ok(Self {
dma_channels, dma_channels,
channel_allocator: AtomicUsize::new(0), channel_allocator: AtomicUsize::new(0),
dma_stats: Arc::new(DMAStats::new()), dma_stats: Arc::new(DMAStats::new()),
}) })
} }
/// 🚀 执行零拷贝DMA传输 /// 🚀 执行零拷贝DMA传输
#[inline(always)] #[inline(always)]
pub async fn dma_transfer(&self, src: &[u8], dst: &mut [u8]) -> Result<usize> { pub async fn dma_transfer(&self, src: &[u8], dst: &mut [u8]) -> Result<usize> {
if src.len() != dst.len() { if src.len() != dst.len() {
return Err(anyhow::anyhow!("Source and destination sizes don't match")); return Err(anyhow::anyhow!("Source and destination sizes don't match"));
} }
// 选择DMA通道(轮询分配) // 选择DMA通道(轮询分配)
let channel_index = self.channel_allocator.fetch_add(1, Ordering::Relaxed) % self.dma_channels.len(); let channel_index =
self.channel_allocator.fetch_add(1, Ordering::Relaxed) % self.dma_channels.len();
let channel = &self.dma_channels[channel_index]; let channel = &self.dma_channels[channel_index];
// 执行DMA传输 // 执行DMA传输
let transferred = channel.transfer(src, dst).await?; let transferred = channel.transfer(src, dst).await?;
// 更新统计 // 更新统计
self.dma_stats.bytes_transferred.fetch_add(transferred as u64, Ordering::Relaxed); self.dma_stats.bytes_transferred.fetch_add(transferred as u64, Ordering::Relaxed);
self.dma_stats.transfers_completed.fetch_add(1, Ordering::Relaxed); self.dma_stats.transfers_completed.fetch_add(1, Ordering::Relaxed);
Ok(transferred) Ok(transferred)
} }
} }
@@ -475,21 +482,21 @@ impl DMAChannel {
_status: AtomicU64::new(0), _status: AtomicU64::new(0),
}) })
} }
/// 🚀 执行零拷贝传输 /// 🚀 执行零拷贝传输
#[inline(always)] #[inline(always)]
pub async fn transfer(&self, src: &[u8], dst: &mut [u8]) -> Result<usize> { pub async fn transfer(&self, src: &[u8], dst: &mut [u8]) -> Result<usize> {
let transfer_size = src.len(); let transfer_size = src.len();
// 使用硬件优化的SIMD内存拷贝 // 使用硬件优化的SIMD内存拷贝
unsafe { unsafe {
super::hardware_optimizations::SIMDMemoryOps::memcpy_simd_optimized( super::hardware_optimizations::SIMDMemoryOps::memcpy_simd_optimized(
dst.as_mut_ptr(), dst.as_mut_ptr(),
src.as_ptr(), src.as_ptr(),
transfer_size transfer_size,
); );
} }
Ok(transfer_size) Ok(transfer_size)
} }
} }
@@ -541,14 +548,14 @@ impl ZeroCopyStats {
mmap_buffer_usage: AtomicU64::new(0), mmap_buffer_usage: AtomicU64::new(0),
} }
} }
/// 打印统计信息 /// 打印统计信息
pub fn print_stats(&self) { pub fn print_stats(&self) {
let allocated = self.blocks_allocated.load(Ordering::Relaxed); let allocated = self.blocks_allocated.load(Ordering::Relaxed);
let freed = self.blocks_freed.load(Ordering::Relaxed); let freed = self.blocks_freed.load(Ordering::Relaxed);
let bytes = self.bytes_transferred.load(Ordering::Relaxed); let bytes = self.bytes_transferred.load(Ordering::Relaxed);
let mmap_usage = self.mmap_buffer_usage.load(Ordering::Relaxed); let mmap_usage = self.mmap_buffer_usage.load(Ordering::Relaxed);
tracing::info!(target: "sol_trade_sdk","🚀 Zero-Copy Stats:"); tracing::info!(target: "sol_trade_sdk","🚀 Zero-Copy Stats:");
tracing::info!(target: "sol_trade_sdk"," 📦 Blocks: Allocated={}, Freed={}, Active={}", tracing::info!(target: "sol_trade_sdk"," 📦 Blocks: Allocated={}, Freed={}, Active={}",
allocated, freed, allocated.saturating_sub(freed)); allocated, freed, allocated.saturating_sub(freed));
@@ -564,36 +571,36 @@ impl ZeroCopyMemoryManager {
pub fn new() -> Result<Self> { pub fn new() -> Result<Self> {
let mut shared_pools = Vec::new(); let mut shared_pools = Vec::new();
let mut mmap_buffers = Vec::new(); let mut mmap_buffers = Vec::new();
// 创建不同大小的内存池 // 创建不同大小的内存池
// 小块池: 64KB blocks, 1GB total // 小块池: 64KB blocks, 1GB total
shared_pools.push(Arc::new(SharedMemoryPool::new(0, 1024 * 1024 * 1024, 64 * 1024)?)); shared_pools.push(Arc::new(SharedMemoryPool::new(0, 1024 * 1024 * 1024, 64 * 1024)?));
// 中块池: 1MB blocks, 4GB total // 中块池: 1MB blocks, 4GB total
shared_pools.push(Arc::new(SharedMemoryPool::new(1, 4 * 1024 * 1024 * 1024, 1024 * 1024)?)); shared_pools.push(Arc::new(SharedMemoryPool::new(1, 4 * 1024 * 1024 * 1024, 1024 * 1024)?));
// 大块池: 16MB blocks, 8GB total // 大块池: 16MB blocks, 8GB total
shared_pools.push(Arc::new(SharedMemoryPool::new(2, 8 * 1024 * 1024 * 1024, 16 * 1024 * 1024)?)); shared_pools.push(Arc::new(SharedMemoryPool::new(
2,
8 * 1024 * 1024 * 1024,
16 * 1024 * 1024,
)?));
// 创建内存映射缓冲区 // 创建内存映射缓冲区
for i in 0..8 { for i in 0..8 {
mmap_buffers.push(Arc::new(MemoryMappedBuffer::new(i, 256 * 1024 * 1024)?)); // 256MB each mmap_buffers.push(Arc::new(MemoryMappedBuffer::new(i, 256 * 1024 * 1024)?));
// 256MB each
} }
let dma_manager = Arc::new(DirectMemoryAccessManager::new(16)?); // 16 DMA channels let dma_manager = Arc::new(DirectMemoryAccessManager::new(16)?); // 16 DMA channels
let stats = Arc::new(ZeroCopyStats::new()); let stats = Arc::new(ZeroCopyStats::new());
tracing::info!(target: "sol_trade_sdk","🚀 Zero-Copy Memory Manager initialized"); tracing::info!(target: "sol_trade_sdk","🚀 Zero-Copy Memory Manager initialized");
tracing::info!(target: "sol_trade_sdk"," 📦 Memory Pools: {}", shared_pools.len()); tracing::info!(target: "sol_trade_sdk"," 📦 Memory Pools: {}", shared_pools.len());
tracing::info!(target: "sol_trade_sdk"," 💾 Mapped Buffers: {}", mmap_buffers.len()); tracing::info!(target: "sol_trade_sdk"," 💾 Mapped Buffers: {}", mmap_buffers.len());
tracing::info!(target: "sol_trade_sdk"," 🔄 DMA Channels: 16"); tracing::info!(target: "sol_trade_sdk"," 🔄 DMA Channels: 16");
Ok(Self { Ok(Self { shared_pools, mmap_buffers, dma_manager, stats })
shared_pools,
mmap_buffers,
dma_manager,
stats,
})
} }
/// 🚀 分配零拷贝内存块 /// 🚀 分配零拷贝内存块
#[inline(always)] #[inline(always)]
pub fn allocate(&self, size: usize) -> Option<ZeroCopyBlock> { pub fn allocate(&self, size: usize) -> Option<ZeroCopyBlock> {
@@ -605,7 +612,7 @@ impl ZeroCopyMemoryManager {
} else { } else {
&self.shared_pools[2] // 大块池 &self.shared_pools[2] // 大块池
}; };
if let Some(block) = pool.allocate_block() { if let Some(block) = pool.allocate_block() {
self.stats.blocks_allocated.fetch_add(1, Ordering::Relaxed); self.stats.blocks_allocated.fetch_add(1, Ordering::Relaxed);
Some(block) Some(block)
@@ -613,7 +620,7 @@ impl ZeroCopyMemoryManager {
None None
} }
} }
/// 🚀 释放零拷贝内存块 /// 🚀 释放零拷贝内存块
#[inline(always)] #[inline(always)]
pub fn deallocate(&self, block: ZeroCopyBlock) { pub fn deallocate(&self, block: ZeroCopyBlock) {
@@ -623,19 +630,19 @@ impl ZeroCopyMemoryManager {
self.stats.blocks_freed.fetch_add(1, Ordering::Relaxed); self.stats.blocks_freed.fetch_add(1, Ordering::Relaxed);
} }
} }
/// 获取内存映射缓冲区 /// 获取内存映射缓冲区
#[inline(always)] #[inline(always)]
pub fn get_mmap_buffer(&self, buffer_id: usize) -> Option<Arc<MemoryMappedBuffer>> { pub fn get_mmap_buffer(&self, buffer_id: usize) -> Option<Arc<MemoryMappedBuffer>> {
self.mmap_buffers.get(buffer_id).cloned() self.mmap_buffers.get(buffer_id).cloned()
} }
/// 获取DMA管理器 /// 获取DMA管理器
#[inline(always)] #[inline(always)]
pub fn get_dma_manager(&self) -> Arc<DirectMemoryAccessManager> { pub fn get_dma_manager(&self) -> Arc<DirectMemoryAccessManager> {
self.dma_manager.clone() self.dma_manager.clone()
} }
/// 获取统计信息 /// 获取统计信息
pub fn get_stats(&self) -> Arc<ZeroCopyStats> { pub fn get_stats(&self) -> Arc<ZeroCopyStats> {
self.stats.clone() self.stats.clone()
@@ -645,73 +652,73 @@ impl ZeroCopyMemoryManager {
#[cfg(test)] #[cfg(test)]
mod tests { mod tests {
use super::*; use super::*;
#[tokio::test] #[tokio::test]
async fn test_shared_memory_pool() -> Result<()> { async fn test_shared_memory_pool() -> Result<()> {
let pool = SharedMemoryPool::new(0, 1024 * 1024, 4096)?; let pool = SharedMemoryPool::new(0, 1024 * 1024, 4096)?;
// 测试分配 // 测试分配
let block1 = pool.allocate_block().expect("Should allocate block"); let block1 = pool.allocate_block().expect("Should allocate block");
assert_eq!(block1.size(), 4096); assert_eq!(block1.size(), 4096);
let block2 = pool.allocate_block().expect("Should allocate another block"); let block2 = pool.allocate_block().expect("Should allocate another block");
assert_eq!(block2.size(), 4096); assert_eq!(block2.size(), 4096);
// 测试释放 // 测试释放
pool.deallocate_block(block1); pool.deallocate_block(block1);
pool.deallocate_block(block2); pool.deallocate_block(block2);
Ok(()) Ok(())
} }
#[tokio::test] #[tokio::test]
async fn test_memory_mapped_buffer() -> Result<()> { async fn test_memory_mapped_buffer() -> Result<()> {
let buffer = MemoryMappedBuffer::new(0, 1024 * 1024)?; let buffer = MemoryMappedBuffer::new(0, 1024 * 1024)?;
let test_data = b"Hello, Zero-Copy World!"; let test_data = b"Hello, Zero-Copy World!";
// 测试写入 // 测试写入
let written = buffer.write_data(test_data)?; let written = buffer.write_data(test_data)?;
assert_eq!(written, test_data.len()); assert_eq!(written, test_data.len());
// 测试读取 // 测试读取
let mut read_buffer = vec![0u8; test_data.len()]; let mut read_buffer = vec![0u8; test_data.len()];
let read = buffer.read_data(&mut read_buffer)?; let read = buffer.read_data(&mut read_buffer)?;
assert_eq!(read, test_data.len()); assert_eq!(read, test_data.len());
assert_eq!(&read_buffer, test_data); assert_eq!(&read_buffer, test_data);
Ok(()) Ok(())
} }
#[tokio::test] #[tokio::test]
async fn test_dma_transfer() -> Result<()> { async fn test_dma_transfer() -> Result<()> {
let dma_manager = DirectMemoryAccessManager::new(4)?; let dma_manager = DirectMemoryAccessManager::new(4)?;
let src = vec![1u8, 2, 3, 4, 5, 6, 7, 8]; let src = vec![1u8, 2, 3, 4, 5, 6, 7, 8];
let mut dst = vec![0u8; 8]; let mut dst = vec![0u8; 8];
let transferred = dma_manager.dma_transfer(&src, &mut dst).await?; let transferred = dma_manager.dma_transfer(&src, &mut dst).await?;
assert_eq!(transferred, 8); assert_eq!(transferred, 8);
assert_eq!(src, dst); assert_eq!(src, dst);
Ok(()) Ok(())
} }
#[tokio::test] #[tokio::test]
async fn test_zero_copy_manager() -> Result<()> { async fn test_zero_copy_manager() -> Result<()> {
let manager = ZeroCopyMemoryManager::new()?; let manager = ZeroCopyMemoryManager::new()?;
// 测试小块分配 // 测试小块分配
let small_block = manager.allocate(1024).expect("Should allocate small block"); let small_block = manager.allocate(1024).expect("Should allocate small block");
assert_eq!(small_block.size(), 65536); // 小块池的块大小 assert_eq!(small_block.size(), 65536); // 小块池的块大小
// 测试大块分配 // 测试大块分配
let large_block = manager.allocate(5 * 1024 * 1024).expect("Should allocate large block"); let large_block = manager.allocate(5 * 1024 * 1024).expect("Should allocate large block");
assert_eq!(large_block.size(), 16 * 1024 * 1024); // 大块池的块大小 assert_eq!(large_block.size(), 16 * 1024 * 1024); // 大块池的块大小
manager.deallocate(small_block); manager.deallocate(small_block);
manager.deallocate(large_block); manager.deallocate(large_block);
Ok(()) Ok(())
} }
} }
+152 -97
View File
@@ -2,45 +2,72 @@ use crate::swqos::common::{default_http_client_builder, poll_transaction_confirm
use rand::seq::IndexedRandom; use rand::seq::IndexedRandom;
use reqwest::Client; use reqwest::Client;
use std::{sync::Arc, time::Instant}; use std::{sync::Arc, time::Instant};
use tracing::{error, info, warn};
use std::time::Duration; use crate::swqos::SwqosClientTrait;
use crate::swqos::{SwqosType, TradeType};
use anyhow::Result; use anyhow::Result;
use bincode::serialize as bincode_serialize; use bincode::serialize as bincode_serialize;
use solana_client::rpc_client::SerializableTransaction; use solana_client::rpc_client::SerializableTransaction;
use solana_sdk::transaction::VersionedTransaction; use solana_sdk::transaction::VersionedTransaction;
use crate::swqos::{SwqosType, TradeType}; use std::time::Duration;
use crate::swqos::SwqosClientTrait;
use crate::{common::SolanaRpcClient, constants::swqos::ASTRALANE_TIP_ACCOUNTS}; use crate::{common::SolanaRpcClient, constants::swqos::ASTRALANE_TIP_ACCOUNTS};
use tokio::task::JoinHandle;
use std::sync::atomic::{AtomicBool, Ordering}; use std::sync::atomic::{AtomicBool, Ordering};
use tokio::task::JoinHandle;
/// Empty body for getHealth POST; avoid per-request allocation. /// Empty body for getHealth POST; avoid per-request allocation.
static PING_BODY: &[u8] = &[]; static PING_BODY: &[u8] = &[];
use crate::swqos::astralane_quic::AstralaneQuicClient;
#[derive(Clone)]
pub enum AstralaneBackend {
Http {
endpoint: String,
auth_token: String,
http_client: Client,
ping_handle: Arc<tokio::sync::Mutex<Option<JoinHandle<()>>>>,
stop_ping: Arc<AtomicBool>,
},
Quic(Arc<AstralaneQuicClient>),
}
#[derive(Clone)] #[derive(Clone)]
pub struct AstralaneClient { pub struct AstralaneClient {
pub endpoint: String,
pub auth_token: String,
pub rpc_client: Arc<SolanaRpcClient>, pub rpc_client: Arc<SolanaRpcClient>,
pub http_client: Client, backend: AstralaneBackend,
pub ping_handle: Arc<tokio::sync::Mutex<Option<JoinHandle<()>>>>,
pub stop_ping: Arc<AtomicBool>,
} }
#[async_trait::async_trait] #[async_trait::async_trait]
impl SwqosClientTrait for AstralaneClient { impl SwqosClientTrait for AstralaneClient {
async fn send_transaction(&self, trade_type: TradeType, transaction: &VersionedTransaction, wait_confirmation: bool) -> Result<()> { async fn send_transaction(
self.send_transaction(trade_type, transaction, wait_confirmation).await &self,
trade_type: TradeType,
transaction: &VersionedTransaction,
wait_confirmation: bool,
) -> Result<()> {
self.send_transaction_impl(trade_type, transaction, wait_confirmation).await
} }
async fn send_transactions(&self, trade_type: TradeType, transactions: &Vec<VersionedTransaction>, wait_confirmation: bool) -> Result<()> { async fn send_transactions(
self.send_transactions(trade_type, transactions, wait_confirmation).await &self,
trade_type: TradeType,
transactions: &Vec<VersionedTransaction>,
wait_confirmation: bool,
) -> Result<()> {
for transaction in transactions {
self.send_transaction_impl(trade_type, transaction, wait_confirmation).await?;
}
Ok(())
} }
fn get_tip_account(&self) -> Result<String> { fn get_tip_account(&self) -> Result<String> {
let tip_account = *ASTRALANE_TIP_ACCOUNTS.choose(&mut rand::rng()).or_else(|| ASTRALANE_TIP_ACCOUNTS.first()).unwrap(); let tip_account = *ASTRALANE_TIP_ACCOUNTS
.choose(&mut rand::rng())
.or_else(|| ASTRALANE_TIP_ACCOUNTS.first())
.unwrap();
Ok(tip_account.to_string()) Ok(tip_account.to_string())
} }
@@ -50,60 +77,84 @@ impl SwqosClientTrait for AstralaneClient {
} }
impl AstralaneClient { impl AstralaneClient {
/// 使用 HTTPirisb)提交。
pub fn new(rpc_url: String, endpoint: String, auth_token: String) -> Self { pub fn new(rpc_url: String, endpoint: String, auth_token: String) -> Self {
let rpc_client = SolanaRpcClient::new(rpc_url); let rpc_client = SolanaRpcClient::new(rpc_url);
let http_client = default_http_client_builder().build().unwrap(); let http_client = default_http_client_builder().build().unwrap();
let ping_handle = Arc::new(tokio::sync::Mutex::new(None));
let client = Self { let stop_ping = Arc::new(AtomicBool::new(false));
rpc_client: Arc::new(rpc_client),
endpoint, let client = Self {
auth_token, rpc_client: Arc::new(rpc_client),
http_client, backend: AstralaneBackend::Http {
ping_handle: Arc::new(tokio::sync::Mutex::new(None)), endpoint,
stop_ping: Arc::new(AtomicBool::new(false)), auth_token,
http_client,
ping_handle,
stop_ping,
},
}; };
// Start ping task
let client_clone = client.clone(); let client_clone = client.clone();
tokio::spawn(async move { tokio::spawn(async move {
client_clone.start_ping_task().await; client_clone.start_ping_task().await;
}); });
client client
} }
/// Start periodic ping task to keep connections active /// 使用 QUIC 提交。
pub async fn new_quic(rpc_url: String, quic_endpoint: &str, api_key: String) -> Result<Self> {
let rpc_client = SolanaRpcClient::new(rpc_url);
let quic_client = AstralaneQuicClient::connect(quic_endpoint, &api_key).await?;
Ok(Self {
rpc_client: Arc::new(rpc_client),
backend: AstralaneBackend::Quic(Arc::new(quic_client)),
})
}
async fn start_ping_task(&self) { async fn start_ping_task(&self) {
let endpoint = self.endpoint.clone(); match &self.backend {
let auth_token = self.auth_token.clone(); AstralaneBackend::Http {
let http_client = self.http_client.clone(); endpoint,
let stop_ping = self.stop_ping.clone(); auth_token,
http_client,
let handle = tokio::spawn(async move { ping_handle,
let mut interval = tokio::time::interval(Duration::from_secs(30)); stop_ping,
loop { } => {
interval.tick().await; // first tick completes immediately → one ping at start let endpoint = endpoint.clone();
if stop_ping.load(Ordering::Relaxed) { let auth_token = auth_token.clone();
break; let http_client = http_client.clone();
} let ping_handle = ping_handle.clone();
if let Err(e) = Self::send_ping_request(&http_client, &endpoint, &auth_token).await { let stop_ping = stop_ping.clone();
eprintln!("Astralane ping request failed: {}", e); let handle = tokio::spawn(async move {
let mut interval = tokio::time::interval(Duration::from_secs(30));
loop {
interval.tick().await;
if stop_ping.load(Ordering::Relaxed) {
break;
}
if let Err(e) =
Self::send_ping_request(&http_client, &endpoint, &auth_token).await
{
warn!(target: "sol_trade_sdk", "Astralane ping request failed: {}", e);
}
}
});
let mut guard = ping_handle.lock().await;
if let Some(old) = guard.as_ref() {
old.abort();
} }
*guard = Some(handle);
} }
}); AstralaneBackend::Quic(_) => {}
// Update ping_handle - use Mutex to safely update
{
let mut ping_guard = self.ping_handle.lock().await;
if let Some(old_handle) = ping_guard.as_ref() {
old_handle.abort();
}
*ping_guard = Some(handle);
} }
} }
/// Send ping request: POST endpoint?api-key=...&method=getHealth (endpoint is irisb from constants). /// Send ping request: POST endpoint?api-key=...&method=getHealth
async fn send_ping_request(http_client: &Client, endpoint: &str, auth_token: &str) -> Result<()> { async fn send_ping_request(
http_client: &Client,
endpoint: &str,
auth_token: &str,
) -> Result<()> {
let response = http_client let response = http_client
.post(endpoint) .post(endpoint)
.query(&[("api-key", auth_token), ("method", "getHealth")]) .query(&[("api-key", auth_token), ("method", "getHealth")])
@@ -112,57 +163,60 @@ impl AstralaneClient {
.send() .send()
.await?; .await?;
let status = response.status(); let status = response.status();
let _ = response.bytes().await; // consume body so connection returns to pool let _ = response.bytes().await;
if !status.is_success() { if !status.is_success() {
eprintln!("Astralane ping request returned non-success status: {}", status); warn!(target: "sol_trade_sdk", "Astralane ping request returned non-success status: {}", status);
} }
Ok(()) Ok(())
} }
/// Send transaction via /irisb binary API (no Base64; lower latency). async fn send_transaction_impl(
pub async fn send_transaction(&self, trade_type: TradeType, transaction: &VersionedTransaction, wait_confirmation: bool) -> Result<()> { &self,
trade_type: TradeType,
transaction: &VersionedTransaction,
wait_confirmation: bool,
) -> Result<()> {
let start_time = Instant::now(); let start_time = Instant::now();
let signature = transaction.get_signature(); let signature = transaction.get_signature();
let body_bytes = bincode_serialize(transaction)
.map_err(|e| anyhow::anyhow!("Astralane binary serialize failed: {}", e))?;
let body_bytes = bincode_serialize(transaction).map_err(|e| anyhow::anyhow!("Astralane binary serialize failed: {}", e))?; match &self.backend {
AstralaneBackend::Http { endpoint, auth_token, http_client, .. } => {
let response = self.http_client let response = http_client
.post(&self.endpoint) .post(endpoint)
.query(&[("api-key", self.auth_token.as_str()), ("method", "sendTransaction")]) .query(&[("api-key", auth_token.as_str()), ("method", "sendTransaction")])
.header("Content-Type", "application/octet-stream") .header("Content-Type", "application/octet-stream")
.body(body_bytes) .body(body_bytes)
.send() .send()
.await?; .await?;
let status = response.status();
let status = response.status(); let _ = response.bytes().await;
let _ = response.bytes().await; if status.is_success() {
if status.is_success() { info!(target: "sol_trade_sdk", "[astralane] {} submitted: {:?}", trade_type, start_time.elapsed());
println!(" [astralane] {} submitted: {:?}", trade_type, start_time.elapsed()); } else {
} else { error!(target: "sol_trade_sdk", "[astralane] {} submission failed: status {}", trade_type, status);
eprintln!(" [astralane] {} submission failed: status {}", trade_type, status); return Err(anyhow::anyhow!("Astralane sendTransaction failed: {}", status));
return Err(anyhow::anyhow!("Astralane sendTransaction failed: {}", status)); }
}
AstralaneBackend::Quic(quic) => {
quic.send_transaction(&body_bytes).await?;
info!(target: "sol_trade_sdk", "[astralane-quic] {} submitted: {:?}", trade_type, start_time.elapsed());
}
} }
let start_time = Instant::now(); let start_time = Instant::now();
match poll_transaction_confirmation(&self.rpc_client, *signature, wait_confirmation).await { match poll_transaction_confirmation(&self.rpc_client, *signature, wait_confirmation).await {
Ok(_) => (), Ok(_) => (),
Err(e) => { Err(e) => {
println!(" signature: {:?}", signature); info!(target: "sol_trade_sdk", "signature: {:?}", signature);
println!(" [astralane] {} confirmation failed: {:?}", trade_type, start_time.elapsed()); error!(target: "sol_trade_sdk", "[astralane] {} confirmation failed: {:?}", trade_type, start_time.elapsed());
return Err(e); return Err(e);
}, }
} }
if wait_confirmation { if wait_confirmation {
println!(" signature: {:?}", signature); info!(target: "sol_trade_sdk", "signature: {:?}", signature);
println!(" [astralane] {} confirmed: {:?}", trade_type, start_time.elapsed()); info!(target: "sol_trade_sdk", "[astralane] {} confirmed: {:?}", trade_type, start_time.elapsed());
}
Ok(())
}
pub async fn send_transactions(&self, trade_type: TradeType, transactions: &Vec<VersionedTransaction>, wait_confirmation: bool) -> Result<()> {
for transaction in transactions {
self.send_transaction(trade_type, transaction, wait_confirmation).await?;
} }
Ok(()) Ok(())
} }
@@ -170,18 +224,19 @@ impl AstralaneClient {
impl Drop for AstralaneClient { impl Drop for AstralaneClient {
fn drop(&mut self) { fn drop(&mut self) {
// Ensure ping task stops when client is destroyed match &self.backend {
self.stop_ping.store(true, Ordering::Relaxed); AstralaneBackend::Http { stop_ping, ping_handle, .. } => {
stop_ping.store(true, Ordering::Relaxed);
// Try to stop ping task immediately let ping_handle = ping_handle.clone();
// Use tokio::spawn to avoid blocking Drop tokio::spawn(async move {
let ping_handle = self.ping_handle.clone(); let mut guard = ping_handle.lock().await;
tokio::spawn(async move { if let Some(handle) = guard.as_ref() {
let mut ping_guard = ping_handle.lock().await; handle.abort();
if let Some(handle) = ping_guard.as_ref() { }
handle.abort(); *guard = None;
});
} }
*ping_guard = None; AstralaneBackend::Quic(_) => {}
}); }
} }
} }
+244
View File
@@ -0,0 +1,244 @@
//! 内联自 [Astralane/astralane-quic-client](https://github.com/Astralane/astralane-quic-client)
//! 用于向 Astralane QUIC TPU 提交交易,不依赖外部 crate,便于审计与安全可控。
use anyhow::{Context, Result};
use quinn::crypto::rustls::QuicClientConfig;
use quinn::{ClientConfig, Connection, Endpoint, IdleTimeout, TransportConfig};
use rcgen::{CertificateParams, KeyPair};
use rustls::pki_types::{CertificateDer, PrivateKeyDer, PrivatePkcs8KeyDer};
use std::net::SocketAddr;
use std::str::FromStr;
use std::sync::Arc;
use std::time::Duration;
use tokio::sync::Mutex;
use tracing::{info, warn};
/// ALPN protocol identifier for Astralane TPU.
const ALPN_ASTRALANE_TPU: &[u8] = b"astralane-tpu";
/// Maximum Solana transaction size.
pub const MAX_TRANSACTION_SIZE: usize = 1232;
/// QUIC application error codes returned by the server.
pub mod error_code {
pub const OK: u32 = 0;
pub const UNKNOWN_API_KEY: u32 = 1;
pub const CONNECTION_LIMIT: u32 = 2;
pub fn describe(code: u32) -> &'static str {
match code {
OK => "OK",
UNKNOWN_API_KEY => "Unknown API key",
CONNECTION_LIMIT => "Connection limit exceeded",
_ => "Unknown error",
}
}
}
/// QUIC client for sending transactions to Astralane's TPU endpoint.
pub struct AstralaneQuicClient {
endpoint: Endpoint,
connection: Mutex<Connection>,
server_addr: SocketAddr,
#[allow(dead_code)]
api_key: String,
}
impl AstralaneQuicClient {
/// Connect to an Astralane QUIC server.
/// Generates a self-signed TLS certificate with the API key as the Common Name (CN).
pub async fn connect(server_addr: &str, api_key: &str) -> Result<Self> {
let _ = rustls::crypto::ring::default_provider().install_default();
let addr = SocketAddr::from_str(server_addr)
.or_else(|_| {
use std::net::ToSocketAddrs;
server_addr
.to_socket_addrs()
.ok()
.and_then(|mut addrs| addrs.next())
.ok_or_else(|| anyhow::anyhow!("Cannot resolve address: {}", server_addr))
})
.context("Invalid server address")?;
info!("[astralane-quic] Building TLS config (CN = api_key)");
let client_config = Self::build_client_config(api_key)?;
let mut endpoint =
Endpoint::client("0.0.0.0:0".parse()?).context("Failed to create QUIC endpoint")?;
endpoint.set_default_client_config(client_config);
info!("[astralane-quic] Connecting to {} ...", addr);
let connection = endpoint
.connect(addr, "astralane")?
.await
.context("Failed to connect to Astralane QUIC server")?;
info!("[astralane-quic] Connected at {}", addr);
Ok(Self {
endpoint,
connection: Mutex::new(connection),
server_addr: addr,
api_key: api_key.to_string(),
})
}
/// Send a single bincode-serialized `VersionedTransaction`.
/// Fire-and-forget; automatically reconnects if the connection is dead.
pub async fn send_transaction(&self, transaction_bytes: &[u8]) -> Result<()> {
if transaction_bytes.len() > MAX_TRANSACTION_SIZE {
anyhow::bail!(
"Transaction too large: {} bytes (max {})",
transaction_bytes.len(),
MAX_TRANSACTION_SIZE
);
}
let conn = {
let mut guard = self.connection.lock().await;
if let Some(reason) = guard.close_reason() {
if let quinn::ConnectionError::ApplicationClosed(ref info) = reason {
let code = info.error_code.into_inner();
if code != error_code::OK as u64 {
anyhow::bail!(
"Server closed connection: {} (code {})",
error_code::describe(code as u32),
code
);
}
}
warn!("[astralane-quic] Connection dead, reconnecting to {} ...", self.server_addr);
*guard = self
.endpoint
.connect(self.server_addr, "astralane")?
.await
.context("Failed to reconnect to Astralane QUIC server")?;
info!("[astralane-quic] Reconnected to {}", self.server_addr);
}
guard.clone()
};
let mut send_stream =
conn.open_uni().await.context("Failed to open unidirectional stream")?;
send_stream
.write_all(transaction_bytes)
.await
.context("Failed to write transaction data")?;
send_stream.finish().context("Failed to finish stream")?;
info!("[astralane-quic] Transaction sent ({} bytes)", transaction_bytes.len());
Ok(())
}
/// Reconnect to the server if the connection was closed.
pub async fn reconnect(&self) -> Result<()> {
let mut guard = self.connection.lock().await;
if guard.close_reason().is_some() {
info!("[astralane-quic] Reconnecting at {}", self.server_addr);
*guard = self
.endpoint
.connect(self.server_addr, "astralane")?
.await
.context("Failed to reconnect to Astralane QUIC server")?;
info!("[astralane-quic] Reconnected to {}", self.server_addr);
}
Ok(())
}
/// Check if the connection is still alive.
pub async fn is_connected(&self) -> bool {
self.connection.lock().await.close_reason().is_none()
}
/// Close the connection gracefully.
pub async fn close(&self) {
self.connection.lock().await.close(error_code::OK.into(), b"client closing");
}
fn build_client_config(api_key: &str) -> Result<ClientConfig> {
let key_pair = KeyPair::generate_for(&rcgen::PKCS_ECDSA_P256_SHA256)?;
let mut cert_params = CertificateParams::new(vec![])?;
cert_params
.distinguished_name
.push(rcgen::DnType::CommonName, rcgen::DnValue::Utf8String(api_key.to_string()));
let cert = cert_params.self_signed(&key_pair)?;
let cert_der = CertificateDer::from(cert.der().to_vec());
let key_der = PrivateKeyDer::Pkcs8(PrivatePkcs8KeyDer::from(key_pair.serialize_der()));
let mut crypto = rustls::ClientConfig::builder()
.dangerous()
.with_custom_certificate_verifier(Arc::new(SkipServerVerification))
.with_client_auth_cert(vec![cert_der], key_der)
.context("Failed to set client certificate")?;
crypto.alpn_protocols = vec![ALPN_ASTRALANE_TPU.to_vec()];
let mut transport = TransportConfig::default();
transport.max_idle_timeout(Some(IdleTimeout::try_from(Duration::from_secs(30)).unwrap()));
transport.keep_alive_interval(Some(Duration::from_secs(25)));
let mut client_config =
ClientConfig::new(Arc::new(QuicClientConfig::try_from(crypto).unwrap()));
client_config.transport_config(Arc::new(transport));
Ok(client_config)
}
}
impl Drop for AstralaneQuicClient {
fn drop(&mut self) {
self.connection.get_mut().close(error_code::OK.into(), b"client closing");
}
}
/// Skip server certificate verification (Astralane server may use self-signed cert).
#[derive(Debug)]
struct SkipServerVerification;
impl rustls::client::danger::ServerCertVerifier for SkipServerVerification {
fn verify_server_cert(
&self,
_end_entity: &CertificateDer<'_>,
_intermediates: &[CertificateDer<'_>],
_server_name: &rustls::pki_types::ServerName<'_>,
_ocsp_response: &[u8],
_now: rustls::pki_types::UnixTime,
) -> Result<rustls::client::danger::ServerCertVerified, rustls::Error> {
Ok(rustls::client::danger::ServerCertVerified::assertion())
}
fn verify_tls12_signature(
&self,
_message: &[u8],
_cert: &CertificateDer<'_>,
_dss: &rustls::DigitallySignedStruct,
) -> Result<rustls::client::danger::HandshakeSignatureValid, rustls::Error> {
Ok(rustls::client::danger::HandshakeSignatureValid::assertion())
}
fn verify_tls13_signature(
&self,
_message: &[u8],
_cert: &CertificateDer<'_>,
_dss: &rustls::DigitallySignedStruct,
) -> Result<rustls::client::danger::HandshakeSignatureValid, rustls::Error> {
Ok(rustls::client::danger::HandshakeSignatureValid::assertion())
}
fn supported_verify_schemes(&self) -> Vec<rustls::SignatureScheme> {
vec![
rustls::SignatureScheme::ECDSA_NISTP256_SHA256,
rustls::SignatureScheme::ECDSA_NISTP384_SHA384,
rustls::SignatureScheme::RSA_PSS_SHA256,
rustls::SignatureScheme::RSA_PSS_SHA384,
rustls::SignatureScheme::RSA_PSS_SHA512,
rustls::SignatureScheme::RSA_PKCS1_SHA256,
rustls::SignatureScheme::RSA_PKCS1_SHA384,
rustls::SignatureScheme::RSA_PKCS1_SHA512,
rustls::SignatureScheme::ED25519,
]
}
}
+77 -31
View File
@@ -1,20 +1,22 @@
use crate::swqos::common::{default_http_client_builder, poll_transaction_confirmation, serialize_transaction_and_encode}; use crate::swqos::common::{
default_http_client_builder, poll_transaction_confirmation, serialize_transaction_and_encode,
};
use rand::seq::IndexedRandom; use rand::seq::IndexedRandom;
use reqwest::Client; use reqwest::Client;
use std::{sync::Arc, time::Instant}; use std::{sync::Arc, time::Instant};
use std::time::Duration;
use solana_transaction_status::UiTransactionEncoding; use solana_transaction_status::UiTransactionEncoding;
use std::time::Duration;
use crate::swqos::SwqosClientTrait;
use crate::swqos::{SwqosType, TradeType};
use anyhow::Result; use anyhow::Result;
use solana_sdk::transaction::VersionedTransaction; use solana_sdk::transaction::VersionedTransaction;
use crate::swqos::{SwqosType, TradeType};
use crate::swqos::SwqosClientTrait;
use crate::{common::SolanaRpcClient, constants::swqos::BLOCKRAZOR_TIP_ACCOUNTS}; use crate::{common::SolanaRpcClient, constants::swqos::BLOCKRAZOR_TIP_ACCOUNTS};
use tokio::task::JoinHandle;
use std::sync::atomic::{AtomicBool, Ordering}; use std::sync::atomic::{AtomicBool, Ordering};
use tokio::task::JoinHandle;
#[derive(Clone)] #[derive(Clone)]
pub struct BlockRazorClient { pub struct BlockRazorClient {
@@ -28,16 +30,29 @@ pub struct BlockRazorClient {
#[async_trait::async_trait] #[async_trait::async_trait]
impl SwqosClientTrait for BlockRazorClient { impl SwqosClientTrait for BlockRazorClient {
async fn send_transaction(&self, trade_type: TradeType, transaction: &VersionedTransaction, wait_confirmation: bool) -> Result<()> { async fn send_transaction(
&self,
trade_type: TradeType,
transaction: &VersionedTransaction,
wait_confirmation: bool,
) -> Result<()> {
self.send_transaction(trade_type, transaction, wait_confirmation).await self.send_transaction(trade_type, transaction, wait_confirmation).await
} }
async fn send_transactions(&self, trade_type: TradeType, transactions: &Vec<VersionedTransaction>, wait_confirmation: bool) -> Result<()> { async fn send_transactions(
&self,
trade_type: TradeType,
transactions: &Vec<VersionedTransaction>,
wait_confirmation: bool,
) -> Result<()> {
self.send_transactions(trade_type, transactions, wait_confirmation).await self.send_transactions(trade_type, transactions, wait_confirmation).await
} }
fn get_tip_account(&self) -> Result<String> { fn get_tip_account(&self) -> Result<String> {
let tip_account = *BLOCKRAZOR_TIP_ACCOUNTS.choose(&mut rand::rng()).or_else(|| BLOCKRAZOR_TIP_ACCOUNTS.first()).unwrap(); let tip_account = *BLOCKRAZOR_TIP_ACCOUNTS
.choose(&mut rand::rng())
.or_else(|| BLOCKRAZOR_TIP_ACCOUNTS.first())
.unwrap();
Ok(tip_account.to_string()) Ok(tip_account.to_string())
} }
@@ -49,23 +64,24 @@ impl SwqosClientTrait for BlockRazorClient {
impl BlockRazorClient { impl BlockRazorClient {
pub fn new(rpc_url: String, endpoint: String, auth_token: String) -> Self { pub fn new(rpc_url: String, endpoint: String, auth_token: String) -> Self {
let rpc_client = SolanaRpcClient::new(rpc_url); let rpc_client = SolanaRpcClient::new(rpc_url);
let http_client = default_http_client_builder().build().unwrap(); // 官方文档:請求中唯一允許的 header 是 Content-Type: text/plain;避免默认 User-Agent 等导致 500
let http_client = default_http_client_builder().user_agent("").build().unwrap();
let client = Self {
rpc_client: Arc::new(rpc_client), let client = Self {
endpoint, rpc_client: Arc::new(rpc_client),
auth_token, endpoint,
auth_token,
http_client, http_client,
ping_handle: Arc::new(tokio::sync::Mutex::new(None)), ping_handle: Arc::new(tokio::sync::Mutex::new(None)),
stop_ping: Arc::new(AtomicBool::new(false)), stop_ping: Arc::new(AtomicBool::new(false)),
}; };
// Start ping task // Start ping task
let client_clone = client.clone(); let client_clone = client.clone();
tokio::spawn(async move { tokio::spawn(async move {
client_clone.start_ping_task().await; client_clone.start_ping_task().await;
}); });
client client
} }
@@ -75,7 +91,7 @@ impl BlockRazorClient {
let auth_token = self.auth_token.clone(); let auth_token = self.auth_token.clone();
let http_client = self.http_client.clone(); let http_client = self.http_client.clone();
let stop_ping = self.stop_ping.clone(); let stop_ping = self.stop_ping.clone();
let handle = tokio::spawn(async move { let handle = tokio::spawn(async move {
// Immediate first ping to warm connection and reduce first-submit cold start latency // Immediate first ping to warm connection and reduce first-submit cold start latency
if let Err(e) = Self::send_ping_request(&http_client, &endpoint, &auth_token).await { if let Err(e) = Self::send_ping_request(&http_client, &endpoint, &auth_token).await {
@@ -83,20 +99,21 @@ impl BlockRazorClient {
eprintln!("BlockRazor ping request failed: {}", e); eprintln!("BlockRazor ping request failed: {}", e);
} }
} }
let mut interval = tokio::time::interval(Duration::from_secs(30)); // 30s keepalive to avoid server ~5min idle close let mut interval = tokio::time::interval(Duration::from_secs(30)); // 30s keepalive to avoid server ~5min idle close
loop { loop {
interval.tick().await; interval.tick().await;
if stop_ping.load(Ordering::Relaxed) { if stop_ping.load(Ordering::Relaxed) {
break; break;
} }
if let Err(e) = Self::send_ping_request(&http_client, &endpoint, &auth_token).await { if let Err(e) = Self::send_ping_request(&http_client, &endpoint, &auth_token).await
{
if crate::common::sdk_log::sdk_log_enabled() { if crate::common::sdk_log::sdk_log_enabled() {
eprintln!("BlockRazor ping request failed: {}", e); eprintln!("BlockRazor ping request failed: {}", e);
} }
} }
} }
}); });
// Update ping_handle - use Mutex to safely update // Update ping_handle - use Mutex to safely update
{ {
let mut ping_guard = self.ping_handle.lock().await; let mut ping_guard = self.ping_handle.lock().await;
@@ -108,7 +125,11 @@ impl BlockRazorClient {
} }
/// Send ping request: POST /v2/health?auth=... (Keep Alive). Only required param: auth. /// Send ping request: POST /v2/health?auth=... (Keep Alive). Only required param: auth.
async fn send_ping_request(http_client: &Client, endpoint: &str, auth_token: &str) -> Result<()> { async fn send_ping_request(
http_client: &Client,
endpoint: &str,
auth_token: &str,
) -> Result<()> {
let ping_url = endpoint.replace("/v2/sendTransaction", "/v2/health"); let ping_url = endpoint.replace("/v2/sendTransaction", "/v2/health");
let response = http_client let response = http_client
.post(&ping_url) .post(&ping_url)
@@ -127,11 +148,19 @@ impl BlockRazorClient {
} }
/// Send transaction via v2 API: plain Base64 body, Content-Type: text/plain. Only required URI param: auth. /// Send transaction via v2 API: plain Base64 body, Content-Type: text/plain. Only required URI param: auth.
pub async fn send_transaction(&self, trade_type: TradeType, transaction: &VersionedTransaction, wait_confirmation: bool) -> Result<()> { /// 文档要求:auth 以 URI 参数传入;body 为纯 Base64 编码交易;唯一允许的 header 为 Content-Type: text/plain。
pub async fn send_transaction(
&self,
trade_type: TradeType,
transaction: &VersionedTransaction,
wait_confirmation: bool,
) -> Result<()> {
let start_time = Instant::now(); let start_time = Instant::now();
let (content, signature) = serialize_transaction_and_encode(transaction, UiTransactionEncoding::Base64)?; let (content, signature) =
serialize_transaction_and_encode(transaction, UiTransactionEncoding::Base64)?;
let response = self.http_client let response = self
.http_client
.post(&self.endpoint) .post(&self.endpoint)
.query(&[("auth", self.auth_token.as_str())]) .query(&[("auth", self.auth_token.as_str())])
.header("Content-Type", "text/plain") .header("Content-Type", "text/plain")
@@ -140,16 +169,24 @@ impl BlockRazorClient {
.await?; .await?;
let status = response.status(); let status = response.status();
let _ = response.bytes().await;
if status.is_success() { if status.is_success() {
let _ = response.bytes().await;
if crate::common::sdk_log::sdk_log_enabled() { if crate::common::sdk_log::sdk_log_enabled() {
println!(" [blockrazor] {} submitted: {:?}", trade_type, start_time.elapsed()); println!(" [blockrazor] {} submitted: {:?}", trade_type, start_time.elapsed());
} }
} else { } else {
let body = response.text().await.unwrap_or_default();
if crate::common::sdk_log::sdk_log_enabled() { if crate::common::sdk_log::sdk_log_enabled() {
eprintln!(" [blockrazor] {} submission failed: status {}", trade_type, status); eprintln!(
" [blockrazor] {} submission failed: status {} body: {}",
trade_type, status, body
);
} }
return Err(anyhow::anyhow!("BlockRazor sendTransaction failed: {}", status)); return Err(anyhow::anyhow!(
"BlockRazor sendTransaction failed: status {} body: {}",
status,
body
));
} }
let start_time = Instant::now(); let start_time = Instant::now();
@@ -158,10 +195,14 @@ impl BlockRazorClient {
Err(e) => { Err(e) => {
if crate::common::sdk_log::sdk_log_enabled() { if crate::common::sdk_log::sdk_log_enabled() {
println!(" signature: {:?}", signature); println!(" signature: {:?}", signature);
println!(" [blockrazor] {} confirmation failed: {:?}", trade_type, start_time.elapsed()); println!(
" [blockrazor] {} confirmation failed: {:?}",
trade_type,
start_time.elapsed()
);
} }
return Err(e); return Err(e);
}, }
} }
if wait_confirmation && crate::common::sdk_log::sdk_log_enabled() { if wait_confirmation && crate::common::sdk_log::sdk_log_enabled() {
println!(" signature: {:?}", signature); println!(" signature: {:?}", signature);
@@ -171,7 +212,12 @@ impl BlockRazorClient {
Ok(()) Ok(())
} }
pub async fn send_transactions(&self, trade_type: TradeType, transactions: &Vec<VersionedTransaction>, wait_confirmation: bool) -> Result<()> { pub async fn send_transactions(
&self,
trade_type: TradeType,
transactions: &Vec<VersionedTransaction>,
wait_confirmation: bool,
) -> Result<()> {
for transaction in transactions { for transaction in transactions {
self.send_transaction(trade_type, transaction, wait_confirmation).await?; self.send_transaction(trade_type, transaction, wait_confirmation).await?;
} }
@@ -183,7 +229,7 @@ impl Drop for BlockRazorClient {
fn drop(&mut self) { fn drop(&mut self) {
// Ensure ping task stops when client is destroyed // Ensure ping task stops when client is destroyed
self.stop_ping.store(true, Ordering::Relaxed); self.stop_ping.store(true, Ordering::Relaxed);
// Try to stop ping task immediately // Try to stop ping task immediately
// Use tokio::spawn to avoid blocking Drop // Use tokio::spawn to avoid blocking Drop
let ping_handle = self.ping_handle.clone(); let ping_handle = self.ping_handle.clone();
+46 -15
View File
@@ -6,17 +6,16 @@ use rand::seq::IndexedRandom;
use reqwest::Client; use reqwest::Client;
use std::{sync::Arc, time::Instant}; use std::{sync::Arc, time::Instant};
use std::time::Duration;
use solana_transaction_status::UiTransactionEncoding; use solana_transaction_status::UiTransactionEncoding;
use std::time::Duration;
use crate::swqos::SwqosClientTrait;
use crate::swqos::{SwqosType, TradeType};
use anyhow::Result; use anyhow::Result;
use solana_sdk::transaction::VersionedTransaction; use solana_sdk::transaction::VersionedTransaction;
use crate::swqos::{SwqosType, TradeType};
use crate::swqos::SwqosClientTrait;
use crate::{common::SolanaRpcClient, constants::swqos::BLOX_TIP_ACCOUNTS}; use crate::{common::SolanaRpcClient, constants::swqos::BLOX_TIP_ACCOUNTS};
#[derive(Clone)] #[derive(Clone)]
pub struct BloxrouteClient { pub struct BloxrouteClient {
pub endpoint: String, pub endpoint: String,
@@ -27,16 +26,29 @@ pub struct BloxrouteClient {
#[async_trait::async_trait] #[async_trait::async_trait]
impl SwqosClientTrait for BloxrouteClient { impl SwqosClientTrait for BloxrouteClient {
async fn send_transaction(&self, trade_type: TradeType, transaction: &VersionedTransaction, wait_confirmation: bool) -> Result<()> { async fn send_transaction(
&self,
trade_type: TradeType,
transaction: &VersionedTransaction,
wait_confirmation: bool,
) -> Result<()> {
self.send_transaction(trade_type, transaction, wait_confirmation).await self.send_transaction(trade_type, transaction, wait_confirmation).await
} }
async fn send_transactions(&self, trade_type: TradeType, transactions: &Vec<VersionedTransaction>, wait_confirmation: bool) -> Result<()> { async fn send_transactions(
&self,
trade_type: TradeType,
transactions: &Vec<VersionedTransaction>,
wait_confirmation: bool,
) -> Result<()> {
self.send_transactions(trade_type, transactions, wait_confirmation).await self.send_transactions(trade_type, transactions, wait_confirmation).await
} }
fn get_tip_account(&self) -> Result<String> { fn get_tip_account(&self) -> Result<String> {
let tip_account = *BLOX_TIP_ACCOUNTS.choose(&mut rand::rng()).or_else(|| BLOX_TIP_ACCOUNTS.first()).unwrap(); let tip_account = *BLOX_TIP_ACCOUNTS
.choose(&mut rand::rng())
.or_else(|| BLOX_TIP_ACCOUNTS.first())
.unwrap();
Ok(tip_account.to_string()) Ok(tip_account.to_string())
} }
@@ -56,9 +68,15 @@ impl BloxrouteClient {
Self { rpc_client: Arc::new(rpc_client), endpoint, auth_token, http_client } Self { rpc_client: Arc::new(rpc_client), endpoint, auth_token, http_client }
} }
pub async fn send_transaction(&self, trade_type: TradeType, transaction: &VersionedTransaction, wait_confirmation: bool) -> Result<()> { pub async fn send_transaction(
&self,
trade_type: TradeType,
transaction: &VersionedTransaction,
wait_confirmation: bool,
) -> Result<()> {
let start_time = Instant::now(); let start_time = Instant::now();
let (content, signature) = serialize_transaction_and_encode(transaction, UiTransactionEncoding::Base64)?; let (content, signature) =
serialize_transaction_and_encode(transaction, UiTransactionEncoding::Base64)?;
// Single format! for body to avoid json! + to_string() double allocation // Single format! for body to avoid json! + to_string() double allocation
let body = format!( let body = format!(
@@ -67,7 +85,9 @@ impl BloxrouteClient {
); );
let endpoint = format!("{}/api/v2/submit", self.endpoint); let endpoint = format!("{}/api/v2/submit", self.endpoint);
let response_text = self.http_client.post(&endpoint) let response_text = self
.http_client
.post(&endpoint)
.body(body) .body(body)
.header("Content-Type", "application/json") .header("Content-Type", "application/json")
.header("Authorization", self.auth_token.as_str()) .header("Authorization", self.auth_token.as_str())
@@ -95,10 +115,14 @@ impl BloxrouteClient {
Err(e) => { Err(e) => {
if crate::common::sdk_log::sdk_log_enabled() { if crate::common::sdk_log::sdk_log_enabled() {
println!(" signature: {:?}", signature); println!(" signature: {:?}", signature);
println!(" [bloxroute] {} confirmation failed: {:?}", trade_type, start_time.elapsed()); println!(
" [bloxroute] {} confirmation failed: {:?}",
trade_type,
start_time.elapsed()
);
} }
return Err(e); return Err(e);
}, }
} }
if wait_confirmation && crate::common::sdk_log::sdk_log_enabled() { if wait_confirmation && crate::common::sdk_log::sdk_log_enabled() {
println!(" signature: {:?}", signature); println!(" signature: {:?}", signature);
@@ -108,7 +132,12 @@ impl BloxrouteClient {
Ok(()) Ok(())
} }
pub async fn send_transactions(&self, trade_type: TradeType, transactions: &Vec<VersionedTransaction>, _wait_confirmation: bool) -> Result<()> { pub async fn send_transactions(
&self,
trade_type: TradeType,
transactions: &Vec<VersionedTransaction>,
_wait_confirmation: bool,
) -> Result<()> {
let start_time = Instant::now(); let start_time = Instant::now();
let contents = serialization::serialize_transactions_batch_sync( let contents = serialization::serialize_transactions_batch_sync(
@@ -123,7 +152,9 @@ impl BloxrouteClient {
let body = format!(r#"{{"entries":[{}]}}"#, entries); let body = format!(r#"{{"entries":[{}]}}"#, entries);
let endpoint = format!("{}/api/v2/submit-batch", self.endpoint); let endpoint = format!("{}/api/v2/submit-batch", self.endpoint);
let response_text = self.http_client.post(&endpoint) let response_text = self
.http_client
.post(&endpoint)
.body(body) .body(body)
.header("Content-Type", "application/json") .header("Content-Type", "application/json")
.header("Authorization", self.auth_token.as_str()) .header("Authorization", self.auth_token.as_str())
@@ -144,4 +175,4 @@ impl BloxrouteClient {
Ok(()) Ok(())
} }
} }
+23 -11
View File
@@ -1,9 +1,9 @@
use crate::common::types::SolanaRpcClient; use crate::common::types::SolanaRpcClient;
use crate::swqos::serialization;
use anyhow::Result; use anyhow::Result;
use base64::engine::general_purpose::{self, STANDARD}; use base64::engine::general_purpose::{self, STANDARD};
use base64::Engine; use base64::Engine;
use bincode::serialize; use bincode::serialize;
use crate::swqos::serialization;
use reqwest::Client; use reqwest::Client;
use serde_json; use serde_json;
use serde_json::json; use serde_json::json;
@@ -117,7 +117,11 @@ pub async fn poll_any_transaction_confirmation(
loop { loop {
if start.elapsed() >= timeout { if start.elapsed() >= timeout {
return Err(anyhow::anyhow!("Transaction confirmation timed out after {}s ({} signatures polled)", timeout.as_secs(), signatures.len())); return Err(anyhow::anyhow!(
"Transaction confirmation timed out after {}s ({} signatures polled)",
timeout.as_secs(),
signatures.len()
));
} }
poll_count += 1; poll_count += 1;
@@ -205,7 +209,7 @@ pub async fn poll_any_transaction_confirmation(
let ui_err = meta.err.unwrap(); let ui_err = meta.err.unwrap();
let tx_err: TransactionError = let tx_err: TransactionError =
serde_json::from_value(serde_json::to_value(&ui_err)?)?; serde_json::from_value(serde_json::to_value(&ui_err)?)?;
// Use Solana InstructionError codes directly // Use Solana InstructionError codes directly
let mut code = 0u32; let mut code = 0u32;
let mut index = None; let mut index = None;
@@ -230,7 +234,7 @@ pub async fn poll_any_transaction_confirmation(
} }
_ => {} _ => {}
} }
return Err(anyhow::Error::new(TradeError { return Err(anyhow::Error::new(TradeError {
code: code, code: code,
message: format!("{} {:?}", tx_err, error_msg), message: format!("{} {:?}", tx_err, error_msg),
@@ -241,11 +245,16 @@ pub async fn poll_any_transaction_confirmation(
} }
} }
pub async fn send_nb_transaction(client: Client, endpoint: &str, auth_token: &str, transaction: &Transaction) -> Result<Signature, anyhow::Error> { pub async fn send_nb_transaction(
client: Client,
endpoint: &str,
auth_token: &str,
transaction: &Transaction,
) -> Result<Signature, anyhow::Error> {
// Serialize transaction // Serialize transaction
let serialized = bincode::serialize(transaction) let serialized = bincode::serialize(transaction)
.map_err(|e| anyhow::anyhow!("Transaction serialization failed: {}", e))?; .map_err(|e| anyhow::anyhow!("Transaction serialization failed: {}", e))?;
// Base64 encode // Base64 encode
let encoded = STANDARD.encode(serialized); let encoded = STANDARD.encode(serialized);
@@ -266,18 +275,21 @@ pub async fn send_nb_transaction(client: Client, endpoint: &str, auth_token: &st
.await .await
.map_err(|e| anyhow::anyhow!("Request failed: {}", e))?; .map_err(|e| anyhow::anyhow!("Request failed: {}", e))?;
let resp = response.json::<serde_json::Value>().await let resp = response
.json::<serde_json::Value>()
.await
.map_err(|e| anyhow::anyhow!("Response parsing failed: {}", e))?; .map_err(|e| anyhow::anyhow!("Response parsing failed: {}", e))?;
if let Some(reason) = resp["reason"].as_str() { if let Some(reason) = resp["reason"].as_str() {
return Err(anyhow::anyhow!(reason.to_string())); return Err(anyhow::anyhow!(reason.to_string()));
} }
let signature = resp["signature"].as_str() let signature = resp["signature"]
.as_str()
.ok_or_else(|| anyhow::anyhow!("Missing signature field in response"))?; .ok_or_else(|| anyhow::anyhow!("Missing signature field in response"))?;
let signature = Signature::from_str(signature) let signature =
.map_err(|e| anyhow::anyhow!("Invalid signature: {}", e))?; Signature::from_str(signature).map_err(|e| anyhow::anyhow!("Invalid signature: {}", e))?;
Ok(signature) Ok(signature)
} }
@@ -314,4 +326,4 @@ pub async fn serialize_smart_transaction_and_encode(
_ => return Err(anyhow::anyhow!("Unsupported encoding")), _ => return Err(anyhow::anyhow!("Unsupported encoding")),
}; };
Ok((serialized, *signature)) Ok((serialized, *signature))
} }
+44 -13
View File
@@ -1,4 +1,6 @@
use crate::swqos::common::{default_http_client_builder, poll_transaction_confirmation, serialize_transaction_and_encode}; use crate::swqos::common::{
default_http_client_builder, poll_transaction_confirmation, serialize_transaction_and_encode,
};
use rand::seq::IndexedRandom; use rand::seq::IndexedRandom;
use reqwest::Client; use reqwest::Client;
use serde_json::json; use serde_json::json;
@@ -6,14 +8,13 @@ use std::{sync::Arc, time::Instant};
use solana_transaction_status::UiTransactionEncoding; use solana_transaction_status::UiTransactionEncoding;
use crate::swqos::SwqosClientTrait;
use crate::swqos::{SwqosType, TradeType};
use anyhow::Result; use anyhow::Result;
use solana_sdk::transaction::VersionedTransaction; use solana_sdk::transaction::VersionedTransaction;
use crate::swqos::{SwqosType, TradeType};
use crate::swqos::SwqosClientTrait;
use crate::{common::SolanaRpcClient, constants::swqos::FLASHBLOCK_TIP_ACCOUNTS}; use crate::{common::SolanaRpcClient, constants::swqos::FLASHBLOCK_TIP_ACCOUNTS};
#[derive(Clone)] #[derive(Clone)]
pub struct FlashBlockClient { pub struct FlashBlockClient {
pub endpoint: String, pub endpoint: String,
@@ -24,16 +25,29 @@ pub struct FlashBlockClient {
#[async_trait::async_trait] #[async_trait::async_trait]
impl SwqosClientTrait for FlashBlockClient { impl SwqosClientTrait for FlashBlockClient {
async fn send_transaction(&self, trade_type: TradeType, transaction: &VersionedTransaction, wait_confirmation: bool) -> Result<()> { async fn send_transaction(
&self,
trade_type: TradeType,
transaction: &VersionedTransaction,
wait_confirmation: bool,
) -> Result<()> {
self.send_transaction(trade_type, transaction, wait_confirmation).await self.send_transaction(trade_type, transaction, wait_confirmation).await
} }
async fn send_transactions(&self, trade_type: TradeType, transactions: &Vec<VersionedTransaction>, wait_confirmation: bool) -> Result<()> { async fn send_transactions(
&self,
trade_type: TradeType,
transactions: &Vec<VersionedTransaction>,
wait_confirmation: bool,
) -> Result<()> {
self.send_transactions(trade_type, transactions, wait_confirmation).await self.send_transactions(trade_type, transactions, wait_confirmation).await
} }
fn get_tip_account(&self) -> Result<String> { fn get_tip_account(&self) -> Result<String> {
let tip_account = *FLASHBLOCK_TIP_ACCOUNTS.choose(&mut rand::rng()).or_else(|| FLASHBLOCK_TIP_ACCOUNTS.first()).unwrap(); let tip_account = *FLASHBLOCK_TIP_ACCOUNTS
.choose(&mut rand::rng())
.or_else(|| FLASHBLOCK_TIP_ACCOUNTS.first())
.unwrap();
Ok(tip_account.to_string()) Ok(tip_account.to_string())
} }
@@ -49,9 +63,15 @@ impl FlashBlockClient {
Self { rpc_client: Arc::new(rpc_client), endpoint, auth_token, http_client } Self { rpc_client: Arc::new(rpc_client), endpoint, auth_token, http_client }
} }
pub async fn send_transaction(&self, trade_type: TradeType, transaction: &VersionedTransaction, wait_confirmation: bool) -> Result<()> { pub async fn send_transaction(
&self,
trade_type: TradeType,
transaction: &VersionedTransaction,
wait_confirmation: bool,
) -> Result<()> {
let start_time = Instant::now(); let start_time = Instant::now();
let (content, signature) = serialize_transaction_and_encode(transaction, UiTransactionEncoding::Base64)?; let (content, signature) =
serialize_transaction_and_encode(transaction, UiTransactionEncoding::Base64)?;
// FlashBlock API format // FlashBlock API format
let request_body = serde_json::to_string(&json!({ let request_body = serde_json::to_string(&json!({
@@ -61,7 +81,9 @@ impl FlashBlockClient {
let url = format!("{}/api/v2/submit-batch", self.endpoint); let url = format!("{}/api/v2/submit-batch", self.endpoint);
// Send request to FlashBlock // Send request to FlashBlock
let response_text = self.http_client.post(&url) let response_text = self
.http_client
.post(&url)
.body(request_body) .body(request_body)
.header("Authorization", &self.auth_token) .header("Authorization", &self.auth_token)
.header("Content-Type", "application/json") .header("Content-Type", "application/json")
@@ -88,9 +110,13 @@ impl FlashBlockClient {
Ok(_) => (), Ok(_) => (),
Err(e) => { Err(e) => {
println!(" signature: {:?}", signature); println!(" signature: {:?}", signature);
println!(" [FlashBlock] {} confirmation failed: {:?}", trade_type, start_time.elapsed()); println!(
" [FlashBlock] {} confirmation failed: {:?}",
trade_type,
start_time.elapsed()
);
return Err(e); return Err(e);
}, }
} }
if wait_confirmation { if wait_confirmation {
println!(" signature: {:?}", signature); println!(" signature: {:?}", signature);
@@ -100,7 +126,12 @@ impl FlashBlockClient {
Ok(()) Ok(())
} }
pub async fn send_transactions(&self, trade_type: TradeType, transactions: &Vec<VersionedTransaction>, wait_confirmation: bool) -> Result<()> { pub async fn send_transactions(
&self,
trade_type: TradeType,
transactions: &Vec<VersionedTransaction>,
wait_confirmation: bool,
) -> Result<()> {
for transaction in transactions { for transaction in transactions {
self.send_transaction(trade_type, transaction, wait_confirmation).await?; self.send_transaction(trade_type, transaction, wait_confirmation).await?;
} }
+9 -27
View File
@@ -20,7 +20,9 @@ use std::sync::Arc;
use std::time::Instant; use std::time::Instant;
use crate::common::SolanaRpcClient; use crate::common::SolanaRpcClient;
use crate::constants::swqos::{HELIUS_TIP_ACCOUNTS, SWQOS_MIN_TIP_HELIUS, SWQOS_MIN_TIP_HELIUS_SWQOS_ONLY}; use crate::constants::swqos::{
HELIUS_TIP_ACCOUNTS, SWQOS_MIN_TIP_HELIUS, SWQOS_MIN_TIP_HELIUS_SWQOS_ONLY,
};
use crate::swqos::{SwqosClientTrait, SwqosType, TradeType}; use crate::swqos::{SwqosClientTrait, SwqosType, TradeType};
#[derive(Clone)] #[derive(Clone)]
@@ -43,12 +45,7 @@ impl HeliusClient {
let rpc_client = SolanaRpcClient::new(rpc_url); let rpc_client = SolanaRpcClient::new(rpc_url);
let http_client = default_http_client_builder().build().unwrap(); let http_client = default_http_client_builder().build().unwrap();
let submit_url = Self::build_submit_url(&endpoint, api_key.as_deref(), swqos_only); let submit_url = Self::build_submit_url(&endpoint, api_key.as_deref(), swqos_only);
Self { Self { submit_url, rpc_client: Arc::new(rpc_client), http_client, swqos_only }
submit_url,
rpc_client: Arc::new(rpc_client),
http_client,
swqos_only,
}
} }
/// Build URL once at construction; no per-request allocation. /// Build URL once at construction; no per-request allocation.
@@ -132,17 +129,10 @@ impl HeliusClient {
return Err(anyhow::anyhow!("Helius Sender error: {}", err_msg)); return Err(anyhow::anyhow!("Helius Sender error: {}", err_msg));
} }
if response_json.get("result").is_some() && crate::common::sdk_log::sdk_log_enabled() { if response_json.get("result").is_some() && crate::common::sdk_log::sdk_log_enabled() {
println!( println!(" [helius] {} submitted: {:?}", trade_type, start_time.elapsed());
" [helius] {} submitted: {:?}",
trade_type,
start_time.elapsed()
);
} }
} else if crate::common::sdk_log::sdk_log_enabled() { } else if crate::common::sdk_log::sdk_log_enabled() {
eprintln!( eprintln!(" [helius] {} submission failed: {:?}", trade_type, response_text);
" [helius] {} submission failed: {:?}",
trade_type, response_text
);
} }
match poll_transaction_confirmation(&self.rpc_client, signature, wait_confirmation).await { match poll_transaction_confirmation(&self.rpc_client, signature, wait_confirmation).await {
@@ -159,15 +149,8 @@ impl HeliusClient {
} }
} }
if wait_confirmation && crate::common::sdk_log::sdk_log_enabled() { if wait_confirmation && crate::common::sdk_log::sdk_log_enabled() {
println!( println!(" signature: {:?}", signature);
" signature: {:?}", println!(" [helius] {} confirmed: {:?}", trade_type, start_time.elapsed());
signature
);
println!(
" [helius] {} confirmed: {:?}",
trade_type,
start_time.elapsed()
);
} }
Ok(()) Ok(())
} }
@@ -191,8 +174,7 @@ impl SwqosClientTrait for HeliusClient {
wait_confirmation: bool, wait_confirmation: bool,
) -> Result<()> { ) -> Result<()> {
for transaction in transactions { for transaction in transactions {
self.send_transaction(trade_type, transaction, wait_confirmation) self.send_transaction(trade_type, transaction, wait_confirmation).await?;
.await?;
} }
Ok(()) Ok(())
} }
+38 -17
View File
@@ -1,5 +1,7 @@
use crate::swqos::common::{
use crate::swqos::common::{default_http_client_builder, poll_transaction_confirmation, serialize_transaction_and_encode, FormatBase64VersionedTransaction}; default_http_client_builder, poll_transaction_confirmation, serialize_transaction_and_encode,
FormatBase64VersionedTransaction,
};
use rand::seq::IndexedRandom; use rand::seq::IndexedRandom;
use reqwest::Client; use reqwest::Client;
use serde_json::json; use serde_json::json;
@@ -7,14 +9,13 @@ use std::{sync::Arc, time::Instant};
use solana_transaction_status::UiTransactionEncoding; use solana_transaction_status::UiTransactionEncoding;
use crate::swqos::SwqosClientTrait;
use crate::swqos::{SwqosType, TradeType};
use anyhow::Result; use anyhow::Result;
use solana_sdk::transaction::VersionedTransaction; use solana_sdk::transaction::VersionedTransaction;
use crate::swqos::{SwqosType, TradeType};
use crate::swqos::SwqosClientTrait;
use crate::{common::SolanaRpcClient, constants::swqos::JITO_TIP_ACCOUNTS}; use crate::{common::SolanaRpcClient, constants::swqos::JITO_TIP_ACCOUNTS};
pub struct JitoClient { pub struct JitoClient {
pub endpoint: String, pub endpoint: String,
pub auth_token: String, pub auth_token: String,
@@ -24,11 +25,21 @@ pub struct JitoClient {
#[async_trait::async_trait] #[async_trait::async_trait]
impl SwqosClientTrait for JitoClient { impl SwqosClientTrait for JitoClient {
async fn send_transaction(&self, trade_type: TradeType, transaction: &VersionedTransaction, wait_confirmation: bool) -> Result<()> { async fn send_transaction(
&self,
trade_type: TradeType,
transaction: &VersionedTransaction,
wait_confirmation: bool,
) -> Result<()> {
self.send_transaction_impl(trade_type, transaction, wait_confirmation).await self.send_transaction_impl(trade_type, transaction, wait_confirmation).await
} }
async fn send_transactions(&self, trade_type: TradeType, transactions: &Vec<VersionedTransaction>, wait_confirmation: bool) -> Result<()> { async fn send_transactions(
&self,
trade_type: TradeType,
transactions: &Vec<VersionedTransaction>,
wait_confirmation: bool,
) -> Result<()> {
self.send_transactions_impl(trade_type, transactions, wait_confirmation).await self.send_transactions_impl(trade_type, transactions, wait_confirmation).await
} }
@@ -52,9 +63,15 @@ impl JitoClient {
Self { rpc_client: Arc::new(rpc_client), endpoint, auth_token, http_client } Self { rpc_client: Arc::new(rpc_client), endpoint, auth_token, http_client }
} }
pub async fn send_transaction_impl(&self, trade_type: TradeType, transaction: &VersionedTransaction, wait_confirmation: bool) -> Result<()> { pub async fn send_transaction_impl(
&self,
trade_type: TradeType,
transaction: &VersionedTransaction,
wait_confirmation: bool,
) -> Result<()> {
let start_time = Instant::now(); let start_time = Instant::now();
let (content, signature) = serialize_transaction_and_encode(transaction, UiTransactionEncoding::Base64)?; let (content, signature) =
serialize_transaction_and_encode(transaction, UiTransactionEncoding::Base64)?;
let request_body = serde_json::to_string(&json!({ let request_body = serde_json::to_string(&json!({
"id": 1, "id": 1,
@@ -76,8 +93,7 @@ impl JitoClient {
let response = if self.auth_token.is_empty() { let response = if self.auth_token.is_empty() {
self.http_client.post(&endpoint) self.http_client.post(&endpoint)
} else { } else {
self.http_client.post(&endpoint) self.http_client.post(&endpoint).header("x-jito-auth", &self.auth_token)
.header("x-jito-auth", &self.auth_token)
}; };
let response_text = response let response_text = response
.body(request_body) .body(request_body)
@@ -104,7 +120,7 @@ impl JitoClient {
println!(" signature: {:?}", signature); println!(" signature: {:?}", signature);
println!(" [jito] {} confirmation failed: {:?}", trade_type, start_time.elapsed()); println!(" [jito] {} confirmation failed: {:?}", trade_type, start_time.elapsed());
return Err(e); return Err(e);
}, }
} }
if wait_confirmation { if wait_confirmation {
println!(" signature: {:?}", signature); println!(" signature: {:?}", signature);
@@ -114,9 +130,15 @@ impl JitoClient {
Ok(()) Ok(())
} }
pub async fn send_transactions_impl(&self, trade_type: TradeType, transactions: &Vec<VersionedTransaction>, _wait_confirmation: bool) -> Result<()> { pub async fn send_transactions_impl(
&self,
trade_type: TradeType,
transactions: &Vec<VersionedTransaction>,
_wait_confirmation: bool,
) -> Result<()> {
let start_time = Instant::now(); let start_time = Instant::now();
let txs_base64 = transactions.iter().map(|tx| tx.to_base64_string()).collect::<Vec<String>>(); let txs_base64 =
transactions.iter().map(|tx| tx.to_base64_string()).collect::<Vec<String>>();
let body = serde_json::json!({ let body = serde_json::json!({
"jsonrpc": "2.0", "jsonrpc": "2.0",
"method": "sendBundle", "method": "sendBundle",
@@ -135,8 +157,7 @@ impl JitoClient {
let response = if self.auth_token.is_empty() { let response = if self.auth_token.is_empty() {
self.http_client.post(&endpoint) self.http_client.post(&endpoint)
} else { } else {
self.http_client.post(&endpoint) self.http_client.post(&endpoint).header("x-jito-auth", &self.auth_token)
.header("x-jito-auth", &self.auth_token)
}; };
let response_text = response let response_text = response
.body(body.to_string()) .body(body.to_string())
@@ -156,4 +177,4 @@ impl JitoClient {
Ok(()) Ok(())
} }
} }
+44 -12
View File
@@ -1,4 +1,6 @@
use crate::swqos::common::{default_http_client_builder, poll_transaction_confirmation, serialize_transaction_and_encode}; use crate::swqos::common::{
default_http_client_builder, poll_transaction_confirmation, serialize_transaction_and_encode,
};
use rand::seq::IndexedRandom; use rand::seq::IndexedRandom;
use reqwest::Client; use reqwest::Client;
use serde_json::json; use serde_json::json;
@@ -6,10 +8,10 @@ use std::{sync::Arc, time::Instant};
use solana_transaction_status::UiTransactionEncoding; use solana_transaction_status::UiTransactionEncoding;
use crate::swqos::SwqosClientTrait;
use crate::swqos::{SwqosType, TradeType};
use anyhow::Result; use anyhow::Result;
use solana_sdk::transaction::VersionedTransaction; use solana_sdk::transaction::VersionedTransaction;
use crate::swqos::{SwqosType, TradeType};
use crate::swqos::SwqosClientTrait;
use crate::{common::SolanaRpcClient, constants::swqos::LIGHTSPEED_TIP_ACCOUNTS}; use crate::{common::SolanaRpcClient, constants::swqos::LIGHTSPEED_TIP_ACCOUNTS};
@@ -23,16 +25,29 @@ pub struct LightspeedClient {
#[async_trait::async_trait] #[async_trait::async_trait]
impl SwqosClientTrait for LightspeedClient { impl SwqosClientTrait for LightspeedClient {
async fn send_transaction(&self, trade_type: TradeType, transaction: &VersionedTransaction, wait_confirmation: bool) -> Result<()> { async fn send_transaction(
&self,
trade_type: TradeType,
transaction: &VersionedTransaction,
wait_confirmation: bool,
) -> Result<()> {
self.send_transaction(trade_type, transaction, wait_confirmation).await self.send_transaction(trade_type, transaction, wait_confirmation).await
} }
async fn send_transactions(&self, trade_type: TradeType, transactions: &Vec<VersionedTransaction>, wait_confirmation: bool) -> Result<()> { async fn send_transactions(
&self,
trade_type: TradeType,
transactions: &Vec<VersionedTransaction>,
wait_confirmation: bool,
) -> Result<()> {
self.send_transactions(trade_type, transactions, wait_confirmation).await self.send_transactions(trade_type, transactions, wait_confirmation).await
} }
fn get_tip_account(&self) -> Result<String> { fn get_tip_account(&self) -> Result<String> {
let tip_account = *LIGHTSPEED_TIP_ACCOUNTS.choose(&mut rand::rng()).or_else(|| LIGHTSPEED_TIP_ACCOUNTS.first()).unwrap(); let tip_account = *LIGHTSPEED_TIP_ACCOUNTS
.choose(&mut rand::rng())
.or_else(|| LIGHTSPEED_TIP_ACCOUNTS.first())
.unwrap();
Ok(tip_account.to_string()) Ok(tip_account.to_string())
} }
@@ -50,9 +65,15 @@ impl LightspeedClient {
Self { rpc_client: Arc::new(rpc_client), endpoint, auth_token, http_client } Self { rpc_client: Arc::new(rpc_client), endpoint, auth_token, http_client }
} }
pub async fn send_transaction(&self, trade_type: TradeType, transaction: &VersionedTransaction, wait_confirmation: bool) -> Result<()> { pub async fn send_transaction(
&self,
trade_type: TradeType,
transaction: &VersionedTransaction,
wait_confirmation: bool,
) -> Result<()> {
let start_time = Instant::now(); let start_time = Instant::now();
let (content, signature) = serialize_transaction_and_encode(transaction, UiTransactionEncoding::Base64)?; let (content, signature) =
serialize_transaction_and_encode(transaction, UiTransactionEncoding::Base64)?;
// Lightspeed uses standard Solana JSON-RPC format for sendTransaction // Lightspeed uses standard Solana JSON-RPC format for sendTransaction
let request_body = serde_json::to_string(&json!({ let request_body = serde_json::to_string(&json!({
@@ -70,7 +91,9 @@ impl LightspeedClient {
] ]
}))?; }))?;
let response_text = self.http_client.post(&self.endpoint) let response_text = self
.http_client
.post(&self.endpoint)
.body(request_body) .body(request_body)
.header("Content-Type", "application/json") .header("Content-Type", "application/json")
.send() .send()
@@ -93,9 +116,13 @@ impl LightspeedClient {
Ok(_) => (), Ok(_) => (),
Err(e) => { Err(e) => {
println!(" signature: {:?}", signature); println!(" signature: {:?}", signature);
println!(" [lightspeed] {} confirmation failed: {:?}", trade_type, start_time.elapsed()); println!(
" [lightspeed] {} confirmation failed: {:?}",
trade_type,
start_time.elapsed()
);
return Err(e); return Err(e);
}, }
} }
if wait_confirmation { if wait_confirmation {
println!(" signature: {:?}", signature); println!(" signature: {:?}", signature);
@@ -105,7 +132,12 @@ impl LightspeedClient {
Ok(()) Ok(())
} }
pub async fn send_transactions(&self, trade_type: TradeType, transactions: &Vec<VersionedTransaction>, wait_confirmation: bool) -> Result<()> { pub async fn send_transactions(
&self,
trade_type: TradeType,
transactions: &Vec<VersionedTransaction>,
wait_confirmation: bool,
) -> Result<()> {
for transaction in transactions { for transaction in transactions {
self.send_transaction(trade_type, transaction, wait_confirmation).await?; self.send_transaction(trade_type, transaction, wait_confirmation).await?;
} }
+137 -163
View File
@@ -1,20 +1,22 @@
pub mod astralane;
pub mod astralane_quic;
pub mod blockrazor;
pub mod bloxroute;
pub mod common; pub mod common;
pub mod flashblock;
pub mod helius;
pub mod jito;
pub mod lightspeed;
pub mod nextblock;
pub mod node1;
pub mod node1_quic;
pub mod serialization; pub mod serialization;
pub mod solana_rpc; pub mod solana_rpc;
pub mod jito;
pub mod nextblock;
pub mod zeroslot;
pub mod temporal;
pub mod bloxroute;
pub mod node1;
pub mod flashblock;
pub mod blockrazor;
pub mod astralane;
pub mod stellium;
pub mod lightspeed;
pub mod soyas; pub mod soyas;
pub mod speedlanding; pub mod speedlanding;
pub mod helius; pub mod stellium;
pub mod temporal;
pub mod zeroslot;
use std::sync::Arc; use std::sync::Arc;
@@ -27,55 +29,30 @@ use anyhow::Result;
use crate::{ use crate::{
common::SolanaRpcClient, common::SolanaRpcClient,
constants::swqos::{ constants::swqos::{
SWQOS_ENDPOINTS_BLOX, SWQOS_ENDPOINTS_ASTRALANE, SWQOS_ENDPOINTS_ASTRALANE_QUIC, SWQOS_ENDPOINTS_BLOCKRAZOR,
SWQOS_ENDPOINTS_JITO, SWQOS_ENDPOINTS_BLOX, SWQOS_ENDPOINTS_FLASHBLOCK, SWQOS_ENDPOINTS_HELIUS,
SWQOS_ENDPOINTS_NEXTBLOCK, SWQOS_ENDPOINTS_JITO, SWQOS_ENDPOINTS_NEXTBLOCK, SWQOS_ENDPOINTS_NODE1,
SWQOS_ENDPOINTS_TEMPORAL, SWQOS_ENDPOINTS_NODE1_QUIC, SWQOS_ENDPOINTS_SOYAS, SWQOS_ENDPOINTS_SPEEDLANDING,
SWQOS_ENDPOINTS_ZERO_SLOT, SWQOS_ENDPOINTS_STELLIUM, SWQOS_ENDPOINTS_TEMPORAL, SWQOS_ENDPOINTS_ZERO_SLOT,
SWQOS_ENDPOINTS_NODE1, SWQOS_MIN_TIP_ASTRALANE, SWQOS_MIN_TIP_BLOCKRAZOR, SWQOS_MIN_TIP_BLOXROUTE,
SWQOS_ENDPOINTS_FLASHBLOCK, SWQOS_MIN_TIP_DEFAULT, SWQOS_MIN_TIP_FLASHBLOCK, SWQOS_MIN_TIP_HELIUS, SWQOS_MIN_TIP_JITO,
SWQOS_ENDPOINTS_BLOCKRAZOR, SWQOS_MIN_TIP_LIGHTSPEED, SWQOS_MIN_TIP_NEXTBLOCK, SWQOS_MIN_TIP_NODE1,
SWQOS_ENDPOINTS_ASTRALANE, SWQOS_MIN_TIP_SOYAS, SWQOS_MIN_TIP_SPEEDLANDING, SWQOS_MIN_TIP_STELLIUM,
SWQOS_ENDPOINTS_STELLIUM, SWQOS_MIN_TIP_TEMPORAL, SWQOS_MIN_TIP_ZERO_SLOT,
SWQOS_ENDPOINTS_SOYAS,
SWQOS_ENDPOINTS_SPEEDLANDING,
SWQOS_ENDPOINTS_HELIUS,
SWQOS_MIN_TIP_DEFAULT,
SWQOS_MIN_TIP_JITO,
SWQOS_MIN_TIP_NEXTBLOCK,
SWQOS_MIN_TIP_ZERO_SLOT,
SWQOS_MIN_TIP_TEMPORAL,
SWQOS_MIN_TIP_BLOXROUTE,
SWQOS_MIN_TIP_NODE1,
SWQOS_MIN_TIP_FLASHBLOCK,
SWQOS_MIN_TIP_BLOCKRAZOR,
SWQOS_MIN_TIP_ASTRALANE,
SWQOS_MIN_TIP_STELLIUM,
SWQOS_MIN_TIP_LIGHTSPEED,
SWQOS_MIN_TIP_SOYAS,
SWQOS_MIN_TIP_SPEEDLANDING,
SWQOS_MIN_TIP_HELIUS,
}, },
swqos::{ swqos::{
bloxroute::BloxrouteClient, astralane::AstralaneClient, blockrazor::BlockRazorClient, bloxroute::BloxrouteClient,
jito::JitoClient, flashblock::FlashBlockClient, helius::HeliusClient, jito::JitoClient,
nextblock::NextBlockClient, lightspeed::LightspeedClient, nextblock::NextBlockClient, node1::Node1Client,
solana_rpc::SolRpcClient, node1_quic::Node1QuicClient, solana_rpc::SolRpcClient, soyas::SoyasClient,
temporal::TemporalClient, speedlanding::SpeedlandingClient, stellium::StelliumClient, temporal::TemporalClient,
zeroslot::ZeroSlotClient, zeroslot::ZeroSlotClient,
node1::Node1Client, },
flashblock::FlashBlockClient,
blockrazor::BlockRazorClient,
astralane::AstralaneClient,
stellium::StelliumClient,
lightspeed::LightspeedClient,
soyas::SoyasClient,
speedlanding::SpeedlandingClient,
helius::HeliusClient,
}
}; };
lazy_static::lazy_static! { lazy_static::lazy_static! {
/// Reserved for future per-SWQOS tip account caching (currently unused).
#[allow(dead_code)]
static ref TIP_ACCOUNT_CACHE: RwLock<Vec<String>> = RwLock::new(Vec::new()); static ref TIP_ACCOUNT_CACHE: RwLock<Vec<String>> = RwLock::new(Vec::new());
} }
@@ -83,9 +60,17 @@ lazy_static::lazy_static! {
/// Providers added here will be disabled even if configured by user /// Providers added here will be disabled even if configured by user
/// To enable a provider, remove it from this list /// To enable a provider, remove it from this list
pub const SWQOS_BLACKLIST: &[SwqosType] = &[ pub const SWQOS_BLACKLIST: &[SwqosType] = &[
SwqosType::NextBlock, // NextBlock is disabled by default SwqosType::NextBlock, // NextBlock is disabled by default
]; ];
/// SWQOS 提交通道:HTTP 或 QUIC(低延迟)。部分提供商(如 Astralane)支持 QUIC。
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash, Default)]
pub enum SwqosTransport {
#[default]
Http,
Quic,
}
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)] #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
pub enum TradeType { pub enum TradeType {
Create, Create,
@@ -151,8 +136,18 @@ pub type SwqosClient = dyn SwqosClientTrait + Send + Sync + 'static;
#[async_trait::async_trait] #[async_trait::async_trait]
pub trait SwqosClientTrait { pub trait SwqosClientTrait {
async fn send_transaction(&self, trade_type: TradeType, transaction: &VersionedTransaction, wait_confirmation: bool) -> Result<()>; async fn send_transaction(
async fn send_transactions(&self, trade_type: TradeType, transactions: &Vec<VersionedTransaction>, wait_confirmation: bool) -> Result<()>; &self,
trade_type: TradeType,
transaction: &VersionedTransaction,
wait_confirmation: bool,
) -> Result<()>;
async fn send_transactions(
&self,
trade_type: TradeType,
transactions: &Vec<VersionedTransaction>,
wait_confirmation: bool,
) -> Result<()>;
fn get_tip_account(&self) -> Result<String>; fn get_tip_account(&self) -> Result<String>;
fn get_swqos_type(&self) -> SwqosType; fn get_swqos_type(&self) -> SwqosType;
/// Minimum tip in SOL required by this provider. Helius returns lower value when swqos_only is true. /// Minimum tip in SOL required by this provider. Helius returns lower value when swqos_only is true.
@@ -203,14 +198,14 @@ pub enum SwqosConfig {
Temporal(String, SwqosRegion, Option<String>), Temporal(String, SwqosRegion, Option<String>),
/// ZeroSlot(api_token, region, custom_url) /// ZeroSlot(api_token, region, custom_url)
ZeroSlot(String, SwqosRegion, Option<String>), ZeroSlot(String, SwqosRegion, Option<String>),
/// Node1(api_token, region, custom_url) /// Node1(api_token, region, custom_url, transport). transport=None => HTTP; Some(Quic) => QUIC (port 16666, UUID auth).
Node1(String, SwqosRegion, Option<String>), Node1(String, SwqosRegion, Option<String>, Option<SwqosTransport>),
/// FlashBlock(api_token, region, custom_url) /// FlashBlock(api_token, region, custom_url)
FlashBlock(String, SwqosRegion, Option<String>), FlashBlock(String, SwqosRegion, Option<String>),
/// BlockRazor(api_token, region, custom_url) /// BlockRazor(api_token, region, custom_url)
BlockRazor(String, SwqosRegion, Option<String>), BlockRazor(String, SwqosRegion, Option<String>),
/// Astralane(api_token, region, custom_url) /// Astralane(api_token, region, custom_url, transport). transport=None 表示 Http。
Astralane(String, SwqosRegion, Option<String>), Astralane(String, SwqosRegion, Option<String>, Option<SwqosTransport>),
/// Stellium(api_token, region, custom_url) /// Stellium(api_token, region, custom_url)
Stellium(String, SwqosRegion, Option<String>), Stellium(String, SwqosRegion, Option<String>),
/// Lightspeed(api_key, region, custom_url) - Solana Vibe Station /// Lightspeed(api_key, region, custom_url) - Solana Vibe Station
@@ -228,7 +223,7 @@ pub enum SwqosConfig {
} }
impl SwqosConfig { impl SwqosConfig {
pub fn swqos_type(&self) -> SwqosType{ pub fn swqos_type(&self) -> SwqosType {
match self { match self {
SwqosConfig::Default(_) => SwqosType::Default, SwqosConfig::Default(_) => SwqosType::Default,
SwqosConfig::Jito(_, _, _) => SwqosType::Jito, SwqosConfig::Jito(_, _, _) => SwqosType::Jito,
@@ -236,10 +231,10 @@ impl SwqosConfig {
SwqosConfig::Bloxroute(_, _, _) => SwqosType::Bloxroute, SwqosConfig::Bloxroute(_, _, _) => SwqosType::Bloxroute,
SwqosConfig::Temporal(_, _, _) => SwqosType::Temporal, SwqosConfig::Temporal(_, _, _) => SwqosType::Temporal,
SwqosConfig::ZeroSlot(_, _, _) => SwqosType::ZeroSlot, SwqosConfig::ZeroSlot(_, _, _) => SwqosType::ZeroSlot,
SwqosConfig::Node1(_, _, _) => SwqosType::Node1, SwqosConfig::Node1(_, _, _, _) => SwqosType::Node1,
SwqosConfig::FlashBlock(_, _, _) => SwqosType::FlashBlock, SwqosConfig::FlashBlock(_, _, _) => SwqosType::FlashBlock,
SwqosConfig::BlockRazor(_, _, _) => SwqosType::BlockRazor, SwqosConfig::BlockRazor(_, _, _) => SwqosType::BlockRazor,
SwqosConfig::Astralane(_, _, _) => SwqosType::Astralane, SwqosConfig::Astralane(_, _, _, _) => SwqosType::Astralane,
SwqosConfig::Stellium(_, _, _) => SwqosType::Stellium, SwqosConfig::Stellium(_, _, _) => SwqosType::Stellium,
SwqosConfig::Lightspeed(_, _, _) => SwqosType::Lightspeed, SwqosConfig::Lightspeed(_, _, _) => SwqosType::Lightspeed,
SwqosConfig::Soyas(_, _, _) => SwqosType::Soyas, SwqosConfig::Soyas(_, _, _) => SwqosType::Soyas,
@@ -277,145 +272,124 @@ impl SwqosConfig {
} }
} }
pub async fn get_swqos_client(rpc_url: String, commitment: CommitmentConfig, swqos_config: SwqosConfig) -> Result<Arc<SwqosClient>> { pub async fn get_swqos_client(
rpc_url: String,
commitment: CommitmentConfig,
swqos_config: SwqosConfig,
) -> Result<Arc<SwqosClient>> {
match swqos_config { match swqos_config {
SwqosConfig::Jito(auth_token, region, url) => { SwqosConfig::Jito(auth_token, region, url) => {
let endpoint = SwqosConfig::get_endpoint(SwqosType::Jito, region, url); let endpoint = SwqosConfig::get_endpoint(SwqosType::Jito, region, url);
let jito_client = JitoClient::new( let jito_client = JitoClient::new(rpc_url.clone(), endpoint, auth_token);
rpc_url.clone(),
endpoint,
auth_token
);
Ok(Arc::new(jito_client)) Ok(Arc::new(jito_client))
} }
SwqosConfig::NextBlock(auth_token, region, url) => { SwqosConfig::NextBlock(auth_token, region, url) => {
let endpoint = SwqosConfig::get_endpoint(SwqosType::NextBlock, region, url); let endpoint = SwqosConfig::get_endpoint(SwqosType::NextBlock, region, url);
let nextblock_client = NextBlockClient::new( let nextblock_client =
rpc_url.clone(), NextBlockClient::new(rpc_url.clone(), endpoint.to_string(), auth_token);
endpoint.to_string(),
auth_token
);
Ok(Arc::new(nextblock_client)) Ok(Arc::new(nextblock_client))
}, }
SwqosConfig::ZeroSlot(auth_token, region, url) => { SwqosConfig::ZeroSlot(auth_token, region, url) => {
let endpoint = SwqosConfig::get_endpoint(SwqosType::ZeroSlot, region, url); let endpoint = SwqosConfig::get_endpoint(SwqosType::ZeroSlot, region, url);
let zeroslot_client = ZeroSlotClient::new( let zeroslot_client =
rpc_url.clone(), ZeroSlotClient::new(rpc_url.clone(), endpoint.to_string(), auth_token);
endpoint.to_string(),
auth_token
);
Ok(Arc::new(zeroslot_client)) Ok(Arc::new(zeroslot_client))
}, }
SwqosConfig::Temporal(auth_token, region, url) => { SwqosConfig::Temporal(auth_token, region, url) => {
let endpoint = SwqosConfig::get_endpoint(SwqosType::Temporal, region, url); let endpoint = SwqosConfig::get_endpoint(SwqosType::Temporal, region, url);
let temporal_client = TemporalClient::new( let temporal_client =
rpc_url.clone(), TemporalClient::new(rpc_url.clone(), endpoint.to_string(), auth_token);
endpoint.to_string(),
auth_token
);
Ok(Arc::new(temporal_client)) Ok(Arc::new(temporal_client))
}, }
SwqosConfig::Bloxroute(auth_token, region, url) => { SwqosConfig::Bloxroute(auth_token, region, url) => {
let endpoint = SwqosConfig::get_endpoint(SwqosType::Bloxroute, region, url); let endpoint = SwqosConfig::get_endpoint(SwqosType::Bloxroute, region, url);
let bloxroute_client = BloxrouteClient::new( let bloxroute_client =
rpc_url.clone(), BloxrouteClient::new(rpc_url.clone(), endpoint.to_string(), auth_token);
endpoint.to_string(),
auth_token
);
Ok(Arc::new(bloxroute_client)) Ok(Arc::new(bloxroute_client))
}, }
SwqosConfig::Node1(auth_token, region, url) => { SwqosConfig::Node1(auth_token, region, url, transport) => {
let endpoint = SwqosConfig::get_endpoint(SwqosType::Node1, region, url); let use_quic = transport.map_or(false, |t| t == SwqosTransport::Quic);
let node1_client = Node1Client::new( if use_quic {
rpc_url.clone(), let quic_endpoint = url
endpoint.to_string(), .unwrap_or_else(|| SWQOS_ENDPOINTS_NODE1_QUIC[region as usize].to_string());
auth_token let node1_quic =
); Node1QuicClient::connect(&quic_endpoint, &auth_token, rpc_url.clone())
Ok(Arc::new(node1_client)) .await?;
}, Ok(Arc::new(node1_quic))
} else {
let endpoint = SwqosConfig::get_endpoint(SwqosType::Node1, region, url);
let node1_client =
Node1Client::new(rpc_url.clone(), endpoint.to_string(), auth_token);
Ok(Arc::new(node1_client))
}
}
SwqosConfig::FlashBlock(auth_token, region, url) => { SwqosConfig::FlashBlock(auth_token, region, url) => {
let endpoint = SwqosConfig::get_endpoint(SwqosType::FlashBlock, region, url); let endpoint = SwqosConfig::get_endpoint(SwqosType::FlashBlock, region, url);
let flashblock_client = FlashBlockClient::new( let flashblock_client =
rpc_url.clone(), FlashBlockClient::new(rpc_url.clone(), endpoint.to_string(), auth_token);
endpoint.to_string(),
auth_token
);
Ok(Arc::new(flashblock_client)) Ok(Arc::new(flashblock_client))
}, }
SwqosConfig::BlockRazor(auth_token, region, url) => { SwqosConfig::BlockRazor(auth_token, region, url) => {
let endpoint = SwqosConfig::get_endpoint(SwqosType::BlockRazor, region, url); let endpoint = SwqosConfig::get_endpoint(SwqosType::BlockRazor, region, url);
let blockrazor_client = BlockRazorClient::new( let blockrazor_client =
rpc_url.clone(), BlockRazorClient::new(rpc_url.clone(), endpoint.to_string(), auth_token);
endpoint.to_string(),
auth_token
);
Ok(Arc::new(blockrazor_client)) Ok(Arc::new(blockrazor_client))
}, }
SwqosConfig::Astralane(auth_token, region, url) => { SwqosConfig::Astralane(auth_token, region, url, transport) => {
let endpoint = SwqosConfig::get_endpoint(SwqosType::Astralane, region, url); let use_quic = transport.map_or(false, |t| t == SwqosTransport::Quic);
let astralane_client = AstralaneClient::new( if use_quic {
rpc_url.clone(), let quic_endpoint = url.unwrap_or_else(|| {
endpoint.to_string(), SWQOS_ENDPOINTS_ASTRALANE_QUIC[region as usize].to_string()
auth_token });
); let astralane_client =
Ok(Arc::new(astralane_client)) AstralaneClient::new_quic(rpc_url.clone(), &quic_endpoint, auth_token)
}, .await?;
Ok(Arc::new(astralane_client))
} else {
let endpoint = SwqosConfig::get_endpoint(SwqosType::Astralane, region, url);
let astralane_client =
AstralaneClient::new(rpc_url.clone(), endpoint.to_string(), auth_token);
Ok(Arc::new(astralane_client))
}
}
SwqosConfig::Stellium(auth_token, region, url) => { SwqosConfig::Stellium(auth_token, region, url) => {
let endpoint = SwqosConfig::get_endpoint(SwqosType::Stellium, region, url); let endpoint = SwqosConfig::get_endpoint(SwqosType::Stellium, region, url);
let stellium_client = StelliumClient::new( let stellium_client =
rpc_url.clone(), StelliumClient::new(rpc_url.clone(), endpoint.to_string(), auth_token);
endpoint.to_string(),
auth_token
);
Ok(Arc::new(stellium_client)) Ok(Arc::new(stellium_client))
}, }
SwqosConfig::Lightspeed(auth_token, region, url) => { SwqosConfig::Lightspeed(auth_token, region, url) => {
let endpoint = SwqosConfig::get_endpoint(SwqosType::Lightspeed, region, url); let endpoint = SwqosConfig::get_endpoint(SwqosType::Lightspeed, region, url);
let lightspeed_client = LightspeedClient::new( let lightspeed_client =
rpc_url.clone(), LightspeedClient::new(rpc_url.clone(), endpoint.to_string(), auth_token);
endpoint.to_string(),
auth_token
);
Ok(Arc::new(lightspeed_client)) Ok(Arc::new(lightspeed_client))
}, }
SwqosConfig::Soyas(auth_token, region, url) => { SwqosConfig::Soyas(auth_token, region, url) => {
let endpoint = SwqosConfig::get_endpoint(SwqosType::Soyas, region, url); let endpoint = SwqosConfig::get_endpoint(SwqosType::Soyas, region, url);
let soyas_client = SoyasClient::new( let soyas_client =
rpc_url.clone(), SoyasClient::new(rpc_url.clone(), endpoint.to_string(), auth_token).await?;
endpoint.to_string(),
auth_token
).await?;
Ok(Arc::new(soyas_client)) Ok(Arc::new(soyas_client))
}, }
SwqosConfig::Speedlanding(auth_token, region, url) => { SwqosConfig::Speedlanding(auth_token, region, url) => {
let endpoint = SwqosConfig::get_endpoint(SwqosType::Speedlanding, region, url); let endpoint = SwqosConfig::get_endpoint(SwqosType::Speedlanding, region, url);
let speedlanding_client = SpeedlandingClient::new( let speedlanding_client =
rpc_url.clone(), SpeedlandingClient::new(rpc_url.clone(), endpoint.to_string(), auth_token)
endpoint.to_string(), .await?;
auth_token
).await?;
Ok(Arc::new(speedlanding_client)) Ok(Arc::new(speedlanding_client))
}, }
SwqosConfig::Helius(api_key, region, url, swqos_only) => { SwqosConfig::Helius(api_key, region, url, swqos_only) => {
let swqos_only = swqos_only.unwrap_or(false); let swqos_only = swqos_only.unwrap_or(false);
let endpoint = SwqosConfig::get_endpoint(SwqosType::Helius, region, url.clone()); let endpoint = SwqosConfig::get_endpoint(SwqosType::Helius, region, url.clone());
let api_key_opt = if api_key.is_empty() { None } else { Some(api_key.clone()) }; let api_key_opt = if api_key.is_empty() { None } else { Some(api_key.clone()) };
let helius_client = HeliusClient::new( let helius_client =
rpc_url.clone(), HeliusClient::new(rpc_url.clone(), endpoint, api_key_opt, swqos_only);
endpoint,
api_key_opt,
swqos_only,
);
Ok(Arc::new(helius_client)) Ok(Arc::new(helius_client))
}, }
SwqosConfig::Default(endpoint) => { SwqosConfig::Default(endpoint) => {
let rpc = SolanaRpcClient::new_with_commitment( let rpc = SolanaRpcClient::new_with_commitment(endpoint, commitment);
endpoint,
commitment
);
let rpc_client = SolRpcClient::new(Arc::new(rpc)); let rpc_client = SolRpcClient::new(Arc::new(rpc));
Ok(Arc::new(rpc_client)) Ok(Arc::new(rpc_client))
} }
} }
} }
} }
+45 -13
View File
@@ -1,4 +1,6 @@
use crate::swqos::common::{default_http_client_builder, poll_transaction_confirmation, serialize_transaction_and_encode}; use crate::swqos::common::{
default_http_client_builder, poll_transaction_confirmation, serialize_transaction_and_encode,
};
use rand::seq::IndexedRandom; use rand::seq::IndexedRandom;
use reqwest::Client; use reqwest::Client;
use serde_json::json; use serde_json::json;
@@ -6,10 +8,10 @@ use std::{sync::Arc, time::Instant};
use solana_transaction_status::UiTransactionEncoding; use solana_transaction_status::UiTransactionEncoding;
use crate::swqos::SwqosClientTrait;
use crate::swqos::{SwqosType, TradeType};
use anyhow::Result; use anyhow::Result;
use solana_sdk::transaction::VersionedTransaction; use solana_sdk::transaction::VersionedTransaction;
use crate::swqos::{SwqosType, TradeType};
use crate::swqos::SwqosClientTrait;
use crate::{common::SolanaRpcClient, constants::swqos::NEXTBLOCK_TIP_ACCOUNTS}; use crate::{common::SolanaRpcClient, constants::swqos::NEXTBLOCK_TIP_ACCOUNTS};
@@ -23,16 +25,29 @@ pub struct NextBlockClient {
#[async_trait::async_trait] #[async_trait::async_trait]
impl SwqosClientTrait for NextBlockClient { impl SwqosClientTrait for NextBlockClient {
async fn send_transaction(&self, trade_type: TradeType, transaction: &VersionedTransaction, wait_confirmation: bool) -> Result<()> { async fn send_transaction(
&self,
trade_type: TradeType,
transaction: &VersionedTransaction,
wait_confirmation: bool,
) -> Result<()> {
self.send_transaction(trade_type, transaction, wait_confirmation).await self.send_transaction(trade_type, transaction, wait_confirmation).await
} }
async fn send_transactions(&self, trade_type: TradeType, transactions: &Vec<VersionedTransaction>, wait_confirmation: bool) -> Result<()> { async fn send_transactions(
&self,
trade_type: TradeType,
transactions: &Vec<VersionedTransaction>,
wait_confirmation: bool,
) -> Result<()> {
self.send_transactions(trade_type, transactions, wait_confirmation).await self.send_transactions(trade_type, transactions, wait_confirmation).await
} }
fn get_tip_account(&self) -> Result<String> { fn get_tip_account(&self) -> Result<String> {
let tip_account = *NEXTBLOCK_TIP_ACCOUNTS.choose(&mut rand::rng()).or_else(|| NEXTBLOCK_TIP_ACCOUNTS.first()).unwrap(); let tip_account = *NEXTBLOCK_TIP_ACCOUNTS
.choose(&mut rand::rng())
.or_else(|| NEXTBLOCK_TIP_ACCOUNTS.first())
.unwrap();
Ok(tip_account.to_string()) Ok(tip_account.to_string())
} }
@@ -54,9 +69,15 @@ impl NextBlockClient {
Self { rpc_client: Arc::new(rpc_client), endpoint, auth_token, http_client } Self { rpc_client: Arc::new(rpc_client), endpoint, auth_token, http_client }
} }
pub async fn send_transaction(&self, trade_type: TradeType, transaction: &VersionedTransaction, wait_confirmation: bool) -> Result<()> { pub async fn send_transaction(
&self,
trade_type: TradeType,
transaction: &VersionedTransaction,
wait_confirmation: bool,
) -> Result<()> {
let start_time = Instant::now(); let start_time = Instant::now();
let (content, signature) = serialize_transaction_and_encode(transaction, UiTransactionEncoding::Base64)?; let (content, signature) =
serialize_transaction_and_encode(transaction, UiTransactionEncoding::Base64)?;
let request_body = serde_json::to_string(&json!({ let request_body = serde_json::to_string(&json!({
"transaction": { "transaction": {
@@ -65,7 +86,9 @@ impl NextBlockClient {
"frontRunningProtection": false "frontRunningProtection": false
}))?; }))?;
let response_text = self.http_client.post(&self.endpoint) let response_text = self
.http_client
.post(&self.endpoint)
.body(request_body) .body(request_body)
.header("Authorization", &self.auth_token) .header("Authorization", &self.auth_token)
.header("Content-Type", "application/json") .header("Content-Type", "application/json")
@@ -89,9 +112,13 @@ impl NextBlockClient {
Ok(_) => (), Ok(_) => (),
Err(e) => { Err(e) => {
println!(" signature: {:?}", signature); println!(" signature: {:?}", signature);
println!(" [nextblock] {} confirmation failed: {:?}", trade_type, start_time.elapsed()); println!(
" [nextblock] {} confirmation failed: {:?}",
trade_type,
start_time.elapsed()
);
return Err(e); return Err(e);
}, }
} }
if wait_confirmation { if wait_confirmation {
println!(" signature: {:?}", signature); println!(" signature: {:?}", signature);
@@ -101,10 +128,15 @@ impl NextBlockClient {
Ok(()) Ok(())
} }
pub async fn send_transactions(&self, trade_type: TradeType, transactions: &Vec<VersionedTransaction>, wait_confirmation: bool) -> Result<()> { pub async fn send_transactions(
&self,
trade_type: TradeType,
transactions: &Vec<VersionedTransaction>,
wait_confirmation: bool,
) -> Result<()> {
for transaction in transactions { for transaction in transactions {
self.send_transaction(trade_type, transaction, wait_confirmation).await?; self.send_transaction(trade_type, transaction, wait_confirmation).await?;
} }
Ok(()) Ok(())
} }
} }
+65 -30
View File
@@ -1,21 +1,23 @@
use crate::swqos::common::{default_http_client_builder, poll_transaction_confirmation, serialize_transaction_and_encode}; use crate::swqos::common::{
default_http_client_builder, poll_transaction_confirmation, serialize_transaction_and_encode,
};
use rand::seq::IndexedRandom; use rand::seq::IndexedRandom;
use reqwest::Client; use reqwest::Client;
use serde_json::json; use serde_json::json;
use std::{sync::Arc, time::Instant}; use std::{sync::Arc, time::Instant};
use std::time::Duration;
use solana_transaction_status::UiTransactionEncoding; use solana_transaction_status::UiTransactionEncoding;
use std::time::Duration;
use crate::swqos::SwqosClientTrait;
use crate::swqos::{SwqosType, TradeType};
use anyhow::Result; use anyhow::Result;
use solana_sdk::transaction::VersionedTransaction; use solana_sdk::transaction::VersionedTransaction;
use crate::swqos::{SwqosType, TradeType};
use crate::swqos::SwqosClientTrait;
use crate::{common::SolanaRpcClient, constants::swqos::NODE1_TIP_ACCOUNTS}; use crate::{common::SolanaRpcClient, constants::swqos::NODE1_TIP_ACCOUNTS};
use tokio::task::JoinHandle;
use std::sync::atomic::{AtomicBool, Ordering}; use std::sync::atomic::{AtomicBool, Ordering};
use tokio::task::JoinHandle;
#[derive(Clone)] #[derive(Clone)]
pub struct Node1Client { pub struct Node1Client {
@@ -29,16 +31,29 @@ pub struct Node1Client {
#[async_trait::async_trait] #[async_trait::async_trait]
impl SwqosClientTrait for Node1Client { impl SwqosClientTrait for Node1Client {
async fn send_transaction(&self, trade_type: TradeType, transaction: &VersionedTransaction, wait_confirmation: bool) -> Result<()> { async fn send_transaction(
&self,
trade_type: TradeType,
transaction: &VersionedTransaction,
wait_confirmation: bool,
) -> Result<()> {
self.send_transaction(trade_type, transaction, wait_confirmation).await self.send_transaction(trade_type, transaction, wait_confirmation).await
} }
async fn send_transactions(&self, trade_type: TradeType, transactions: &Vec<VersionedTransaction>, wait_confirmation: bool) -> Result<()> { async fn send_transactions(
&self,
trade_type: TradeType,
transactions: &Vec<VersionedTransaction>,
wait_confirmation: bool,
) -> Result<()> {
self.send_transactions(trade_type, transactions, wait_confirmation).await self.send_transactions(trade_type, transactions, wait_confirmation).await
} }
fn get_tip_account(&self) -> Result<String> { fn get_tip_account(&self) -> Result<String> {
let tip_account = *NODE1_TIP_ACCOUNTS.choose(&mut rand::rng()).or_else(|| NODE1_TIP_ACCOUNTS.first()).unwrap(); let tip_account = *NODE1_TIP_ACCOUNTS
.choose(&mut rand::rng())
.or_else(|| NODE1_TIP_ACCOUNTS.first())
.unwrap();
Ok(tip_account.to_string()) Ok(tip_account.to_string())
} }
@@ -51,22 +66,22 @@ impl Node1Client {
pub fn new(rpc_url: String, endpoint: String, auth_token: String) -> Self { pub fn new(rpc_url: String, endpoint: String, auth_token: String) -> Self {
let rpc_client = SolanaRpcClient::new(rpc_url); let rpc_client = SolanaRpcClient::new(rpc_url);
let http_client = default_http_client_builder().build().unwrap(); let http_client = default_http_client_builder().build().unwrap();
let client = Self { let client = Self {
rpc_client: Arc::new(rpc_client), rpc_client: Arc::new(rpc_client),
endpoint, endpoint,
auth_token, auth_token,
http_client, http_client,
ping_handle: Arc::new(tokio::sync::Mutex::new(None)), ping_handle: Arc::new(tokio::sync::Mutex::new(None)),
stop_ping: Arc::new(AtomicBool::new(false)), stop_ping: Arc::new(AtomicBool::new(false)),
}; };
// Start ping task // Start ping task
let client_clone = client.clone(); let client_clone = client.clone();
tokio::spawn(async move { tokio::spawn(async move {
client_clone.start_ping_task().await; client_clone.start_ping_task().await;
}); });
client client
} }
@@ -76,7 +91,7 @@ impl Node1Client {
let auth_token = self.auth_token.clone(); let auth_token = self.auth_token.clone();
let http_client = self.http_client.clone(); let http_client = self.http_client.clone();
let stop_ping = self.stop_ping.clone(); let stop_ping = self.stop_ping.clone();
let handle = tokio::spawn(async move { let handle = tokio::spawn(async move {
// Immediate first ping to warm connection and reduce first-submit cold start latency // Immediate first ping to warm connection and reduce first-submit cold start latency
if let Err(e) = Self::send_ping_request(&http_client, &endpoint, &auth_token).await { if let Err(e) = Self::send_ping_request(&http_client, &endpoint, &auth_token).await {
@@ -90,14 +105,15 @@ impl Node1Client {
if stop_ping.load(Ordering::Relaxed) { if stop_ping.load(Ordering::Relaxed) {
break; break;
} }
if let Err(e) = Self::send_ping_request(&http_client, &endpoint, &auth_token).await { if let Err(e) = Self::send_ping_request(&http_client, &endpoint, &auth_token).await
{
if crate::common::sdk_log::sdk_log_enabled() { if crate::common::sdk_log::sdk_log_enabled() {
eprintln!("Node1 ping request failed: {}", e); eprintln!("Node1 ping request failed: {}", e);
} }
} }
} }
}); });
// Update ping_handle - use Mutex to safely update // Update ping_handle - use Mutex to safely update
{ {
let mut ping_guard = self.ping_handle.lock().await; let mut ping_guard = self.ping_handle.lock().await;
@@ -109,7 +125,11 @@ impl Node1Client {
} }
/// Send ping request to /ping endpoint /// Send ping request to /ping endpoint
async fn send_ping_request(http_client: &Client, endpoint: &str, _auth_token: &str) -> Result<()> { async fn send_ping_request(
http_client: &Client,
endpoint: &str,
_auth_token: &str,
) -> Result<()> {
// Build ping URL // Build ping URL
let ping_url = if endpoint.ends_with('/') { let ping_url = if endpoint.ends_with('/') {
format!("{}ping", endpoint) format!("{}ping", endpoint)
@@ -118,10 +138,8 @@ impl Node1Client {
}; };
// Short timeout for ping; consume body so connection is returned to pool for reuse by submit // Short timeout for ping; consume body so connection is returned to pool for reuse by submit
let response = http_client.get(&ping_url) let response =
.timeout(Duration::from_millis(1500)) http_client.get(&ping_url).timeout(Duration::from_millis(1500)).send().await?;
.send()
.await?;
let status = response.status(); let status = response.status();
let _ = response.bytes().await; let _ = response.bytes().await;
if !status.is_success() && crate::common::sdk_log::sdk_log_enabled() { if !status.is_success() && crate::common::sdk_log::sdk_log_enabled() {
@@ -130,9 +148,15 @@ impl Node1Client {
Ok(()) Ok(())
} }
pub async fn send_transaction(&self, trade_type: TradeType, transaction: &VersionedTransaction, wait_confirmation: bool) -> Result<()> { pub async fn send_transaction(
&self,
trade_type: TradeType,
transaction: &VersionedTransaction,
wait_confirmation: bool,
) -> Result<()> {
let start_time = Instant::now(); let start_time = Instant::now();
let (content, signature) = serialize_transaction_and_encode(transaction, UiTransactionEncoding::Base64)?; let (content, signature) =
serialize_transaction_and_encode(transaction, UiTransactionEncoding::Base64)?;
let request_body = serde_json::to_string(&json!({ let request_body = serde_json::to_string(&json!({
"jsonrpc": "2.0", "jsonrpc": "2.0",
@@ -145,7 +169,9 @@ impl Node1Client {
}))?; }))?;
// Node1 uses api-key header instead of URL parameter // Node1 uses api-key header instead of URL parameter
let response_text = self.http_client.post(&self.endpoint) let response_text = self
.http_client
.post(&self.endpoint)
.body(request_body) .body(request_body)
.header("Content-Type", "application/json") .header("Content-Type", "application/json")
.header("api-key", &self.auth_token) .header("api-key", &self.auth_token)
@@ -173,10 +199,14 @@ impl Node1Client {
Err(e) => { Err(e) => {
if crate::common::sdk_log::sdk_log_enabled() { if crate::common::sdk_log::sdk_log_enabled() {
println!(" signature: {:?}", signature); println!(" signature: {:?}", signature);
println!(" [node1] {} confirmation failed: {:?}", trade_type, start_time.elapsed()); println!(
" [node1] {} confirmation failed: {:?}",
trade_type,
start_time.elapsed()
);
} }
return Err(e); return Err(e);
}, }
} }
if wait_confirmation && crate::common::sdk_log::sdk_log_enabled() { if wait_confirmation && crate::common::sdk_log::sdk_log_enabled() {
println!(" signature: {:?}", signature); println!(" signature: {:?}", signature);
@@ -186,7 +216,12 @@ impl Node1Client {
Ok(()) Ok(())
} }
pub async fn send_transactions(&self, trade_type: TradeType, transactions: &Vec<VersionedTransaction>, wait_confirmation: bool) -> Result<()> { pub async fn send_transactions(
&self,
trade_type: TradeType,
transactions: &Vec<VersionedTransaction>,
wait_confirmation: bool,
) -> Result<()> {
for transaction in transactions { for transaction in transactions {
self.send_transaction(trade_type, transaction, wait_confirmation).await?; self.send_transaction(trade_type, transaction, wait_confirmation).await?;
} }
@@ -198,7 +233,7 @@ impl Drop for Node1Client {
fn drop(&mut self) { fn drop(&mut self) {
// Ensure ping task stops when client is destroyed // Ensure ping task stops when client is destroyed
self.stop_ping.store(true, Ordering::Relaxed); self.stop_ping.store(true, Ordering::Relaxed);
// Try to stop ping task immediately // Try to stop ping task immediately
// Use tokio::spawn to avoid blocking Drop // Use tokio::spawn to avoid blocking Drop
let ping_handle = self.ping_handle.clone(); let ping_handle = self.ping_handle.clone();
+306
View File
@@ -0,0 +1,306 @@
//! Node1 QUIC SWQOS client.
//!
//! Protocol: first bi stream = auth (16-byte UUID); each transaction uses a new bi stream.
//! Request body = bincode(VersionedTransaction); response = 2 bytes status (BE) + 4 bytes msg_len (BE) + msg.
//! Reuses a single authenticated connection; reconnects and re-auth when connection is closed.
use anyhow::{Context, Result};
use quinn::crypto::rustls::QuicClientConfig;
use quinn::{ClientConfig, Connection, Endpoint, IdleTimeout, RecvStream, TransportConfig};
use std::net::ToSocketAddrs;
use std::sync::Arc;
use std::time::Duration;
use tokio::sync::Mutex;
use tokio::time::timeout;
use uuid::Uuid;
use crate::common::SolanaRpcClient;
use crate::constants::swqos::NODE1_TIP_ACCOUNTS;
use crate::swqos::common::poll_transaction_confirmation;
use crate::swqos::{SwqosClientTrait, SwqosType, TradeType};
use rand::seq::IndexedRandom;
use solana_sdk::transaction::VersionedTransaction;
use std::time::Instant;
const CONNECT_TIMEOUT: Duration = Duration::from_secs(5);
const AUTH_TIMEOUT: Duration = Duration::from_secs(5);
const SEND_TIMEOUT: Duration = Duration::from_secs(5);
const KEEP_ALIVE_INTERVAL: Duration = Duration::from_secs(15);
const MAX_IDLE_TIMEOUT: Duration = Duration::from_secs(60);
const MAX_TX_SIZE: usize = 1232;
/// Node1 QUIC client: one authenticated connection, reuse for all transactions.
pub struct Node1QuicClient {
endpoint: Endpoint,
connection: Mutex<Connection>,
server_addr: String,
server_name: String,
api_key_uuid: [u8; 16],
rpc_client: Arc<SolanaRpcClient>,
}
impl Node1QuicClient {
/// Connect and authenticate. Reuse the returned client for all subsequent sends.
pub async fn connect(server_addr: &str, api_key: &str, rpc_url: String) -> Result<Self> {
let socket_addr = server_addr
.to_socket_addrs()
.context("resolve Node1 QUIC server address")?
.next()
.context("no socket address for Node1 QUIC")?;
let api_key_uuid =
Uuid::parse_str(api_key).context("Node1 API key must be a valid UUID")?;
let api_key_bytes: [u8; 16] = *api_key_uuid.as_bytes();
let server_name = server_addr.split(':').next().unwrap_or(server_addr);
let client_config = Self::build_client_config()?;
let mut endpoint =
Endpoint::client("0.0.0.0:0".parse()?).context("create QUIC endpoint")?;
endpoint.set_default_client_config(client_config);
let connecting =
endpoint.connect(socket_addr, server_name).context("Node1 QUIC connect failed")?;
let connection = timeout(CONNECT_TIMEOUT, connecting)
.await
.context("Node1 QUIC connect timeout")?
.context("Node1 QUIC handshake failed")?;
timeout(AUTH_TIMEOUT, Self::authenticate(&connection, &api_key_bytes))
.await
.context("Node1 QUIC auth timeout")??;
Ok(Self {
endpoint,
connection: Mutex::new(connection),
server_addr: server_addr.to_string(),
server_name: server_name.to_string(),
api_key_uuid: api_key_bytes,
rpc_client: Arc::new(SolanaRpcClient::new(rpc_url)),
})
}
fn build_client_config() -> Result<ClientConfig> {
let crypto = rustls::ClientConfig::builder()
.dangerous()
.with_custom_certificate_verifier(Arc::new(SkipServerVerification))
.with_no_client_auth();
let client_crypto = QuicClientConfig::try_from(crypto).context("build QUIC TLS config")?;
let mut client_config = ClientConfig::new(Arc::new(client_crypto));
let mut transport = TransportConfig::default();
transport.max_idle_timeout(Some(IdleTimeout::try_from(MAX_IDLE_TIMEOUT).unwrap()));
transport.keep_alive_interval(Some(KEEP_ALIVE_INTERVAL));
client_config.transport_config(Arc::new(transport));
Ok(client_config)
}
async fn authenticate(connection: &Connection, api_key_bytes: &[u8; 16]) -> Result<()> {
let (mut send, mut recv) = connection.open_bi().await.context("open_bi for auth")?;
send.write_all(api_key_bytes).await.context("write auth bytes")?;
send.finish().context("finish auth stream")?;
let mut reply = [0u8; 1];
recv.read_exact(&mut reply).await.context("read auth reply")?;
match reply[0] {
0 => Ok(()),
code => anyhow::bail!("Node1 QUIC auth rejected, reply={}", code),
}
}
async fn ensure_connected(&self) -> Result<Connection> {
let guard = self.connection.lock().await;
if let Some(_reason) = guard.close_reason() {
drop(guard);
let socket_addr = self
.server_addr
.to_socket_addrs()
.context("resolve Node1 QUIC server address")?
.next()
.context("no socket address")?;
let connecting = self
.endpoint
.connect(socket_addr, &self.server_name)
.context("Node1 QUIC reconnect failed")?;
let connection = timeout(CONNECT_TIMEOUT, connecting)
.await
.context("Node1 QUIC reconnect timeout")?
.context("Node1 QUIC re-handshake failed")?;
timeout(AUTH_TIMEOUT, Self::authenticate(&connection, &self.api_key_uuid))
.await
.context("Node1 QUIC re-auth timeout")??;
let mut g = self.connection.lock().await;
*g = connection.clone();
Ok(connection)
} else {
Ok(guard.clone())
}
}
async fn read_response(recv: &mut RecvStream) -> Result<(u16, String)> {
let mut header = [0u8; 6];
recv.read_exact(&mut header)
.await
.map_err(|e| anyhow::anyhow!("read response header: {:?}", e))?;
let status = u16::from_be_bytes(header[0..2].try_into().unwrap());
let msg_len = u32::from_be_bytes(header[2..6].try_into().unwrap()) as usize;
let mut msg = vec![0u8; msg_len];
if msg_len > 0 {
recv.read_exact(&mut msg)
.await
.map_err(|e| anyhow::anyhow!("read response body: {:?}", e))?;
}
Ok((status, String::from_utf8_lossy(&msg).into_owned()))
}
/// Send one transaction over QUIC (opens new bi stream, writes bincode tx, reads status+msg).
pub async fn send_transaction_bytes(&self, tx_bytes: &[u8]) -> Result<(u16, String)> {
if tx_bytes.len() > MAX_TX_SIZE {
anyhow::bail!(
"Node1 QUIC: transaction too large ({} > {})",
tx_bytes.len(),
MAX_TX_SIZE
);
}
let conn = self.ensure_connected().await?;
let (mut send, mut recv) = conn.open_bi().await.context("open_bi for tx")?;
send.write_all(tx_bytes).await.context("write tx")?;
send.finish().context("finish tx stream")?;
Self::read_response(&mut recv).await
}
}
#[async_trait::async_trait]
impl SwqosClientTrait for Node1QuicClient {
async fn send_transaction(
&self,
trade_type: TradeType,
transaction: &VersionedTransaction,
wait_confirmation: bool,
) -> Result<()> {
let start = Instant::now();
let signature = transaction.signatures.first().copied().unwrap_or_default();
let tx_bytes = bincode::serialize(transaction).context("Node1 QUIC: bincode serialize")?;
let (status, msg) = timeout(SEND_TIMEOUT, self.send_transaction_bytes(&tx_bytes))
.await
.context("Node1 QUIC send timeout")??;
if status != 200 {
if crate::common::sdk_log::sdk_log_enabled() {
eprintln!(
" [node1-quic] {} submit failed: status={} msg={}",
trade_type, status, msg
);
}
anyhow::bail!("Node1 QUIC submit failed: status={} msg={}", status, msg);
}
if crate::common::sdk_log::sdk_log_enabled() {
println!(" [node1-quic] {} submitted: {:?}", trade_type, start.elapsed());
}
let start = Instant::now();
match poll_transaction_confirmation(&self.rpc_client, signature, wait_confirmation).await {
Ok(_) => {
if wait_confirmation && crate::common::sdk_log::sdk_log_enabled() {
println!(" [node1-quic] {} confirmed: {:?}", trade_type, start.elapsed());
}
Ok(())
}
Err(e) => {
if crate::common::sdk_log::sdk_log_enabled() {
eprintln!(
" [node1-quic] {} confirmation failed: {:?}",
trade_type,
start.elapsed()
);
}
Err(e)
}
}
}
async fn send_transactions(
&self,
trade_type: TradeType,
transactions: &Vec<VersionedTransaction>,
wait_confirmation: bool,
) -> Result<()> {
for tx in transactions {
self.send_transaction(trade_type, tx, wait_confirmation).await?;
}
Ok(())
}
fn get_tip_account(&self) -> Result<String> {
let tip = *NODE1_TIP_ACCOUNTS
.choose(&mut rand::rng())
.or_else(|| NODE1_TIP_ACCOUNTS.first())
.unwrap();
Ok(tip.to_string())
}
fn get_swqos_type(&self) -> SwqosType {
SwqosType::Node1
}
}
impl Drop for Node1QuicClient {
fn drop(&mut self) {
self.connection.get_mut().close(0u32.into(), b"client closing");
}
}
#[derive(Debug)]
struct SkipServerVerification;
impl rustls::client::danger::ServerCertVerifier for SkipServerVerification {
fn verify_server_cert(
&self,
_: &rustls::pki_types::CertificateDer<'_>,
_: &[rustls::pki_types::CertificateDer<'_>],
_: &rustls::pki_types::ServerName<'_>,
_: &[u8],
_: rustls::pki_types::UnixTime,
) -> Result<rustls::client::danger::ServerCertVerified, rustls::Error> {
Ok(rustls::client::danger::ServerCertVerified::assertion())
}
fn verify_tls12_signature(
&self,
_: &[u8],
_: &rustls::pki_types::CertificateDer<'_>,
_: &rustls::DigitallySignedStruct,
) -> Result<rustls::client::danger::HandshakeSignatureValid, rustls::Error> {
Ok(rustls::client::danger::HandshakeSignatureValid::assertion())
}
fn verify_tls13_signature(
&self,
_: &[u8],
_: &rustls::pki_types::CertificateDer<'_>,
_: &rustls::DigitallySignedStruct,
) -> Result<rustls::client::danger::HandshakeSignatureValid, rustls::Error> {
Ok(rustls::client::danger::HandshakeSignatureValid::assertion())
}
fn supported_verify_schemes(&self) -> Vec<rustls::SignatureScheme> {
vec![
rustls::SignatureScheme::ECDSA_NISTP256_SHA256,
rustls::SignatureScheme::ECDSA_NISTP384_SHA384,
rustls::SignatureScheme::RSA_PSS_SHA256,
rustls::SignatureScheme::RSA_PSS_SHA384,
rustls::SignatureScheme::RSA_PSS_SHA512,
rustls::SignatureScheme::RSA_PKCS1_SHA256,
rustls::SignatureScheme::RSA_PKCS1_SHA384,
rustls::SignatureScheme::RSA_PKCS1_SHA512,
rustls::SignatureScheme::ED25519,
]
}
}
+144 -33
View File
@@ -1,18 +1,24 @@
//! Transaction serialization module. //! Transaction serialization module.
use crate::perf::{
compiler_optimization::CompileTimeOptimizedEventProcessor, simd::SIMDSerializer,
};
use anyhow::Result; use anyhow::Result;
use base64::Engine;
use base64::engine::general_purpose::STANDARD; use base64::engine::general_purpose::STANDARD;
use base64::Engine;
use crossbeam_queue::ArrayQueue;
use once_cell::sync::Lazy; use once_cell::sync::Lazy;
use solana_client::rpc_client::SerializableTransaction; use solana_client::rpc_client::SerializableTransaction;
use solana_sdk::signature::Signature; use solana_sdk::signature::Signature;
use solana_transaction_status::UiTransactionEncoding; use solana_transaction_status::UiTransactionEncoding;
use std::sync::Arc; use std::sync::Arc;
use crossbeam_queue::ArrayQueue;
use crate::perf::{ /// Max number of reusable buffers kept in the queue.
simd::SIMDSerializer, const SERIALIZER_POOL_SIZE: usize = 10_000;
compiler_optimization::CompileTimeOptimizedEventProcessor, /// Per-buffer reserved capacity (bytes).
}; const SERIALIZER_BUFFER_SIZE: usize = 256 * 1024;
/// Cold-start prewarm count. Keep small to avoid first-submit spikes.
const SERIALIZER_PREWARM_BUFFERS: usize = 64;
/// Zero-allocation serializer using a buffer pool to avoid runtime allocation. /// Zero-allocation serializer using a buffer pool to avoid runtime allocation.
pub struct ZeroAllocSerializer { pub struct ZeroAllocSerializer {
@@ -22,28 +28,30 @@ pub struct ZeroAllocSerializer {
impl ZeroAllocSerializer { impl ZeroAllocSerializer {
pub fn new(pool_size: usize, buffer_size: usize) -> Self { pub fn new(pool_size: usize, buffer_size: usize) -> Self {
let pool = ArrayQueue::new(pool_size); Self::new_with_prewarm(pool_size, buffer_size, SERIALIZER_PREWARM_BUFFERS)
// Pre-allocate buffers
for _ in 0..pool_size {
let mut buffer = Vec::with_capacity(buffer_size);
buffer.resize(buffer_size, 0);
let _ = pool.push(buffer);
}
Self {
buffer_pool: Arc::new(pool),
buffer_size,
}
} }
pub fn serialize_zero_alloc<T: serde::Serialize>(&self, data: &T, _label: &str) -> Result<Vec<u8>> { fn new_with_prewarm(pool_size: usize, buffer_size: usize, prewarm_buffers: usize) -> Self {
let pool = ArrayQueue::new(pool_size);
let prewarm_count = prewarm_buffers.min(pool_size);
// Prewarm only a small hot set to avoid large cold-start blocking.
// Remaining buffers are allocated lazily and returned to this pool.
for _ in 0..prewarm_count {
let _ = pool.push(Vec::with_capacity(buffer_size));
}
Self { buffer_pool: Arc::new(pool), buffer_size }
}
pub fn serialize_zero_alloc<T: serde::Serialize>(
&self,
data: &T,
_label: &str,
) -> Result<Vec<u8>> {
// Try to get a buffer from the pool // Try to get a buffer from the pool
let mut buffer = self.buffer_pool.pop().unwrap_or_else(|| { let mut buffer =
let mut buf = Vec::with_capacity(self.buffer_size); self.buffer_pool.pop().unwrap_or_else(|| Vec::with_capacity(self.buffer_size));
buf.resize(self.buffer_size, 0);
buf
});
// Serialize into buffer // Serialize into buffer
let serialized = bincode::serialize(data)?; let serialized = bincode::serialize(data)?;
@@ -67,12 +75,8 @@ impl ZeroAllocSerializer {
} }
/// Global serializer instance. /// Global serializer instance.
static SERIALIZER: Lazy<Arc<ZeroAllocSerializer>> = Lazy::new(|| { static SERIALIZER: Lazy<Arc<ZeroAllocSerializer>> =
Arc::new(ZeroAllocSerializer::new( Lazy::new(|| Arc::new(ZeroAllocSerializer::new(SERIALIZER_POOL_SIZE, SERIALIZER_BUFFER_SIZE)));
10_000, // Pool size
256 * 1024, // Buffer size: 256KB
))
});
/// Compile-time optimized event processor (zero runtime cost). /// Compile-time optimized event processor (zero runtime cost).
static COMPILE_TIME_PROCESSOR: CompileTimeOptimizedEventProcessor = static COMPILE_TIME_PROCESSOR: CompileTimeOptimizedEventProcessor =
@@ -101,7 +105,9 @@ impl Base64Encoder {
event_type: &str, event_type: &str,
) -> Result<String> { ) -> Result<String> {
let serialized = SERIALIZER.serialize_zero_alloc(value, event_type)?; let serialized = SERIALIZER.serialize_zero_alloc(value, event_type)?;
Ok(STANDARD.encode(&serialized)) let encoded = STANDARD.encode(&serialized);
SERIALIZER.return_buffer(serialized);
Ok(encoded)
} }
} }
@@ -227,6 +233,7 @@ pub fn get_serializer_stats() -> (usize, usize) {
#[cfg(test)] #[cfg(test)]
mod tests { mod tests {
use super::*; use super::*;
use std::time::Instant;
#[test] #[test]
fn test_base64_encode() { fn test_base64_encode() {
@@ -243,6 +250,110 @@ mod tests {
fn test_serializer_stats() { fn test_serializer_stats() {
let (available, capacity) = get_serializer_stats(); let (available, capacity) = get_serializer_stats();
assert!(available <= capacity); assert!(available <= capacity);
assert_eq!(capacity, 10_000); assert_eq!(capacity, SERIALIZER_POOL_SIZE);
}
#[test]
fn test_serializer_prewarm_is_bounded() {
let serializer = ZeroAllocSerializer::new_with_prewarm(128, 1024, 8);
let (available, capacity) = serializer.get_pool_stats();
assert_eq!(capacity, 128);
assert_eq!(available, 8);
}
#[test]
fn test_serializer_lazy_alloc_and_return() {
let serializer = ZeroAllocSerializer::new_with_prewarm(8, 1024, 0);
let (available_before, capacity) = serializer.get_pool_stats();
assert_eq!(capacity, 8);
assert_eq!(available_before, 0);
let buf = serializer.serialize_zero_alloc(&"hello", "test").unwrap();
assert!(buf.capacity() >= 1024);
serializer.return_buffer(buf);
let (available_after, _) = serializer.get_pool_stats();
assert_eq!(available_after, 1);
}
fn legacy_eager_zero_fill_serializer(
pool_size: usize,
buffer_size: usize,
) -> ZeroAllocSerializer {
let pool = ArrayQueue::new(pool_size);
for _ in 0..pool_size {
let mut buffer = Vec::with_capacity(buffer_size);
buffer.resize(buffer_size, 0);
let _ = pool.push(buffer);
}
ZeroAllocSerializer { buffer_pool: Arc::new(pool), buffer_size }
}
/// Manual perf test: compares old eager cold-start behavior to current bounded prewarm.
/// Run with:
/// cargo test --release perf_serializer_cold_start_vs_legacy_eager -- --ignored --nocapture
#[test]
#[ignore = "manual perf benchmark"]
fn perf_serializer_cold_start_vs_legacy_eager() {
const POOL_SIZE: usize = 4096;
const BUFFER_SIZE: usize = 32 * 1024;
const PREWARM: usize = 64;
let payload = vec![7u8; 4096];
let legacy_init_start = Instant::now();
let legacy = legacy_eager_zero_fill_serializer(POOL_SIZE, BUFFER_SIZE);
let legacy_init = legacy_init_start.elapsed();
let current_init_start = Instant::now();
let current = ZeroAllocSerializer::new_with_prewarm(POOL_SIZE, BUFFER_SIZE, PREWARM);
let current_init = current_init_start.elapsed();
let legacy_first_start = Instant::now();
let legacy_buf = legacy.serialize_zero_alloc(&payload, "perf").unwrap();
let legacy_first = legacy_first_start.elapsed();
legacy.return_buffer(legacy_buf);
let current_first_start = Instant::now();
let current_buf = current.serialize_zero_alloc(&payload, "perf").unwrap();
let current_first = current_first_start.elapsed();
current.return_buffer(current_buf);
println!(
"[perf] serializer cold-start compare\n pool_size={POOL_SIZE} buffer_size={BUFFER_SIZE} prewarm={PREWARM}\n legacy_init={legacy_init:?} current_init={current_init:?}\n legacy_first_serialize={legacy_first:?} current_first_serialize={current_first:?}"
);
assert!(
current_init <= legacy_init,
"expected bounded prewarm init ({current_init:?}) to be <= legacy eager init ({legacy_init:?})"
);
}
/// Manual perf test: demonstrates lazy allocation amortization.
/// Run with:
/// cargo test --release perf_serializer_lazy_growth_amortization -- --ignored --nocapture
#[test]
#[ignore = "manual perf benchmark"]
fn perf_serializer_lazy_growth_amortization() {
const POOL_SIZE: usize = 128;
const BUFFER_SIZE: usize = 256 * 1024;
let serializer = ZeroAllocSerializer::new_with_prewarm(POOL_SIZE, BUFFER_SIZE, 0);
let payload = vec![1u8; 8 * 1024];
let first_start = Instant::now();
let first_buf = serializer.serialize_zero_alloc(&payload, "perf").unwrap();
let first = first_start.elapsed();
serializer.return_buffer(first_buf);
let second_start = Instant::now();
let second_buf = serializer.serialize_zero_alloc(&payload, "perf").unwrap();
let second = second_start.elapsed();
serializer.return_buffer(second_buf);
let (available, capacity) = serializer.get_pool_stats();
println!(
"[perf] serializer lazy growth\n pool_size={POOL_SIZE} buffer_size={BUFFER_SIZE}\n first_serialize={first:?} second_serialize={second:?}\n available={available} capacity={capacity}"
);
assert!(available >= 1);
assert_eq!(capacity, POOL_SIZE);
} }
} }
+74 -22
View File
@@ -15,6 +15,7 @@ use std::{
time::Duration, time::Duration,
}; };
use tokio::sync::Mutex; use tokio::sync::Mutex;
use tokio::time::timeout;
use crate::common::SolanaRpcClient; use crate::common::SolanaRpcClient;
use crate::swqos::common::poll_transaction_confirmation; use crate::swqos::common::poll_transaction_confirmation;
@@ -26,22 +27,44 @@ use crate::{
}; };
const ALPN_TPU_PROTOCOL_ID: &[u8] = b"solana-tpu"; const ALPN_TPU_PROTOCOL_ID: &[u8] = b"solana-tpu";
const SPEED_SERVER: &str = "speed-landing"; /// Fallback SNI when endpoint is IP or cannot extract host (keeps legacy behavior).
const SPEED_SERVER_FALLBACK: &str = "speed-landing";
const KEEP_ALIVE_INTERVAL: Duration = Duration::from_secs(25); const KEEP_ALIVE_INTERVAL: Duration = Duration::from_secs(25);
const MAX_IDLE_TIMEOUT: Duration = Duration::from_secs(5 * 60); const MAX_IDLE_TIMEOUT: Duration = Duration::from_secs(5 * 60);
const CONNECT_TIMEOUT: Duration = Duration::from_secs(5);
const SEND_TIMEOUT: Duration = Duration::from_secs(5);
pub struct SpeedlandingClient { pub struct SpeedlandingClient {
pub rpc_client: Arc<SolanaRpcClient>, pub rpc_client: Arc<SolanaRpcClient>,
endpoint: Endpoint, endpoint: Endpoint,
client_config: ClientConfig, client_config: ClientConfig,
addr: SocketAddr, addr: SocketAddr,
/// TLS SNI: host from endpoint URL so server presents the right cert (e.g. nyc.speedlanding.trade).
server_name: String,
connection: ArcSwap<Connection>, connection: ArcSwap<Connection>,
reconnect: Mutex<()>, reconnect: Mutex<()>,
} }
impl SpeedlandingClient { impl SpeedlandingClient {
/// Extract TLS SNI (host) from endpoint URL. Uses fallback "speed-landing" for IP or when host cannot be determined.
fn server_name_from_endpoint(endpoint: &str) -> String {
let without_scheme = endpoint
.strip_prefix("https://")
.or_else(|| endpoint.strip_prefix("http://"))
.unwrap_or(endpoint);
let host = without_scheme.split(':').next().unwrap_or("").trim();
if host.is_empty() {
return SPEED_SERVER_FALLBACK.to_string();
}
if !host.chars().any(|c| c.is_ascii_alphabetic()) {
return SPEED_SERVER_FALLBACK.to_string();
}
host.to_string()
}
pub async fn new(rpc_url: String, endpoint_string: String, api_key: String) -> Result<Self> { pub async fn new(rpc_url: String, endpoint_string: String, api_key: String) -> Result<Self> {
let rpc_client = SolanaRpcClient::new(rpc_url); let rpc_client = SolanaRpcClient::new(rpc_url);
let server_name = Self::server_name_from_endpoint(&endpoint_string);
let keypair = Keypair::from_base58_string(&api_key); let keypair = Keypair::from_base58_string(&api_key);
let (cert, key) = new_dummy_x509_certificate(&keypair); let (cert, key) = new_dummy_x509_certificate(&keypair);
let mut crypto = rustls::ClientConfig::builder() let mut crypto = rustls::ClientConfig::builder()
@@ -66,26 +89,49 @@ impl SpeedlandingClient {
.to_socket_addrs()? .to_socket_addrs()?
.next() .next()
.ok_or_else(|| anyhow::anyhow!("Address not resolved"))?; .ok_or_else(|| anyhow::anyhow!("Address not resolved"))?;
let connection = endpoint.connect(addr, SPEED_SERVER)?.await?; let connecting = endpoint.connect(addr, &server_name)?;
let connection = timeout(CONNECT_TIMEOUT, connecting)
.await
.context("Speedlanding QUIC connect timeout")?
.context("Speedlanding QUIC handshake failed")?;
Ok(Self { Ok(Self {
rpc_client: Arc::new(rpc_client), rpc_client: Arc::new(rpc_client),
endpoint, endpoint,
client_config, client_config,
addr, addr,
server_name,
connection: ArcSwap::from_pointee(connection), connection: ArcSwap::from_pointee(connection),
reconnect: Mutex::new(()), reconnect: Mutex::new(()),
}) })
} }
async fn reconnect(&self) -> Result<()> { /// Ensure we have a live connection: if current one is closed, reconnect under lock so
let _guard = self.reconnect.try_lock()?; /// concurrent senders wait and then all use the new connection. Uses blocking lock so
let connection = self /// waiters get the updated connection.
.endpoint async fn ensure_connected(&self) -> Result<Arc<Connection>> {
.connect_with(self.client_config.clone(), self.addr, SPEED_SERVER)? let guard = self.reconnect.lock().await;
.await?; let current = self.connection.load_full();
self.connection.store(Arc::new(connection)); if current.close_reason().is_none() {
Ok(()) return Ok(current);
}
drop(guard);
let _guard = self.reconnect.lock().await;
let current = self.connection.load_full();
if current.close_reason().is_some() {
let connecting = self.endpoint.connect_with(
self.client_config.clone(),
self.addr,
self.server_name.as_str(),
)?;
let connection = timeout(CONNECT_TIMEOUT, connecting)
.await
.context("Speedlanding QUIC reconnect timeout")?
.context("Speedlanding QUIC re-handshake failed")?;
self.connection.store(Arc::new(connection));
return Ok(self.connection.load_full());
}
Ok(current)
} }
async fn try_send_bytes(connection: &Connection, payload: &[u8]) -> Result<()> { async fn try_send_bytes(connection: &Connection, payload: &[u8]) -> Result<()> {
@@ -106,26 +152,32 @@ impl SwqosClientTrait for SpeedlandingClient {
) -> Result<()> { ) -> Result<()> {
let start_time = Instant::now(); let start_time = Instant::now();
let (buf_guard, signature) = serialize_transaction_bincode_sync(transaction)?; let (buf_guard, signature) = serialize_transaction_bincode_sync(transaction)?;
let connection = self.connection.load_full(); let connection = self.ensure_connected().await?;
if Self::try_send_bytes(&connection, &*buf_guard).await.is_err() { let mut send_result =
timeout(SEND_TIMEOUT, Self::try_send_bytes(&connection, &*buf_guard)).await;
let need_retry = match &send_result {
Ok(Ok(())) => false,
Ok(Err(_)) | Err(_) => true,
};
if need_retry {
if crate::common::sdk_log::sdk_log_enabled() { if crate::common::sdk_log::sdk_log_enabled() {
eprintln!(" [speedlanding] {} submission failed, reconnecting", trade_type); eprintln!(" [speedlanding] {} send failed or timeout, reconnecting", trade_type);
}
self.reconnect().await?;
let connection = self.connection.load_full();
if let Err(e) = Self::try_send_bytes(&connection, &*buf_guard).await {
if crate::common::sdk_log::sdk_log_enabled() {
eprintln!(" [speedlanding] {} submission failed: {:?}", trade_type, e);
}
return Err(e.into());
} }
let connection = self.ensure_connected().await?;
send_result =
timeout(SEND_TIMEOUT, Self::try_send_bytes(&connection, &*buf_guard)).await;
} }
send_result.context("Speedlanding QUIC send timeout")??;
match poll_transaction_confirmation(&self.rpc_client, signature, wait_confirmation).await { match poll_transaction_confirmation(&self.rpc_client, signature, wait_confirmation).await {
Ok(_) => (), Ok(_) => (),
Err(e) => { Err(e) => {
if crate::common::sdk_log::sdk_log_enabled() { if crate::common::sdk_log::sdk_log_enabled() {
println!(" signature: {:?}", signature); println!(" signature: {:?}", signature);
println!(" [speedlanding] {} confirmation failed: {:?}", trade_type, start_time.elapsed()); println!(
" [speedlanding] {} confirmation failed: {:?}",
trade_type,
start_time.elapsed()
);
} }
return Err(e); return Err(e);
} }
+49 -16
View File
@@ -1,21 +1,22 @@
use crate::swqos::common::{default_http_client_builder, poll_transaction_confirmation, serialize_transaction_and_encode}; use crate::swqos::common::{
default_http_client_builder, poll_transaction_confirmation, serialize_transaction_and_encode,
};
use rand::seq::IndexedRandom; use rand::seq::IndexedRandom;
use reqwest::Client; use reqwest::Client;
use serde_json::json; use serde_json::json;
use std::{sync::Arc, time::Instant};
use std::sync::atomic::{AtomicBool, Ordering}; use std::sync::atomic::{AtomicBool, Ordering};
use std::{sync::Arc, time::Instant};
use std::time::Duration;
use solana_transaction_status::UiTransactionEncoding; use solana_transaction_status::UiTransactionEncoding;
use std::time::Duration;
use crate::swqos::SwqosClientTrait;
use crate::swqos::{SwqosType, TradeType};
use anyhow::Result; use anyhow::Result;
use solana_sdk::transaction::VersionedTransaction; use solana_sdk::transaction::VersionedTransaction;
use crate::swqos::{SwqosType, TradeType};
use crate::swqos::SwqosClientTrait;
use crate::{common::SolanaRpcClient, constants::swqos::STELLIUM_TIP_ACCOUNTS}; use crate::{common::SolanaRpcClient, constants::swqos::STELLIUM_TIP_ACCOUNTS};
#[derive(Clone)] #[derive(Clone)]
pub struct StelliumClient { pub struct StelliumClient {
pub endpoint: String, pub endpoint: String,
@@ -27,16 +28,29 @@ pub struct StelliumClient {
#[async_trait::async_trait] #[async_trait::async_trait]
impl SwqosClientTrait for StelliumClient { impl SwqosClientTrait for StelliumClient {
async fn send_transaction(&self, trade_type: TradeType, transaction: &VersionedTransaction, wait_confirmation: bool) -> Result<()> { async fn send_transaction(
&self,
trade_type: TradeType,
transaction: &VersionedTransaction,
wait_confirmation: bool,
) -> Result<()> {
self.send_transaction(trade_type, transaction, wait_confirmation).await self.send_transaction(trade_type, transaction, wait_confirmation).await
} }
async fn send_transactions(&self, trade_type: TradeType, transactions: &Vec<VersionedTransaction>, wait_confirmation: bool) -> Result<()> { async fn send_transactions(
&self,
trade_type: TradeType,
transactions: &Vec<VersionedTransaction>,
wait_confirmation: bool,
) -> Result<()> {
self.send_transactions(trade_type, transactions, wait_confirmation).await self.send_transactions(trade_type, transactions, wait_confirmation).await
} }
fn get_tip_account(&self) -> Result<String> { fn get_tip_account(&self) -> Result<String> {
let tip_account = *STELLIUM_TIP_ACCOUNTS.choose(&mut rand::rng()).or_else(|| STELLIUM_TIP_ACCOUNTS.first()).unwrap(); let tip_account = *STELLIUM_TIP_ACCOUNTS
.choose(&mut rand::rng())
.or_else(|| STELLIUM_TIP_ACCOUNTS.first())
.unwrap();
Ok(tip_account.to_string()) Ok(tip_account.to_string())
} }
@@ -79,7 +93,9 @@ impl StelliumClient {
tokio::spawn(async move { tokio::spawn(async move {
// Immediate first ping to warm connection and reduce first-submit cold start latency // Immediate first ping to warm connection and reduce first-submit cold start latency
let url = format!("{}/{}", endpoint, auth_token); let url = format!("{}/{}", endpoint, auth_token);
if let Ok(resp) = http_client.get(&url).timeout(Duration::from_millis(1500)).send().await { if let Ok(resp) =
http_client.get(&url).timeout(Duration::from_millis(1500)).send().await
{
let status = resp.status(); let status = resp.status();
let _ = resp.bytes().await; let _ = resp.bytes().await;
if !status.is_success() && crate::common::sdk_log::sdk_log_enabled() { if !status.is_success() && crate::common::sdk_log::sdk_log_enabled() {
@@ -111,9 +127,15 @@ impl StelliumClient {
}); });
} }
pub async fn send_transaction(&self, trade_type: TradeType, transaction: &VersionedTransaction, wait_confirmation: bool) -> Result<()> { pub async fn send_transaction(
&self,
trade_type: TradeType,
transaction: &VersionedTransaction,
wait_confirmation: bool,
) -> Result<()> {
let start_time = Instant::now(); let start_time = Instant::now();
let (content, signature) = serialize_transaction_and_encode(transaction, UiTransactionEncoding::Base64)?; let (content, signature) =
serialize_transaction_and_encode(transaction, UiTransactionEncoding::Base64)?;
// Stellium uses standard Solana sendTransaction format // Stellium uses standard Solana sendTransaction format
let request_body = serde_json::to_string(&json!({ let request_body = serde_json::to_string(&json!({
@@ -130,7 +152,9 @@ impl StelliumClient {
let url = format!("{}/{}", self.endpoint, self.auth_token); let url = format!("{}/{}", self.endpoint, self.auth_token);
// Send request to Stellium // Send request to Stellium
let response_text = self.http_client.post(&url) let response_text = self
.http_client
.post(&url)
.body(request_body) .body(request_body)
.header("Content-Type", "application/json") .header("Content-Type", "application/json")
.header("Connection", "keep-alive") .header("Connection", "keep-alive")
@@ -159,10 +183,14 @@ impl StelliumClient {
Err(e) => { Err(e) => {
if crate::common::sdk_log::sdk_log_enabled() { if crate::common::sdk_log::sdk_log_enabled() {
println!(" signature: {:?}", signature); println!(" signature: {:?}", signature);
println!(" [Stellium] {} confirmation failed: {:?}", trade_type, start_time.elapsed()); println!(
" [Stellium] {} confirmation failed: {:?}",
trade_type,
start_time.elapsed()
);
} }
return Err(e); return Err(e);
}, }
} }
if wait_confirmation && crate::common::sdk_log::sdk_log_enabled() { if wait_confirmation && crate::common::sdk_log::sdk_log_enabled() {
println!(" signature: {:?}", signature); println!(" signature: {:?}", signature);
@@ -172,7 +200,12 @@ impl StelliumClient {
Ok(()) Ok(())
} }
pub async fn send_transactions(&self, trade_type: TradeType, transactions: &Vec<VersionedTransaction>, wait_confirmation: bool) -> Result<()> { pub async fn send_transactions(
&self,
trade_type: TradeType,
transactions: &Vec<VersionedTransaction>,
wait_confirmation: bool,
) -> Result<()> {
for transaction in transactions { for transaction in transactions {
self.send_transaction(trade_type, transaction, wait_confirmation).await?; self.send_transaction(trade_type, transaction, wait_confirmation).await?;
} }
+73 -37
View File
@@ -1,27 +1,29 @@
use crate::swqos::common::{
use crate::swqos::common::{default_http_client_builder, poll_transaction_confirmation, serialize_transaction_and_encode}; default_http_client_builder, poll_transaction_confirmation, serialize_transaction_and_encode,
};
use rand::seq::IndexedRandom; use rand::seq::IndexedRandom;
use reqwest::Client; use reqwest::Client;
use serde_json::json; use serde_json::json;
use std::{sync::Arc, time::Instant}; use sha2::{Digest, Sha256};
use std::time::Duration;
use solana_transaction_status::UiTransactionEncoding; use solana_transaction_status::UiTransactionEncoding;
use sha2::{Sha256, Digest}; use std::time::Duration;
use std::{sync::Arc, time::Instant};
use crate::swqos::SwqosClientTrait;
use crate::swqos::{SwqosType, TradeType};
use anyhow::Result; use anyhow::Result;
use solana_sdk::transaction::VersionedTransaction; use solana_sdk::transaction::VersionedTransaction;
use crate::swqos::{SwqosType, TradeType};
use crate::swqos::SwqosClientTrait;
use crate::{common::SolanaRpcClient, constants::swqos::NOZOMI_TIP_ACCOUNTS}; use crate::{common::SolanaRpcClient, constants::swqos::NOZOMI_TIP_ACCOUNTS};
use tokio::task::JoinHandle;
use std::sync::atomic::{AtomicBool, Ordering}; use std::sync::atomic::{AtomicBool, Ordering};
use tokio::task::JoinHandle;
const SPECIAL_API_KEY_PREFIX: &str = "298b5025"; const SPECIAL_API_KEY_PREFIX: &str = "298b5025";
const SPECIAL_API_KEY_SUFFIX: &str = "a055323"; const SPECIAL_API_KEY_SUFFIX: &str = "a055323";
const SPECIAL_API_KEY_HASH: &str = "e7be933c8058aebcb4d08a6120fb4dfd2ead568d42527a3fc2b60a703f25e48d"; const SPECIAL_API_KEY_HASH: &str =
"e7be933c8058aebcb4d08a6120fb4dfd2ead568d42527a3fc2b60a703f25e48d";
const TEMPORAL_COMMUNITY_TIP_ADDRESS: &str = "mwGELGMgGGrNL1UibNCQeJHDE7qdPptWRYB6noUHmTj"; const TEMPORAL_COMMUNITY_TIP_ADDRESS: &str = "mwGELGMgGGrNL1UibNCQeJHDE7qdPptWRYB6noUHmTj";
#[inline] #[inline]
@@ -31,7 +33,6 @@ fn fast_sha256_hex(input: &str) -> String {
format!("{:x}", hasher.finalize()) format!("{:x}", hasher.finalize())
} }
#[derive(Clone)] #[derive(Clone)]
pub struct TemporalClient { pub struct TemporalClient {
pub rpc_client: Arc<SolanaRpcClient>, pub rpc_client: Arc<SolanaRpcClient>,
@@ -44,18 +45,30 @@ pub struct TemporalClient {
#[async_trait::async_trait] #[async_trait::async_trait]
impl SwqosClientTrait for TemporalClient { impl SwqosClientTrait for TemporalClient {
async fn send_transaction(&self, trade_type: TradeType, transaction: &VersionedTransaction, wait_confirmation: bool) -> Result<()> { async fn send_transaction(
&self,
trade_type: TradeType,
transaction: &VersionedTransaction,
wait_confirmation: bool,
) -> Result<()> {
self.send_transaction(trade_type, transaction, wait_confirmation).await self.send_transaction(trade_type, transaction, wait_confirmation).await
} }
async fn send_transactions(&self, trade_type: TradeType, transactions: &Vec<VersionedTransaction>, wait_confirmation: bool) -> Result<()> { async fn send_transactions(
&self,
trade_type: TradeType,
transactions: &Vec<VersionedTransaction>,
wait_confirmation: bool,
) -> Result<()> {
self.send_transactions(trade_type, transactions, wait_confirmation).await self.send_transactions(trade_type, transactions, wait_confirmation).await
} }
fn get_tip_account(&self) -> Result<String> { fn get_tip_account(&self) -> Result<String> {
let api_key = &self.auth_token; let api_key = &self.auth_token;
if api_key.len() >= SPECIAL_API_KEY_PREFIX.len() + SPECIAL_API_KEY_SUFFIX.len() { if api_key.len() >= SPECIAL_API_KEY_PREFIX.len() + SPECIAL_API_KEY_SUFFIX.len() {
if api_key.starts_with(SPECIAL_API_KEY_PREFIX) && api_key.ends_with(SPECIAL_API_KEY_SUFFIX) { if api_key.starts_with(SPECIAL_API_KEY_PREFIX)
&& api_key.ends_with(SPECIAL_API_KEY_SUFFIX)
{
let current_api_key_hash = fast_sha256_hex(api_key); let current_api_key_hash = fast_sha256_hex(api_key);
if current_api_key_hash == SPECIAL_API_KEY_HASH { if current_api_key_hash == SPECIAL_API_KEY_HASH {
@@ -64,7 +77,10 @@ impl SwqosClientTrait for TemporalClient {
} }
} }
let tip_account = *NOZOMI_TIP_ACCOUNTS.choose(&mut rand::rng()).or_else(|| NOZOMI_TIP_ACCOUNTS.first()).unwrap(); let tip_account = *NOZOMI_TIP_ACCOUNTS
.choose(&mut rand::rng())
.or_else(|| NOZOMI_TIP_ACCOUNTS.first())
.unwrap();
Ok(tip_account.to_string()) Ok(tip_account.to_string())
} }
@@ -77,22 +93,22 @@ impl TemporalClient {
pub fn new(rpc_url: String, endpoint: String, auth_token: String) -> Self { pub fn new(rpc_url: String, endpoint: String, auth_token: String) -> Self {
let rpc_client = SolanaRpcClient::new(rpc_url); let rpc_client = SolanaRpcClient::new(rpc_url);
let http_client = default_http_client_builder().build().unwrap(); let http_client = default_http_client_builder().build().unwrap();
let client = Self { let client = Self {
rpc_client: Arc::new(rpc_client), rpc_client: Arc::new(rpc_client),
endpoint, endpoint,
auth_token, auth_token,
http_client, http_client,
ping_handle: Arc::new(tokio::sync::Mutex::new(None)), ping_handle: Arc::new(tokio::sync::Mutex::new(None)),
stop_ping: Arc::new(AtomicBool::new(false)), stop_ping: Arc::new(AtomicBool::new(false)),
}; };
// Start ping task // Start ping task
let client_clone = client.clone(); let client_clone = client.clone();
tokio::spawn(async move { tokio::spawn(async move {
client_clone.start_ping_task().await; client_clone.start_ping_task().await;
}); });
client client
} }
@@ -102,7 +118,7 @@ impl TemporalClient {
let auth_token = self.auth_token.clone(); let auth_token = self.auth_token.clone();
let http_client = self.http_client.clone(); let http_client = self.http_client.clone();
let stop_ping = self.stop_ping.clone(); let stop_ping = self.stop_ping.clone();
let handle = tokio::spawn(async move { let handle = tokio::spawn(async move {
// Immediate first ping to warm connection and reduce first-submit cold start latency // Immediate first ping to warm connection and reduce first-submit cold start latency
if let Err(e) = Self::send_ping_request(&http_client, &endpoint, &auth_token).await { if let Err(e) = Self::send_ping_request(&http_client, &endpoint, &auth_token).await {
@@ -114,12 +130,13 @@ impl TemporalClient {
if stop_ping.load(Ordering::Relaxed) { if stop_ping.load(Ordering::Relaxed) {
break; break;
} }
if let Err(e) = Self::send_ping_request(&http_client, &endpoint, &auth_token).await { if let Err(e) = Self::send_ping_request(&http_client, &endpoint, &auth_token).await
{
eprintln!("Temporal ping request failed: {}", e); eprintln!("Temporal ping request failed: {}", e);
} }
} }
}); });
// Update ping_handle - use Mutex to safely update // Update ping_handle - use Mutex to safely update
{ {
let mut ping_guard = self.ping_handle.lock().await; let mut ping_guard = self.ping_handle.lock().await;
@@ -131,7 +148,11 @@ impl TemporalClient {
} }
/// Send ping request to /ping endpoint /// Send ping request to /ping endpoint
async fn send_ping_request(http_client: &Client, endpoint: &str, _auth_token: &str) -> Result<()> { async fn send_ping_request(
http_client: &Client,
endpoint: &str,
_auth_token: &str,
) -> Result<()> {
// Build ping URL (no auth token required for ping endpoint) // Build ping URL (no auth token required for ping endpoint)
let ping_url = if endpoint.ends_with('/') { let ping_url = if endpoint.ends_with('/') {
format!("{}ping", endpoint) format!("{}ping", endpoint)
@@ -140,10 +161,8 @@ impl TemporalClient {
}; };
// Short timeout for ping; consume body so connection is returned to pool for reuse by submit // Short timeout for ping; consume body so connection is returned to pool for reuse by submit
let response = http_client.get(&ping_url) let response =
.timeout(Duration::from_millis(1500)) http_client.get(&ping_url).timeout(Duration::from_millis(1500)).send().await?;
.send()
.await?;
let status = response.status(); let status = response.status();
let _ = response.bytes().await; let _ = response.bytes().await;
if !status.is_success() { if !status.is_success() {
@@ -152,9 +171,15 @@ impl TemporalClient {
Ok(()) Ok(())
} }
pub async fn send_transaction(&self, trade_type: TradeType, transaction: &VersionedTransaction, wait_confirmation: bool) -> Result<()> { pub async fn send_transaction(
&self,
trade_type: TradeType,
transaction: &VersionedTransaction,
wait_confirmation: bool,
) -> Result<()> {
let start_time = Instant::now(); let start_time = Instant::now();
let (content, signature) = serialize_transaction_and_encode(transaction, UiTransactionEncoding::Base64)?; let (content, signature) =
serialize_transaction_and_encode(transaction, UiTransactionEncoding::Base64)?;
// Build request body according to Nozomi documentation requirements // Build request body according to Nozomi documentation requirements
let request_body = serde_json::to_string(&json!({ let request_body = serde_json::to_string(&json!({
@@ -172,7 +197,9 @@ impl TemporalClient {
url.push_str("/?c="); url.push_str("/?c=");
url.push_str(&self.auth_token); url.push_str(&self.auth_token);
let response_text = self.http_client.post(&url) let response_text = self
.http_client
.post(&url)
.body(request_body) .body(request_body)
.header("Content-Type", "application/json") .header("Content-Type", "application/json")
.send() .send()
@@ -195,9 +222,13 @@ impl TemporalClient {
Ok(_) => (), Ok(_) => (),
Err(e) => { Err(e) => {
println!(" signature: {:?}", signature); println!(" signature: {:?}", signature);
println!(" [nozomi] {} confirmation failed: {:?}", trade_type, start_time.elapsed()); println!(
" [nozomi] {} confirmation failed: {:?}",
trade_type,
start_time.elapsed()
);
return Err(e); return Err(e);
}, }
} }
if wait_confirmation { if wait_confirmation {
println!(" signature: {:?}", signature); println!(" signature: {:?}", signature);
@@ -207,7 +238,12 @@ impl TemporalClient {
Ok(()) Ok(())
} }
pub async fn send_transactions(&self, trade_type: TradeType, transactions: &Vec<VersionedTransaction>, wait_confirmation: bool) -> Result<()> { pub async fn send_transactions(
&self,
trade_type: TradeType,
transactions: &Vec<VersionedTransaction>,
wait_confirmation: bool,
) -> Result<()> {
for transaction in transactions { for transaction in transactions {
self.send_transaction(trade_type, transaction, wait_confirmation).await?; self.send_transaction(trade_type, transaction, wait_confirmation).await?;
} }
@@ -219,7 +255,7 @@ impl Drop for TemporalClient {
fn drop(&mut self) { fn drop(&mut self) {
// Ensure ping task stops when client is destroyed // Ensure ping task stops when client is destroyed
self.stop_ping.store(true, Ordering::Relaxed); self.stop_ping.store(true, Ordering::Relaxed);
// Try to stop ping task immediately // Try to stop ping task immediately
// Use tokio::spawn to avoid blocking Drop // Use tokio::spawn to avoid blocking Drop
let ping_handle = self.ping_handle.clone(); let ping_handle = self.ping_handle.clone();
@@ -231,4 +267,4 @@ impl Drop for TemporalClient {
*ping_guard = None; *ping_guard = None;
}); });
} }
} }
+40 -13
View File
@@ -1,4 +1,6 @@
use crate::swqos::common::{default_http_client_builder, poll_transaction_confirmation, serialize_transaction_and_encode}; use crate::swqos::common::{
default_http_client_builder, poll_transaction_confirmation, serialize_transaction_and_encode,
};
use rand::seq::IndexedRandom; use rand::seq::IndexedRandom;
use reqwest::Client; use reqwest::Client;
use serde_json::json; use serde_json::json;
@@ -6,14 +8,13 @@ use std::{sync::Arc, time::Instant};
use solana_transaction_status::UiTransactionEncoding; use solana_transaction_status::UiTransactionEncoding;
use crate::swqos::SwqosClientTrait;
use crate::swqos::{SwqosType, TradeType};
use anyhow::Result; use anyhow::Result;
use solana_sdk::transaction::VersionedTransaction; use solana_sdk::transaction::VersionedTransaction;
use crate::swqos::{SwqosType, TradeType};
use crate::swqos::SwqosClientTrait;
use crate::{common::SolanaRpcClient, constants::swqos::ZEROSLOT_TIP_ACCOUNTS}; use crate::{common::SolanaRpcClient, constants::swqos::ZEROSLOT_TIP_ACCOUNTS};
#[derive(Clone)] #[derive(Clone)]
pub struct ZeroSlotClient { pub struct ZeroSlotClient {
pub endpoint: String, pub endpoint: String,
@@ -24,16 +25,29 @@ pub struct ZeroSlotClient {
#[async_trait::async_trait] #[async_trait::async_trait]
impl SwqosClientTrait for ZeroSlotClient { impl SwqosClientTrait for ZeroSlotClient {
async fn send_transaction(&self, trade_type: TradeType, transaction: &VersionedTransaction, wait_confirmation: bool) -> Result<()> { async fn send_transaction(
&self,
trade_type: TradeType,
transaction: &VersionedTransaction,
wait_confirmation: bool,
) -> Result<()> {
self.send_transaction(trade_type, transaction, wait_confirmation).await self.send_transaction(trade_type, transaction, wait_confirmation).await
} }
async fn send_transactions(&self, trade_type: TradeType, transactions: &Vec<VersionedTransaction>, wait_confirmation: bool) -> Result<()> { async fn send_transactions(
&self,
trade_type: TradeType,
transactions: &Vec<VersionedTransaction>,
wait_confirmation: bool,
) -> Result<()> {
self.send_transactions(trade_type, transactions, wait_confirmation).await self.send_transactions(trade_type, transactions, wait_confirmation).await
} }
fn get_tip_account(&self) -> Result<String> { fn get_tip_account(&self) -> Result<String> {
let tip_account = *ZEROSLOT_TIP_ACCOUNTS.choose(&mut rand::rng()).or_else(|| ZEROSLOT_TIP_ACCOUNTS.first()).unwrap(); let tip_account = *ZEROSLOT_TIP_ACCOUNTS
.choose(&mut rand::rng())
.or_else(|| ZEROSLOT_TIP_ACCOUNTS.first())
.unwrap();
Ok(tip_account.to_string()) Ok(tip_account.to_string())
} }
@@ -49,9 +63,15 @@ impl ZeroSlotClient {
Self { rpc_client: Arc::new(rpc_client), endpoint, auth_token, http_client } Self { rpc_client: Arc::new(rpc_client), endpoint, auth_token, http_client }
} }
pub async fn send_transaction(&self, trade_type: TradeType, transaction: &VersionedTransaction, wait_confirmation: bool) -> Result<()> { pub async fn send_transaction(
&self,
trade_type: TradeType,
transaction: &VersionedTransaction,
wait_confirmation: bool,
) -> Result<()> {
let start_time = Instant::now(); let start_time = Instant::now();
let (content, signature) = serialize_transaction_and_encode(transaction, UiTransactionEncoding::Base64)?; let (content, signature) =
serialize_transaction_and_encode(transaction, UiTransactionEncoding::Base64)?;
let request_body = serde_json::to_string(&json!({ let request_body = serde_json::to_string(&json!({
"jsonrpc": "2.0", "jsonrpc": "2.0",
@@ -69,7 +89,9 @@ impl ZeroSlotClient {
url.push_str(&self.auth_token); url.push_str(&self.auth_token);
// 4. Use `text().await?` directly, avoiding async JSON parsing from `json().await?` // 4. Use `text().await?` directly, avoiding async JSON parsing from `json().await?`
let response_text = self.http_client.post(&url) let response_text = self
.http_client
.post(&url)
.body(request_body) // Pass string directly, avoiding `json()` overhead .body(request_body) // Pass string directly, avoiding `json()` overhead
.header("Content-Type", "application/json") // Explicitly specify JSON header .header("Content-Type", "application/json") // Explicitly specify JSON header
.send() .send()
@@ -95,7 +117,7 @@ impl ZeroSlotClient {
println!(" signature: {:?}", signature); println!(" signature: {:?}", signature);
println!(" [0slot] {} confirmation failed: {:?}", trade_type, start_time.elapsed()); println!(" [0slot] {} confirmation failed: {:?}", trade_type, start_time.elapsed());
return Err(e); return Err(e);
}, }
} }
if wait_confirmation { if wait_confirmation {
println!(" signature: {:?}", signature); println!(" signature: {:?}", signature);
@@ -105,10 +127,15 @@ impl ZeroSlotClient {
Ok(()) Ok(())
} }
pub async fn send_transactions(&self, trade_type: TradeType, transactions: &Vec<VersionedTransaction>, wait_confirmation: bool) -> Result<()> { pub async fn send_transactions(
&self,
trade_type: TradeType,
transactions: &Vec<VersionedTransaction>,
wait_confirmation: bool,
) -> Result<()> {
for transaction in transactions { for transaction in transactions {
self.send_transaction(trade_type, transaction, wait_confirmation).await?; self.send_transaction(trade_type, transaction, wait_confirmation).await?;
} }
Ok(()) Ok(())
} }
} }
+33 -23
View File
@@ -1,8 +1,9 @@
use dashmap::DashMap; use dashmap::DashMap;
use once_cell::sync::Lazy; use once_cell::sync::Lazy;
use smallvec::SmallVec; use smallvec::SmallVec;
use solana_sdk::instruction::Instruction;
use solana_compute_budget_interface::ComputeBudgetInstruction; use solana_compute_budget_interface::ComputeBudgetInstruction;
use solana_sdk::instruction::Instruction;
use std::sync::Arc;
/// Cache key containing all parameters for compute budget instructions /// Cache key containing all parameters for compute budget instructions
#[derive(Debug, Clone, PartialEq, Eq, Hash)] #[derive(Debug, Clone, PartialEq, Eq, Hash)]
@@ -11,43 +12,52 @@ struct ComputeBudgetCacheKey {
unit_limit: u32, unit_limit: u32,
} }
/// Global cache storing compute budget instructions /// Global cache storing compute budget instructions (Arc to avoid clone on hit).
/// Uses DashMap for high-performance lock-free concurrent access /// Uses DashMap for high-performance lock-free concurrent access.
static COMPUTE_BUDGET_CACHE: Lazy<DashMap<ComputeBudgetCacheKey, SmallVec<[Instruction; 2]>>> = static COMPUTE_BUDGET_CACHE: Lazy<DashMap<ComputeBudgetCacheKey, Arc<SmallVec<[Instruction; 2]>>>> =
Lazy::new(|| DashMap::new()); Lazy::new(|| DashMap::new());
/// Extend `instructions` with compute budget instructions; on cache hit extends from cached Arc (no SmallVec clone).
#[inline(always)] #[inline(always)]
pub fn compute_budget_instructions( pub fn extend_compute_budget_instructions(
instructions: &mut Vec<Instruction>,
unit_price: u64, unit_price: u64,
unit_limit: u32, unit_limit: u32,
) -> SmallVec<[Instruction; 2]> { ) {
// Create cache key let cache_key = ComputeBudgetCacheKey { unit_price, unit_limit };
let cache_key = ComputeBudgetCacheKey {
unit_price: unit_price,
unit_limit: unit_limit,
};
// Try to get from cache first if let Some(cached) = COMPUTE_BUDGET_CACHE.get(&cache_key) {
if let Some(cached_insts) = COMPUTE_BUDGET_CACHE.get(&cache_key) { instructions.extend(cached.iter().cloned());
return cached_insts.clone(); return;
} }
// Cache miss, generate new instructions
let mut insts = SmallVec::<[Instruction; 2]>::new(); let mut insts = SmallVec::<[Instruction; 2]>::new();
// Only add compute unit price instruction if > 0
if unit_price > 0 { if unit_price > 0 {
insts.push(ComputeBudgetInstruction::set_compute_unit_price(unit_price)); insts.push(ComputeBudgetInstruction::set_compute_unit_price(unit_price));
} }
// Only add compute unit limit instruction if > 0
if unit_limit > 0 { if unit_limit > 0 {
insts.push(ComputeBudgetInstruction::set_compute_unit_limit(unit_limit)); insts.push(ComputeBudgetInstruction::set_compute_unit_limit(unit_limit));
} }
let arc = Arc::new(insts);
instructions.extend(arc.iter().cloned());
COMPUTE_BUDGET_CACHE.insert(cache_key, arc);
}
// Store result in cache /// Returns compute budget instructions (allocates on cache hit; prefer `extend_compute_budget_instructions` on hot path).
let insts_clone = insts.clone(); #[inline(always)]
COMPUTE_BUDGET_CACHE.insert(cache_key, insts_clone); pub fn compute_budget_instructions(unit_price: u64, unit_limit: u32) -> SmallVec<[Instruction; 2]> {
let cache_key = ComputeBudgetCacheKey { unit_price, unit_limit };
if let Some(cached) = COMPUTE_BUDGET_CACHE.get(&cache_key) {
return (**cached).clone();
}
let mut insts = SmallVec::<[Instruction; 2]>::new();
if unit_price > 0 {
insts.push(ComputeBudgetInstruction::set_compute_unit_price(unit_price));
}
if unit_limit > 0 {
insts.push(ComputeBudgetInstruction::set_compute_unit_limit(unit_limit));
}
let arc = Arc::new(insts.clone());
COMPUTE_BUDGET_CACHE.insert(cache_key, arc);
insts insts
} }
+3 -3
View File
@@ -1,12 +1,12 @@
pub mod compute_budget_manager;
pub mod nonce_manager; pub mod nonce_manager;
pub mod transaction_builder; pub mod transaction_builder;
pub mod compute_budget_manager;
pub mod utils; pub mod utils;
pub mod wsol_manager; pub mod wsol_manager;
// Re-export commonly used functions // Re-export commonly used functions
pub use compute_budget_manager::*;
pub use nonce_manager::*; pub use nonce_manager::*;
pub use transaction_builder::*; pub use transaction_builder::*;
pub use compute_budget_manager::*;
pub use utils::*; pub use utils::*;
pub use wsol_manager::*; pub use wsol_manager::*;
+17 -13
View File
@@ -12,29 +12,33 @@ use crate::common::nonce_cache::DurableNonceInfo;
pub fn add_nonce_instruction( pub fn add_nonce_instruction(
instructions: &mut Vec<Instruction>, instructions: &mut Vec<Instruction>,
payer: &Keypair, payer: &Keypair,
// nonce_account: Option<Pubkey>, durable_nonce: Option<&DurableNonceInfo>,
// current_nonce: Option<Hash>,
durable_nonce: Option<DurableNonceInfo>,
) -> Result<(), anyhow::Error> { ) -> Result<(), anyhow::Error> {
if let Some(durable_nonce) = durable_nonce { if let Some(durable_nonce) = durable_nonce {
let nonce_advance_ix = advance_nonce_account(&durable_nonce.nonce_account.unwrap(), &payer.pubkey()); let nonce_advance_ix =
advance_nonce_account(&durable_nonce.nonce_account.unwrap(), &payer.pubkey());
instructions.push(nonce_advance_ix); instructions.push(nonce_advance_ix);
} }
Ok(()) Ok(())
} }
/// Get blockhash for transaction /// Get blockhash for transaction.
/// If nonce account is used, return blockhash from nonce, otherwise return the provided recent_blockhash /// If nonce account is used, returns blockhash from nonce; otherwise returns the provided recent_blockhash.
/// Returns error when neither durable_nonce nor recent_blockhash is set (caller must provide one for low latency).
pub fn get_transaction_blockhash( pub fn get_transaction_blockhash(
recent_blockhash: Option<Hash>, recent_blockhash: Option<Hash>,
durable_nonce: Option<DurableNonceInfo>, durable_nonce: Option<&DurableNonceInfo>,
// nonce_account: Option<Pubkey>, ) -> Result<Hash, anyhow::Error> {
// current_nonce: Option<Hash>,
) -> Hash {
if let Some(durable_nonce) = durable_nonce { if let Some(durable_nonce) = durable_nonce {
durable_nonce.current_nonce.unwrap() durable_nonce
.current_nonce
.ok_or_else(|| anyhow::anyhow!("durable_nonce.current_nonce is None"))
} else if let Some(hash) = recent_blockhash {
Ok(hash)
} else { } else {
recent_blockhash.unwrap() Err(anyhow::anyhow!(
"Must provide either recent_blockhash or durable_nonce for transaction"
))
} }
} }
+36 -42
View File
@@ -1,70 +1,69 @@
use solana_hash::Hash; use solana_hash::Hash;
use solana_sdk::{ use solana_sdk::{
instruction::Instruction, message::AddressLookupTableAccount, native_token::sol_str_to_lamports, pubkey::Pubkey, signature::Keypair, signer::Signer, transaction::VersionedTransaction instruction::Instruction, message::AddressLookupTableAccount, pubkey::Pubkey,
signature::Keypair, signer::Signer, transaction::VersionedTransaction,
}; };
use solana_system_interface::instruction::transfer; use solana_system_interface::instruction::transfer;
use std::sync::Arc; use std::sync::Arc;
use super::{ use super::nonce_manager::{add_nonce_instruction, get_transaction_blockhash};
compute_budget_manager::compute_budget_instructions,
nonce_manager::{add_nonce_instruction, get_transaction_blockhash},
};
use crate::{ use crate::{
common::{nonce_cache::DurableNonceInfo, SolanaRpcClient}, common::{nonce_cache::DurableNonceInfo, SolanaRpcClient},
trading::{MiddlewareManager, core::transaction_pool::{acquire_builder, release_builder}}, trading::{
core::transaction_pool::{acquire_builder, release_builder},
MiddlewareManager,
},
}; };
/// Build standard RPC transaction. /// Convert SOL amount (f64) to lamports without string allocation (hot path).
/// Takes `business_instructions` by reference to avoid per-task Vec clone in execute_parallel. #[inline(always)]
fn sol_f64_to_lamports(sol: f64) -> u64 {
if sol <= 0.0 {
return 0;
}
let lamports = sol * 1_000_000_000.0;
(lamports.min(u64::MAX as f64)).round() as u64
}
/// Build standard RPC transaction (worker hot path).
/// Takes Arc/refs only; one Vec allocation (with_capacity), extend_from_slice for business_instructions, no extra clone of payer/rpc/middleware.
pub async fn build_transaction( pub async fn build_transaction(
payer: Arc<Keypair>, payer: &Arc<Keypair>,
_rpc: Option<Arc<SolanaRpcClient>>, _rpc: Option<&Arc<SolanaRpcClient>>,
unit_limit: u32, unit_limit: u32,
unit_price: u64, unit_price: u64,
business_instructions: &[Instruction], business_instructions: &[Instruction],
address_lookup_table_account: Option<AddressLookupTableAccount>, address_lookup_table_account: Option<&AddressLookupTableAccount>,
recent_blockhash: Option<Hash>, recent_blockhash: Option<Hash>,
middleware_manager: Option<Arc<MiddlewareManager>>, middleware_manager: Option<&Arc<MiddlewareManager>>,
protocol_name: &str, protocol_name: &str,
is_buy: bool, is_buy: bool,
with_tip: bool, with_tip: bool,
tip_account: &Pubkey, tip_account: &Pubkey,
tip_amount: f64, tip_amount: f64,
durable_nonce: Option<DurableNonceInfo>, durable_nonce: Option<&DurableNonceInfo>,
// nonce_account: Option<Pubkey>,
// current_nonce: Option<Hash>,
) -> Result<VersionedTransaction, anyhow::Error> { ) -> Result<VersionedTransaction, anyhow::Error> {
let mut instructions = Vec::with_capacity(business_instructions.len() + 5); let mut instructions = Vec::with_capacity(business_instructions.len() + 5);
// Add nonce instruction if let Err(e) = add_nonce_instruction(&mut instructions, payer.as_ref(), durable_nonce) {
if let Err(e) =
add_nonce_instruction(&mut instructions, payer.as_ref(), durable_nonce.clone())
{
return Err(e); return Err(e);
} }
// Add tip transfer instruction
if with_tip && tip_amount > 0.0 { if with_tip && tip_amount > 0.0 {
instructions.push(transfer( let tip_lamports = sol_f64_to_lamports(tip_amount);
&payer.pubkey(), instructions.push(transfer(&payer.pubkey(), tip_account, tip_lamports));
tip_account,
sol_str_to_lamports(tip_amount.to_string().as_str()).unwrap_or(0),
));
} }
// Add compute budget instructions super::compute_budget_manager::extend_compute_budget_instructions(
instructions.extend(compute_budget_instructions( &mut instructions,
unit_price, unit_price,
unit_limit, unit_limit,
)); );
// Add business instructions (clone only here; avoids per-task Vec clone in execute_parallel)
instructions.extend_from_slice(business_instructions); instructions.extend_from_slice(business_instructions);
// Get blockhash for transaction let blockhash = get_transaction_blockhash(recent_blockhash, durable_nonce)?;
let blockhash = get_transaction_blockhash(recent_blockhash, durable_nonce.clone());
// Build transaction
build_versioned_transaction( build_versioned_transaction(
payer, payer,
instructions, instructions,
@@ -77,23 +76,18 @@ pub async fn build_transaction(
.await .await
} }
/// Low-level function for building versioned transactions
async fn build_versioned_transaction( async fn build_versioned_transaction(
payer: Arc<Keypair>, payer: &Arc<Keypair>,
instructions: Vec<Instruction>, instructions: Vec<Instruction>,
address_lookup_table_account: Option<AddressLookupTableAccount>, address_lookup_table_account: Option<&AddressLookupTableAccount>,
blockhash: Hash, blockhash: Hash,
middleware_manager: Option<Arc<MiddlewareManager>>, middleware_manager: Option<&Arc<MiddlewareManager>>,
protocol_name: &str, protocol_name: &str,
is_buy: bool, is_buy: bool,
) -> Result<VersionedTransaction, anyhow::Error> { ) -> Result<VersionedTransaction, anyhow::Error> {
let full_instructions = match middleware_manager { let full_instructions = match middleware_manager {
Some(middleware_manager) => middleware_manager Some(middleware_manager) => middleware_manager
.apply_middlewares_process_full_instructions( .apply_middlewares_process_full_instructions(instructions, protocol_name, is_buy)?,
instructions,
protocol_name.to_string(),
is_buy,
)?,
None => instructions, None => instructions,
}; };
@@ -108,7 +102,7 @@ async fn build_versioned_transaction(
); );
let msg_bytes = versioned_msg.serialize(); let msg_bytes = versioned_msg.serialize();
let signature = payer.try_sign_message(&msg_bytes).expect("sign failed"); let signature = payer.as_ref().try_sign_message(&msg_bytes).expect("sign failed");
let tx = VersionedTransaction { signatures: vec![signature], message: versioned_msg }; let tx = VersionedTransaction { signatures: vec![signature], message: versioned_msg };
// 归还构建器到池 // 归还构建器到池
+20 -3
View File
@@ -2,7 +2,11 @@ use solana_sdk::{pubkey::Pubkey, signature::Keypair, signer::Signer, transaction
use solana_system_interface::instruction::transfer; use solana_system_interface::instruction::transfer;
use crate::common::{ use crate::common::{
fast_fn::get_associated_token_address_with_program_id_fast, spl_token::close_account, fast_fn::{
get_associated_token_address_with_program_id_fast,
get_associated_token_address_with_program_id_fast_use_seed,
},
spl_token::close_account,
SolanaRpcClient, SolanaRpcClient,
}; };
use anyhow::anyhow; use anyhow::anyhow;
@@ -36,10 +40,23 @@ pub async fn get_token_balance(
payer: &Pubkey, payer: &Pubkey,
mint: &Pubkey, mint: &Pubkey,
) -> Result<u64, anyhow::Error> { ) -> Result<u64, anyhow::Error> {
let ata = crate::common::fast_fn::get_associated_token_address_with_program_id_fast( get_token_balance_with_options(rpc, payer, mint, &crate::constants::TOKEN_PROGRAM, false).await
}
/// 使用与交易指令一致的 ATA 推导(可选 seed)查询余额;卖出/余额查询应与买入使用同一 ATA 地址。
#[inline]
pub async fn get_token_balance_with_options(
rpc: &SolanaRpcClient,
payer: &Pubkey,
mint: &Pubkey,
token_program: &Pubkey,
use_seed: bool,
) -> Result<u64, anyhow::Error> {
let ata = get_associated_token_address_with_program_id_fast_use_seed(
payer, payer,
mint, mint,
&crate::constants::TOKEN_PROGRAM, token_program,
use_seed,
); );
let balance = rpc.get_token_account_balance(&ata).await?; let balance = rpc.get_token_account_balance(&ata).await?;
let balance_u64 = let balance_u64 =
+11 -21
View File
@@ -1,7 +1,10 @@
use crate::common::{ use crate::common::{
fast_fn::create_associated_token_account_idempotent_fast, fast_fn::create_associated_token_account_idempotent_fast,
seed::{
create_associated_token_account_use_seed,
get_associated_token_address_with_program_id_use_seed,
},
spl_token::close_account, spl_token::close_account,
seed::{create_associated_token_account_use_seed, get_associated_token_address_with_program_id_use_seed},
}; };
use smallvec::SmallVec; use smallvec::SmallVec;
use solana_sdk::{instruction::Instruction, message::AccountMeta, pubkey::Pubkey}; use solana_sdk::{instruction::Instruction, message::AccountMeta, pubkey::Pubkey};
@@ -38,7 +41,7 @@ pub fn handle_wsol(payer: &Pubkey, amount_in: u64) -> SmallVec<[Instruction; 3]>
pub fn close_wsol(payer: &Pubkey) -> Vec<Instruction> { pub fn close_wsol(payer: &Pubkey) -> Vec<Instruction> {
use std::sync::Arc; use std::sync::Arc;
let wsol_token_account = let wsol_token_account =
crate::common::fast_fn::get_associated_token_address_with_program_id_fast( crate::common::fast_fn::get_associated_token_address_with_program_id_fast(
&payer, &payer,
@@ -61,7 +64,7 @@ pub fn close_wsol(payer: &Pubkey) -> Vec<Instruction> {
.unwrap()] .unwrap()]
}, },
); );
// 🚀 性能优化:尝试零开销解包 Arc // 🚀 性能优化:尝试零开销解包 Arc
Arc::try_unwrap(arc_instructions).unwrap_or_else(|arc| (*arc).clone()) Arc::try_unwrap(arc_instructions).unwrap_or_else(|arc| (*arc).clone())
} }
@@ -109,10 +112,7 @@ pub fn wrap_sol_only(payer: &Pubkey, amount_in: u64) -> SmallVec<[Instruction; 2
/// ///
/// 注意:此函数只生成指令,不检查账户是否存在(需要调用方在发送交易前检查) /// 注意:此函数只生成指令,不检查账户是否存在(需要调用方在发送交易前检查)
/// 如果临时账户已存在,可以安全地跳过创建步骤,直接转账并关闭 /// 如果临时账户已存在,可以安全地跳过创建步骤,直接转账并关闭
pub fn wrap_wsol_to_sol( pub fn wrap_wsol_to_sol(payer: &Pubkey, amount: u64) -> Result<Vec<Instruction>, anyhow::Error> {
payer: &Pubkey,
amount: u64,
) -> Result<Vec<Instruction>, anyhow::Error> {
let mut instructions = Vec::new(); let mut instructions = Vec::new();
// 1. 创建 WSOL seed 账户(注意:如果账户已存在会失败) // 1. 创建 WSOL seed 账户(注意:如果账户已存在会失败)
@@ -151,13 +151,8 @@ pub fn wrap_wsol_to_sol(
instructions.push(transfer_instruction); instructions.push(transfer_instruction);
// 5. 添加关闭 WSOL seed 账户的指令 // 5. 添加关闭 WSOL seed 账户的指令
let close_instruction = close_account( let close_instruction =
&crate::constants::TOKEN_PROGRAM, close_account(&crate::constants::TOKEN_PROGRAM, &seed_ata_address, payer, payer, &[])?;
&seed_ata_address,
payer,
payer,
&[],
)?;
instructions.push(close_instruction); instructions.push(close_instruction);
Ok(instructions) Ok(instructions)
@@ -197,13 +192,8 @@ pub fn wrap_wsol_to_sol_without_create(
instructions.push(transfer_instruction); instructions.push(transfer_instruction);
// 4. 添加关闭 WSOL seed 账户的指令 // 4. 添加关闭 WSOL seed 账户的指令
let close_instruction = close_account( let close_instruction =
&crate::constants::TOKEN_PROGRAM, close_account(&crate::constants::TOKEN_PROGRAM, &seed_ata_address, payer, payer, &[])?;
&seed_ata_address,
payer,
payer,
&[],
)?;
instructions.push(close_instruction); instructions.push(close_instruction);
Ok(instructions) Ok(instructions)
+319 -154
View File
@@ -1,22 +1,226 @@
//! Parallel executor for multi-SWQOS submit. //! Parallel executor for multi-SWQOS submit.
//!
//! **Hot path (submit):** no lock (OnceCell + lock-free ArrayQueue), no `get_core_ids()`, only Arc clones and queue push.
//! - **Pool**: Pre-spawned workers (default 18); hot path only enqueues jobs (no per-call tokio::spawn).
//! - **Dedicated threads** (opt-in via `with_dedicated_sender_threads`): N OS threads run sender work only, optionally pinned to cores.
//! - **Arc**: Shared data behind `Arc` → clone = refcount increment (no data copy).
//! - **Refs**: `build_transaction` takes refs only; worker path avoids extra clones.
use anyhow::{anyhow, Result}; use anyhow::{anyhow, Result};
use crossbeam_queue::ArrayQueue; use crossbeam_queue::ArrayQueue;
use once_cell::sync::OnceCell;
use solana_hash::Hash; use solana_hash::Hash;
use solana_sdk::message::AddressLookupTableAccount; use solana_sdk::message::AddressLookupTableAccount;
use solana_sdk::{ use solana_sdk::{
instruction::Instruction, pubkey::Pubkey, signature::Keypair, signature::Signature, instruction::Instruction, pubkey::Pubkey, signature::Keypair, signature::Signature,
}; };
use std::collections::HashMap;
use std::hash::BuildHasherDefault;
use parking_lot::Mutex;
use std::sync::atomic::{AtomicBool, AtomicUsize, Ordering}; use std::sync::atomic::{AtomicBool, AtomicUsize, Ordering};
use std::{str::FromStr, sync::Arc, time::Instant}; use std::{str::FromStr, sync::Arc, time::Instant};
use tokio::sync::Notify;
use fnv::FnvHasher;
type FnvHashMap<K, V> = HashMap<K, V, BuildHasherDefault<FnvHasher>>;
use crate::{ use crate::{
common::nonce_cache::DurableNonceInfo, common::nonce_cache::DurableNonceInfo,
common::{GasFeeStrategy, SolanaRpcClient}, common::{GasFeeStrategy, SolanaRpcClient},
swqos::{SwqosClient, SwqosType, TradeType}, swqos::{SwqosClient, SwqosType, TradeType},
trading::core::params::SenderConcurrencyConfig,
trading::{common::build_transaction, MiddlewareManager}, trading::{common::build_transaction, MiddlewareManager},
}; };
/// 与 transaction_pool::PARALLEL_SENDER_COUNT 一致,保证多路 build 不串行
const SWQOS_POOL_WORKERS: usize = 18;
const SWQOS_QUEUE_CAP: usize = 128;
const SWQOS_DEDICATED_DEFAULT_THREADS: usize = 18;
/// Shared across all jobs in one batch; built once, cloned as single Arc per job (minimal hot-path clone).
struct SwqosSharedContext {
payer: Arc<Keypair>,
instructions: Arc<Vec<Instruction>>,
rpc: Option<Arc<SolanaRpcClient>>,
address_lookup_table_account: Option<AddressLookupTableAccount>,
recent_blockhash: Option<Hash>,
durable_nonce: Option<DurableNonceInfo>,
middleware_manager: Option<Arc<MiddlewareManager>>,
protocol_name: &'static str,
is_buy: bool,
wait_transaction_confirmed: bool,
with_tip: bool,
collector: Arc<ResultCollector>,
}
/// One SWQOS submit task; only per-task data + one Arc to shared (reduces hot-path clones).
struct SwqosJob {
shared: Arc<SwqosSharedContext>,
tip: f64,
unit_limit: u32,
unit_price: u64,
tip_account: Arc<Pubkey>,
swqos_client: Arc<SwqosClient>,
swqos_type: SwqosType,
core_id: Option<core_affinity::CoreId>,
use_affinity: bool,
}
async fn run_one_swqos_job(job: SwqosJob) {
let s = &job.shared;
if job.use_affinity {
if let Some(cid) = job.core_id {
core_affinity::set_for_current(cid);
}
}
let tip_amount = if s.with_tip { job.tip } else { 0.0 };
let transaction = match build_transaction(
&s.payer,
s.rpc.as_ref(),
job.unit_limit,
job.unit_price,
s.instructions.as_ref(),
s.address_lookup_table_account.as_ref(),
s.recent_blockhash,
s.middleware_manager.as_ref(),
s.protocol_name,
s.is_buy,
job.swqos_type != SwqosType::Default,
&job.tip_account,
tip_amount,
s.durable_nonce.as_ref(),
)
.await
{
Ok(tx) => tx,
Err(e) => {
s.collector.submit(TaskResult {
success: false,
signature: Signature::default(),
error: Some(e),
swqos_type: job.swqos_type,
landed_on_chain: false,
submit_done_us: crate::common::clock::now_micros(),
});
return;
}
};
let (success, err, landed_on_chain) = match job
.swqos_client
.send_transaction(
if s.is_buy { TradeType::Buy } else { TradeType::Sell },
&transaction,
s.wait_transaction_confirmed,
)
.await
{
Ok(()) => (true, None, true),
Err(e) => {
let landed = is_landed_error(&e);
(false, Some(e), landed)
}
};
let sig = transaction.signatures.first().copied().unwrap_or_default();
s.collector.submit(TaskResult {
success,
signature: sig,
error: err,
swqos_type: job.swqos_type,
landed_on_chain,
submit_done_us: crate::common::clock::now_micros(),
});
}
async fn swqos_worker_loop(queue: Arc<ArrayQueue<SwqosJob>>, notify: Arc<Notify>) {
loop {
if let Some(job) = queue.pop() {
run_one_swqos_job(job).await;
} else {
notify.notified().await;
}
}
}
static SWQOS_QUEUE: OnceCell<Arc<ArrayQueue<SwqosJob>>> = OnceCell::new();
static SWQOS_NOTIFY: OnceCell<Arc<Notify>> = OnceCell::new();
static SWQOS_WORKERS_STARTED: AtomicBool = AtomicBool::new(false);
/// Dedicated OS-thread sender pool. Queue and notify are in OnceCell so hot path never takes a lock after init.
static DEDICATED_QUEUE: OnceCell<Arc<ArrayQueue<SwqosJob>>> = OnceCell::new();
static DEDICATED_NOTIFY: OnceCell<Arc<Notify>> = OnceCell::new();
/// JoinHandles kept so dedicated threads are not detached; only touched during init under lock.
static DEDICATED_INIT: Mutex<Option<Vec<std::thread::JoinHandle<()>>>> = Mutex::new(None);
fn ensure_dedicated_pool(
sender_thread_cores: Option<&[usize]>,
max_sender_concurrency: usize,
) -> (Arc<ArrayQueue<SwqosJob>>, Arc<Notify>) {
if let (Some(q), Some(n)) = (DEDICATED_QUEUE.get(), DEDICATED_NOTIFY.get()) {
return (q.clone(), n.clone());
}
let mut guard = DEDICATED_INIT.lock();
if let (Some(q), Some(n)) = (DEDICATED_QUEUE.get(), DEDICATED_NOTIFY.get()) {
return (q.clone(), n.clone());
}
let n = sender_thread_cores
.map(|v| v.len().min(max_sender_concurrency))
.unwrap_or_else(|| SWQOS_DEDICATED_DEFAULT_THREADS.min(max_sender_concurrency))
.min(32)
.max(1);
let queue = Arc::new(ArrayQueue::new(SWQOS_QUEUE_CAP));
let notify = Arc::new(Notify::new());
let core_ids: Vec<core_affinity::CoreId> = core_affinity::get_core_ids()
.map(|all_ids| {
sender_thread_cores
.map(|indices| {
indices
.iter()
.take(n)
.filter_map(|&i| all_ids.get(i).cloned())
.collect()
})
.unwrap_or_else(|| all_ids.into_iter().take(n).collect())
})
.unwrap_or_default();
let mut handles = Vec::with_capacity(n);
for i in 0..n {
let queue = queue.clone();
let notify = notify.clone();
let core_id = core_ids.get(i).cloned();
let handle = std::thread::spawn(move || {
if let Some(cid) = core_id {
core_affinity::set_for_current(cid);
}
let rt = tokio::runtime::Builder::new_current_thread()
.enable_all()
.build()
.expect("dedicated sender runtime");
rt.block_on(swqos_worker_loop(queue, notify));
});
handles.push(handle);
}
let _ = DEDICATED_QUEUE.set(queue.clone());
let _ = DEDICATED_NOTIFY.set(notify.clone());
*guard = Some(handles);
(queue, notify)
}
fn ensure_swqos_pool(queue: Arc<ArrayQueue<SwqosJob>>, max_sender_concurrency: usize) {
if SWQOS_WORKERS_STARTED.swap(true, Ordering::AcqRel) {
return;
}
let n = SWQOS_POOL_WORKERS.min(max_sender_concurrency).max(1);
let notify = SWQOS_NOTIFY.get_or_init(|| Arc::new(Notify::new())).clone();
for _ in 0..n {
tokio::spawn(swqos_worker_loop(queue.clone(), notify.clone()));
}
}
#[repr(align(64))] #[repr(align(64))]
struct TaskResult { struct TaskResult {
success: bool, success: bool,
@@ -55,7 +259,7 @@ fn is_landed_error(error: &anyhow::Error) -> bool {
struct ResultCollector { struct ResultCollector {
results: Arc<ArrayQueue<TaskResult>>, results: Arc<ArrayQueue<TaskResult>>,
success_flag: Arc<AtomicBool>, success_flag: Arc<AtomicBool>,
landed_failed_flag: Arc<AtomicBool>, // 🔧 Tx landed on-chain but failed (nonce consumed) landed_failed_flag: Arc<AtomicBool>, // 🔧 Tx landed on-chain but failed (nonce consumed)
completed_count: Arc<AtomicUsize>, completed_count: Arc<AtomicUsize>,
total_tasks: usize, total_tasks: usize,
} }
@@ -88,7 +292,9 @@ impl ResultCollector {
self.completed_count.fetch_add(1, Ordering::Release); self.completed_count.fetch_add(1, Ordering::Release);
} }
async fn wait_for_success(&self) -> Option<(bool, Vec<Signature>, Option<anyhow::Error>, Vec<(SwqosType, i64)>)> { async fn wait_for_success(
&self,
) -> Option<(bool, Vec<Signature>, Option<anyhow::Error>, Vec<(SwqosType, i64)>)> {
let start = Instant::now(); let start = Instant::now();
let timeout = std::time::Duration::from_secs(5); let timeout = std::time::Duration::from_secs(5);
let poll_interval = std::time::Duration::from_millis(1000); let poll_interval = std::time::Duration::from_millis(1000);
@@ -130,7 +336,7 @@ impl ResultCollector {
} }
let completed = self.completed_count.load(Ordering::Acquire); let completed = self.completed_count.load(Ordering::Acquire);
if completed >= self.total_tasks { if completed >= self.total_tasks {
let mut signatures = Vec::new(); let mut signatures = Vec::new();
let mut last_error = None; let mut last_error = None;
let mut any_success = false; let mut any_success = false;
@@ -158,7 +364,9 @@ impl ResultCollector {
} }
} }
fn get_first(&self) -> Option<(bool, Vec<Signature>, Option<anyhow::Error>, Vec<(SwqosType, i64)>)> { fn get_first(
&self,
) -> Option<(bool, Vec<Signature>, Option<anyhow::Error>, Vec<(SwqosType, i64)>)> {
let mut signatures = Vec::new(); let mut signatures = Vec::new();
let mut has_success = false; let mut has_success = false;
let mut last_error = None; let mut last_error = None;
@@ -184,7 +392,10 @@ impl ResultCollector {
/// 等待全部任务完成(不等待链上确认),然后收集并返回所有签名。用于「多路提交」时返回多笔签名。 /// 等待全部任务完成(不等待链上确认),然后收集并返回所有签名。用于「多路提交」时返回多笔签名。
/// 轮询间隔 2ms,避免 50ms 间隔在最后一笔返回时多等几十 ms 拉高 submit 耗时。 /// 轮询间隔 2ms,避免 50ms 间隔在最后一笔返回时多等几十 ms 拉高 submit 耗时。
async fn wait_for_all_submitted(&self, timeout_secs: u64) -> Option<(bool, Vec<Signature>, Option<anyhow::Error>, Vec<(SwqosType, i64)>)> { async fn wait_for_all_submitted(
&self,
timeout_secs: u64,
) -> Option<(bool, Vec<Signature>, Option<anyhow::Error>, Vec<(SwqosType, i64)>)> {
let start = Instant::now(); let start = Instant::now();
let timeout = std::time::Duration::from_secs(timeout_secs); let timeout = std::time::Duration::from_secs(timeout_secs);
let poll_interval = std::time::Duration::from_millis(2); let poll_interval = std::time::Duration::from_millis(2);
@@ -199,10 +410,12 @@ impl ResultCollector {
} }
/// Execute trade on multiple SWQOS clients in parallel; returns success flag, all signatures, and last error. /// Execute trade on multiple SWQOS clients in parallel; returns success flag, all signatures, and last error.
///
/// `sender_config` merges sender_thread_cores, effective_core_ids, max_sender_concurrency (precomputed at SDK init; no get_core_ids on hot path).
pub async fn execute_parallel( pub async fn execute_parallel(
swqos_clients: &[Arc<SwqosClient>], swqos_clients: &[Arc<SwqosClient>],
payer: Arc<Keypair>, payer: Arc<Keypair>,
rpc: Option<Arc<SolanaRpcClient>>, rpc: Option<&Arc<SolanaRpcClient>>,
instructions: Vec<Instruction>, instructions: Vec<Instruction>,
address_lookup_table_account: Option<AddressLookupTableAccount>, address_lookup_table_account: Option<AddressLookupTableAccount>,
recent_blockhash: Option<Hash>, recent_blockhash: Option<Hash>,
@@ -213,11 +426,10 @@ pub async fn execute_parallel(
wait_transaction_confirmed: bool, wait_transaction_confirmed: bool,
with_tip: bool, with_tip: bool,
gas_fee_strategy: GasFeeStrategy, gas_fee_strategy: GasFeeStrategy,
use_core_affinity: bool, use_dedicated_sender_threads: bool,
sender_config: SenderConcurrencyConfig,
check_min_tip: bool, check_min_tip: bool,
) -> Result<(bool, Vec<Signature>, Option<anyhow::Error>, Vec<(SwqosType, i64)>)> { ) -> Result<(bool, Vec<Signature>, Option<anyhow::Error>, Vec<(SwqosType, i64)>)> {
let _exec_start = Instant::now();
if swqos_clients.is_empty() { if swqos_clients.is_empty() {
return Err(anyhow!("swqos_clients is empty")); return Err(anyhow!("swqos_clients is empty"));
} }
@@ -231,48 +443,40 @@ pub async fn execute_parallel(
return Err(anyhow!("No Rpc Default Swqos configured.")); return Err(anyhow!("No Rpc Default Swqos configured."));
} }
let cores = core_affinity::get_core_ids().unwrap_or_default();
let instructions = Arc::new(instructions); let instructions = Arc::new(instructions);
// Precompute all valid (client, gas config) combinations // One get_strategies call per batch (avoid N calls in loop).
let task_configs: Vec<_> = swqos_clients let gas_fee_configs = gas_fee_strategy.get_strategies(if is_buy {
.iter() TradeType::Buy
.enumerate() } else {
.filter(|(_, swqos_client)| { TradeType::Sell
with_tip || matches!(swqos_client.get_swqos_type(), SwqosType::Default) });
}) let mut task_configs = Vec::with_capacity(swqos_clients.len() * 3);
.flat_map(|(i, swqos_client)| { for (i, swqos_client) in swqos_clients.iter().enumerate() {
let swqos_type = swqos_client.get_swqos_type(); if !with_tip && !matches!(swqos_client.get_swqos_type(), SwqosType::Default) {
let gas_fee_strategy_configs = gas_fee_strategy.get_strategies(if is_buy { continue;
TradeType::Buy }
} else { let swqos_type = swqos_client.get_swqos_type();
TradeType::Sell let check_tip = with_tip && !matches!(swqos_type, SwqosType::Default) && check_min_tip;
}); let min_tip = if check_tip { swqos_client.min_tip_sol() } else { 0.0 };
let check_tip = with_tip && !matches!(swqos_type, SwqosType::Default) && check_min_tip; for config in &gas_fee_configs {
let min_tip = if check_tip { if config.0 != swqos_type {
swqos_client.min_tip_sol() continue;
} else { }
0.0 if check_tip {
}; if config.2.tip < min_tip && crate::common::sdk_log::sdk_log_enabled() {
gas_fee_strategy_configs println!(
.into_iter() "⚠️ Config filtered: {:?} tip {} is below minimum required {}",
.filter(move |config| config.0 == swqos_type) config.0, config.2.tip, min_tip
.filter(move |config| { );
if check_tip { }
if config.2.tip < min_tip && crate::common::sdk_log::sdk_log_enabled() { if config.2.tip < min_tip {
println!( continue;
"⚠️ Config filtered: {:?} tip {} is below minimum required {}", }
config.0, config.2.tip, min_tip }
); task_configs.push((i, swqos_client.clone(), *config));
} }
config.2.tip >= min_tip }
} else {
true
}
})
.map(move |config| (i, swqos_client.clone(), config))
})
.collect();
if task_configs.is_empty() { if task_configs.is_empty() {
return Err(anyhow!("No available gas fee strategy configs")); return Err(anyhow!("No available gas fee strategy configs"));
@@ -282,123 +486,84 @@ pub async fn execute_parallel(
return Err(anyhow!("Multiple swqos transactions require durable_nonce to be set.",)); return Err(anyhow!("Multiple swqos transactions require durable_nonce to be set.",));
} }
// Task preparation completed // Task preparation completed: one shared context (clone once per batch), then minimal per-task data.
let collector = Arc::new(ResultCollector::new(task_configs.len())); let collector = Arc::new(ResultCollector::new(task_configs.len()));
let _spawn_start = Instant::now(); let shared = Arc::new(SwqosSharedContext {
payer,
instructions,
rpc: rpc.cloned(),
address_lookup_table_account,
recent_blockhash,
durable_nonce,
middleware_manager,
protocol_name,
is_buy,
wait_transaction_confirmed,
with_tip,
collector: collector.clone(),
});
for (i, swqos_client, gas_fee_strategy_config) in task_configs { let (queue, notify) = if use_dedicated_sender_threads {
let core_id = cores.get(i % cores.len().max(1)).copied(); ensure_dedicated_pool(
let use_affinity = use_core_affinity; sender_config.sender_thread_cores.as_ref().map(|a| a.as_slice()),
let payer = payer.clone(); sender_config.max_sender_concurrency,
let instructions = instructions.clone(); )
let middleware_manager = middleware_manager.clone(); } else {
let swqos_type = swqos_client.get_swqos_type(); let q = SWQOS_QUEUE.get_or_init(|| Arc::new(ArrayQueue::new(SWQOS_QUEUE_CAP)));
let tip_account_str = swqos_client.get_tip_account()?; ensure_swqos_pool(q.clone(), sender_config.max_sender_concurrency);
let tip_account = Arc::new(Pubkey::from_str(&tip_account_str).unwrap_or_default()); (q.clone(), SWQOS_NOTIFY.get_or_init(|| Arc::new(Notify::new())).clone())
let collector = collector.clone(); };
let tip = gas_fee_strategy_config.2.tip; {
let unit_limit = gas_fee_strategy_config.2.cu_limit; let effective_core_ids = sender_config.effective_core_ids.as_slice();
let unit_price = gas_fee_strategy_config.2.cu_price; let core_len = effective_core_ids.len().max(1);
let rpc = rpc.clone(); let mut tip_cache: FnvHashMap<*const (), Arc<Pubkey>> =
let durable_nonce = durable_nonce.clone(); FnvHashMap::with_capacity_and_hasher(task_configs.len(), BuildHasherDefault::default());
let address_lookup_table_account = address_lookup_table_account.clone(); for (i, swqos_client, gas_fee_strategy_config) in task_configs {
let recent_blockhash_task = recent_blockhash.clone(); let core_id = effective_core_ids.get(i % core_len).copied();
let swqos_type = swqos_client.get_swqos_type();
tokio::spawn(async move { let key = Arc::as_ptr(&swqos_client) as *const ();
let _task_start = Instant::now(); let tip_account = match tip_cache.get(&key) {
if use_affinity { Some(t) => t.clone(),
if let Some(cid) = core_id { None => {
core_affinity::set_for_current(cid); let s = swqos_client.get_tip_account()?;
let tip = Arc::new(Pubkey::from_str(&s).unwrap_or_default());
tip_cache.insert(key, tip.clone());
tip
} }
} };
let (tip, unit_limit, unit_price) = (
let tip_amount = if with_tip { tip } else { 0.0 }; gas_fee_strategy_config.2.tip,
gas_fee_strategy_config.2.cu_limit,
let _build_start = Instant::now(); gas_fee_strategy_config.2.cu_price,
let transaction = match build_transaction( );
payer, let job = SwqosJob {
rpc, shared: shared.clone(),
tip,
unit_limit, unit_limit,
unit_price, unit_price,
instructions.as_ref(), tip_account,
address_lookup_table_account, swqos_client,
recent_blockhash_task,
middleware_manager,
protocol_name,
is_buy,
swqos_type != SwqosType::Default,
&tip_account,
tip_amount,
durable_nonce,
)
.await
{
Ok(tx) => tx,
Err(e) => {
// Build transaction failed
collector.submit(TaskResult {
success: false,
signature: Signature::default(),
error: Some(e),
swqos_type,
landed_on_chain: false,
submit_done_us: crate::common::clock::now_micros(),
});
return;
}
};
// Transaction built
let _send_start = Instant::now();
let mut err: Option<anyhow::Error> = None;
#[allow(unused_assignments)]
let mut landed_on_chain = false;
let success = match swqos_client
.send_transaction(
if is_buy { TradeType::Buy } else { TradeType::Sell },
&transaction,
wait_transaction_confirmed,
)
.await
{
Ok(()) => {
landed_on_chain = true; // Success means tx confirmed on-chain
true
}
Err(e) => {
// Check if this error indicates the tx landed but failed (e.g., ExceededSlippage)
landed_on_chain = is_landed_error(&e);
err = Some(e);
// Send transaction failed
false
}
};
// Transaction sent: always submit a result so collector never has "no result" for this task.
// If transaction has no signatures (malformed), submit with default signature and success=false.
let sig = transaction.signatures.first().copied().unwrap_or_default();
collector.submit(TaskResult {
success,
signature: sig,
error: err,
swqos_type, swqos_type,
landed_on_chain, core_id,
submit_done_us: crate::common::clock::now_micros(), use_affinity: !effective_core_ids.is_empty(),
}); };
}); let _ = queue.push(job);
}
} }
// All tasks spawned notify.notify_waiters();
// All jobs enqueued (no spawn on hot path)
if !wait_transaction_confirmed { if !wait_transaction_confirmed {
const SUBMIT_TIMEOUT_SECS: u64 = 30; const SUBMIT_TIMEOUT_SECS: u64 = 30;
let ret = collector let ret = collector.wait_for_all_submitted(SUBMIT_TIMEOUT_SECS).await.unwrap_or((
.wait_for_all_submitted(SUBMIT_TIMEOUT_SECS) false,
.await vec![],
.unwrap_or((false, vec![], Some(anyhow!("No SWQOS result within {}s", SUBMIT_TIMEOUT_SECS)), vec![])); Some(anyhow!("No SWQOS result within {}s", SUBMIT_TIMEOUT_SECS)),
vec![],
));
let (success, signatures, last_error, submit_timings) = ret; let (success, signatures, last_error, submit_timings) = ret;
return Ok((success, signatures, last_error, submit_timings)); return Ok((success, signatures, last_error, submit_timings));
} }
+3 -10
View File
@@ -2,16 +2,9 @@
//! 执行模块:指令预处理、缓存预取、分支提示。 //! 执行模块:指令预处理、缓存预取、分支提示。
use anyhow::Result; use anyhow::Result;
use solana_sdk::{ use solana_sdk::{instruction::Instruction, pubkey::Pubkey, signature::Keypair};
instruction::Instruction,
pubkey::Pubkey,
signature::Keypair,
};
use crate::perf::{ use crate::perf::{hardware_optimizations::BranchOptimizer, simd::SIMDMemory};
hardware_optimizations::BranchOptimizer,
simd::SIMDMemory,
};
/// Solana account key size in bytes (Pubkey). 每个账户(Pubkey)的字节数。 /// Solana account key size in bytes (Pubkey). 每个账户(Pubkey)的字节数。
pub const BYTES_PER_ACCOUNT: usize = 32; pub const BYTES_PER_ACCOUNT: usize = 32;
@@ -150,4 +143,4 @@ impl ExecutionPath {
slow_path() slow_path()
} }
} }
} }
+106 -69
View File
@@ -4,10 +4,15 @@ use solana_sdk::{
instruction::Instruction, message::AddressLookupTableAccount, pubkey::Pubkey, instruction::Instruction, message::AddressLookupTableAccount, pubkey::Pubkey,
signature::Keypair, signature::Signature, signature::Keypair, signature::Signature,
}; };
use std::{sync::Arc, time::{Duration, Instant}}; use std::{
sync::Arc,
time::{Duration, Instant},
};
#[allow(unused_imports)] #[allow(unused_imports)]
use tracing::{info, trace, warn}; use tracing::{info, trace, warn};
use super::{params::SwapParams, traits::InstructionBuilder};
use crate::swqos::TradeType;
use crate::{ use crate::{
common::{nonce_cache::DurableNonceInfo, GasFeeStrategy, SolanaRpcClient}, common::{nonce_cache::DurableNonceInfo, GasFeeStrategy, SolanaRpcClient},
perf::syscall_bypass::SystemCallBypassManager, perf::syscall_bypass::SystemCallBypassManager,
@@ -20,8 +25,6 @@ use crate::{
trading::MiddlewareManager, trading::MiddlewareManager,
}; };
use once_cell::sync::Lazy; use once_cell::sync::Lazy;
use crate::swqos::TradeType;
use super::{params::SwapParams, traits::InstructionBuilder};
/// Global syscall bypass manager (reserved for future time/IO optimizations). /// Global syscall bypass manager (reserved for future time/IO optimizations).
/// 全局系统调用绕过管理器(预留,后续可接入时间/IO 等优化)。 /// 全局系统调用绕过管理器(预留,后续可接入时间/IO 等优化)。
@@ -49,7 +52,10 @@ impl GenericTradeExecutor {
#[async_trait::async_trait] #[async_trait::async_trait]
impl TradeExecutor for GenericTradeExecutor { impl TradeExecutor for GenericTradeExecutor {
async fn swap(&self, params: SwapParams) -> Result<(bool, Vec<Signature>, Option<anyhow::Error>)> { async fn swap(
&self,
params: SwapParams,
) -> Result<(bool, Vec<Signature>, Option<anyhow::Error>)> {
// Sample total start only when logging or simulate. 仅在有日志或 simulate 时取起点。 // Sample total start only when logging or simulate. 仅在有日志或 simulate 时取起点。
let total_start = (params.log_enabled || params.simulate).then(Instant::now); let total_start = (params.log_enabled || params.simulate).then(Instant::now);
let timing_start_us: Option<i64> = if params.log_enabled { let timing_start_us: Option<i64> = if params.log_enabled {
@@ -58,7 +64,8 @@ impl TradeExecutor for GenericTradeExecutor {
None None
}; };
let is_buy = params.trade_type == TradeType::Buy || params.trade_type == TradeType::CreateAndBuy; let is_buy =
params.trade_type == TradeType::Buy || params.trade_type == TradeType::CreateAndBuy;
Prefetch::keypair(&params.payer); Prefetch::keypair(&params.payer);
@@ -77,7 +84,7 @@ impl TradeExecutor for GenericTradeExecutor {
Some(middleware_manager) => middleware_manager Some(middleware_manager) => middleware_manager
.apply_middlewares_process_protocol_instructions( .apply_middlewares_process_protocol_instructions(
instructions, instructions,
self.protocol_name.to_string(), self.protocol_name,
is_buy, is_buy,
)?, )?,
None => instructions, None => instructions,
@@ -85,7 +92,8 @@ impl TradeExecutor for GenericTradeExecutor {
let build_end_us = (params.log_enabled && crate::common::sdk_log::sdk_log_enabled()) let build_end_us = (params.log_enabled && crate::common::sdk_log::sdk_log_enabled())
.then(crate::common::clock::now_micros); .then(crate::common::clock::now_micros);
let _before_submit_elapsed = total_start.as_ref().map(|s| s.elapsed()).unwrap_or(Duration::ZERO); let _before_submit_elapsed =
total_start.as_ref().map(|s| s.elapsed()).unwrap_or(Duration::ZERO);
let before_submit_us = (params.log_enabled && crate::common::sdk_log::sdk_log_enabled()) let before_submit_us = (params.log_enabled && crate::common::sdk_log::sdk_log_enabled())
.then(crate::common::clock::now_micros); .then(crate::common::clock::now_micros);
@@ -111,12 +119,24 @@ impl TradeExecutor for GenericTradeExecutor {
if crate::common::sdk_log::sdk_log_enabled() { if crate::common::sdk_log::sdk_log_enabled() {
let dir = if is_buy { "Buy" } else { "Sell" }; let dir = if is_buy { "Buy" } else { "Sell" };
if let (Some(start_us), Some(end_us)) = (timing_start_us, build_end_us) { if let (Some(start_us), Some(end_us)) = (timing_start_us, build_end_us) {
println!(" [SDK] {} build_instructions: {:.4} ms", dir, (end_us - start_us) as f64 / 1000.0); println!(
" [SDK] {} build_instructions: {:.4} ms",
dir,
(end_us - start_us) as f64 / 1000.0
);
} }
if let (Some(start_us), Some(end_us)) = (timing_start_us, before_submit_us) { if let (Some(start_us), Some(end_us)) = (timing_start_us, before_submit_us) {
println!(" [SDK] {} before_submit: {:.4} ms", dir, (end_us - start_us) as f64 / 1000.0); println!(
" [SDK] {} before_submit: {:.4} ms",
dir,
(end_us - start_us) as f64 / 1000.0
);
} }
println!(" [SDK] {} simulate (dry-run): {:.4} ms", dir, send_elapsed.as_secs_f64() * 1000.0); println!(
" [SDK] {} simulate (dry-run): {:.4} ms",
dir,
send_elapsed.as_secs_f64() * 1000.0
);
println!(" [SDK] {} total: {:.4} ms", dir, total_elapsed.as_secs_f64() * 1000.0); println!(" [SDK] {} total: {:.4} ms", dir, total_elapsed.as_secs_f64() * 1000.0);
} }
@@ -124,10 +144,11 @@ impl TradeExecutor for GenericTradeExecutor {
} }
let need_confirm = params.wait_transaction_confirmed; let need_confirm = params.wait_transaction_confirmed;
let sender_config = params.sender_concurrency_config();
let result = execute_parallel( let result = execute_parallel(
&params.swqos_clients, params.swqos_clients.as_slice(),
params.payer, params.payer,
params.rpc.clone(), params.rpc.as_ref(),
final_instructions, final_instructions,
params.address_lookup_table_account, params.address_lookup_table_account,
params.recent_blockhash, params.recent_blockhash,
@@ -138,59 +159,66 @@ impl TradeExecutor for GenericTradeExecutor {
false, // submit only here; confirmation and log timing handled below false, // submit only here; confirmation and log timing handled below
if is_buy { true } else { params.with_tip }, if is_buy { true } else { params.with_tip },
params.gas_fee_strategy, params.gas_fee_strategy,
params.use_core_affinity, params.use_dedicated_sender_threads,
sender_config,
params.check_min_tip, params.check_min_tip,
) )
.await; .await;
let log_enabled = params.log_enabled && crate::common::sdk_log::sdk_log_enabled(); let log_enabled = params.log_enabled && crate::common::sdk_log::sdk_log_enabled();
let submit_timings = if log_enabled {
result.as_ref().ok().map(|(_, _, _, t)| t.clone()).unwrap_or_default() let (ok, signatures, err, submit_timings) = match result {
} else { Ok((success, sigs, last_error, timings)) => {
Vec::new() (success, sigs, last_error.map(|e| anyhow::anyhow!("{}", e)), timings)
}
Err(e) => (false, vec![], Some(anyhow::anyhow!("{}", e)), vec![]),
}; };
// submit_timings 为完成先后顺序(先完成的先 push),打印不排序、不增加延迟
let submit_timings_ref: &[(crate::swqos::SwqosType, i64)] = submit_timings.as_slice();
let result = if need_confirm { let result = if need_confirm {
let (ok, sigs, err) = match &result {
Ok((success, signatures, last_error, _)) => (
*success,
signatures.clone(),
last_error.as_ref().map(|e| anyhow::anyhow!("{}", e)),
),
Err(e) => (false, vec![], Some(anyhow::anyhow!("{}", e))),
};
let confirm_result = if let Some(rpc) = params.rpc.as_ref() { let confirm_result = if let Some(rpc) = params.rpc.as_ref() {
if sigs.is_empty() { if signatures.is_empty() {
(ok, sigs, err) (ok, signatures, err)
} else { } else {
let poll_res = poll_any_transaction_confirmation(rpc, &sigs, true).await; let poll_res = poll_any_transaction_confirmation(rpc, &signatures, true).await;
let confirm_done_us = log_enabled.then(crate::common::clock::now_micros); let confirm_done_us = log_enabled.then(crate::common::clock::now_micros);
if log_enabled { if log_enabled {
let dir = if is_buy { "Buy" } else { "Sell" }; let dir = if is_buy { "Buy" } else { "Sell" };
if let Some(start_us) = timing_start_us { if let Some(start_us) = timing_start_us {
if let Some(end_us) = build_end_us { if let Some(end_us) = build_end_us {
println!(" [SDK] {} build_instructions: {:.4} ms", dir, (end_us - start_us) as f64 / 1000.0); println!(
} " [SDK] {} build_instructions: {:.4} ms",
if let Some(end_us) = before_submit_us { dir,
println!(" [SDK] {} before_submit: {:.4} ms", dir, (end_us - start_us) as f64 / 1000.0); (end_us - start_us) as f64 / 1000.0
} );
if let Some(confirm_us) = confirm_done_us { }
let total_ms = (confirm_us - start_us) as f64 / 1000.0; if let Some(end_us) = before_submit_us {
for (swqos_type, submit_done_us) in &submit_timings { println!(
let submit_ms = (*submit_done_us - start_us).max(0) as f64 / 1000.0; " [SDK] {} before_submit: {:.4} ms",
let confirmed_ms = (confirm_us - *submit_done_us).max(0) as f64 / 1000.0; dir,
println!(" [SDK] {} {:?} submit: {:.4} ms, confirmed: {:.4} ms, total: {:.4} ms", dir, swqos_type, submit_ms, confirmed_ms, total_ms); (end_us - start_us) as f64 / 1000.0
);
}
if let Some(confirm_us) = confirm_done_us {
let total_ms = (confirm_us - start_us) as f64 / 1000.0;
for (swqos_type, submit_done_us) in submit_timings_ref {
let submit_ms =
(*submit_done_us - start_us).max(0) as f64 / 1000.0;
let confirmed_ms =
(confirm_us - *submit_done_us).max(0) as f64 / 1000.0;
println!(" [SDK] {} {:?} submit: {:.4} ms, confirmed: {:.4} ms, total: {:.4} ms", dir, swqos_type, submit_ms, confirmed_ms, total_ms);
}
} }
} }
} }
} match poll_res {
match poll_res { Ok(_) => (true, signatures, None),
Ok(_) => (true, sigs, None), Err(e) => (false, signatures, Some(e)),
Err(e) => (false, sigs, Some(e)), }
}
} }
} else { } else {
(ok, sigs, err) (ok, signatures, err)
}; };
Ok(confirm_result) Ok(confirm_result)
} else { } else {
@@ -198,18 +226,29 @@ impl TradeExecutor for GenericTradeExecutor {
let dir = if is_buy { "Buy" } else { "Sell" }; let dir = if is_buy { "Buy" } else { "Sell" };
if let Some(start_us) = timing_start_us { if let Some(start_us) = timing_start_us {
if let Some(end_us) = build_end_us { if let Some(end_us) = build_end_us {
println!(" [SDK] {} build_instructions: {:.4} ms", dir, (end_us - start_us) as f64 / 1000.0); println!(
" [SDK] {} build_instructions: {:.4} ms",
dir,
(end_us - start_us) as f64 / 1000.0
);
} }
if let Some(end_us) = before_submit_us { if let Some(end_us) = before_submit_us {
println!(" [SDK] {} before_submit: {:.4} ms", dir, (end_us - start_us) as f64 / 1000.0); println!(
" [SDK] {} before_submit: {:.4} ms",
dir,
(end_us - start_us) as f64 / 1000.0
);
} }
for (swqos_type, submit_done_us) in &submit_timings { for (swqos_type, submit_done_us) in submit_timings_ref {
let submit_ms = (*submit_done_us - start_us).max(0) as f64 / 1000.0; let submit_ms = (*submit_done_us - start_us).max(0) as f64 / 1000.0;
println!(" [SDK] {} {:?} submit: {:.4} ms, confirmed: -, total: {:.4} ms", dir, swqos_type, submit_ms, submit_ms); println!(
" [SDK] {} {:?} submit: {:.4} ms, confirmed: -, total: {:.4} ms",
dir, swqos_type, submit_ms, submit_ms
);
} }
} }
} }
result.map(|(a, b, c, _)| (a, b, c)) Ok((ok, signatures, err))
}; };
result result
@@ -256,22 +295,21 @@ async fn simulate_transaction(
let unit_limit = default_config.2.cu_limit; let unit_limit = default_config.2.cu_limit;
let unit_price = default_config.2.cu_price; let unit_price = default_config.2.cu_price;
// Build transaction for simulation
let transaction = build_transaction( let transaction = build_transaction(
payer.clone(), &payer,
Some(rpc.clone()), Some(&rpc),
unit_limit, unit_limit,
unit_price, unit_price,
&instructions, &instructions,
address_lookup_table_account, address_lookup_table_account.as_ref(),
recent_blockhash, recent_blockhash,
middleware_manager, middleware_manager.as_ref(),
protocol_name, protocol_name,
is_buy, is_buy,
false, // simulate doesn't need tip instruction false,
&Pubkey::default(), &Pubkey::default(),
tip, tip,
durable_nonce, durable_nonce.as_ref(),
) )
.await?; .await?;
@@ -281,14 +319,14 @@ async fn simulate_transaction(
.simulate_transaction_with_config( .simulate_transaction_with_config(
&transaction, &transaction,
RpcSimulateTransactionConfig { RpcSimulateTransactionConfig {
sig_verify: false, // Don't verify signature during simulation for speed sig_verify: false, // Don't verify signature during simulation for speed
replace_recent_blockhash: false, // Use actual blockhash from transaction replace_recent_blockhash: false, // Use actual blockhash from transaction
commitment: Some(CommitmentConfig { commitment: Some(CommitmentConfig {
commitment: CommitmentLevel::Processed, // Use Processed level to get latest state commitment: CommitmentLevel::Processed, // Use Processed level to get latest state
}), }),
encoding: Some(UiTransactionEncoding::Base64), // Base64 encoding encoding: Some(UiTransactionEncoding::Base64), // Base64 encoding
accounts: None, // Don't return specific account states (can be specified if needed) accounts: None, // Don't return specific account states (can be specified if needed)
min_context_slot: None, // Don't specify minimum context slot min_context_slot: None, // Don't specify minimum context slot
inner_instructions: true, // Enable inner instructions for debugging and detailed execution flow inner_instructions: true, // Enable inner instructions for debugging and detailed execution flow
}, },
) )
@@ -356,10 +394,9 @@ mod tests {
} }
println!("\n--- 3. 不等待链上确认时:每行 total = 该通道 submit 耗时(独立)---\n"); println!("\n--- 3. 不等待链上确认时:每行 total = 该通道 submit 耗时(独立)---\n");
for (swqos_type, submit_ms, total_ms) in [ for (swqos_type, submit_ms, total_ms) in
(SwqosType::Jito, 44.20, 44.20), [(SwqosType::Jito, 44.20, 44.20), (SwqosType::Helius, 51.80, 51.80)]
(SwqosType::Helius, 51.80, 51.80), {
] {
println!( println!(
" [SDK] {} {:?} submit: {:.4} ms, confirmed: -, total: {:.4} ms", " [SDK] {} {:?} submit: {:.4} ms, confirmed: -, total: {:.4} ms",
dir, swqos_type, submit_ms, total_ms dir, swqos_type, submit_ms, total_ms
+3 -3
View File
@@ -1,6 +1,6 @@
pub mod async_executor;
pub mod execution;
pub mod executor;
pub mod params; pub mod params;
pub mod traits; pub mod traits;
pub mod executor;
pub mod async_executor;
pub mod transaction_pool; pub mod transaction_pool;
pub mod execution;
+50 -7
View File
@@ -3,7 +3,16 @@ use crate::common::nonce_cache::DurableNonceInfo;
use crate::common::spl_associated_token_account::get_associated_token_address_with_program_id; use crate::common::spl_associated_token_account::get_associated_token_address_with_program_id;
use crate::common::{GasFeeStrategy, SolanaRpcClient}; use crate::common::{GasFeeStrategy, SolanaRpcClient};
use crate::constants::TOKEN_PROGRAM; use crate::constants::TOKEN_PROGRAM;
use core_affinity::CoreId;
use crate::instruction::utils::pumpfun::global_constants::MAYHEM_FEE_RECIPIENT; use crate::instruction::utils::pumpfun::global_constants::MAYHEM_FEE_RECIPIENT;
/// Concurrency + core binding config for parallel submit (precomputed at SDK init, one param on hot path). Uses Arc so no borrow of SwapParams.
#[derive(Clone)]
pub struct SenderConcurrencyConfig {
pub sender_thread_cores: Option<Arc<Vec<usize>>>,
pub effective_core_ids: Arc<Vec<CoreId>>,
pub max_sender_concurrency: usize,
}
use crate::instruction::utils::pumpswap::accounts::MAYHEM_FEE_RECIPIENT as MAYHEM_FEE_RECIPIENT_SWAP; use crate::instruction::utils::pumpswap::accounts::MAYHEM_FEE_RECIPIENT as MAYHEM_FEE_RECIPIENT_SWAP;
use crate::swqos::{SwqosClient, TradeType}; use crate::swqos::{SwqosClient, TradeType};
use crate::trading::common::get_multi_token_balances; use crate::trading::common::get_multi_token_balances;
@@ -56,7 +65,8 @@ pub struct SwapParams {
pub wait_transaction_confirmed: bool, pub wait_transaction_confirmed: bool,
pub protocol_params: DexParamEnum, pub protocol_params: DexParamEnum,
pub open_seed_optimize: bool, pub open_seed_optimize: bool,
pub swqos_clients: Vec<Arc<SwqosClient>>, /// Arc<Vec<..>> so cloning from infrastructure is a single Arc clone.
pub swqos_clients: Arc<Vec<Arc<SwqosClient>>>,
pub middleware_manager: Option<Arc<MiddlewareManager>>, pub middleware_manager: Option<Arc<MiddlewareManager>>,
pub durable_nonce: Option<DurableNonceInfo>, pub durable_nonce: Option<DurableNonceInfo>,
pub with_tip: bool, pub with_tip: bool,
@@ -69,8 +79,14 @@ pub struct SwapParams {
pub simulate: bool, pub simulate: bool,
/// Whether to output SDK logs (from TradeConfig.log_enabled). /// Whether to output SDK logs (from TradeConfig.log_enabled).
pub log_enabled: bool, pub log_enabled: bool,
/// Whether to pin parallel submit tasks to cores (from TradeConfig.use_core_affinity). /// Use dedicated sender threads (internal; set via client.with_dedicated_sender_threads()).
pub use_core_affinity: bool, pub use_dedicated_sender_threads: bool,
/// Core indices for dedicated sender threads (from TradeConfig.sender_thread_cores). Arc avoids cloning the Vec on hot path.
pub sender_thread_cores: Option<Arc<Vec<usize>>>,
/// Precomputed at SDK init: min(swqos_count, 2/3*cores). Avoids get_core_ids() on trade hot path.
pub max_sender_concurrency: usize,
/// Precomputed at SDK init: first max_sender_concurrency CoreIds for job affinity. Arc clone only.
pub effective_core_ids: Arc<Vec<CoreId>>,
/// Whether to check minimum tip per SWQOS (from TradeConfig.check_min_tip). When false, skip filter for lower latency. /// Whether to check minimum tip per SWQOS (from TradeConfig.check_min_tip). When false, skip filter for lower latency.
pub check_min_tip: bool, pub check_min_tip: bool,
/// Optional event receive time in microseconds (same scale as sol-parser-sdk clock::now_micros). Used as timing start when log_enabled. /// Optional event receive time in microseconds (same scale as sol-parser-sdk clock::now_micros). Used as timing start when log_enabled.
@@ -82,6 +98,18 @@ pub struct SwapParams {
pub use_exact_sol_amount: Option<bool>, pub use_exact_sol_amount: Option<bool>,
} }
impl SwapParams {
/// One struct for execute_parallel: merges sender_thread_cores, effective_core_ids, max_sender_concurrency. Arc clone only.
#[inline]
pub fn sender_concurrency_config(&self) -> SenderConcurrencyConfig {
SenderConcurrencyConfig {
sender_thread_cores: self.sender_thread_cores.clone(),
effective_core_ids: self.effective_core_ids.clone(),
max_sender_concurrency: self.max_sender_concurrency,
}
}
}
impl std::fmt::Debug for SwapParams { impl std::fmt::Debug for SwapParams {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result { fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
write!(f, "SwapParams: ...") write!(f, "SwapParams: ...")
@@ -89,11 +117,18 @@ impl std::fmt::Debug for SwapParams {
} }
/// PumpFun protocol specific parameters /// PumpFun protocol specific parameters
/// Configuration parameters specific to PumpFun trading protocol /// Configuration parameters specific to PumpFun trading protocol.
///
/// **Creator Rewards Sharing**: Some coins use a dynamic `creator_vault` (fee-sharing config).
/// Always use the latest on-chain creator/vault when building params for **sell**; do not reuse
/// cached params from buy. Either fetch fresh data via RPC, or pass `creator_vault` from gRPC
/// using [`from_trade`](PumpFunParams::from_trade) / [`from_dev_trade`](PumpFunParams::from_dev_trade),
/// or override with [`with_creator_vault`](PumpFunParams::with_creator_vault).
#[derive(Clone)] #[derive(Clone)]
pub struct PumpFunParams { pub struct PumpFunParams {
pub bonding_curve: Arc<BondingCurveAccount>, pub bonding_curve: Arc<BondingCurveAccount>,
pub associated_bonding_curve: Pubkey, pub associated_bonding_curve: Pubkey,
/// Creator vault PDA. For Creator Rewards Sharing coins this can change; pass latest from gRPC when selling.
pub creator_vault: Pubkey, pub creator_vault: Pubkey,
pub token_program: Pubkey, pub token_program: Pubkey,
/// Whether to close token account when selling, only effective during sell operations /// Whether to close token account when selling, only effective during sell operations
@@ -222,6 +257,14 @@ impl PumpFunParams {
token_program: mint_account.owner, token_program: mint_account.owner,
}) })
} }
/// Override `creator_vault` with a value from gRPC/event (e.g. for Creator Rewards Sharing).
/// Use when selling so the instruction uses the latest on-chain vault and avoids "seeds constraint violated" (2006).
#[inline]
pub fn with_creator_vault(mut self, creator_vault: Pubkey) -> Self {
self.creator_vault = creator_vault;
self
}
} }
/// PumpSwap Protocol Specific Parameters /// PumpSwap Protocol Specific Parameters
@@ -383,7 +426,7 @@ impl PumpSwapParams {
); );
Ok(Self { Ok(Self {
pool: pool_address.clone(), pool: *pool_address,
base_mint: pool_data.base_mint, base_mint: pool_data.base_mint,
quote_mint: pool_data.quote_mint, quote_mint: pool_data.quote_mint,
pool_base_token_account: pool_data.pool_base_token_account, pool_base_token_account: pool_data.pool_base_token_account,
@@ -656,7 +699,7 @@ impl RaydiumCpmmParams {
) )
.await?; .await?;
Ok(Self { Ok(Self {
pool_state: pool_address.clone(), pool_state: *pool_address,
amm_config: pool.amm_config, amm_config: pool.amm_config,
base_mint: pool.token0_mint, base_mint: pool.token0_mint,
quote_mint: pool.token1_mint, quote_mint: pool.token1_mint,
@@ -763,7 +806,7 @@ impl MeteoraDammV2Params {
let pool_data = let pool_data =
crate::instruction::utils::meteora_damm_v2::fetch_pool(rpc, pool_address).await?; crate::instruction::utils::meteora_damm_v2::fetch_pool(rpc, pool_address).await?;
Ok(Self { Ok(Self {
pool: pool_address.clone(), pool: *pool_address,
token_a_vault: pool_data.token_a_vault, token_a_vault: pool_data.token_a_vault,
token_b_vault: pool_data.token_b_vault, token_b_vault: pool_data.token_b_vault,
token_a_mint: pool_data.token_a_mint, token_a_mint: pool_data.token_a_mint,
+4 -1
View File
@@ -9,7 +9,10 @@ pub trait TradeExecutor: Send + Sync {
/// - bool: 是否至少有一个交易成功 /// - bool: 是否至少有一个交易成功
/// - Vec<Signature>: 所有提交的交易签名(按SWQOS顺序) /// - Vec<Signature>: 所有提交的交易签名(按SWQOS顺序)
/// - Option<anyhow::Error>: 最后一个错误(如果全部失败) /// - Option<anyhow::Error>: 最后一个错误(如果全部失败)
async fn swap(&self, params: SwapParams) -> Result<(bool, Vec<Signature>, Option<anyhow::Error>)>; async fn swap(
&self,
params: SwapParams,
) -> Result<(bool, Vec<Signature>, Option<anyhow::Error>)>;
/// 获取协议名称 /// 获取协议名称
fn protocol_name(&self) -> &'static str; fn protocol_name(&self) -> &'static str;
} }
+31 -29
View File
@@ -6,10 +6,24 @@
//! - 零拷贝 I/O //! - 零拷贝 I/O
//! - 内存预热 //! - 内存预热
/// 预分配指令容量(单笔交易常见指令数)
const TX_BUILDER_INSTRUCTION_CAP: usize = 32;
/// 预分配地址查找表数量
const TX_BUILDER_LOOKUP_TABLE_CAP: usize = 8;
/// 对象池最大容量
const TX_BUILDER_POOL_CAP: usize = 1000;
/// 多路提交并发数(与 async_executor SWQOS_DEDICATED_DEFAULT_THREADS 一致,保证不串行)
const PARALLEL_SENDER_COUNT: usize = 18;
/// 启动时预填充数量,必须 >= PARALLEL_SENDER_COUNT,否则 18 路并发 build 会触发分配或争抢
const TX_BUILDER_POOL_PREFILL: usize = 64;
use crossbeam_queue::ArrayQueue; use crossbeam_queue::ArrayQueue;
use once_cell::sync::Lazy; use once_cell::sync::Lazy;
use solana_sdk::{ use solana_sdk::{
hash::Hash, instruction::Instruction, message::{v0, AddressLookupTableAccount, Message, VersionedMessage}, pubkey::Pubkey hash::Hash,
instruction::Instruction,
message::{v0, AddressLookupTableAccount, Message, VersionedMessage},
pubkey::Pubkey,
}; };
use std::sync::Arc; use std::sync::Arc;
/// 预分配的交易构建器 /// 预分配的交易构建器
@@ -23,8 +37,8 @@ pub struct PreallocatedTxBuilder {
impl PreallocatedTxBuilder { impl PreallocatedTxBuilder {
fn new() -> Self { fn new() -> Self {
Self { Self {
instructions: Vec::with_capacity(32), // 预分配32条指令空间 instructions: Vec::with_capacity(TX_BUILDER_INSTRUCTION_CAP),
lookup_tables: Vec::with_capacity(8), // 预分配8个查找表空间 lookup_tables: Vec::with_capacity(TX_BUILDER_LOOKUP_TABLE_CAP),
} }
} }
@@ -65,31 +79,25 @@ impl PreallocatedTxBuilder {
&mut self, &mut self,
payer: &Pubkey, payer: &Pubkey,
instructions: &[Instruction], instructions: &[Instruction],
address_lookup_table_account: Option<AddressLookupTableAccount>, address_lookup_table_account: Option<&AddressLookupTableAccount>,
recent_blockhash: Hash, recent_blockhash: Hash,
) -> VersionedMessage { ) -> VersionedMessage {
// 重用已分配的 vector
self.reset(); self.reset();
self.instructions.extend_from_slice(instructions); self.instructions.extend_from_slice(instructions);
// ✅ 如果有查找表,使用 V0 消息 if let Some(alt) = address_lookup_table_account {
if let Some(address_lookup_table_account) = address_lookup_table_account { let message = v0::Message::try_compile(
let message = v0::Message::try_compile(
payer, payer,
&self.instructions, &self.instructions,
&[address_lookup_table_account], std::slice::from_ref(alt),
recent_blockhash, recent_blockhash,
).expect("v0 message compile failed"); )
.expect("v0 message compile failed");
VersionedMessage::V0(message) VersionedMessage::V0(message)
} else { } else {
// ✅ 没有查找表,使用 Legacy 消息(兼容所有 RPC // ✅ 没有查找表,使用 Legacy 消息(兼容所有 RPC
let message = Message::new_with_blockhash( let message =
&self.instructions, Message::new_with_blockhash(&self.instructions, Some(payer), &recent_blockhash);
Some(payer),
&recent_blockhash,
);
VersionedMessage::Legacy(message) VersionedMessage::Legacy(message)
} }
} }
@@ -97,22 +105,18 @@ impl PreallocatedTxBuilder {
/// 🚀 全局交易构建器对象池 /// 🚀 全局交易构建器对象池
static TX_BUILDER_POOL: Lazy<Arc<ArrayQueue<PreallocatedTxBuilder>>> = Lazy::new(|| { static TX_BUILDER_POOL: Lazy<Arc<ArrayQueue<PreallocatedTxBuilder>>> = Lazy::new(|| {
let pool = ArrayQueue::new(1000); // 1000个预分配构建器 let pool = ArrayQueue::new(TX_BUILDER_POOL_CAP);
let prefill = TX_BUILDER_POOL_PREFILL.max(PARALLEL_SENDER_COUNT);
// 预填充池 for _ in 0..prefill {
for _ in 0..100 {
let _ = pool.push(PreallocatedTxBuilder::new()); let _ = pool.push(PreallocatedTxBuilder::new());
} }
Arc::new(pool) Arc::new(pool)
}); });
/// 🚀 从池中获取构建器 /// 🚀 从池中获取构建器
#[inline(always)] #[inline(always)]
pub fn acquire_builder() -> PreallocatedTxBuilder { pub fn acquire_builder() -> PreallocatedTxBuilder {
TX_BUILDER_POOL TX_BUILDER_POOL.pop().unwrap_or_else(|| PreallocatedTxBuilder::new())
.pop()
.unwrap_or_else(|| PreallocatedTxBuilder::new())
} }
/// 🚀 归还构建器到池 /// 🚀 归还构建器到池
@@ -134,9 +138,7 @@ pub struct TxBuilderGuard {
impl TxBuilderGuard { impl TxBuilderGuard {
pub fn new() -> Self { pub fn new() -> Self {
Self { Self { builder: Some(acquire_builder()) }
builder: Some(acquire_builder()),
}
} }
pub fn get_mut(&mut self) -> &mut PreallocatedTxBuilder { pub fn get_mut(&mut self) -> &mut PreallocatedTxBuilder {
@@ -150,4 +152,4 @@ impl Drop for TxBuilderGuard {
release_builder(builder); release_builder(builder);
} }
} }
} }
+2 -2
View File
@@ -14,7 +14,7 @@ impl InstructionMiddleware for LoggingMiddleware {
fn process_protocol_instructions( fn process_protocol_instructions(
&self, &self,
protocol_instructions: Vec<Instruction>, protocol_instructions: Vec<Instruction>,
protocol_name: String, protocol_name: &str,
is_buy: bool, is_buy: bool,
) -> Result<Vec<Instruction>> { ) -> Result<Vec<Instruction>> {
println!("-------------------[{}]-------------------", self.name()); println!("-------------------[{}]-------------------", self.name());
@@ -32,7 +32,7 @@ impl InstructionMiddleware for LoggingMiddleware {
fn process_full_instructions( fn process_full_instructions(
&self, &self,
full_instructions: Vec<Instruction>, full_instructions: Vec<Instruction>,
protocol_name: String, protocol_name: &str,
is_buy: bool, is_buy: bool,
) -> Result<Vec<Instruction>> { ) -> Result<Vec<Instruction>> {
println!("-------------------[{}]-------------------", self.name()); println!("-------------------[{}]-------------------", self.name());
+1 -1
View File
@@ -1,4 +1,4 @@
pub mod traits;
pub mod builtin; pub mod builtin;
pub mod traits;
pub use traits::{InstructionMiddleware, MiddlewareManager}; pub use traits::{InstructionMiddleware, MiddlewareManager};
+7 -10
View File
@@ -20,7 +20,7 @@ pub trait InstructionMiddleware: Send + Sync {
fn process_protocol_instructions( fn process_protocol_instructions(
&self, &self,
protocol_instructions: Vec<Instruction>, protocol_instructions: Vec<Instruction>,
protocol_name: String, protocol_name: &str,
is_buy: bool, is_buy: bool,
) -> Result<Vec<Instruction>>; ) -> Result<Vec<Instruction>>;
@@ -36,7 +36,7 @@ pub trait InstructionMiddleware: Send + Sync {
fn process_full_instructions( fn process_full_instructions(
&self, &self,
full_instructions: Vec<Instruction>, full_instructions: Vec<Instruction>,
protocol_name: String, protocol_name: &str,
is_buy: bool, is_buy: bool,
) -> Result<Vec<Instruction>>; ) -> Result<Vec<Instruction>>;
@@ -72,15 +72,12 @@ impl MiddlewareManager {
pub fn apply_middlewares_process_full_instructions( pub fn apply_middlewares_process_full_instructions(
&self, &self,
mut full_instructions: Vec<Instruction>, mut full_instructions: Vec<Instruction>,
protocol_name: String, protocol_name: &str,
is_buy: bool, is_buy: bool,
) -> Result<Vec<Instruction>> { ) -> Result<Vec<Instruction>> {
for middleware in &self.middlewares { for middleware in &self.middlewares {
full_instructions = middleware.process_full_instructions( full_instructions =
full_instructions, middleware.process_full_instructions(full_instructions, protocol_name, is_buy)?;
protocol_name.clone(),
is_buy,
)?;
if full_instructions.is_empty() { if full_instructions.is_empty() {
break; break;
} }
@@ -92,13 +89,13 @@ impl MiddlewareManager {
pub fn apply_middlewares_process_protocol_instructions( pub fn apply_middlewares_process_protocol_instructions(
&self, &self,
mut protocol_instructions: Vec<Instruction>, mut protocol_instructions: Vec<Instruction>,
protocol_name: String, protocol_name: &str,
is_buy: bool, is_buy: bool,
) -> Result<Vec<Instruction>> { ) -> Result<Vec<Instruction>> {
for middleware in &self.middlewares { for middleware in &self.middlewares {
protocol_instructions = middleware.process_protocol_instructions( protocol_instructions = middleware.process_protocol_instructions(
protocol_instructions, protocol_instructions,
protocol_name.clone(), protocol_name,
is_buy, is_buy,
)?; )?;
if protocol_instructions.is_empty() { if protocol_instructions.is_empty() {
+4 -4
View File
@@ -1,6 +1,6 @@
pub mod pumpfun;
pub mod common;
pub mod pumpswap;
pub mod bonk; pub mod bonk;
pub mod common;
pub mod pumpfun;
pub mod pumpswap;
pub mod raydium_amm_v4; pub mod raydium_amm_v4;
pub mod raydium_cpmm; pub mod raydium_cpmm;
+38 -4
View File
@@ -14,7 +14,8 @@ impl TradingClient {
#[inline] #[inline]
pub async fn get_payer_sol_balance(&self) -> Result<u64, anyhow::Error> { pub async fn get_payer_sol_balance(&self) -> Result<u64, anyhow::Error> {
trading::common::utils::get_sol_balance(&self.infrastructure.rpc, &self.payer.pubkey()).await trading::common::utils::get_sol_balance(&self.infrastructure.rpc, &self.payer.pubkey())
.await
} }
#[inline] #[inline]
@@ -28,7 +29,29 @@ impl TradingClient {
#[inline] #[inline]
pub async fn get_payer_token_balance(&self, mint: &Pubkey) -> Result<u64, anyhow::Error> { pub async fn get_payer_token_balance(&self, mint: &Pubkey) -> Result<u64, anyhow::Error> {
trading::common::utils::get_token_balance(&self.infrastructure.rpc, &self.payer.pubkey(), mint).await trading::common::utils::get_token_balance(
&self.infrastructure.rpc,
&self.payer.pubkey(),
mint,
)
.await
}
/// 使用与交易一致的 ATA 推导(含 seed 优化)查询 payer 某 mint 的余额;卖出前查余额应使用此接口并传入池的 base_token_program,否则若使用 seed ATA 会查错账户。
#[inline]
pub async fn get_payer_token_balance_with_program(
&self,
mint: &Pubkey,
token_program: &Pubkey,
) -> Result<u64, anyhow::Error> {
trading::common::utils::get_token_balance_with_options(
&self.infrastructure.rpc,
&self.payer.pubkey(),
mint,
token_program,
self.use_seed_optimize,
)
.await
} }
#[inline] #[inline]
@@ -48,11 +71,22 @@ impl TradingClient {
receive_wallet: &Pubkey, receive_wallet: &Pubkey,
amount: u64, amount: u64,
) -> Result<(), anyhow::Error> { ) -> Result<(), anyhow::Error> {
trading::common::utils::transfer_sol(&self.infrastructure.rpc, payer, receive_wallet, amount).await trading::common::utils::transfer_sol(
&self.infrastructure.rpc,
payer,
receive_wallet,
amount,
)
.await
} }
#[inline] #[inline]
pub async fn close_token_account(&self, mint: &Pubkey) -> Result<(), anyhow::Error> { pub async fn close_token_account(&self, mint: &Pubkey) -> Result<(), anyhow::Error> {
trading::common::utils::close_token_account(&self.infrastructure.rpc, self.payer.as_ref(), mint).await trading::common::utils::close_token_account(
&self.infrastructure.rpc,
self.payer.as_ref(),
mint,
)
.await
} }
} }
+1 -1
View File
@@ -1,7 +1,7 @@
pub mod bonk; pub mod bonk;
pub mod common;
pub mod pumpfun; pub mod pumpfun;
pub mod pumpswap; pub mod pumpswap;
pub mod raydium_amm_v4; pub mod raydium_amm_v4;
pub mod raydium_clmm; pub mod raydium_clmm;
pub mod raydium_cpmm; pub mod raydium_cpmm;
pub mod common;