exposure_participation.go

package keeper

import (
	sdk "github.com/cosmos/cosmos-sdk/types"

	"github.com/furya-network/furya/x/strategicreserve/types"
)

// SetParticipationExposure sets a participation exposure.
func (k Keeper) SetParticipationExposure(ctx sdk.Context, pe types.ParticipationExposure) {
	peKey := types.GetParticipationExposureKey(pe.BookUID, pe.OddsUID, pe.ParticipationIndex)

	store := k.getParticipationExposureStore(ctx)
	b := k.cdc.MustMarshal(&pe)
	store.Set(peKey, b)

	k.SetParticipationExposureByIndex(ctx, pe)
}

// removeParticipationExposure removes a participation exposure.
func (k Keeper) removeParticipationExposure(ctx sdk.Context, pe types.ParticipationExposure) {
	store := k.getParticipationExposureStore(ctx)
	store.Delete(types.GetParticipationExposureKey(pe.BookUID, pe.OddsUID, pe.ParticipationIndex))
}

// GetExposureByBookAndOdds returns all exposures for a book id and odds id
func (k Keeper) GetExposureByBookAndOdds(ctx sdk.Context, bookUID, oddsUID string) (list []types.ParticipationExposure, err error) {
	store := k.getParticipationExposureStore(ctx)
	iterator := sdk.KVStorePrefixIterator(store, types.GetParticipationExposuresKey(bookUID, oddsUID))

	defer func() {
		err = iterator.Close()
	}()

	for ; iterator.Valid(); iterator.Next() {
		var val types.ParticipationExposure
		k.cdc.MustUnmarshal(iterator.Value(), &val)
		list = append(list, val)
	}

	return
}

// GetAllParticipationExposures returns all participation exposures used during genesis dump.
func (k Keeper) GetAllParticipationExposures(ctx sdk.Context) (list []types.ParticipationExposure, err error) {
	store := k.getParticipationExposureStore(ctx)
	iterator := sdk.KVStorePrefixIterator(store, []byte{})

	defer func() {
		err = iterator.Close()
	}()

	for ; iterator.Valid(); iterator.Next() {
		var val types.ParticipationExposure
		k.cdc.MustUnmarshal(iterator.Value(), &val)
		list = append(list, val)
	}

	return
}

// SetParticipationExposureByIndex sets a participation exposure by index.
func (k Keeper) SetParticipationExposureByIndex(ctx sdk.Context, pe types.ParticipationExposure) {
	peKey := types.GetParticipationExposureByIndexKey(pe.BookUID, pe.OddsUID, pe.ParticipationIndex)

	store := k.getParticipationExposureByIndexStore(ctx)
	b := k.cdc.MustMarshal(&pe)
	store.Set(peKey, b)
}

// GetExposureByBookAndParticipationIndex returns all exposures for a book id and participation index
func (k Keeper) GetExposureByBookAndParticipationIndex(ctx sdk.Context, bookUID string, participationIndex uint64) (list []types.ParticipationExposure, err error) {
	store := k.getParticipationExposureByIndexStore(ctx)
	iterator := sdk.KVStorePrefixIterator(store, types.GetParticipationByIndexKey(bookUID, participationIndex))

	defer func() {
		err = iterator.Close()
	}()

	for ; iterator.Valid(); iterator.Next() {
		var val types.ParticipationExposure
		k.cdc.MustUnmarshal(iterator.Value(), &val)
		list = append(list, val)
	}

	return
}

// removeParticipationExposureByIndex removes a participation exposure by index.
func (k Keeper) removeParticipationExposureByIndex(ctx sdk.Context, pe types.ParticipationExposure) {
	store := k.getParticipationExposureByIndexStore(ctx)
	store.Delete(types.GetParticipationExposureByIndexKey(pe.BookUID, pe.OddsUID, pe.ParticipationIndex))
}

// SetHistoricalParticipationExposure sets a historical participation exposure.
func (k Keeper) SetHistoricalParticipationExposure(ctx sdk.Context, pe types.ParticipationExposure) {
	peKey := types.GetHistoricalParticipationExposureKey(pe.BookUID, pe.OddsUID, pe.ParticipationIndex, pe.Round)

	store := k.getHistoricalParticipationExposureStore(ctx)
	b := k.cdc.MustMarshal(&pe)
	store.Set(peKey, b)
}

// GetAllHistoricalParticipationExposures returns all participation exposures used during genesis dump.
func (k Keeper) GetAllHistoricalParticipationExposures(ctx sdk.Context) (list []types.ParticipationExposure, err error) {
	store := k.getHistoricalParticipationExposureStore(ctx)
	iterator := sdk.KVStorePrefixIterator(store, []byte{})

	defer func() {
		err = iterator.Close()
	}()

	for ; iterator.Valid(); iterator.Next() {
		var val types.ParticipationExposure
		k.cdc.MustUnmarshal(iterator.Value(), &val)
		list = append(list, val)
	}

	return
}

// MoveToHistoricalParticipationExposure removes the participation exposures and indices
// and sets historical participation exposures.
func (k Keeper) MoveToHistoricalParticipationExposure(ctx sdk.Context, pe types.ParticipationExposure) {
	k.SetHistoricalParticipationExposure(ctx, pe)
	k.removeParticipationExposure(ctx, pe)
	k.removeParticipationExposureByIndex(ctx, pe)
}