update response

This commit is contained in:
Sinh 2021-12-03 16:45:23 +07:00
parent 282a94a2ef
commit b1202dc4e3
2 changed files with 21 additions and 7 deletions

View File

@ -21,7 +21,6 @@ type PushRequest struct {
// PushResponse ... // PushResponse ...
type PushResponse struct { type PushResponse struct {
RequestID string `json:"requestId"` RequestID string `json:"requestId"`
Error string `json:"error"`
} }
type pushRequest struct { type pushRequest struct {

View File

@ -64,14 +64,17 @@ func (c *Client) PushToUsers(payload PushRequest) (requestID string, err error)
if err != nil { if err != nil {
return "", err return "", err
} }
var res PushResponse var res struct {
Data PushResponse `json:"data"`
Error string `json:"error"`
}
if err := json.Unmarshal(msg.Data, &res); err != nil { if err := json.Unmarshal(msg.Data, &res); err != nil {
return "", err return "", err
} }
if res.Error != "" { if res.Error != "" {
return "", errors.New(res.Error) return "", errors.New(res.Error)
} }
return res.RequestID, nil return res.Data.RequestID, nil
} }
// Query get list notification by user id // Query get list notification by user id
@ -87,11 +90,17 @@ func (c *Client) Query(q Query) (ListNotificationResponse, error) {
if err != nil { if err != nil {
return ListNotificationResponse{}, err return ListNotificationResponse{}, err
} }
var res ListNotificationResponse var res struct {
Data ListNotificationResponse `json:"data"`
Error string `json:"error"`
}
if err := json.Unmarshal(msg.Data, &res); err != nil { if err := json.Unmarshal(msg.Data, &res); err != nil {
return ListNotificationResponse{}, err return ListNotificationResponse{}, err
} }
return res, nil if res.Error != "" {
return ListNotificationResponse{}, errors.New(res.Error)
}
return res.Data, nil
} }
// CountUnread count total unread notification // CountUnread count total unread notification
@ -105,11 +114,17 @@ func (c *Client) CountUnread(q CountUnread) (int64, error) {
if err != nil { if err != nil {
return 0, err return 0, err
} }
var res CountUnreadResponse var res struct {
Data CountUnreadResponse `json:"data"`
Error string `json:"error"`
}
if err := json.Unmarshal(msg.Data, &res); err != nil { if err := json.Unmarshal(msg.Data, &res); err != nil {
return 0, err return 0, err
} }
return res.Total, nil if res.Error != "" {
return 0, errors.New(res.Error)
}
return res.Data.Total, nil
} }
// Read mark notification as read // Read mark notification as read