[LeverageLp]: add position usd value (#764)

* feat: add position usd value * add test --------- Co-authored-by: Cosmic Vagabond <121588426+cosmic-vagabond@users.noreply.github.com>
elys-network · Sep 3, 2024 · 07ad6ea · 07ad6ea
1 parent 6333cac
commit 07ad6ea
Show file tree

Hide file tree

Showing 4 changed files with 98 additions and 44 deletions.
diff --git a/proto/elys/leveragelp/types.proto b/proto/elys/leveragelp/types.proto
@@ -73,6 +73,10 @@ message LegacyPosition {
 message QueryPosition {
   Position position        = 1;
   string   updated_leverage = 2 [(gogoproto.customtype) = "github.com/cosmos/cosmos-sdk/types.Dec", (gogoproto.nullable) = false];
+  string position_usd_value = 3 [
+    (gogoproto.customtype) = "github.com/cosmos/cosmos-sdk/types.Int",
+    (gogoproto.nullable) = false
+  ];
 }
 
 message PositionAndInterest {

diff --git a/x/leveragelp/keeper/query_get_position_test.go b/x/leveragelp/keeper/query_get_position_test.go
@@ -101,17 +101,18 @@ func (suite KeeperTestSuite) TestQueryGetPosition() {
 	suite.Require().Equal(position, res.Position.Position)
 	suite.Require().Equal(updated_leverage, res.Position.UpdatedLeverage)
 
-
 	expected := types.PositionAndInterest{
 		Position: &types.QueryPosition{
 			Position: position,
 			UpdatedLeverage: updated_leverage,
+			PositionUsdValue: sdk.NewInt(5000),
 		},
 		InterestRateHour: sdk.MustNewDecFromStr("0.000017123287671233"),
 		InterestRateHourUsd: sdk.ZeroDec(),
 	}
 	pos_for_address_res, _ := k.QueryPositionsForAddress(suite.ctx, &types.PositionsForAddressRequest{Address: addr.String(), Pagination: nil} )
 
 	suite.Require().Equal(expected.Position, pos_for_address_res.Positions[0].Position)
-	suite.Require().Equal(expected.InterestRateHour, pos_for_address_res.Positions[0].InterestRateHour)
+	suite.Require().True(expected.InterestRateHour.Equal(pos_for_address_res.Positions[0].InterestRateHour))
+	suite.Require().True(expected.InterestRateHourUsd.Equal(pos_for_address_res.Positions[0].InterestRateHourUsd))
 }
diff --git a/x/leveragelp/keeper/utils.go b/x/leveragelp/keeper/utils.go
@@ -92,6 +92,7 @@ func (k Keeper) GetLeverageLpUpdatedLeverage(ctx sdk.Context, positions []*types
 		updatedLeveragePositions = append(updatedLeveragePositions, &types.QueryPosition{
 			Position:        position,
 			UpdatedLeverage: updated_leverage,
+			PositionUsdValue: exitAmountAfterFee,
 		})
 	}
 	return updatedLeveragePositions, nil

diff --git a/x/leveragelp/types/types.pb.go b/x/leveragelp/types/types.pb.go