99 lines
2.5 KiB
Go
99 lines
2.5 KiB
Go
package types
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
|
|
"github.com/cosmos/cosmos-sdk/codec"
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
)
|
|
|
|
// Pool - dynamic parameters of the current state
|
|
type Pool struct {
|
|
LooseTokens sdk.Dec `json:"loose_tokens"` // tokens which are not bonded in a validator
|
|
BondedTokens sdk.Dec `json:"bonded_tokens"` // reserve of bonded tokens
|
|
}
|
|
|
|
// nolint
|
|
func (p Pool) Equal(p2 Pool) bool {
|
|
bz1 := MsgCdc.MustMarshalBinary(&p)
|
|
bz2 := MsgCdc.MustMarshalBinary(&p2)
|
|
return bytes.Equal(bz1, bz2)
|
|
}
|
|
|
|
// initial pool for testing
|
|
func InitialPool() Pool {
|
|
return Pool{
|
|
LooseTokens: sdk.ZeroDec(),
|
|
BondedTokens: sdk.ZeroDec(),
|
|
}
|
|
}
|
|
|
|
//____________________________________________________________________
|
|
|
|
// Sum total of all staking tokens in the pool
|
|
func (p Pool) TokenSupply() sdk.Dec {
|
|
return p.LooseTokens.Add(p.BondedTokens)
|
|
}
|
|
|
|
//____________________________________________________________________
|
|
|
|
// get the bond ratio of the global state
|
|
func (p Pool) BondedRatio() sdk.Dec {
|
|
supply := p.TokenSupply()
|
|
if supply.GT(sdk.ZeroDec()) {
|
|
return p.BondedTokens.Quo(supply)
|
|
}
|
|
return sdk.ZeroDec()
|
|
}
|
|
|
|
//_______________________________________________________________________
|
|
|
|
func (p Pool) looseTokensToBonded(bondedTokens sdk.Dec) Pool {
|
|
p.BondedTokens = p.BondedTokens.Add(bondedTokens)
|
|
p.LooseTokens = p.LooseTokens.Sub(bondedTokens)
|
|
if p.LooseTokens.LT(sdk.ZeroDec()) {
|
|
panic(fmt.Sprintf("sanity check: loose tokens negative, pool: %v", p))
|
|
}
|
|
return p
|
|
}
|
|
|
|
func (p Pool) bondedTokensToLoose(bondedTokens sdk.Dec) Pool {
|
|
p.BondedTokens = p.BondedTokens.Sub(bondedTokens)
|
|
p.LooseTokens = p.LooseTokens.Add(bondedTokens)
|
|
if p.BondedTokens.LT(sdk.ZeroDec()) {
|
|
panic(fmt.Sprintf("sanity check: bonded tokens negative, pool: %v", p))
|
|
}
|
|
return p
|
|
}
|
|
|
|
// HumanReadableString returns a human readable string representation of a
|
|
// pool.
|
|
func (p Pool) HumanReadableString() string {
|
|
|
|
resp := "Pool \n"
|
|
resp += fmt.Sprintf("Loose Tokens: %s\n", p.LooseTokens)
|
|
resp += fmt.Sprintf("Bonded Tokens: %s\n", p.BondedTokens)
|
|
resp += fmt.Sprintf("Token Supply: %s\n", p.TokenSupply())
|
|
resp += fmt.Sprintf("Bonded Ratio: %v\n", p.BondedRatio())
|
|
return resp
|
|
}
|
|
|
|
// unmarshal the current pool value from store key or panics
|
|
func MustUnmarshalPool(cdc *codec.Codec, value []byte) Pool {
|
|
pool, err := UnmarshalPool(cdc, value)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return pool
|
|
}
|
|
|
|
// unmarshal the current pool value from store key
|
|
func UnmarshalPool(cdc *codec.Codec, value []byte) (pool Pool, err error) {
|
|
err = cdc.UnmarshalBinary(value, &pool)
|
|
if err != nil {
|
|
return
|
|
}
|
|
return
|
|
}
|