Method HandleTask

remote.go:446–502

HandleTask is an implementation of the cbgt.TaskRequestHandler interface

(in []byte)

// HandleTask is an implementation of the cbgt.TaskRequestHandler interface
func (r *IndexClient) HandleTask(in []byte) (*cbgt.TaskRequestStatus, error) {
	var treq cbgt.TaskRequest
	err := json.Unmarshal(in, &treq)
	if err != nil {
		return nil, err
	}
	// populate the target partitions
	treq.PartitionNames = r.PIndexNames
	buf, err := MarshalJSON(&treq)
	if err != nil {
		return nil, err
	}

	u, err := UrlWithAuth(r.AuthType(), r.TaskRequestURL)
	if err != nil {
		return nil, fmt.Errorf("remote: auth for HandleTask,"+
			" TaskRequestURL: %s, authType: %s, err: %v",
			r.TaskRequestURL, r.AuthType(), err)
	}

	req, err := http.NewRequest("POST", u, bytes.NewReader(buf))
	if err != nil {
		return nil, err
	}
	req.Header.Add("Content-Type", "application/json")

	resp, err := r.httpClient.Do(req)
	if err != nil {
		return nil, err
	}
	defer resp.Body.Close()

	respBuf, err := ioutil.ReadAll(resp.Body)
	if err != nil {
		return nil, fmt.Errorf("remote: HandleTask error reading resp.Body,"+
			" TaskRequestURL: %s, resp: %#v, err: %v", r.TaskRequestURL,
			resp, err)
	}

	r.lastMutex.Lock()
	defer r.lastMutex.Unlock()

	rv := &cbgt.TaskRequestStatus{Errors: make(map[string]error)}
	err = json.Unmarshal(respBuf, rv)
	if err != nil {
		return completeTaskStatus(&treq, err, r.PIndexNames), nil
	}
	if resp.StatusCode != http.StatusOK {
		return completeTaskStatus(&treq,
			fmt.Errorf("remote: HandleTask got status code: %d,"+
				" TaskRequestURL: %s, buf: %s, resp: %#v, err: %v",
				resp.StatusCode, r.TaskRequestURL, buf, resp, err),
			r.PIndexNames), nil
	}

	return rv, nil
}