Function getRowData

pkg/planner/core/plan_cache_partition_table_test.go:609–663

(rowData map[any]string, filler string, cols []string, isCaseSensitive bool, id ...any)

}

func getRowData(rowData map[any]string, filler string, cols []string, isCaseSensitive bool, id ...any) []string {
	maxRange := 2000000
	ret := make([]string, 0, len(id))
	dup := make(map[any]struct{}, len(id))
	for i := range id {
		isStr := false
		switch x := id[i].(type) {
		case int:
			if x >= maxRange {
				continue
			}
		case string:
			if isCaseSensitive {
				if x >= "x" {
					continue
				}
			} else {
				lc := strings.ToLower(x)
				if lc >= "x" {
					continue
				}
			}
			isStr = true
		default:
			panic("Unsupported type")
		}
		if _, ok := dup[id[i]]; ok {
			continue
		}
		var row string
		for j, col := range cols {
			if j > 0 {
				row += " "
			}
			switch col {
			case "a":
				if isStr {
					row += id[i].(string)
				} else {
					row += strconv.Itoa(id[i].(int))
				}
			case "b":
				row += rowData[id[i]]
			case "c":
				row += filler
			case "space(1)":
				row += " "
			}
		}
		ret = append(ret, row)
		dup[id[i]] = struct{}{}
	}
	sort.Strings(ret)
	return ret
}

func getRandCols(seededRand *rand.Rand) ([]string, bool) {
	allCols := []string{"a", "b", "c", "space(1)"}