/
keeper.go
94 lines (78 loc) · 2.5 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
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
93
94
package keeper
import (
"fmt"
"github.com/tendermint/tendermint/libs/log"
"github.com/cosmos/cosmos-sdk/codec"
storetypes "github.com/cosmos/cosmos-sdk/store/types"
sdk "github.com/cosmos/cosmos-sdk/types"
paramtypes "github.com/cosmos/cosmos-sdk/x/params/types"
"github.com/sge-network/sge/x/orderbook/types"
)
// Keeper of the orderbook store
type Keeper struct {
storeKey storetypes.StoreKey
cdc codec.BinaryCodec
paramstore paramtypes.Subspace
bankKeeper types.BankKeeper
accountKeeper types.AccountKeeper
BetKeeper types.BetKeeper
marketKeeper types.MarketKeeper
houseKeeper types.HouseKeeper
ovmKeeper types.OVMKeeper
feeGrantKeeper types.FeeGrantKeeper
hooks types.OrderBookHooks
}
// SdkExpectedKeepers contains expected keepers parameter needed by NewKeeper
type SdkExpectedKeepers struct {
BankKeeper types.BankKeeper
AccountKeeper types.AccountKeeper
FeeGrantKeeper types.FeeGrantKeeper
}
// NewKeeper creates a new orderbook Keeper instance
func NewKeeper(
cdc codec.BinaryCodec,
key storetypes.StoreKey,
ps paramtypes.Subspace,
expectedKeepers SdkExpectedKeepers,
) *Keeper {
// set KeyTable if it has not already been set
if !ps.HasKeyTable() {
ps = ps.WithKeyTable(types.ParamKeyTable())
}
return &Keeper{
storeKey: key,
cdc: cdc,
paramstore: ps,
bankKeeper: expectedKeepers.BankKeeper,
accountKeeper: expectedKeepers.AccountKeeper,
feeGrantKeeper: expectedKeepers.FeeGrantKeeper,
}
}
// SetBetKeeper sets the bet module keeper to the order book keeper.
func (k *Keeper) SetBetKeeper(betKeeper types.BetKeeper) {
k.BetKeeper = betKeeper
}
// SetMarketKeeper sets the market module keeper to the order book keeper.
func (k *Keeper) SetMarketKeeper(marketKeeper types.MarketKeeper) {
k.marketKeeper = marketKeeper
}
// SetHouseKeeper sets the market module keeper to the order book keeper.
func (k *Keeper) SetHouseKeeper(houseKeeper types.HouseKeeper) {
k.houseKeeper = houseKeeper
}
// SetOVMKeeper sets the ovm module keeper to the market keeper.
func (k *Keeper) SetOVMKeeper(ovmKeeper types.OVMKeeper) {
k.ovmKeeper = ovmKeeper
}
// SetHooks sets the hooks for governance
func (k *Keeper) SetHooks(gh types.OrderBookHooks) *Keeper {
if k.hooks != nil {
panic("cannot set orderbook hooks twice")
}
k.hooks = gh
return k
}
// Logger returns the logger of the keeper
func (Keeper) Logger(ctx sdk.Context) log.Logger {
return ctx.Logger().With("module", fmt.Sprintf("x/%s", types.ModuleName))
}