1use std::fmt::Display;
17
18use nautilus_core::UnixNanos;
19use serde::{Deserialize, Serialize};
20use ustr::Ustr;
21
22use super::{OrderEvent, OrderEventType};
23use crate::{
24 events::{
25 OrderAccepted, OrderCancelRejected, OrderCanceled, OrderDenied, OrderEmulated,
26 OrderExpired, OrderFilled, OrderInitialized, OrderModifyRejected, OrderPendingCancel,
27 OrderPendingUpdate, OrderRejected, OrderReleased, OrderSubmitted, OrderTriggered,
28 OrderUpdated,
29 },
30 identifiers::{AccountId, ClientOrderId, InstrumentId, StrategyId, TraderId, VenueOrderId},
31};
32
33#[allow(
35 clippy::large_enum_variant,
36 reason = "lint fires only with high-precision feature"
37)]
38#[derive(Clone, PartialEq, Eq, Debug, Serialize, Deserialize)]
39pub enum OrderEventAny {
40 Initialized(OrderInitialized),
41 Denied(OrderDenied),
42 Emulated(OrderEmulated),
43 Released(OrderReleased),
44 Submitted(OrderSubmitted),
45 Accepted(OrderAccepted),
46 Rejected(OrderRejected),
47 Canceled(OrderCanceled),
48 Expired(OrderExpired),
49 Triggered(OrderTriggered),
50 PendingUpdate(OrderPendingUpdate),
51 PendingCancel(OrderPendingCancel),
52 ModifyRejected(OrderModifyRejected),
53 CancelRejected(OrderCancelRejected),
54 Updated(OrderUpdated),
55 Filled(OrderFilled),
56}
57
58impl OrderEventAny {
59 #[must_use]
60 pub fn into_boxed(self) -> Box<dyn OrderEvent> {
61 match self {
62 Self::Initialized(event) => Box::new(event),
63 Self::Denied(event) => Box::new(event),
64 Self::Emulated(event) => Box::new(event),
65 Self::Released(event) => Box::new(event),
66 Self::Submitted(event) => Box::new(event),
67 Self::Accepted(event) => Box::new(event),
68 Self::Rejected(event) => Box::new(event),
69 Self::Canceled(event) => Box::new(event),
70 Self::Expired(event) => Box::new(event),
71 Self::Triggered(event) => Box::new(event),
72 Self::PendingUpdate(event) => Box::new(event),
73 Self::PendingCancel(event) => Box::new(event),
74 Self::ModifyRejected(event) => Box::new(event),
75 Self::CancelRejected(event) => Box::new(event),
76 Self::Updated(event) => Box::new(event),
77 Self::Filled(event) => Box::new(event),
78 }
79 }
80
81 #[must_use]
82 pub fn event_type(&self) -> OrderEventType {
83 match self {
84 Self::Initialized(_) => OrderEventType::Initialized,
85 Self::Denied(_) => OrderEventType::Denied,
86 Self::Emulated(_) => OrderEventType::Emulated,
87 Self::Released(_) => OrderEventType::Released,
88 Self::Submitted(_) => OrderEventType::Submitted,
89 Self::Accepted(_) => OrderEventType::Accepted,
90 Self::Rejected(_) => OrderEventType::Rejected,
91 Self::Canceled(_) => OrderEventType::Canceled,
92 Self::Expired(_) => OrderEventType::Expired,
93 Self::Triggered(_) => OrderEventType::Triggered,
94 Self::PendingUpdate(_) => OrderEventType::PendingUpdate,
95 Self::PendingCancel(_) => OrderEventType::PendingCancel,
96 Self::ModifyRejected(_) => OrderEventType::ModifyRejected,
97 Self::CancelRejected(_) => OrderEventType::CancelRejected,
98 Self::Updated(_) => OrderEventType::Updated,
99 Self::Filled(_) => OrderEventType::Filled,
100 }
101 }
102
103 #[must_use]
104 pub fn trader_id(&self) -> TraderId {
105 match self {
106 Self::Initialized(event) => event.trader_id,
107 Self::Denied(event) => event.trader_id,
108 Self::Emulated(event) => event.trader_id,
109 Self::Released(event) => event.trader_id,
110 Self::Submitted(event) => event.trader_id,
111 Self::Accepted(event) => event.trader_id,
112 Self::Rejected(event) => event.trader_id,
113 Self::Canceled(event) => event.trader_id,
114 Self::Expired(event) => event.trader_id,
115 Self::Triggered(event) => event.trader_id,
116 Self::PendingUpdate(event) => event.trader_id,
117 Self::PendingCancel(event) => event.trader_id,
118 Self::ModifyRejected(event) => event.trader_id,
119 Self::CancelRejected(event) => event.trader_id,
120 Self::Updated(event) => event.trader_id,
121 Self::Filled(event) => event.trader_id,
122 }
123 }
124
125 #[must_use]
126 pub fn client_order_id(&self) -> ClientOrderId {
127 match self {
128 Self::Initialized(event) => event.client_order_id,
129 Self::Denied(event) => event.client_order_id,
130 Self::Emulated(event) => event.client_order_id,
131 Self::Released(event) => event.client_order_id,
132 Self::Submitted(event) => event.client_order_id,
133 Self::Accepted(event) => event.client_order_id,
134 Self::Rejected(event) => event.client_order_id,
135 Self::Canceled(event) => event.client_order_id,
136 Self::Expired(event) => event.client_order_id,
137 Self::Triggered(event) => event.client_order_id,
138 Self::PendingUpdate(event) => event.client_order_id,
139 Self::PendingCancel(event) => event.client_order_id,
140 Self::ModifyRejected(event) => event.client_order_id,
141 Self::CancelRejected(event) => event.client_order_id,
142 Self::Updated(event) => event.client_order_id,
143 Self::Filled(event) => event.client_order_id,
144 }
145 }
146
147 #[must_use]
148 pub fn venue_order_id(&self) -> Option<VenueOrderId> {
149 match self {
150 Self::Initialized(event) => event.venue_order_id(),
151 Self::Denied(event) => event.venue_order_id(),
152 Self::Emulated(event) => event.venue_order_id(),
153 Self::Released(event) => event.venue_order_id(),
154 Self::Submitted(event) => event.venue_order_id(),
155 Self::Accepted(event) => event.venue_order_id(),
156 Self::Rejected(event) => event.venue_order_id(),
157 Self::Canceled(event) => event.venue_order_id(),
158 Self::Expired(event) => event.venue_order_id(),
159 Self::Triggered(event) => event.venue_order_id(),
160 Self::PendingUpdate(event) => event.venue_order_id(),
161 Self::PendingCancel(event) => event.venue_order_id(),
162 Self::ModifyRejected(event) => event.venue_order_id(),
163 Self::CancelRejected(event) => event.venue_order_id(),
164 Self::Updated(event) => event.venue_order_id(),
165 Self::Filled(event) => event.venue_order_id(),
166 }
167 }
168
169 #[must_use]
170 pub fn account_id(&self) -> Option<AccountId> {
171 match self {
172 Self::Initialized(event) => event.account_id(),
173 Self::Denied(event) => event.account_id(),
174 Self::Emulated(event) => event.account_id(),
175 Self::Released(event) => event.account_id(),
176 Self::Submitted(event) => event.account_id(),
177 Self::Accepted(event) => event.account_id(),
178 Self::Rejected(event) => event.account_id(),
179 Self::Canceled(event) => event.account_id(),
180 Self::Expired(event) => event.account_id(),
181 Self::Triggered(event) => event.account_id(),
182 Self::PendingUpdate(event) => event.account_id(),
183 Self::PendingCancel(event) => event.account_id(),
184 Self::ModifyRejected(event) => event.account_id(),
185 Self::CancelRejected(event) => event.account_id(),
186 Self::Updated(event) => event.account_id(),
187 Self::Filled(event) => event.account_id(),
188 }
189 }
190
191 #[must_use]
192 pub fn instrument_id(&self) -> InstrumentId {
193 match self {
194 Self::Initialized(event) => event.instrument_id(),
195 Self::Denied(event) => event.instrument_id(),
196 Self::Emulated(event) => event.instrument_id(),
197 Self::Released(event) => event.instrument_id(),
198 Self::Submitted(event) => event.instrument_id(),
199 Self::Accepted(event) => event.instrument_id(),
200 Self::Rejected(event) => event.instrument_id(),
201 Self::Canceled(event) => event.instrument_id(),
202 Self::Expired(event) => event.instrument_id(),
203 Self::Triggered(event) => event.instrument_id(),
204 Self::PendingUpdate(event) => event.instrument_id(),
205 Self::PendingCancel(event) => event.instrument_id(),
206 Self::ModifyRejected(event) => event.instrument_id(),
207 Self::CancelRejected(event) => event.instrument_id(),
208 Self::Updated(event) => event.instrument_id(),
209 Self::Filled(event) => event.instrument_id(),
210 }
211 }
212
213 #[must_use]
214 pub fn strategy_id(&self) -> StrategyId {
215 match self {
216 Self::Initialized(event) => event.strategy_id,
217 Self::Denied(event) => event.strategy_id,
218 Self::Emulated(event) => event.strategy_id,
219 Self::Released(event) => event.strategy_id,
220 Self::Submitted(event) => event.strategy_id,
221 Self::Accepted(event) => event.strategy_id,
222 Self::Rejected(event) => event.strategy_id,
223 Self::Canceled(event) => event.strategy_id,
224 Self::Expired(event) => event.strategy_id,
225 Self::Triggered(event) => event.strategy_id,
226 Self::PendingUpdate(event) => event.strategy_id,
227 Self::PendingCancel(event) => event.strategy_id,
228 Self::ModifyRejected(event) => event.strategy_id,
229 Self::CancelRejected(event) => event.strategy_id,
230 Self::Updated(event) => event.strategy_id,
231 Self::Filled(event) => event.strategy_id,
232 }
233 }
234
235 #[must_use]
236 pub fn ts_event(&self) -> UnixNanos {
237 match self {
238 Self::Initialized(event) => event.ts_event,
239 Self::Denied(event) => event.ts_event,
240 Self::Emulated(event) => event.ts_event,
241 Self::Released(event) => event.ts_event,
242 Self::Submitted(event) => event.ts_event,
243 Self::Accepted(event) => event.ts_event,
244 Self::Rejected(event) => event.ts_event,
245 Self::Canceled(event) => event.ts_event,
246 Self::Expired(event) => event.ts_event,
247 Self::Triggered(event) => event.ts_event,
248 Self::PendingUpdate(event) => event.ts_event,
249 Self::PendingCancel(event) => event.ts_event,
250 Self::ModifyRejected(event) => event.ts_event,
251 Self::CancelRejected(event) => event.ts_event,
252 Self::Updated(event) => event.ts_event,
253 Self::Filled(event) => event.ts_event,
254 }
255 }
256
257 #[must_use]
258 pub fn message(&self) -> Option<Ustr> {
259 match self {
260 Self::Initialized(_) => None,
261 Self::Denied(event) => Some(event.reason),
262 Self::Emulated(_) => None,
263 Self::Released(_) => None,
264 Self::Submitted(_) => None,
265 Self::Accepted(_) => None,
266 Self::Rejected(event) => Some(event.reason),
267 Self::Canceled(_) => None,
268 Self::Expired(_) => None,
269 Self::Triggered(_) => None,
270 Self::PendingUpdate(_) => None,
271 Self::PendingCancel(_) => None,
272 Self::ModifyRejected(event) => Some(event.reason),
273 Self::CancelRejected(event) => Some(event.reason),
274 Self::Updated(_) => None,
275 Self::Filled(_) => None,
276 }
277 }
278}
279
280impl From<OrderEventAny> for OrderFilled {
286 #[inline]
287 fn from(event: OrderEventAny) -> Self {
288 match event {
289 OrderEventAny::Filled(event) => event,
290 _ => panic!("Invalid `OrderEventAny` not `OrderFilled`, was {event:?}"),
291 }
292 }
293}
294
295impl Display for OrderEventAny {
296 fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
297 match self {
298 Self::Initialized(e) => write!(f, "{e}"),
299 Self::Denied(e) => write!(f, "{e}"),
300 Self::Emulated(e) => write!(f, "{e}"),
301 Self::Released(e) => write!(f, "{e}"),
302 Self::Submitted(e) => write!(f, "{e}"),
303 Self::Accepted(e) => write!(f, "{e}"),
304 Self::Rejected(e) => write!(f, "{e}"),
305 Self::Canceled(e) => write!(f, "{e}"),
306 Self::Expired(e) => write!(f, "{e}"),
307 Self::Triggered(e) => write!(f, "{e}"),
308 Self::PendingUpdate(e) => write!(f, "{e}"),
309 Self::PendingCancel(e) => write!(f, "{e}"),
310 Self::ModifyRejected(e) => write!(f, "{e}"),
311 Self::CancelRejected(e) => write!(f, "{e}"),
312 Self::Updated(e) => write!(f, "{e}"),
313 Self::Filled(e) => write!(f, "{e}"),
314 }
315 }
316}
317
318#[cfg(test)]
319mod tests {
320 use rstest::rstest;
321
322 use super::OrderEventAny;
323 use crate::events::{OrderAccepted, OrderFilled, order::stubs::*};
324
325 #[rstest]
326 fn test_from_order_event_any_to_filled(order_filled: OrderFilled) {
327 let expected_trade_id = order_filled.trade_id;
328 let event = OrderEventAny::Filled(order_filled);
329 let filled: OrderFilled = event.into();
330 assert_eq!(filled.trade_id, expected_trade_id);
331 }
332
333 #[rstest]
334 #[should_panic(expected = "Invalid `OrderEventAny` not `OrderFilled`")]
335 fn test_from_order_event_any_to_filled_panics_on_wrong_variant(order_accepted: OrderAccepted) {
336 let event = OrderEventAny::Accepted(order_accepted);
337 let _filled: OrderFilled = event.into();
338 }
339
340 #[rstest]
341 fn test_display_delegates_to_inner(order_filled: OrderFilled) {
342 let inner_display = format!("{order_filled}");
343 let event = OrderEventAny::Filled(order_filled);
344 assert_eq!(format!("{event}"), inner_display);
345 }
346}