From dc528b3f9c8262cbaa1a6f91b51775a9fa66519a Mon Sep 17 00:00:00 2001
From: Sushant Chandla <sushantchandla@gmail.com>
Date: Wed, 16 Oct 2024 20:32:31 +0530
Subject: [PATCH] fix account needed

---
 lib/dex-gamma/src/edge.rs           | 2 +-
 lib/router-lib/src/dex/interface.rs | 3 +++
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/lib/dex-gamma/src/edge.rs b/lib/dex-gamma/src/edge.rs
index ed4e3b2..cafe6a5 100644
--- a/lib/dex-gamma/src/edge.rs
+++ b/lib/dex-gamma/src/edge.rs
@@ -42,7 +42,7 @@ impl DexEdgeIdentifier for GammaEdgeIdentifier {
     }
 
     fn accounts_needed(&self) -> usize {
-        13
+        11
     }
 
     fn as_any(&self) -> &dyn Any {
diff --git a/lib/router-lib/src/dex/interface.rs b/lib/router-lib/src/dex/interface.rs
index d79a560..a59fefe 100644
--- a/lib/router-lib/src/dex/interface.rs
+++ b/lib/router-lib/src/dex/interface.rs
@@ -117,6 +117,9 @@ pub trait DexEdgeIdentifier: Sync + Send {
     fn desc(&self) -> String;
     fn input_mint(&self) -> Pubkey;
     fn output_mint(&self) -> Pubkey;
+    /// This should count all account needed to make a swap, excluding:
+    /// - user wallet address
+    /// - user output ATA
     fn accounts_needed(&self) -> usize;
     fn as_any(&self) -> &dyn Any;
 }