/
keeper.go
46 lines (39 loc) · 1.26 KB
/
keeper.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
package keeper
import (
"github.com/cosmos/cosmos-sdk/codec"
storetypes "github.com/cosmos/cosmos-sdk/store/types"
sdk "github.com/cosmos/cosmos-sdk/types"
timerstoretypes "github.com/lavanet/lava/x/timerstore/types"
)
func NewKeeper(cdc codec.BinaryCodec) *Keeper {
return &Keeper{
cdc: cdc,
}
}
// Keeper is the timerstore keeper. The keeper retains all the fixation stores used by modules,
// it also manages their lifecycle.
type Keeper struct {
timerStoresBegin []*timerstoretypes.TimerStore
timerStoresEnd []*timerstoretypes.TimerStore
cdc codec.BinaryCodec
}
func (k *Keeper) NewTimerStoreBeginBlock(storeKey storetypes.StoreKey, prefix string) *timerstoretypes.TimerStore {
ts := timerstoretypes.NewTimerStore(storeKey, k.cdc, prefix)
k.timerStoresBegin = append(k.timerStoresBegin, ts)
return ts
}
func (k *Keeper) NewTimerStoreEndBlock(storeKey storetypes.StoreKey, prefix string) *timerstoretypes.TimerStore {
ts := timerstoretypes.NewTimerStore(storeKey, k.cdc, prefix)
k.timerStoresEnd = append(k.timerStoresEnd, ts)
return ts
}
func (k *Keeper) BeginBlock(ctx sdk.Context) {
for _, ts := range k.timerStoresBegin {
ts.Tick(ctx)
}
}
func (k *Keeper) EndBlock(ctx sdk.Context) {
for _, ts := range k.timerStoresEnd {
ts.Tick(ctx)
}
}