-
Notifications
You must be signed in to change notification settings - Fork 28
/
denom.go
92 lines (76 loc) · 2.56 KB
/
denom.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
package keeper
import (
"crypto/sha256"
"fmt"
errorsmod "cosmossdk.io/errors"
sdk "github.com/cosmos/cosmos-sdk/types"
sdkerrors "github.com/cosmos/cosmos-sdk/types/errors"
"github.com/irisnet/irismod/modules/mt/types"
)
const denomIdPrefix = "mt-denom-%d"
// genDenomID generate a denom ID by auto increment sequence
func (k Keeper) genDenomID(ctx sdk.Context) string {
sequence := k.GetDenomSequence(ctx)
denomID := fmt.Sprintf(denomIdPrefix, sequence)
hash := fmt.Sprintf("%x", sha256.Sum256([]byte(denomID)))
k.SetDenomSequence(ctx, sequence+1)
return hash
}
// HasDenom returns whether the specified denom ID exists
func (k Keeper) HasDenom(ctx sdk.Context, id string) bool {
store := ctx.KVStore(k.storeKey)
return store.Has(types.KeyDenom(id))
}
// SetDenom is responsible for saving the definition of denom
func (k Keeper) SetDenom(ctx sdk.Context, denom types.Denom) {
store := ctx.KVStore(k.storeKey)
bz := k.cdc.MustMarshal(&denom)
store.Set(types.KeyDenom(denom.Id), bz)
}
// GetDenom returns the denom by id
func (k Keeper) GetDenom(ctx sdk.Context, id string) (denom types.Denom, found bool) {
store := ctx.KVStore(k.storeKey)
bz := store.Get(types.KeyDenom(id))
if len(bz) == 0 {
return denom, false
}
k.cdc.MustUnmarshal(bz, &denom)
return denom, true
}
// GetDenoms returns all the denoms
func (k Keeper) GetDenoms(ctx sdk.Context) (denoms []types.Denom) {
store := ctx.KVStore(k.storeKey)
iterator := sdk.KVStorePrefixIterator(store, types.KeyDenom(""))
defer iterator.Close()
for ; iterator.Valid(); iterator.Next() {
var denom types.Denom
k.cdc.MustUnmarshal(iterator.Value(), &denom)
denoms = append(denoms, denom)
}
return denoms
}
// UpdateDenom is responsible for updating the definition of denom
func (k Keeper) UpdateDenom(ctx sdk.Context, denom types.Denom) error {
if !k.HasDenom(ctx, denom.Id) {
return errorsmod.Wrapf(sdkerrors.ErrNotFound, "denom not found (%s)", denom.Id)
}
store := ctx.KVStore(k.storeKey)
bz := k.cdc.MustMarshal(&denom)
store.Set(types.KeyDenom(denom.Id), bz)
return nil
}
// GetDenomSequence gets the next denom sequence from the store.
func (k Keeper) GetDenomSequence(ctx sdk.Context) uint64 {
store := ctx.KVStore(k.storeKey)
bz := store.Get([]byte(types.KeyNextDenomSequence))
if bz == nil {
return 1
}
return sdk.BigEndianToUint64(bz)
}
// SetDenomSequence sets the next denom sequence to the store.
func (k Keeper) SetDenomSequence(ctx sdk.Context, sequence uint64) {
store := ctx.KVStore(k.storeKey)
bz := sdk.Uint64ToBigEndian(sequence)
store.Set([]byte(types.KeyNextDenomSequence), bz)
}