func ShockGet(host string, nodeid string, token string) (node *ShockNode, err error) { if host == "" || nodeid == "" { return nil, errors.New("empty shock host or node id") } //logger.Debug(1, fmt.Sprintf("ShockGet: %s %s %s", host, nodeid, token)) var res *http.Response shockurl := fmt.Sprintf("%s/node/%s", host, nodeid) var user *httpclient.Auth if token != "" { user = httpclient.GetUserByTokenAuth(token) } c := make(chan int, 1) go func() { res, err = httpclient.Get(shockurl, httpclient.Header{}, nil, user) c <- 1 //we are ending }() select { case <-c: //go ahead case <-time.After(SHOCK_TIMEOUT): return nil, errors.New("timeout when getting node from shock, url=" + shockurl) } if err != nil { return } defer res.Body.Close() jsonstream, err := ioutil.ReadAll(res.Body) if err != nil { return nil, err } response := new(ShockResponse) if err := json.Unmarshal(jsonstream, response); err != nil { return nil, err } if len(response.Errs) > 0 { return nil, errors.New(strings.Join(response.Errs, ",")) } node = &response.Data if node == nil { err = errors.New("empty node got from Shock") } return }
func FetchShockStream(url string, token string) (r io.ReadCloser, err error) { var user *httpclient.Auth if token != "" { user = httpclient.GetUserByTokenAuth(token) } //download file from Shock res, err := httpclient.Get(url, httpclient.Header{}, nil, user) if err != nil { //return nil, err return nil, errors.New("httpclient.Get returned: " + err.Error()) } if res.StatusCode != 200 { //err in fetching data resbody, _ := ioutil.ReadAll(res.Body) return nil, errors.New(fmt.Sprintf("op=fetchFile, url=%s, res=%s", url, resbody)) } return res.Body, err }