genesis.go

package keeper

import (
	sdk "github.com/cosmos/cosmos-sdk/types"
	"github.com/umee-network/umee/v4/x/oracle/types"
)

// IterateAllHistoricPrices iterates over all historic prices.
// Iterator stops when exhausting the source, or when the handler returns `true`.
func (k Keeper) IterateAllHistoricPrices(
	ctx sdk.Context,
	handler func(types.Price) bool,
) {
	store := ctx.KVStore(k.storeKey)
	iter := sdk.KVStorePrefixIterator(store, types.KeyPrefixHistoricPrice)
	defer iter.Close()

	for ; iter.Valid(); iter.Next() {
		var decProto sdk.DecProto
		k.cdc.MustUnmarshal(iter.Value(), &decProto)
		denom, blockNum := types.ParseDenomAndBlockFromKey(iter.Key(), types.KeyPrefixHistoricPrice)
		historicPrice := types.Price{
			ExchangeRateTuple: types.ExchangeRateTuple{ExchangeRate: decProto.Dec, Denom: denom},
			BlockNum:          blockNum,
		}
		if handler(historicPrice) {
			break
		}
	}
}

// AllHistoricPrices is a helper function that collects and returns all
// median prices using the IterateAllHistoricPrices iterator
func (k Keeper) AllHistoricPrices(ctx sdk.Context) types.Prices {
	prices := types.Prices{}
	k.IterateAllHistoricPrices(ctx, func(median types.Price) (stop bool) {
		prices = append(prices, median)
		return false
	})
	return prices
}

// IterateAllMedianPrices iterates over all median prices.
// Iterator stops when exhausting the source, or when the handler returns `true`.
func (k Keeper) IterateAllMedianPrices(
	ctx sdk.Context,
	handler func(types.Price) bool,
) {
	store := ctx.KVStore(k.storeKey)
	iter := sdk.KVStorePrefixIterator(store, types.KeyPrefixMedian)
	defer iter.Close()

	for ; iter.Valid(); iter.Next() {
		var decProto sdk.DecProto
		k.cdc.MustUnmarshal(iter.Value(), &decProto)
		denom, blockNum := types.ParseDenomAndBlockFromKey(iter.Key(), types.KeyPrefixMedian)
		median := types.Price{
			ExchangeRateTuple: types.ExchangeRateTuple{ExchangeRate: decProto.Dec, Denom: denom},
			BlockNum:          blockNum,
		}

		if handler(median) {
			break
		}
	}
}

// AllMedianPrices is a helper function that collects and returns all
// median prices using the IterateAllMedianPrices iterator
func (k Keeper) AllMedianPrices(ctx sdk.Context) types.Prices {
	prices := types.Prices{}
	k.IterateAllMedianPrices(ctx, func(median types.Price) (stop bool) {
		prices = append(prices, median)
		return false
	})
	return prices
}

// IterateAllMedianDeviationPrices iterates over all median deviation prices.
// Iterator stops when exhausting the source, or when the handler returns `true`.
func (k Keeper) IterateAllMedianDeviationPrices(
	ctx sdk.Context,
	handler func(types.Price) bool,
) {
	store := ctx.KVStore(k.storeKey)
	iter := sdk.KVStorePrefixIterator(store, types.KeyPrefixMedianDeviation)
	defer iter.Close()

	for ; iter.Valid(); iter.Next() {
		var decProto sdk.DecProto
		k.cdc.MustUnmarshal(iter.Value(), &decProto)
		denom, blockNum := types.ParseDenomAndBlockFromKey(iter.Key(), types.KeyPrefixMedianDeviation)
		medianDeviation := types.Price{
			ExchangeRateTuple: types.ExchangeRateTuple{ExchangeRate: decProto.Dec, Denom: denom},
			BlockNum:          blockNum,
		}

		if handler(medianDeviation) {
			break
		}
	}
}

// AllMedianDeviationPrices is a helper function that collects and returns
// all median prices using the IterateAllMedianDeviationPrices iterator
func (k Keeper) AllMedianDeviationPrices(ctx sdk.Context) types.Prices {
	prices := types.Prices{}
	k.IterateAllMedianDeviationPrices(ctx, func(median types.Price) (stop bool) {
		prices = append(prices, median)
		return false
	})
	return prices
}