snapshot of https://github.com/sqlc-dev/sqlc.git (main, 2026-09-16; blobs > 131072B omitted)

internal/sql/lang/operator.go

@@ -0,0 +1,63 @@
+package lang
+
+// TODO: This logic is PostgreSQL-specific and needs to be refactored to support MySQL
+
+func IsComparisonOperator(s string) bool {
+	switch s {
+	case ">":
+	case "<":
+	case "<=":
+	case ">=":
+	case "=":
+	case "==":
+	case "<>":
+	case "!=":
+	// Comparisons an engine spells as a keyword rather than as punctuation.
+	// They are written upper case, which is how SQLite's parser records
+	// them; PostgreSQL and MySQL use distinct node types or their own
+	// operator names for these, so the spellings do not overlap.
+	case "IS":
+	case "IS NOT":
+	case "IS DISTINCT FROM":
+	case "IS NOT DISTINCT FROM":
+	case "ISNULL":
+	case "NOTNULL":
+	case "NOT NULL":
+	case "LIKE":
+	case "NOT LIKE":
+	case "GLOB":
+	case "NOT GLOB":
+	case "REGEXP":
+	case "NOT REGEXP":
+	case "MATCH":
+	case "NOT MATCH":
+	default:
+		return false
+	}
+	return true
+}
+
+func IsMathematicalOperator(s string) bool {
+	switch s {
+	case "+":
+	case "-":
+	case "*":
+	case "/":
+	case "%":
+	case "^":
+	case "|/":
+	case "||/":
+	case "!":
+	case "!!":
+	case "@":
+	case "&":
+	case "|":
+	case "#":
+	case "~":
+	case "<<":
+	case ">>":
+	default:
+		return false
+	}
+	return true
+}