planetmint-go/x/dao/keeper/mint_request.go
Lorenz Herzberger 0fe015a308
add MintRequestInvariants
Signed-off-by: Lorenz Herzberger <lorenzherzberger@gmail.com>
2023-10-10 12:11:03 +02:00

63 lines
2.1 KiB
Go

package keeper
import (
"github.com/cosmos/cosmos-sdk/store/prefix"
sdk "github.com/cosmos/cosmos-sdk/types"
"github.com/planetmint/planetmint-go/x/dao/types"
)
func (k Keeper) StoreMintRequest(ctx sdk.Context, mintRequest types.MintRequest) {
hashStore := prefix.NewStore(ctx.KVStore(k.storeKey), types.KeyPrefix(types.MintRequestHashKey))
hashAppendValue := k.cdc.MustMarshal(&mintRequest)
hashStore.Set(getMintRequestKeyBytes(mintRequest.LiquidTxHash), hashAppendValue)
addressStore := prefix.NewStore(ctx.KVStore(k.storeKey), types.KeyPrefix(types.MintRequestAddressKey))
mintRequests, _ := k.GetMintRequestsByAddress(ctx, mintRequest.Beneficiary)
mintRequests.Requests = append(mintRequests.Requests, &mintRequest)
addressAppendValue := k.cdc.MustMarshal(&mintRequests)
addressStore.Set(getMintRequestKeyBytes(mintRequest.Beneficiary), addressAppendValue)
}
func (k Keeper) GetMintRequestsByAddress(ctx sdk.Context, address string) (val types.MintRequests, found bool) {
store := prefix.NewStore(ctx.KVStore(k.storeKey), types.KeyPrefix(types.MintRequestAddressKey))
mintRequests := store.Get(getMintRequestKeyBytes(address))
if mintRequests == nil {
return val, false
}
k.cdc.MustUnmarshal(mintRequests, &val)
return val, true
}
func (k Keeper) GetMintRequestByHash(ctx sdk.Context, hash string) (val types.MintRequest, found bool) {
store := prefix.NewStore(ctx.KVStore(k.storeKey), types.KeyPrefix(types.MintRequestHashKey))
mintRequest := store.Get(getMintRequestKeyBytes(hash))
if mintRequest == nil {
return val, false
}
k.cdc.MustUnmarshal(mintRequest, &val)
return val, true
}
func (k Keeper) IterateAllMintRequests(ctx sdk.Context, cb func(types.MintRequest) bool) {
store := prefix.NewStore(ctx.KVStore(k.storeKey), types.KeyPrefix(types.MintRequestHashKey))
iterator := store.Iterator(nil, nil)
defer iterator.Close()
for ; iterator.Valid(); iterator.Next() {
var mintRequest types.MintRequest
val := iterator.Value()
k.cdc.MustUnmarshal(val, &mintRequest)
if cb(mintRequest) {
break
}
}
}
func getMintRequestKeyBytes(key string) []byte {
bz := []byte(key)
return bz
}