digdigdig3_station/data/
block_trade.rs1use digdigdig3::core::types::{StreamEvent, TradeSide};
2use serde::{Deserialize, Serialize};
3
4use crate::series::DataPoint;
5
6#[derive(Debug, Clone, Serialize, Deserialize)]
12pub struct BlockTradePoint {
13 pub ts_ms: i64,
14 pub block_id: String,
15 pub price: f64,
16 pub quantity: f64,
17 pub side: TradeSide,
18 pub is_iv: bool,
19}
20
21const TAIL_OFFSET: usize = 32;
22
23impl DataPoint for BlockTradePoint {
24 const RECORD_SIZE: usize = 44;
25
26 fn encode(&self, out: &mut [u8]) {
27 out[0..8].copy_from_slice(&(self.ts_ms as u64).to_le_bytes());
28 out[8..16].copy_from_slice(&self.price.to_le_bytes());
29 out[16..24].copy_from_slice(&self.quantity.to_le_bytes());
30 out[24] = match self.side {
31 TradeSide::Buy => 0,
32 TradeSide::Sell => 1,
33 };
34 out[25] = u8::from(self.is_iv);
35 }
36
37 fn decode(bytes: &[u8]) -> Option<Self> {
38 if bytes.len() != Self::RECORD_SIZE { return None; }
39 Some(Self {
40 ts_ms: u64::from_le_bytes(bytes[0..8].try_into().ok()?) as i64,
41 block_id: String::new(),
42 price: f64::from_le_bytes(bytes[8..16].try_into().ok()?),
43 quantity: f64::from_le_bytes(bytes[16..24].try_into().ok()?),
44 side: match bytes[24] { 0 => TradeSide::Buy, _ => TradeSide::Sell },
45 is_iv: bytes[25] != 0,
46 })
47 }
48
49 fn timestamp_ms(&self) -> i64 { self.ts_ms }
50
51 fn from_stream_event(ev: &StreamEvent) -> Option<Self> {
52 if let StreamEvent::BlockTrade {
53 symbol: _,
54 block_id,
55 price,
56 quantity,
57 side,
58 timestamp,
59 is_iv,
60 } = ev {
61 Some(Self {
62 ts_ms: *timestamp,
63 block_id: block_id.clone(),
64 price: *price,
65 quantity: *quantity,
66 side: *side,
67 is_iv: *is_iv,
68 })
69 } else {
70 None
71 }
72 }
73
74 fn encode_blob(&self) -> Option<Vec<u8>> {
75 let bytes = self.block_id.as_bytes();
76 let mut out = Vec::with_capacity(2 + bytes.len());
77 out.extend_from_slice(&(bytes.len() as u16).to_le_bytes());
78 out.extend_from_slice(bytes);
79 Some(out)
80 }
81
82 fn decode_blob(header: &[u8], blob: &[u8]) -> Option<Self> {
83 let mut p = Self::decode(header)?;
84 if blob.len() >= 2 {
85 let len = u16::from_le_bytes(blob[0..2].try_into().ok()?) as usize;
86 if blob.len() >= 2 + len {
87 p.block_id = String::from_utf8_lossy(&blob[2..2 + len]).into_owned();
88 }
89 }
90 Some(p)
91 }
92
93 fn blob_pointer_offset() -> Option<usize> { Some(TAIL_OFFSET) }
94}