/
keeper.go
172 lines (150 loc) · 5.18 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
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
package keeper
import (
"github.com/cosmos/cosmos-sdk/codec"
sdk "github.com/cosmos/cosmos-sdk/types"
"github.com/medibloc/panacea-core/x/aol/types"
)
// Keeper maintains the link to data storage
// and exposes getter/setter methods for
// the various parts of the state machine
type Keeper struct {
// Unexposed key to access store from sdk.Context
storeKey sdk.StoreKey
cdc *codec.Codec
}
// NewKeeper creates new instances of the aol Keeper
func NewKeeper(storeKey sdk.StoreKey, cdc *codec.Codec) Keeper {
return Keeper{
storeKey: storeKey,
cdc: cdc,
}
}
func (k Keeper) SetOwner(ctx sdk.Context, ownerAddr sdk.AccAddress, owner types.Owner) {
store := ctx.KVStore(k.storeKey)
ownerKey := OwnerKey(ownerAddr)
bz := k.cdc.MustMarshalBinaryLengthPrefixed(owner)
store.Set(ownerKey, bz)
}
func (k Keeper) GetOwner(ctx sdk.Context, ownerAddr sdk.AccAddress) types.Owner {
store := ctx.KVStore(k.storeKey)
ownerKey := OwnerKey(ownerAddr)
bz := store.Get(ownerKey)
if bz == nil {
return types.Owner{}
}
var owner types.Owner
k.cdc.MustUnmarshalBinaryLengthPrefixed(bz, &owner)
return owner
}
func (k Keeper) ListOwner(ctx sdk.Context) []sdk.AccAddress {
store := ctx.KVStore(k.storeKey)
addrs := make([]sdk.AccAddress, 0)
prefix := OwnerKey(sdk.AccAddress{})
iter := sdk.KVStorePrefixIterator(store, prefix)
defer iter.Close()
for ; iter.Valid(); iter.Next() {
addr := getLastElement(iter.Key(), prefix)
addrs = append(addrs, addr)
}
return addrs
}
func (k Keeper) SetTopic(ctx sdk.Context, ownerAddr sdk.AccAddress, topicName string, topic types.Topic) {
store := ctx.KVStore(k.storeKey)
topicKey := TopicKey(ownerAddr, topicName)
bz := k.cdc.MustMarshalBinaryLengthPrefixed(topic)
store.Set(topicKey, bz)
}
func (k Keeper) GetTopic(ctx sdk.Context, ownerAddr sdk.AccAddress, topicName string) types.Topic {
store := ctx.KVStore(k.storeKey)
topicKey := TopicKey(ownerAddr, topicName)
bz := store.Get(topicKey)
if bz == nil {
return types.Topic{}
}
var topic types.Topic
k.cdc.MustUnmarshalBinaryLengthPrefixed(bz, &topic)
return topic
}
func (k Keeper) HasTopic(ctx sdk.Context, ownerAddr sdk.AccAddress, topicName string) bool {
store := ctx.KVStore(k.storeKey)
topicKey := TopicKey(ownerAddr, topicName)
return store.Has(topicKey)
}
func (k Keeper) ListTopic(ctx sdk.Context, ownerAddr sdk.AccAddress) []string {
store := ctx.KVStore(k.storeKey)
topics := make([]string, 0)
prefix := TopicKey(ownerAddr, "")
iter := sdk.KVStorePrefixIterator(store, prefix)
defer iter.Close()
for ; iter.Valid(); iter.Next() {
topic := getLastElement(iter.Key(), prefix)
topics = append(topics, string(topic))
}
return topics
}
func (k Keeper) SetWriter(ctx sdk.Context, ownerAddr sdk.AccAddress, topic string, writerAddr sdk.AccAddress, writer types.Writer) {
store := ctx.KVStore(k.storeKey)
writerKey := ACLWriterKey(ownerAddr, topic, writerAddr)
bz := k.cdc.MustMarshalBinaryLengthPrefixed(writer)
store.Set(writerKey, bz)
}
func (k Keeper) GetWriter(ctx sdk.Context, ownerAddr sdk.AccAddress, topic string, writerAddr sdk.AccAddress) types.Writer {
store := ctx.KVStore(k.storeKey)
writerKey := ACLWriterKey(ownerAddr, topic, writerAddr)
bz := store.Get(writerKey)
if bz == nil {
return types.Writer{}
}
var writer types.Writer
k.cdc.MustUnmarshalBinaryLengthPrefixed(bz, &writer)
return writer
}
func (k Keeper) HasWriter(ctx sdk.Context, ownerAddr sdk.AccAddress, topic string, writerAddr sdk.AccAddress) bool {
store := ctx.KVStore(k.storeKey)
writerKey := ACLWriterKey(ownerAddr, topic, writerAddr)
return store.Has(writerKey)
}
func (k Keeper) DeleteWriter(ctx sdk.Context, ownerAddr sdk.AccAddress, topic string, writerAddr sdk.AccAddress) {
store := ctx.KVStore(k.storeKey)
writerKey := ACLWriterKey(ownerAddr, topic, writerAddr)
store.Delete(writerKey)
}
//TODO Max Retrieve
func (k Keeper) ListWriter(ctx sdk.Context, ownerAddr sdk.AccAddress, topic string) []sdk.AccAddress {
store := ctx.KVStore(k.storeKey)
writers := make([]sdk.AccAddress, 0)
prefix := ACLWriterKey(ownerAddr, topic, sdk.AccAddress{})
iter := sdk.KVStorePrefixIterator(store, prefix)
defer iter.Close()
for ; iter.Valid(); iter.Next() {
writer := sdk.AccAddress{}
data := getLastElement(iter.Key(), prefix)
if err := writer.Unmarshal(data); err != nil {
return []sdk.AccAddress{}
}
writers = append(writers, writer)
}
return writers
}
func (k Keeper) SetRecord(ctx sdk.Context, ownerAddr sdk.AccAddress, topic string, offset uint64, record types.Record) {
store := ctx.KVStore(k.storeKey)
recordKey := RecordKey(ownerAddr, topic, offset)
bz := k.cdc.MustMarshalBinaryLengthPrefixed(record)
store.Set(recordKey, bz)
}
func (k Keeper) GetRecord(ctx sdk.Context, ownerAddr sdk.AccAddress, topic string, offset uint64) types.Record {
store := ctx.KVStore(k.storeKey)
recordKey := RecordKey(ownerAddr, topic, offset)
bz := store.Get(recordKey)
if bz == nil {
return types.Record{}
}
var record types.Record
k.cdc.MustUnmarshalBinaryLengthPrefixed(bz, &record)
return record
}
func (k Keeper) HasRecord(ctx sdk.Context, ownerAddr sdk.AccAddress, topic string, offset uint64) bool {
store := ctx.KVStore(k.storeKey)
recordKey := RecordKey(ownerAddr, topic, offset)
return store.Has(recordKey)
}