Skip to content

Commit e0bbc76

Browse files
committed
feat: added omitZeroBalances param for the GetAccountService
1 parent 5905362 commit e0bbc76

File tree

2 files changed

+20
-3
lines changed

2 files changed

+20
-3
lines changed

v2/account_service.go

+13-1
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,15 @@ import (
77

88
// GetAccountService get account info
99
type GetAccountService struct {
10-
c *Client
10+
c *Client
11+
omitZeroBalances *bool
12+
}
13+
14+
// OmitZeroBalances sets the omitZeroBalances parameter on the request.
15+
// When set to true, the API will return the non-zero balances of an account.
16+
func (s *GetAccountService) OmitZeroBalances(v bool) *GetAccountService {
17+
s.omitZeroBalances = &v
18+
return s
1119
}
1220

1321
// Do send request
@@ -17,6 +25,10 @@ func (s *GetAccountService) Do(ctx context.Context, opts ...RequestOption) (res
1725
endpoint: "/api/v3/account",
1826
secType: secTypeSigned,
1927
}
28+
if s.omitZeroBalances != nil {
29+
r.setParam("omitZeroBalances", *s.omitZeroBalances)
30+
}
31+
2032
data, err := s.c.callAPI(ctx, r, opts...)
2133
if err != nil {
2234
return nil, err

v2/account_service_test.go

+7-2
Original file line numberDiff line numberDiff line change
@@ -50,12 +50,17 @@ func (s *accountServiceTestSuite) TestGetAccount() {
5050
}`)
5151
s.mockDo(data, nil)
5252
defer s.assertDo()
53+
54+
omitZeroBalances := true
55+
5356
s.assertReq(func(r *request) {
54-
e := newSignedRequest()
57+
e := newSignedRequest().setParams(params{
58+
"omitZeroBalances": omitZeroBalances,
59+
})
5560
s.assertRequestEqual(e, r)
5661
})
5762

58-
res, err := s.client.NewGetAccountService().Do(newContext())
63+
res, err := s.client.NewGetAccountService().OmitZeroBalances(omitZeroBalances).Do(newContext())
5964
s.r().NoError(err)
6065
e := &Account{
6166
MakerCommission: 15,

0 commit comments

Comments
 (0)