0g-chain/x/pricefeed/genesis.go
2019-11-25 14:46:02 -05:00

52 lines
1.3 KiB
Go

package pricefeed
import (
sdk "github.com/cosmos/cosmos-sdk/types"
)
// InitGenesis sets distribution information for genesis.
func InitGenesis(ctx sdk.Context, keeper Keeper, data GenesisState) {
// Set the assets and oracles from params
keeper.SetAssetParams(ctx, data.AssetParams)
keeper.SetOracleParams(ctx ,data.OracleParams)
// Iterate through the posted prices and set them in the store
for _, pp := range data.PostedPrices {
addr, err := sdk.AccAddressFromBech32(pp.OracleAddress)
if err != nil {
panic(err)
}
_, err = keeper.SetPrice(ctx, addr, pp.AssetCode, pp.Price, pp.Expiry)
if err != nil {
panic(err)
}
}
// Set the current price (if any) based on what's now in the store
if err := keeper.SetCurrentPrices(ctx); err != nil {
panic(err)
}
}
// ExportGenesis returns a GenesisState for a given context and keeper.
func ExportGenesis(ctx sdk.Context, keeper Keeper) GenesisState {
// Get the params for assets and oracles
assetParams := keeper.GetAssetParams(ctx)
oracleParams := keeper.GetOracleParams(ctx)
var postedPrices []PostedPrice
for _, asset := range keeper.GetAssets(ctx) {
pp := keeper.GetRawPrices(ctx, asset.AssetCode)
postedPrices = append(postedPrices, pp...)
}
return GenesisState{
AssetParams: assetParams,
OracleParams: oracleParams,
PostedPrices: postedPrices,
}
}