fix: prevent unnecessary calculation

p2p-org · Jan 24, 2024 · 4cf5ff7 · 4cf5ff7
1 parent 69d8871
commit 4cf5ff7
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/Packages/KeyAppKit/Sources/Send/Input/SendInputState.swift b/Packages/KeyAppKit/Sources/Send/Input/SendInputState.swift
@@ -235,6 +235,11 @@ public struct SendInputState: Equatable {
 
 public extension SendInputState {
     var totalAmount: UInt64 {
+        // assertion to prevent unneccessary calculation that can lead to slightly different results
+        if isSendingMaxAmount {
+            return token.lamports
+        }
+
         var totalAmount: UInt64 = 0
 
         // get receivingAmount