1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
use serde::{Deserialize, Serialize};

use crate::portfolio::{OrderEvent, OrderType};

/// Evaluates the risk associated with an [`OrderEvent`] to determine if it should be actioned. It
/// can also amend the order (eg/ [`OrderType`]) to better fit the risk strategy required for
/// profitability.
pub trait OrderEvaluator {
    const DEFAULT_ORDER_TYPE: OrderType;

    /// May return an amended [`OrderEvent`] if the associated risk is appropriate. Returns `None`
    /// if the risk is too high.
    fn evaluate_order(&self, order: OrderEvent) -> Option<OrderEvent>;
}

/// Default risk manager that implements [`OrderEvaluator`].
#[derive(Copy, Clone, Eq, PartialEq, Ord, PartialOrd, Hash, Debug, Deserialize, Serialize)]
pub struct DefaultRisk {}

impl OrderEvaluator for DefaultRisk {
    const DEFAULT_ORDER_TYPE: OrderType = OrderType::Market;

    fn evaluate_order(&self, mut order: OrderEvent) -> Option<OrderEvent> {
        if self.risk_too_high(&order) {
            return None;
        }
        order.order_type = DefaultRisk::DEFAULT_ORDER_TYPE;
        Some(order)
    }
}

impl DefaultRisk {
    fn risk_too_high(&self, _: &OrderEvent) -> bool {
        false
    }
}