fix balance sum

MinterTeam · Mar 25, 2021 · 5390ff1 · 5390ff1
1 parent 5666b8a
commit 5390ff1
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 8 deletions.
diff --git a/address/resource.go b/address/resource.go
@@ -18,9 +18,9 @@ type Resource struct {
 }
 
 type Params struct {
-	TotalBalanceSum    *big.Float
+	TotalBalanceSum    *big.Int
 	TotalBalanceSumUSD *big.Float
-	StakeBalanceSum    *big.Float
+	StakeBalanceSum    *big.Int
 	StakeBalanceSumUSD *big.Float
 }
 

diff --git a/balance/service.go b/balance/service.go
@@ -17,7 +17,7 @@ func NewService(baseCoin string, marketService *market.Service) *Service {
 	return &Service{baseCoin, marketService}
 }
 
-func (s *Service) GetTotalBalance(address *models.Address) *big.Float {
+func (s *Service) GetTotalBalance(address *models.Address) *big.Int {
 	sum := big.NewInt(0)
 	for _, balance := range address.Balances {
 		if balance.Coin.Crr == 0 {
@@ -39,10 +39,10 @@ func (s *Service) GetTotalBalance(address *models.Address) *big.Float {
 		))
 	}
 
-	return new(big.Float).SetInt(sum)
+	return sum
 }
 
-func (s *Service) GetStakeBalance(stakes []models.Stake) *big.Float {
+func (s *Service) GetStakeBalance(stakes []models.Stake) *big.Int {
 	sum := big.NewInt(0)
 
 	for _, stake := range stakes {
@@ -65,9 +65,9 @@ func (s *Service) GetStakeBalance(stakes []models.Stake) *big.Float {
 		))
 	}
 
-	return new(big.Float).SetInt(sum)
+	return sum
 }
 
-func (s *Service) GetTotalBalanceInUSD(sumInBasecoin *big.Float) *big.Float {
-	return new(big.Float).Mul(sumInBasecoin, big.NewFloat(s.marketService.PriceChange.Price))
+func (s *Service) GetTotalBalanceInUSD(sumInBasecoin *big.Int) *big.Float {
+	return new(big.Float).Mul(new(big.Float).SetInt(sumInBasecoin), big.NewFloat(s.marketService.PriceChange.Price))
 }