mirror of
https://github.com/floor-licker/polyfill-rs.git
synced 2026-08-16 14:08:07 +00:00
Initial commit (clean, no build artifacts)
This commit is contained in:
+203
@@ -0,0 +1,203 @@
|
||||
//! Polyfill-rs: High-performance Rust client for Polymarket
|
||||
//!
|
||||
//! A production-ready Rust client for Polymarket optimized for high-frequency trading.
|
||||
//!
|
||||
//! # Features
|
||||
//!
|
||||
//! - **High-performance order book management** with optimized data structures
|
||||
//! - **Real-time market data streaming** with WebSocket support
|
||||
//! - **Trade execution simulation** with slippage protection
|
||||
//! - **Comprehensive error handling** with specific error types
|
||||
//! - **Rate limiting and retry logic** for robust API interactions
|
||||
//! - **Ethereum integration** with EIP-712 signing support
|
||||
//! - **Benchmarking tools** for performance analysis
|
||||
//!
|
||||
//! # Quick Start
|
||||
//!
|
||||
//! ```rust
|
||||
//! use polyfill_rs::{ClobClient, OrderArgs, Side};
|
||||
//! use rust_decimal::Decimal;
|
||||
//!
|
||||
//! #[tokio::main]
|
||||
//! async fn main() -> Result<(), Box<dyn std::error::Error>> {
|
||||
//! // Create client (compatible with polymarket-rs-client)
|
||||
//! let mut client = ClobClient::with_l1_headers(
|
||||
//! "https://clob.polymarket.com",
|
||||
//! "your_private_key",
|
||||
//! 137,
|
||||
//! );
|
||||
//!
|
||||
//! // Get API credentials
|
||||
//! let api_creds = client.create_or_derive_api_key(None).await?;
|
||||
//! client.set_api_creds(api_creds);
|
||||
//!
|
||||
//! // Create and post order
|
||||
//! let order_args = OrderArgs::new(
|
||||
//! "token_id",
|
||||
//! Decimal::from_str("0.75")?,
|
||||
//! Decimal::from_str("100.0")?,
|
||||
//! Side::BUY,
|
||||
//! );
|
||||
//!
|
||||
//! let result = client.create_and_post_order(&order_args).await?;
|
||||
//! println!("Order posted: {:?}", result);
|
||||
//!
|
||||
//! Ok(())
|
||||
//! }
|
||||
//! ```
|
||||
//!
|
||||
//! # Advanced Usage
|
||||
//!
|
||||
//! ```rust
|
||||
//! use polyfill_rs::{PolyfillClient, ClientConfig};
|
||||
//!
|
||||
//! #[tokio::main]
|
||||
//! async fn main() -> Result<(), Box<dyn std::error::Error>> {
|
||||
//! // Advanced configuration
|
||||
//! let config = ClientConfig {
|
||||
//! base_url: "https://clob.polymarket.com".to_string(),
|
||||
//! chain_id: 137,
|
||||
//! private_key: Some("your_private_key".to_string()),
|
||||
//! max_slippage: Some(Decimal::from_str("0.001")?),
|
||||
//! fee_rate: Some(Decimal::from_str("0.02")?),
|
||||
//! ..Default::default()
|
||||
//! };
|
||||
//!
|
||||
//! let mut client = PolyfillClient::with_config(config)?;
|
||||
//!
|
||||
//! // Subscribe to real-time order book updates
|
||||
//! client.subscribe_to_order_book("token_id").await?;
|
||||
//!
|
||||
//! // Process incoming messages
|
||||
//! while let Some(message) = client.get_next_message().await? {
|
||||
//! println!("Received: {:?}", message);
|
||||
//! }
|
||||
//!
|
||||
//! Ok(())
|
||||
//! }
|
||||
//! ```
|
||||
|
||||
use tracing::info;
|
||||
|
||||
|
||||
// Global constants
|
||||
pub const DEFAULT_CHAIN_ID: u64 = 137; // Polygon
|
||||
pub const DEFAULT_BASE_URL: &str = "https://clob.polymarket.com";
|
||||
pub const DEFAULT_TIMEOUT_SECS: u64 = 30;
|
||||
pub const DEFAULT_MAX_RETRIES: u32 = 3;
|
||||
pub const DEFAULT_RATE_LIMIT_RPS: u32 = 100;
|
||||
|
||||
// Initialize logging
|
||||
pub fn init() {
|
||||
tracing_subscriber::fmt::init();
|
||||
info!("Polyfill-rs initialized");
|
||||
}
|
||||
|
||||
// Re-export main types
|
||||
pub use crate::types::{
|
||||
ApiCredentials, Balance, ClientConfig, FillEvent, MarketSnapshot, Order, OrderBook,
|
||||
OrderDelta, OrderRequest, OrderStatus, OrderType, Side, StreamMessage, WssAuth,
|
||||
WssSubscription, WssChannelType,
|
||||
};
|
||||
|
||||
// Re-export client
|
||||
pub use crate::client::{ClobClient, PolyfillClient};
|
||||
|
||||
// Re-export compatibility types (for easy migration from polymarket-rs-client)
|
||||
pub use crate::client::{
|
||||
OrderArgs, OrderBookSummary,
|
||||
};
|
||||
|
||||
// Re-export error types
|
||||
pub use crate::errors::{PolyfillError, Result};
|
||||
|
||||
// Re-export advanced components
|
||||
pub use crate::book::{OrderBook as OrderBookImpl, OrderBookManager};
|
||||
pub use crate::fill::{FillEngine, FillResult};
|
||||
pub use crate::stream::{MarketStream, StreamManager, WebSocketStream};
|
||||
pub use crate::decode::Decoder;
|
||||
|
||||
// Re-export utilities
|
||||
pub use crate::utils::{
|
||||
crypto, math, retry, time, url, rate_limit,
|
||||
};
|
||||
|
||||
// Module declarations
|
||||
pub mod book;
|
||||
pub mod client;
|
||||
pub mod decode;
|
||||
pub mod errors;
|
||||
pub mod fill;
|
||||
pub mod stream;
|
||||
pub mod types;
|
||||
pub mod utils;
|
||||
|
||||
// Benchmarks
|
||||
#[cfg(test)]
|
||||
mod benches {
|
||||
use criterion::{criterion_group, criterion_main};
|
||||
use crate::{OrderBookManager, OrderDelta, Side};
|
||||
use rust_decimal::Decimal;
|
||||
use chrono::Utc;
|
||||
use std::str::FromStr;
|
||||
|
||||
fn order_book_benchmark(c: &mut criterion::Criterion) {
|
||||
let mut book_manager = OrderBookManager::new(100);
|
||||
|
||||
c.bench_function("apply_order_delta", |b| {
|
||||
b.iter(|| {
|
||||
let delta = OrderDelta {
|
||||
token_id: "test_token".to_string(),
|
||||
timestamp: Utc::now(),
|
||||
side: Side::BUY,
|
||||
price: Decimal::from_str("0.75").unwrap(),
|
||||
size: Decimal::from_str("100.0").unwrap(),
|
||||
sequence: 1,
|
||||
};
|
||||
|
||||
let _ = book_manager.apply_delta(delta);
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
criterion_group!(benches, order_book_benchmark);
|
||||
criterion_main!(benches);
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
use rust_decimal::Decimal;
|
||||
use std::str::FromStr;
|
||||
use alloy_primitives::U256;
|
||||
|
||||
#[test]
|
||||
fn test_client_creation() {
|
||||
let client = ClobClient::new("https://test.example.com");
|
||||
// Test that the client was created successfully
|
||||
// We can't test private fields, but we can verify the client exists
|
||||
assert!(true); // Client creation successful
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_order_args_creation() {
|
||||
let args = OrderArgs::new(
|
||||
"test_token",
|
||||
Decimal::from_str("0.75").unwrap(),
|
||||
Decimal::from_str("100.0").unwrap(),
|
||||
Side::BUY,
|
||||
);
|
||||
|
||||
assert_eq!(args.token_id, "test_token");
|
||||
assert_eq!(args.side, Side::BUY);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_order_args_default() {
|
||||
let args = OrderArgs::default();
|
||||
assert_eq!(args.token_id, "");
|
||||
assert_eq!(args.price, Decimal::ZERO);
|
||||
assert_eq!(args.size, Decimal::ZERO);
|
||||
assert_eq!(args.side, Side::BUY);
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user