-
Notifications
You must be signed in to change notification settings - Fork 18
/
rewards.go
80 lines (67 loc) · 2.92 KB
/
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
68
69
70
71
72
73
74
75
76
77
78
79
80
// Copyright (c) 2022 Gobalsky Labs Limited
//
// Use of this software is governed by the Business Source License included
// in the LICENSE.DATANODE file and at https://www.mariadb.com/bsl11.
//
// Change Date: 18 months from the later of the date of the first publicly
// available Distribution of this version of the repository, and 25 June 2022.
//
// On the date above, in accordance with the Business Source License, use
// of this software will be governed by version 3 or later of the GNU General
// Public License.
package service
import (
"context"
"code.vegaprotocol.io/vega/datanode/entities"
"code.vegaprotocol.io/vega/datanode/utils"
"code.vegaprotocol.io/vega/logging"
)
type rewardStore interface {
Add(ctx context.Context, r entities.Reward) error
GetAll(ctx context.Context) ([]entities.Reward, error)
GetByCursor(ctx context.Context, partyID *string, assetID *string, fromEpoch, toEpoch *uint64, p entities.CursorPagination) ([]entities.Reward, entities.PageInfo, error)
GetSummaries(ctx context.Context, partyID *string, assetID *string) ([]entities.RewardSummary, error)
GetEpochSummaries(ctx context.Context, filter entities.RewardSummaryFilter, p entities.CursorPagination) ([]entities.EpochRewardSummary, entities.PageInfo, error)
}
type Reward struct {
store rewardStore
observer utils.Observer[entities.Reward]
}
func NewReward(store rewardStore, log *logging.Logger) *Reward {
return &Reward{
store: store,
observer: utils.NewObserver[entities.Reward]("reward", log, 0, 0),
}
}
func (r *Reward) Add(ctx context.Context, reward entities.Reward) error {
err := r.store.Add(ctx, reward)
if err != nil {
return err
}
r.observer.Notify([]entities.Reward{reward})
return nil
}
func (r *Reward) GetAll(ctx context.Context) ([]entities.Reward, error) {
return r.store.GetAll(ctx)
}
func (r *Reward) GetByCursor(ctx context.Context, partyID, assetID *string, fromEpoch, toEpoch *uint64, p entities.CursorPagination) ([]entities.Reward, entities.PageInfo, error) {
return r.store.GetByCursor(ctx, partyID, assetID, fromEpoch, toEpoch, p)
}
func (r *Reward) GetSummaries(ctx context.Context, partyID *string, assetID *string) ([]entities.RewardSummary, error) {
return r.store.GetSummaries(ctx, partyID, assetID)
}
func (r *Reward) GetEpochRewardSummaries(ctx context.Context, filter entities.RewardSummaryFilter, p entities.CursorPagination) ([]entities.EpochRewardSummary, entities.PageInfo, error) {
return r.store.GetEpochSummaries(ctx, filter, p)
}
func (r *Reward) Observe(ctx context.Context, retries int, assetID, partyID string) (rewardCh <-chan []entities.Reward, ref uint64) {
ch, ref := r.observer.Observe(ctx,
retries,
func(reward entities.Reward) bool {
return (len(assetID) == 0 || assetID == reward.AssetID.String()) &&
(len(partyID) == 0 || partyID == reward.PartyID.String())
})
return ch, ref
}
func (r *Reward) GetRewardSubscribersCount() int32 {
return r.observer.GetSubscribersCount()
}