postgreSQL 中的自定义操作符示例详解

SELECT 3 OPERATOR(pg_catalog.+) 4 sum; -- 1️⃣

SELECT 3 + 4; -- 2️⃣

SELECT 3 OPERATOR(+) 4; -- 可以省略 schema，默认 pg_catalog

CREATE OPERATOR operator_name (
{ LEFTARG = left_type -- 左操作数类型（单目操作符可省略）
| RIGHTARG = right_type -- 右操作数类型（单目操作符可省略）
| BOTHARG = both_type } -- 左右类型相同时代替上面两个
[, PROCEDURE = function_name ] -- 必须：真正执行的函数
[, COMMUTATOR = com_op ] -- 可选：交换律操作符（如 + 和 + 本身）
[, NEGATOR = neg_op ] -- 可选：取反操作符（如 = 的取反是 <>）
[, RESTRICT = res_proc ] -- 可选：用于优化器选择性估计
[, JOIN = join_proc ] -- 可选：用于优化器连接估计
[, HASHES ] -- 可选：支持 HASH JOIN 和 hash 聚合
[, MERGES ] -- 可选：支持 MERGE JOIN
);

-- 第1步：先创建一个底层函数
CREATE OR REPLACE FUNCTION immutable_upper(text)
RETURNS text AS $$
SELECT upper($1);
$$ LANGUAGE sql IMMUTABLE STRICT;
-- 第2步：创建前缀操作符（右操作数，没有左操作数）
CREATE OPERATOR !! (
RIGHTARG = text, -- 只有右操作数，在右边 → 前缀操作符
PROCEDURE = immutable_upper -- 调用上面那个函数
);
-- 第3步：试用
SELECT !! 'hello'; -- 返回 HELLO
SELECT !! column_name FROM users;

CREATE OR REPLACE FUNCTION geometry_strict_equal(anyelement, anyelement)
RETURNS boolean AS $$
SELECT $1 IS NOT DISTINCT FROM $2;
$$ LANGUAGE sql IMMUTABLE;

CREATE OPERATOR === (
LEFTARG = anyelement,
RIGHTARG = anyelement,
PROCEDURE = geometry_strict_equal,
COMMUTATOR = ===, -- 自己和自己交换律
NEGATOR = !==, -- 稍后会创建它的取反
HASHES, -- 支持 hash join / hash agg
MERGES -- 支持 merge join
);
-- 创建取反操作符 !==
CREATE OPERATOR !== (
LEFTARG = anyelement,
RIGHTARG = anyelement,
PROCEDURE = geometry_strict_equal,
NEGATOR = === -- 互相指向对方
);

SELECT
n.nspname AS schema,
o.oprname AS operator, -- 操作符名称
format_type(o.oprleft, NULL) AS left_type,
format_type(o.oprright, NULL) AS right_type,
p.proname AS function_name -- 函数名称
FROM pg_operator o
JOIN pg_namespace n ON n.oid = o.oprnamespace
JOIN pg_proc p ON p.oid = o.oprcode
WHERE n.nspname NOT IN ('pg_catalog')
and o.oprname = '!!'; -- 可以去掉过滤看看

DROP OPERATOR IF EXISTS !! (NONE, text); -- 先删除操作符，必须传左右两个参数，没有的写NONE
DROP FUNCTION public.immutable_upper(text); -- 函数如果还要用可以不删

CREATE OR REPLACE FUNCTION chinese_ilike(text, text)
RETURNS boolean AS $$
SELECT $1 ILIKE $2;
$$ LANGUAGE sql IMMUTABLE STRICT;