-
Notifications
You must be signed in to change notification settings - Fork 2
/
smesher.go
70 lines (60 loc) · 2.44 KB
/
smesher.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
package service
import (
"context"
"fmt"
"go.mongodb.org/mongo-driver/bson"
"go.mongodb.org/mongo-driver/mongo/options"
"github.com/spacemeshos/explorer-backend/model"
)
// GetSmesher returns smesher by id.
func (e *Service) GetSmesher(ctx context.Context, smesherID string) (*model.Smesher, error) {
smesher, err := e.storage.GetSmesher(ctx, smesherID)
if err != nil {
return nil, err
}
if smesher == nil {
return nil, ErrNotFound
}
smesher.Rewards, _, err = e.CountSmesherRewards(ctx, smesherID)
return smesher, err
}
// GetSmeshers returns smeshers by filter.
func (e *Service) GetSmeshers(ctx context.Context, page, perPage int64) (smeshers []*model.Smesher, total int64, err error) {
total, err = e.storage.CountSmeshers(ctx, &bson.D{})
if err != nil {
return nil, 0, fmt.Errorf("failed to count total smeshers: %w", err)
}
if total == 0 {
return []*model.Smesher{}, 0, nil
}
smeshers, err = e.storage.GetSmeshers(ctx, &bson.D{}, e.getFindOptions("timestamp", page, perPage))
if err != nil {
return nil, 0, fmt.Errorf("failed to get smeshers: %w", err)
}
return smeshers, total, nil
}
// GetSmesherActivations returns smesher activations by filter.
func (e *Service) GetSmesherActivations(ctx context.Context, smesherID string, page, perPage int64) (atxs []*model.Activation, total int64, err error) {
return e.getActivations(ctx, &bson.D{{Key: "smesher", Value: smesherID}}, e.getFindOptions("layer", page, perPage))
}
// GetSmesherRewards returns smesher rewards by filter.
func (e *Service) GetSmesherRewards(ctx context.Context, smesherID string, page, perPage int64) (rewards []*model.Reward, total int64, err error) {
opts := e.getFindOptions("layer", page, perPage)
opts.SetProjection(bson.D{})
return e.getRewards(ctx, &bson.D{{Key: "smesher", Value: smesherID}}, opts)
}
// CountSmesherRewards returns smesher rewards count by filter.
func (e *Service) CountSmesherRewards(ctx context.Context, smesherID string) (total, count int64, err error) {
return e.storage.CountSmesherRewards(ctx, smesherID)
}
func (e *Service) getSmeshers(ctx context.Context, filter *bson.D, options *options.FindOptions) (smeshers []*model.Smesher, total int64, err error) {
total, err = e.storage.CountEpochSmeshers(ctx, filter)
if err != nil {
return []*model.Smesher{}, 0, err
}
smeshers, err = e.storage.GetEpochSmeshers(ctx, filter, options)
if err != nil {
return nil, 0, fmt.Errorf("error load smeshers: %w", err)
}
return smeshers, total, nil
}