0g-chain/x/auction/keeper/store.go

149 lines
4.8 KiB
Go
Raw Normal View History

2019-12-12 00:16:10 +00:00
package keeper
import (
"time"
2019-12-21 01:04:04 +00:00
"github.com/cosmos/cosmos-sdk/store/prefix"
2019-12-12 00:16:10 +00:00
sdk "github.com/cosmos/cosmos-sdk/types"
"github.com/kava-labs/kava/x/auction/types"
)
2019-12-21 01:04:04 +00:00
// SetNextAuctionID stores an ID to be used for the next created auction
func (k Keeper) SetNextAuctionID(ctx sdk.Context, id types.ID) {
store := ctx.KVStore(k.storeKey)
store.Set(types.NextAuctionIDKey, id.Bytes())
2019-12-12 00:16:10 +00:00
}
2019-12-21 01:04:04 +00:00
// GetNextAuctionID reads the next available global ID from store
func (k Keeper) GetNextAuctionID(ctx sdk.Context) (types.ID, sdk.Error) {
2019-12-12 00:16:10 +00:00
store := ctx.KVStore(k.storeKey)
2019-12-21 01:04:04 +00:00
bz := store.Get(types.NextAuctionIDKey)
2019-12-12 00:16:10 +00:00
if bz == nil {
2019-12-21 01:04:04 +00:00
//return 0, types.ErrInvalidGenesis(k.codespace, "initial auction ID hasn't been set") // TODO create error
return 0, sdk.ErrInternal("initial auction ID hasn't been set")
2019-12-12 00:16:10 +00:00
}
2019-12-21 01:04:04 +00:00
return types.NewIDFromBytes(bz), nil
2019-12-12 00:16:10 +00:00
}
// incrementNextAuctionID increments the global ID in the store by 1
2019-12-21 01:04:04 +00:00
func (k Keeper) IncrementNextAuctionID(ctx sdk.Context) sdk.Error {
id, err := k.GetNextAuctionID(ctx)
if err != nil {
return err
2019-12-12 00:16:10 +00:00
}
2019-12-21 01:04:04 +00:00
k.SetNextAuctionID(ctx, id+1)
return nil
}
// storeNewAuction stores an auction, adding a new ID, and setting indexes
func (k Keeper) storeNewAuction(ctx sdk.Context, auction types.Auction) (types.ID, sdk.Error) {
newAuctionID, err := k.GetNextAuctionID(ctx)
if err != nil {
return 0, err
}
auction.SetID(newAuctionID)
2019-12-12 00:16:10 +00:00
2019-12-21 01:04:04 +00:00
k.SetAuction(ctx, auction)
2019-12-12 00:16:10 +00:00
2019-12-21 01:04:04 +00:00
err = k.IncrementNextAuctionID(ctx)
if err != nil {
return 0, err
}
return newAuctionID, nil
2019-12-12 00:16:10 +00:00
}
2019-12-21 01:04:04 +00:00
// TODO should get/set/delete be responsible for updating auctionByTime index?
2019-12-12 00:16:10 +00:00
// SetAuction puts the auction into the database and adds it to the queue
// it overwrites any pre-existing auction with same ID
func (k Keeper) SetAuction(ctx sdk.Context, auction types.Auction) {
// remove the auction from the queue if it is already in there
2019-12-21 01:04:04 +00:00
// existingAuction, found := k.GetAuction(ctx, auction.GetID())
// if found {
// k.removeFromQueue(ctx, existingAuction.GetEndTime(), existingAuction.GetID())
// }
2019-12-12 00:16:10 +00:00
// store auction
2019-12-21 01:04:04 +00:00
store := prefix.NewStore(ctx.KVStore(k.storeKey), types.AuctionKeyPrefix)
2019-12-12 00:16:10 +00:00
bz := k.cdc.MustMarshalBinaryLengthPrefixed(auction)
2019-12-21 01:04:04 +00:00
store.Set(types.GetAuctionKey(auction.GetID()), bz)
2019-12-12 00:16:10 +00:00
// add to the queue
2019-12-21 01:04:04 +00:00
//k.InsertIntoQueue(ctx, auction.GetEndTime(), auction.GetID())
2019-12-12 00:16:10 +00:00
}
// getAuction gets an auction from the store by auctionID
func (k Keeper) GetAuction(ctx sdk.Context, auctionID types.ID) (types.Auction, bool) {
var auction types.Auction
2019-12-21 01:04:04 +00:00
store := prefix.NewStore(ctx.KVStore(k.storeKey), types.AuctionKeyPrefix)
bz := store.Get(types.GetAuctionKey(auctionID))
2019-12-12 00:16:10 +00:00
if bz == nil {
return auction, false
}
k.cdc.MustUnmarshalBinaryLengthPrefixed(bz, &auction)
return auction, true
}
// DeleteAuction removes an auction from the store without any validation
func (k Keeper) DeleteAuction(ctx sdk.Context, auctionID types.ID) {
// remove from queue
2019-12-21 01:04:04 +00:00
//auction, found := k.GetAuction(ctx, auctionID)
// if found {
// k.removeFromQueue(ctx, auction.GetEndTime(), auctionID)
// }
2019-12-12 00:16:10 +00:00
// delete auction
2019-12-21 01:04:04 +00:00
store := prefix.NewStore(ctx.KVStore(k.storeKey), types.AuctionKeyPrefix)
store.Delete(types.GetAuctionKey(auctionID))
2019-12-12 00:16:10 +00:00
}
// Inserts a AuctionID into the queue at endTime
func (k Keeper) InsertIntoQueue(ctx sdk.Context, endTime time.Time, auctionID types.ID) {
2019-12-21 01:04:04 +00:00
store := prefix.NewStore(ctx.KVStore(k.storeKey), types.AuctionByTimeKeyPrefix)
store.Set(types.GetAuctionByTimeKey(endTime, auctionID), auctionID.Bytes())
2019-12-12 00:16:10 +00:00
}
// removes an auctionID from the queue
2019-12-21 01:04:04 +00:00
func (k Keeper) RemoveFromQueue(ctx sdk.Context, endTime time.Time, auctionID types.ID) {
store := prefix.NewStore(ctx.KVStore(k.storeKey), types.AuctionByTimeKeyPrefix)
store.Delete(types.GetAuctionByTimeKey(endTime, auctionID))
2019-12-12 00:16:10 +00:00
}
2019-12-21 01:04:04 +00:00
func (k Keeper) IterateAuctionsByTime(ctx sdk.Context, inclusiveCutoffTime time.Time, cb func(auctionID types.ID) (stop bool)) {
store := prefix.NewStore(ctx.KVStore(k.storeKey), types.AuctionByTimeKeyPrefix)
iterator := store.Iterator(
nil, // start at the very start of the prefix store
sdk.PrefixEndBytes(sdk.FormatTimeBytes(inclusiveCutoffTime)), // include any keys with times equal to inclusiveCutoffTime
2019-12-12 00:16:10 +00:00
)
2019-12-21 01:04:04 +00:00
defer iterator.Close()
for ; iterator.Valid(); iterator.Next() {
// TODO get the auction ID - either read from store, or extract from key
auctionID := types.NewIDFromBytes(iterator.Value())
2019-12-12 00:16:10 +00:00
2019-12-21 01:04:04 +00:00
if cb(auctionID) {
break
}
}
2019-12-12 00:16:10 +00:00
}
2019-12-21 01:04:04 +00:00
// IterateAuctions provides an iterator over all stored auctions. For
// each auction, cb will be called. If the cb returns true, the iterator
// will close and stop.
func (k Keeper) IterateAuctions(ctx sdk.Context, cb func(auction types.Auction) (stop bool)) {
iterator := sdk.KVStorePrefixIterator(ctx.KVStore(k.storeKey), types.AuctionKeyPrefix)
2019-12-12 00:16:10 +00:00
2019-12-21 01:04:04 +00:00
defer iterator.Close()
for ; iterator.Valid(); iterator.Next() {
var auction types.Auction
k.cdc.MustUnmarshalBinaryLengthPrefixed(iterator.Value(), &auction)
2019-12-12 00:16:10 +00:00
2019-12-21 01:04:04 +00:00
if cb(auction) {
break
}
}
2019-12-12 00:16:10 +00:00
}