尝试优化代码

This commit is contained in:
wyx2685
2024-01-25 23:06:37 +09:00
parent e292b3b0e7
commit 91e78fbc20
7 changed files with 66 additions and 24 deletions

View File

@@ -134,8 +134,18 @@ func (c *Client) GetNodeInfo() (node *NodeInfo, err error) {
if err = c.checkResponse(r, path, err); err != nil {
return nil, err
}
if r.StatusCode() == 304 {
return nil, nil
if r != nil {
defer func() {
if r.RawBody() != nil {
r.RawBody().Close()
}
}()
if r.StatusCode() == 304 {
return nil, nil
}
} else {
return nil, fmt.Errorf("received nil response")
}
node = &NodeInfo{
Id: c.NodeId,

View File

@@ -30,13 +30,21 @@ func (c *Client) GetUserList() (UserList []UserInfo, err error) {
r, err := c.client.R().
SetHeader("If-None-Match", c.userEtag).
Get(path)
err = c.checkResponse(r, path, err)
if err != nil {
if err = c.checkResponse(r, path, err); err != nil {
return nil, err
}
if r.StatusCode() == 304 {
return nil, nil
if r != nil {
defer func() {
if r.RawBody() != nil {
r.RawBody().Close()
}
}()
if r.StatusCode() == 304 {
return nil, nil
}
} else {
return nil, fmt.Errorf("received nil response")
}
var userList *UserListBody
err = json.Unmarshal(r.Body(), &userList)