program: flip auction flag when trigger order adds auction

drift-labs · Dec 18, 2023 · 8beb3f2 · 8beb3f2
1 parent 06664a9
commit 8beb3f2
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 2 deletions.
diff --git a/programs/drift/src/controller/orders.rs b/programs/drift/src/controller/orders.rs
@@ -2579,6 +2579,10 @@ pub fn trigger_order(
             state.min_perp_auction_duration,
         )?;
 
+        if user.orders[order_index].has_auction() {
+            user.increment_open_auctions();
+        }
+
         let direction = user.orders[order_index].direction;
         let base_asset_amount = user.orders[order_index].base_asset_amount;
 
@@ -4570,6 +4574,10 @@ pub fn trigger_spot_order(
             state.default_spot_auction_duration,
         )?;
 
+        if user.orders[order_index].has_auction() {
+            user.increment_open_auctions();
+        }
+
         let direction = user.orders[order_index].direction;
         let base_asset_amount = user.orders[order_index].base_asset_amount;
 

diff --git a/programs/drift/src/state/user.rs b/programs/drift/src/state/user.rs
@@ -355,11 +355,15 @@ impl User {
         self.open_orders = self.open_orders.saturating_add(1);
         self.has_open_order = self.open_orders > 0;
         if is_auction {
-            self.open_auctions = self.open_auctions.saturating_add(1);
-            self.has_open_auction = self.open_auctions > 0;
+            self.increment_open_auctions();
         }
     }
 
+    pub fn increment_open_auctions(&mut self) {
+        self.open_auctions = self.open_auctions.saturating_add(1);
+        self.has_open_auction = self.open_auctions > 0;
+    }
+
     pub fn decrement_open_orders(&mut self, is_auction: bool) {
         self.open_orders = self.open_orders.saturating_sub(1);
         self.has_open_order = self.open_orders > 0;