/
api_deals.go
79 lines (71 loc) · 2.21 KB
/
api_deals.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
package api
import (
"fmt"
"github.com/textileio/powergate/deals"
)
// StorageDealRecords lists storage deals for this FFS instance according to the provided options.
func (i *API) StorageDealRecords(opts ...deals.DealRecordsOption) ([]deals.StorageDealRecord, error) {
i.lock.Lock()
defer i.lock.Unlock()
c := deals.DealRecordsConfig{}
for _, opt := range opts {
opt(&c)
}
finalAddrs, err := i.finalAddresses(c.FromAddrs)
if err != nil {
return nil, fmt.Errorf("getting final addrs: %v", err)
}
recs, err := i.drm.ListStorageDealRecords(
deals.WithFromAddrs(finalAddrs...),
deals.WithAscending(c.Ascending),
deals.WithDataCids(c.DataCids...),
deals.WithIncludeFinal(c.IncludeFinal),
deals.WithIncludePending(c.IncludePending),
)
if err != nil {
return nil, fmt.Errorf("calling ListStorageDealRecords: %v", err)
}
return recs, nil
}
// RetrievalDealRecords returns a list of retrieval deals for this FFS instance according to the provided options.
func (i *API) RetrievalDealRecords(opts ...deals.DealRecordsOption) ([]deals.RetrievalDealRecord, error) {
i.lock.Lock()
defer i.lock.Unlock()
c := deals.DealRecordsConfig{}
for _, opt := range opts {
opt(&c)
}
finalAddrs, err := i.finalAddresses(c.FromAddrs)
if err != nil {
return nil, fmt.Errorf("getting final addrs: %v", err)
}
recs, err := i.drm.ListRetrievalDealRecords(
deals.WithFromAddrs(finalAddrs...),
deals.WithAscending(c.Ascending),
deals.WithDataCids(c.DataCids...),
)
if err != nil {
return nil, fmt.Errorf("calling dm.ListRetrievalDealRecords: %v", err)
}
return recs, nil
}
func (i *API) finalAddresses(fromAddrs []string) ([]string, error) {
instanceAddrs := make([]string, 0, len(i.cfg.Addrs))
instanceAddrsFilter := make(map[string]struct{})
for _, addrInfo := range i.cfg.Addrs {
instanceAddrs = append(instanceAddrs, addrInfo.Addr)
instanceAddrsFilter[addrInfo.Addr] = struct{}{}
}
var finalAddrs []string
if len(fromAddrs) > 0 {
for _, addr := range fromAddrs {
if _, ok := instanceAddrsFilter[addr]; !ok {
return nil, fmt.Errorf("address %s is not managed by this ffs instance", addr)
}
finalAddrs = append(finalAddrs, addr)
}
} else {
finalAddrs = instanceAddrs
}
return finalAddrs, nil
}