Method Visit

optimizer/count_threshold.go:16–54

(node *Node)

type countThreshold struct{}

func (*countThreshold) Visit(node *Node) {
	binary, ok := (*node).(*BinaryNode)
	if !ok {
		return
	}

	count, ok := binary.Left.(*BuiltinNode)
	if !ok || count.Name != "count" || len(count.Arguments) != 2 {
		return
	}

	integer, ok := binary.Right.(*IntegerNode)
	if !ok || integer.Value < 0 {
		return
	}

	var threshold int
	switch binary.Operator {
	case ">":
		threshold = integer.Value + 1
	case ">=":
		threshold = integer.Value
	case "<":
		threshold = integer.Value
	case "<=":
		threshold = integer.Value + 1
	default:
		return
	}

	// Skip if threshold is 0 or 1 (handled by count_any optimizer)
	if threshold <= 1 {
		return
	}

	// Set threshold on the count node for early termination
	// The original comparison remains unchanged
	count.Threshold = &threshold
}