forked from InjectiveLabs/sdk-go
/
trading_rewards.go
67 lines (54 loc) · 1.22 KB
/
trading_rewards.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
package types
import (
"sort"
sdk "github.com/cosmos/cosmos-sdk/types"
)
type TradingRewardPoints map[string]sdk.Dec
func NewTradingRewardPoints() TradingRewardPoints {
tradingRewardPoints := make(TradingRewardPoints)
return tradingRewardPoints
}
func (l TradingRewardPoints) GetSortedAccountKeys() []string {
accountKeys := make([]string, 0, len(l))
for k := range l {
accountKeys = append(accountKeys, k)
}
sort.Strings(accountKeys)
return accountKeys
}
func (l TradingRewardPoints) AddPointsForAddress(addr string, newPoints sdk.Dec) {
if !newPoints.IsPositive() {
return
}
v, found := l[addr]
if !found {
l[addr] = newPoints
} else {
l[addr] = v.Add(newPoints)
}
}
func (l *TradingRewardPoints) GetAllAccountAddresses() []string {
accountAddresses := make([]string, 0)
for k := range *l {
accountAddresses = append(accountAddresses, k)
}
return accountAddresses
}
func MergeTradingRewardPoints(m1, m2 TradingRewardPoints) TradingRewardPoints {
if len(m1) == 0 {
return m2
} else if len(m2) == 0 {
return m1
}
if len(m1) >= len(m2) {
for k, v := range m2 {
m1.AddPointsForAddress(k, v)
}
return m1
} else {
for k, v := range m1 {
m2.AddPointsForAddress(k, v)
}
return m2
}
}