postgresql操作符重载如何实现_postgresqloperator扩展_技术教程

PostgreSQL的操作符重载基于类型系统与函数绑定实现，而非C++式签名区分；通过为不同数据类型定义对应操作函数（如+对complex类型绑定complex_add），并支持操作符族与索引策略，实现多义性。

PostgreSQL 的操作符重载（Operator Overloading）并不是像 C++ 那样通过“函数签名不同”来实现的，而是通过定义不同的操作符族、索引策略以及结合函数和数据类型的绑定机制来支持对相同符号（如 +, =, 等）在不同数据类型上的多义性。这种“重载”本质上是基于类型系统的函数多态实现。

1. 操作符重载的基本原理

在 PostgreSQL 中，操作符（如 +）本身只是一个符号，它背后关联的是一个具体的函数。同一个操作符可以用于多种数据类型，只要为每种类型组合定义了对应的操作函数即可。这就是所谓的“操作符重载”。

例如：

integer + integer 调用内置的整数加法函数
text || text 表示字符串拼接
你可以定义 point + point 实现几何点相加

PostgreSQL 根据输入参数的数据类型选择正确的函数执行，从而实现“重载”效果。

2. 如何自定义操作符（实现“重载”）

假设你想为两个自定义类型 complex（复数）定义加法操作符 +。

步骤一：创建复合类型

CREATE TYPE complex AS (
    re double precision,
    im double precision
);

步骤二：创建处理加法的函数

CREATE OR REPLACE FUNCTION complex_add(a complex, b complex)
RETURNS complex AS $$
BEGIN
    RETURN ROW(a.re + b.re, a.im + b.im)::complex;
END;
$$ LANGUAGE plpgsql IMMUTABLE;

步骤三：定义操作符

CREATE OPERATOR + (
    LEFTARG = complex,
    RIGHTARG = complex,
    PROCEDURE = complex_add,
    COMMUTATOR = +
);

现在你可以这样使用：

SELECT ROW(1.0, 2.0)::complex + ROW(3.0, 4.0)::complex;
-- 结果: (4,6)

这就实现了对 + 操作符在 complex 类型上的“重载”。

3. 支持索引：操作符族与 B-Tree/GiST

如果希望你的操作符能用于索引（比如 =, ），需要将其组织进 操作符族（operator family） 并声明其支持的索引策略。

例如，为 complex 定义大小比较操作符以支持 B-Tree 索引：

CREATE OR REPLACE FUNCTION complex_lt(a complex, b complex) 
RETURNS boolean AS $$
BEGIN
    RETURN a.re < b.re OR (a.re = b.re AND a.im < b.im);
END;
$$ LANGUAGE plpgsql IMMUTABLE;
CREATE OPERATOR < (
LEFTARG = complex,
RIGHTARG = complex,
PROCEDURE = complex_lt
);
-- 创建操作符类
CREATE OPERATOR CLASS complex_ops
DEFAULT FOR TYPE complex USING btree AS
OPERATOR    1   < ,
OPERATOR    3   = ,
FUNCTION    1   complex_cmp(complex, complex);