fix

waves-exchange · Aug 16, 2023 · abd2eb9 · abd2eb9
1 parent 602d3ef
commit abd2eb9
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 2 deletions.
diff --git a/ride/lp.ride b/ride/lp.ride
@@ -1021,7 +1021,17 @@ func calculateAmountOutForSwapAndSendTokens(cleanAmountIn: Int, isReverse: Boole
   strict checkK = newK >= oldK || "new K is fewer error".throw()
   strict checkMin = amountOutMin <= amountOut || "Exchange result is fewer coins than expected".throw()
 
-  strict refreshKLpActions = refreshKLpInternal(0, 0, 0)._1
+  let amountAssetBalanceDelta = if (isReverse) then {
+    -amountOut
+  } else {
+    feePoolAmount
+  }
+  let priceAssetBalanceDelta = if (isReverse) then {
+    feePoolAmount
+  } else {
+    -amountOut
+  }
+  strict refreshKLpActions = refreshKLpInternal(amountAssetBalanceDelta, priceAssetBalanceDelta, 0)._1
 
   strict rebalanceState = rebalanceAsset(assetIn)
   strict withdrawState = withdrawAndRebalanceAsset(assetOut, amountOut)

diff --git a/ride/lp_stable.ride b/ride/lp_stable.ride
@@ -1062,7 +1062,17 @@ func calculateAmountOutForSwapAndSendTokens(cleanAmountIn: Int, isReverse: Boole
   let newD = newXp.getD()
   strict checkD = newD >= D || "new D is fewer error".throw()
 
-  strict refreshDLpActions = refreshDLpInternal(0, 0, 0)._1
+  let amountAssetBalanceDelta = if (isReverse) then {
+    -totalGetRaw
+  } else {
+    feePoolAmount
+  }
+  let priceAssetBalanceDelta = if (isReverse) then {
+    feePoolAmount
+  } else {
+    -totalGetRaw
+  }
+  strict refreshDLpActions = refreshDLpInternal(amountAssetBalanceDelta, priceAssetBalanceDelta, 0)._1
 
   (
     [