flip/go-data-bridge-client
1
0
Fork 0
mirror of https://github.com/FlipsideCrypto/go-data-bridge-client.git synced 2026-02-06 10:48:12 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

check for unread count instead of relying on 404

Browse Source
This commit is contained in:
Don Cote 2019-09-13 14:29:05 -04:00
parent c3b653b305
commit e5a9b7fa61
1 changed files with 47 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

53
record.go
View File

@ -13,9 +13,48 @@ type Record struct {
Data []map[string]interface{} `json:"data"`
}
func getNextRecord(c Client) (*Record, error) {
url := fmt.Sprintf("%s/topics/%s/records/next?consumer_id=%s&api_key=%s", c.BaseURL, c.TopicSlug, c.ConsumerID, c.APIKey)
func getUnreadCount(c Client) (*int32, error) {
url := fmt.Sprintf("%s/topics/%s?api_key=%s", c.BaseURL, c.TopicSlug, c.APIKey)
req, _ := http.NewRequest("GET", url, nil)
req.Header.Add("content-type", "application/json")
req.Header.Add("api_key", c.APIKey)
res, err := http.DefaultClient.Do(req)
if err != nil {
return nil, errors.Wrap(err, fmt.Sprintf("error making databridge http request for %s", url))
}
if res.StatusCode != 200 {
return nil, errors.Wrap(err, fmt.Sprintf("databridge - error getting unread count for %s", url))
}
defer res.Body.Close()
body, err := ioutil.ReadAll(res.Body)
if err != nil {
return nil, errors.Wrap(err, "error trying to read databridge response body")
}
var js struct {
Count int32 `json:"unread_count"`
}
err = json.Unmarshal([]byte(body), &js)
if err != nil {
return nil, errors.Wrap(err, "error trying to unmarshal response body to json")
}
return &js.Count, nil
}
func getNextRecord(c Client) (*Record, error) {
count, err := getUnreadCount(c)
if err != nil {
return nil, err
} else if *count == 0 {
return nil, nil
}
url := fmt.Sprintf("%s/topics/%s/records/next?consumer_id=%s&api_key=%s", c.BaseURL, c.TopicSlug, c.ConsumerID, c.APIKey)
req, _ := http.NewRequest("GET", url, nil)
req.Header.Add("content-type", "application/json")
req.Header.Add("api_key", c.APIKey)
@ -25,8 +64,8 @@ func getNextRecord(c Client) (*Record, error) {
return nil, err
}
if res.StatusCode == 404 {
return nil, nil
if res.StatusCode != 200 {
return nil, errors.New(fmt.Sprintf("databridge responded with non-200 for %s", url))
}
defer res.Body.Close()
@ -51,12 +90,14 @@ func (r Record) updateRecordState(c Client, state string) error {
req.Header.Add("content-type", "application/json")
req.Header.Add("api_key", c.APIKey)
_, err := http.DefaultClient.Do(req)
res, err := http.DefaultClient.Do(req)
if err != nil {
return err
}
// todo - probably get more granular on checking this response
if res.StatusCode != 200 {
return errors.New(fmt.Sprintf("databridge update record state responded with non-200 for %s", url))
}
return nil
}