Function getNodesHostingPIndex

pindex_copy_request.go:211–328

getNodesHostingPIndex confirms the list of node UUIDs hosting the given pindex.

(uuids []string,
	pindex, authType string, nodeDefs *cbgt.NodeDefs)

// getNodesHostingPIndex confirms the list of node UUIDs
// hosting the given pindex.
func getNodesHostingPIndex(uuids []string,
	pindex, authType string, nodeDefs *cbgt.NodeDefs) []string {
	if len(uuids) == 0 {
		return nil
	}
	urlMap := getStatsUrls(uuids, authType, nodeDefs)
	var wg sync.WaitGroup
	size := len(urlMap)

	type statsReq struct {
		uuid string
		url  string
	}

	type statsResp struct {
		uuid   string
		status bool
		err    error
	}

	requestCh := make(chan *statsReq, size)
	responseCh := make(chan *statsResp, size)
	nWorkers := getWorkerCount(size)
	// spawn the stats get workers
	for i := 0; i < nWorkers; i++ {
		wg.Add(1)
		go func() {
			for reqs := range requestCh {
				ctx, cancel := context.WithTimeout(context.Background(),
					60*time.Second)
				defer cancel()
				req, _ := http.NewRequestWithContext(ctx, "GET", reqs.url, nil)

				httpClient := cbgt.HttpClient()
				res, err := httpClient.Do(req)
				if err != nil {
					responseCh <- &statsResp{
						uuid:   reqs.uuid,
						status: false,
						err: fmt.Errorf("pindex_copy_request:"+
							" getNodesHostingPIndex, get stats for pindex: %s,"+
							" err: %v", reqs.url, err),
					}
					continue
				}
				// parse the response to get the pindex status
				pindexesData := struct {
					PIndexes map[string]struct {
						Partitions map[string]struct {
							UUID string `json:"uuid"`
							Seq  uint64 `json:"seq"`
						} `json:"partitions"`
						Basic struct {
							DocCount uint64 `json:"DocCount"`
						} `json:"basic"`
					} `json:"pindexes"`
				}{}
				data, derr := ioutil.ReadAll(res.Body)