/
resources.go
54 lines (45 loc) · 1.37 KB
/
resources.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
package client
import (
"context"
"github.com/Paven-Org/gotron-sdk/pkg/common"
"github.com/Paven-Org/gotron-sdk/pkg/proto/api"
"github.com/Paven-Org/gotron-sdk/pkg/proto/core"
)
// GetAccountResource from BASE58 address
func (g *GrpcClient) GetAccountResource(addr string) (*api.AccountResourceMessage, error) {
account := new(core.Account)
var err error
account.Address, err = common.DecodeCheck(addr)
if err != nil {
return nil, err
}
ctx, cancel := context.WithTimeout(context.Background(), g.grpcTimeout)
defer cancel()
return g.Client.GetAccountResource(ctx, account)
}
// GetDelegatedResources from BASE58 address
func (g *GrpcClient) GetDelegatedResources(address string) ([]*api.DelegatedResourceList, error) {
addrBytes, err := common.DecodeCheck(address)
if err != nil {
return nil, err
}
ctx, cancel := context.WithTimeout(context.Background(), g.grpcTimeout)
defer cancel()
ai, err := g.Client.GetDelegatedResourceAccountIndex(ctx, GetMessageBytes(addrBytes))
if err != nil {
return nil, err
}
result := make([]*api.DelegatedResourceList, len(ai.GetToAccounts()))
for i, addrTo := range ai.GetToAccounts() {
dm := &api.DelegatedResourceMessage{
FromAddress: addrBytes,
ToAddress: addrTo,
}
resource, err := g.Client.GetDelegatedResource(ctx, dm)
if err != nil {
return nil, err
}
result[i] = resource
}
return result, nil
}