Function buildCopyPartitionRequestUtil

pindex_copy_request.go:153–207

buildCopyPartitionRequestUtil creates the partition copy request with a sorted list of source nodeDefs for the given pindex.

(planPIndexes *cbgt.PlanPIndexes,
	pindexName string, stats *CopyPartitionStats, nodeDefs *cbgt.NodeDefs,
	sourceUUIDs []string, formerPrimary string, mgr *cbgt.Manager,
	stopCh chan struct{})

// buildCopyPartitionRequestUtil creates the partition copy request
// with a sorted list of source nodeDefs for the given pindex.
func buildCopyPartitionRequestUtil(planPIndexes *cbgt.PlanPIndexes,
	pindexName string, stats *CopyPartitionStats, nodeDefs *cbgt.NodeDefs,
	sourceUUIDs []string, formerPrimary string, mgr *cbgt.Manager,
	stopCh chan struct{}) (*CopyPartitionRequest, error) {
	// compute the given partition load on the cluster nodes.
	nodeParitionCount := make(map[string]int)
	for _, pp := range planPIndexes.PlanPIndexes {
		for node := range pp.Nodes {
			if _, ok := nodeParitionCount[node]; !ok {
				nodeParitionCount[node] = 1
			} else {
				nodeParitionCount[node]++
			}
		}
	}

	// form the source partition candidates.
	planPIndex := planPIndexes.PlanPIndexes[pindexName]
	var sps []*srcPartition
	for _, uuid := range sourceUUIDs {
		if nodeDef, ok := nodeDefs.NodeDefs[uuid]; ok {
			node := planPIndex.Nodes[uuid]
			sp := &srcPartition{nodeDef: nodeDef,
				partitionCountOnNode: nodeParitionCount[uuid]}
			if node.Priority == 0 {
				sp.state = "primary"
			} else if node.Priority > 0 {
				sp.state = "replica"
			}
			sps = append(sps, sp)
		}
	}

	// sort them in the order of preference.
	sort.Sort(&srcPartitionSorter{
		sps:   sps,
		state: "primary",
		rack:  mgr.Container()})

	if len(sps) > 0 {
		var sourceNodes []*cbgt.NodeDef
		for _, sp := range sps {
			sourceNodes = append(sourceNodes, sp.nodeDef)
		}
		return &CopyPartitionRequest{
			SourceNodes:      sourceNodes,
			SourcePIndexName: planPIndex.Name,
			SourceUUID:       planPIndex.UUID,
			Stats:            stats,
			CancelCh:         stopCh,
		}, nil
	}

	return nil, nil
}

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