StakingTokenSupply no longer includes Dao Treasury

onomyprotocol · Aug 22, 2023 · 13ddd90 · 13ddd90
1 parent 6a3986f
commit 13ddd90
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/x/staking/keeper/pool.go b/x/staking/keeper/pool.go
@@ -64,7 +64,14 @@ func (k Keeper) TotalBondedTokens(ctx sdk.Context) sdk.Int {
 
 // StakingTokenSupply staking tokens from the total supply
 func (k Keeper) StakingTokenSupply(ctx sdk.Context) sdk.Int {
-	return k.bankKeeper.GetSupply(ctx, k.BondDenom(ctx)).Amount
+	stakeSupply := k.bankKeeper.GetSupply(ctx, k.BondDenom(ctx)).Amount
+	daoAddr := k.authKeeper.GetModuleAddress("dao")
+	if daoAddr != nil {
+		daoSupply := k.bankKeeper.GetBalance(ctx, daoAddr, k.BondDenom(ctx))
+		stakeSupply = stakeSupply.Sub(daoSupply.Amount)
+	}
+
+	return stakeSupply
 }
 
 // BondedRatio the fraction of the staking tokens which are currently bonded