rustybot/src/events.rs

@@ -15,14 +15,9 @@ pub struct ActorMessage {
 
 #[derive(Debug)]
 pub enum Message {
-    Update {
-        tick: u64,
-    },
-    ClosePosition {
-        position: Position,
-        order_form: OrderForm,
-    },
-    OpenPosition,
+    Update { tick: u64 },
+    ClosePosition { position_id: u64 },
+    SubmitOrder { order: OrderForm },
 }
 
 #[derive(Copy, Clone, Debug, Eq, PartialEq, Hash)]

rustybot/src/managers.rs

@@ -356,19 +356,12 @@ impl OrderManagerHandle {
         Ok(recv.await?)
     }
 
-    pub async fn close_position(
-        &mut self,
-        position: Position,
-        order_form: OrderForm,
-    ) -> Result<OptionUpdate, BoxError> {
+    pub async fn close_position(&mut self, position_id: u64) -> Result<OptionUpdate, BoxError> {
         let (send, recv) = oneshot::channel();
 
         self.sender
             .send(ActorMessage {
-                message: Message::ClosePosition {
-                    position,
-                    order_form,
-                },
+                message: Message::ClosePosition { position_id },
                 respond_to: send,
             })
             .await?;
@@ -409,9 +402,8 @@ impl OrderManager {
                 self.update().await?;
             }
             Message::ClosePosition {
-                position,
-                order_form,
-            } => self.close_position(&position, &order_form).await?,
+                position_id: position_id,
+            } => self.close_position(position_id).await?,
             _ => {}
         };
 
@@ -420,17 +412,14 @@ impl OrderManager {
         Ok(())
     }
 
-    pub async fn close_position(
-        &mut self,
-        position: &Position,
-        order_form: &OrderForm,
-    ) -> Result<(), BoxError> {
-        info!("Closing position #{}", position.id());
+    pub async fn close_position(&mut self, position_id: u64) -> Result<(), BoxError> {
+        info!("Closing position #{}", position_id);
 
-        debug!("Retrieving open orders and current prices...");
-        let (open_orders, order_book) = tokio::join!(
+        debug!("Retrieving open orders, positions and current prices...");
+        let (open_orders, order_book, open_positions) = tokio::join!(
             self.client.active_orders(&self.pair),
-            self.client.order_book(&self.pair)
+            self.client.order_book(&self.pair),
+            self.client.active_positions(&self.pair)
         );
 
         let open_orders = match open_orders {
@@ -449,6 +438,22 @@ impl OrderManager {
             }
         };
 
+        let position = match open_positions {
+            Ok(opt_positions) => {
+                let positions = opt_positions.ok_or::<BoxError>("No open positions".into())?;
+
+                positions
+                    .into_iter()
+                    .find(|x| x.id() == position_id)
+                    .ok_or::<BoxError>("Position #{} not found in open positions".into())?
+            }
+
+            Err(e) => {
+                error!("Could not retrieve open positions: {}", e);
+                return Err(e);
+            }
+        };
+
         let opt_position_order = open_orders
             .iter()
             .find(|x| x.current_form.amount().neg() == position.amount());
@@ -490,15 +495,15 @@ impl OrderManager {
 
                 let (_, strat_messages) =
                     self.strategy
-                        .on_position_close(active_order, position, &order_book)?;
+                        .on_position_close(active_order, &position, &order_book)?;
 
                 if let Some(messages) = strat_messages {
                     for m in messages {
                         match m {
-                            Message::ClosePosition { order_form, .. } => {
-                                info!("Closing open order with a {} order", order_form.kind());
+                            Message::SubmitOrder { order } => {
+                                info!("Closing open order with a {} order", order.kind());
 
-                                if let Ok(_) = self.client.submit_order(&order_form).await {
+                                if let Ok(_) = self.client.submit_order(&order).await {
                                     info!("Cancelling open order #{}", active_order.id);
                                     self.client.cancel_order(active_order).await?;
                                 }
@@ -591,13 +596,8 @@ impl PairManager {
         if let Some(messages) = opt_pos_messages {
             for m in messages {
                 match m {
-                    Message::ClosePosition {
-                        position,
-                        order_form,
-                    } => {
-                        self.order_manager
-                            .close_position(position, order_form)
-                            .await?;
+                    Message::ClosePosition { position_id } => {
+                        self.order_manager.close_position(position_id).await?;
                     }
                     _ => {}
                 }

rustybot/src/strategy.rs

@@ -112,15 +112,7 @@ impl PositionStrategy for TrailingStop {
             } else {
                 debug!("Inserting close position message...");
                 messages.push(Message::ClosePosition {
-                    position: position.clone(),
-                    order_form: OrderForm::new(
-                        position.pair().clone(),
-                        OrderKind::Limit {
-                            price: 0.0,
-                            amount: position.amount(),
-                        },
-                        TradingPlatform::Margin,
-                    ),
+                    position_id: position.id(),
                 });
                 PositionProfitState::Critical
             }
@@ -249,9 +241,8 @@ impl OrderStrategy for FastOrderStrategy {
         let delta = (1.0 - (offer_comparison / order_price)) * 100.0;
 
         if delta > self.threshold {
-            messages.push(Message::ClosePosition {
-                position: active_position.clone(),
-                order_form: OrderForm::new(
+            messages.push(Message::SubmitOrder {
+                order: OrderForm::new(
                     order.symbol.clone(),
                     OrderKind::Market {
                         amount: order.current_form.amount(),