close all positions return payload

dydxprotocol · Sep 20, 2024 · c68efa0 · c68efa0
1 parent 0871947
commit c68efa0
Show file tree

Hide file tree

Showing 5 changed files with 10 additions and 9 deletions.
diff --git a/src/commonMain/kotlin/exchange.dydx.abacus/state/manager/AsyncAbacusStateManagerProtocol.kt b/src/commonMain/kotlin/exchange.dydx.abacus/state/manager/AsyncAbacusStateManagerProtocol.kt
@@ -88,7 +88,7 @@ interface AsyncAbacusStateManagerProtocol {
     fun faucet(amount: Double, callback: TransactionCallback)
     fun cancelOrder(orderId: String, callback: TransactionCallback)
     fun cancelAllOrders(marketId: String?, callback: TransactionCallback)
-    fun closeAllPositions(callback: TransactionCallback)
+    fun closeAllPositions(callback: TransactionCallback): HumanReadableCloseAllPositionsPayload?
 
     // Bridge functions.
     // If client is not using cancelOrder function, it should call orderCanceled function with

diff --git a/src/commonMain/kotlin/exchange.dydx.abacus/state/v2/manager/AsyncAbacusStateManagerV2.kt b/src/commonMain/kotlin/exchange.dydx.abacus/state/v2/manager/AsyncAbacusStateManagerV2.kt
@@ -630,12 +630,13 @@ class AsyncAbacusStateManagerV2(
         }
     }
 
-    override fun closeAllPositions(callback: TransactionCallback) {
-        try {
+    override fun closeAllPositions(callback: TransactionCallback): HumanReadableCloseAllPositionsPayload? {
+        return try {
             adaptor?.closeAllPositions(callback)
         } catch (e: Exception) {
             val error = V4TransactionErrors.error(null, e.toString())
             callback(false, error, null)
+            null
         }
     }
 

diff --git a/src/commonMain/kotlin/exchange.dydx.abacus/state/v2/manager/StateManagerAdaptorV2.kt b/src/commonMain/kotlin/exchange.dydx.abacus/state/v2/manager/StateManagerAdaptorV2.kt
@@ -615,8 +615,8 @@ internal class StateManagerAdaptorV2(
         accounts.cancelAllOrders(marketId, callback)
     }
 
-    internal fun closeAllPositions(callback: TransactionCallback) {
-        accounts.closeAllPositions(currentHeight, callback)
+    internal fun closeAllPositions(callback: TransactionCallback): HumanReadableCloseAllPositionsPayload? {
+        return accounts.closeAllPositions(currentHeight, callback)
     }
 
     internal fun orderCanceled(orderId: String) {

diff --git a/src/commonMain/kotlin/exchange.dydx.abacus/state/v2/supervisor/AccountSupervisor.kt b/src/commonMain/kotlin/exchange.dydx.abacus/state/v2/supervisor/AccountSupervisor.kt
@@ -1214,8 +1214,8 @@ internal fun AccountSupervisor.cancelAllOrders(marketId: String?, callback: Tran
     subaccount?.cancelAllOrders(marketId, callback)
 }
 
-internal fun AccountSupervisor.closeAllPositions(currentHeight: Int?, callback: TransactionCallback) {
-    subaccount?.closeAllPositions(currentHeight, callback)
+internal fun AccountSupervisor.closeAllPositions(currentHeight: Int?, callback: TransactionCallback): HumanReadableCloseAllPositionsPayload? {
+    return subaccount?.closeAllPositions(currentHeight, callback)
 }
 
 internal fun AccountSupervisor.orderCanceled(orderId: String) {

diff --git a/src/commonMain/kotlin/exchange.dydx.abacus/state/v2/supervisor/AccountsSupervisor.kt b/src/commonMain/kotlin/exchange.dydx.abacus/state/v2/supervisor/AccountsSupervisor.kt
@@ -366,8 +366,8 @@ internal fun AccountsSupervisor.cancelAllOrders(marketId: String?, callback: Tra
     account?.cancelAllOrders(marketId, callback)
 }
 
-internal fun AccountsSupervisor.closeAllPositions(currentHeight: Int?, callback: TransactionCallback) {
-    account?.closeAllPositions(currentHeight, callback)
+internal fun AccountsSupervisor.closeAllPositions(currentHeight: Int?, callback: TransactionCallback): HumanReadableCloseAllPositionsPayload? {
+    return account?.closeAllPositions(currentHeight, callback)
 }
 
 internal fun AccountsSupervisor.orderCanceled(orderId: String) {