Function MemoizeComparisonExprOp

pkg/sql/sem/tree/expr.go:460–493

MemoizeComparisonExprOp populates the Op field of the ComparisonExpr. TODO(ajwerner): It feels dangerous to leave this to the caller to set. Should we rework the construction and access to the underlying Op to enforce safety?

(node *ComparisonExpr)

// Should we rework the construction and access to the underlying Op to
// enforce safety?
func MemoizeComparisonExprOp(node *ComparisonExpr) {
	fOp, fLeft, fRight, _, _ := FoldComparisonExpr(node.Operator, node.Left, node.Right)
	leftRet, rightRet := fLeft.(TypedExpr).ResolvedType(), fRight.(TypedExpr).ResolvedType()
	switch node.Operator.Symbol {
	case treecmp.Any, treecmp.Some, treecmp.All:
		// Array operators memoize the SubOperator's CmpOp.
		fOp, _, _, _, _ = FoldComparisonExpr(node.SubOperator, nil, nil)
		// The right operand is either an array or a tuple/subquery.
		switch rightRet.Family() {
		case types.ArrayFamily:
			// For example:
			//   x = ANY(ARRAY[1,2])
			rightRet = rightRet.ArrayContents()
		case types.TupleFamily:
			// For example:
			//   x = ANY(SELECT y FROM t)
			//   x = ANY(1,2)
			if len(rightRet.TupleContents()) > 0 {
				rightRet = rightRet.TupleContents()[0]
			} else {
				rightRet = leftRet
			}
		}
	}

	fn, ok := CmpOps[fOp.Symbol].LookupImpl(leftRet, rightRet)
	if !ok {
		panic(errors.AssertionFailedf("lookup for ComparisonExpr %s's CmpOp failed (%s(%s,%s))",
			AsStringWithFlags(node, FmtShowTypes), redact.Safe(fOp.String()),
			leftRet.SQLStringForError(), rightRet.SQLStringForError(),
		))
	}
	node.Op = fn
}

// TypedLeft returns the ComparisonExpr's left expression as a TypedExpr.
func (node *ComparisonExpr) TypedLeft() TypedExpr {