package heliumapi import ( "encoding/json" "fmt" "io/ioutil" "net/http" ) var ( ApiUrl = "https://api.helium.io" client = &http.Client{} ) // createGetRequest create a GET request to the helium api func createGetRequest(path string, params map[string]string) (*http.Request, error) { req, err := http.NewRequest("GET", fmt.Sprintf("%s/%s", ApiUrl, path), nil) // setup headers req.Header.Add("Accept", "application/json") // setup query param if there are any if params != nil { query := req.URL.Query() for k, v := range params { query.Add(k, v) } if err != nil { return nil, err } req.URL.RawQuery = query.Encode() } return req, err } // getHeliumApi query a regular helium api endpoint func getHeliumApi(path string, params *map[string]string) ([]byte, error) { // if params is nil, set it to an empty map if params == nil { params = &map[string]string{} } req, err := createGetRequest(path, *params) if err != nil { return nil, fmt.Errorf("failed to create query request %v: %v", path, err) } // query the api resp, err := client.Do(req) if err != nil { return nil, fmt.Errorf("failed to query %v: %v", path, err) } defer resp.Body.Close() // read the response body body, err := ioutil.ReadAll(resp.Body) if err != nil { return nil, fmt.Errorf("failed to read response body of %v: %v", path, err) } return body, nil } // getHeliumApiWithCursor query a helium api with a cursor // It will continue to query the api until all the cursors are exhausted. Combine all the // responses in a single object. func getHeliumApiWithCursor(path string, params *map[string]string) ([][]byte, error) { // if params is nil, set it to an empty map if params == nil { params = &map[string]string{} } type ResponseWithCursor struct { Cursor string `json:"cursor"` } res := [][]byte{} respCursor := ResponseWithCursor{} req, err := createGetRequest(path, *params) if err != nil { return nil, fmt.Errorf("failed to create query request %v: %v", path, err) } for { // query the api resp, err := client.Do(req) if err != nil { return nil, fmt.Errorf("failed to query %v: %v", path, err) } defer resp.Body.Close() // read the response body and add it to the result array body, err := ioutil.ReadAll(resp.Body) if err != nil { return nil, fmt.Errorf("failed to read response body of %v: %v", path, err) } res = append(res, body) // parse the response body for a cursor respCursor.Cursor = "" err = json.Unmarshal(body, &respCursor) if err != nil { return nil, fmt.Errorf("failed to unmarshal response from %v: %v", path, err) } // continue querying until there is no longer a cursor if respCursor.Cursor != "" { params = &map[string]string{"cursor": respCursor.Cursor} req, err = createGetRequest(path, *params) if err != nil { return nil, fmt.Errorf("failed to create query request %v: %v", path, err) } } else { break } } return res, nil }