扩展性开发实战：自定义类型与函数的构建方法 = 飞龙博客

扩展性开发实战：自定义类型与函数的构建方法

(6) feilong.org 修订于2026-08-21 08:49:24 PostgreSQL教程

自定义数据类型的创建与应用

PostgreSQL提供了强大的扩展能力，允许开发者通过自定义数据类型（Custom Types）和函数（Functions）满足特定业务需求。本文将深入讲解如何在PostgreSQL中实现这两种扩展机制，并结合实际案例演示其应用场景。

1. 自定义数据类型
1.1 基本概念
自定义类型用于描述复杂的数据结构，可分为：
- 复合类型（Composite Types）：由多个字段组成的结构体
- 数组类型（Array Types）：基于现有类型的多维集合
- 范围类型（Range Types）：表示连续值的区间

1.2 创建复合类型
使用

CREATE TYPE

1	CREATE TYPE

语句定义复合类型，语法如下：

CREATE TYPE type_name AS (  
    field1 type1,  
    field2 type2,  
    ...  
);

CREATE TYPE type_name AS (

field1 type1,

field2 type2,

...

);

示例：创建一个表示地址信息的复合类型

CREATE TYPE address_type AS (
    street TEXT,
    city VARCHAR(50),
    postal_code CHAR(6)
);

CREATE TYPE address_type AS (

street TEXT,

city VARCHAR(50),

postal_code CHAR(6)

);

1.3 使用自定义类型
- 作为表字段：直接在表中使用自定义类型

CREATE TABLE customer(
    id SERIAL PRIMARY KEY,
    name TEXT,
    contact address_type
);

CREATE TABLE customer(

id SERIAL PRIMARY KEY,

name TEXT,

contact address_type

);

- 查询操作：通过字段名访问复合类型的成员

SELECT name, contact.street FROM customer WHERE contact.city = '北京';

1	SELECT name, contact.street FROM customer WHERE contact.city = '北京';

2. 自定义函数的构建
2.1 函数基础语法
PostgreSQL支持多种语言编写函数，常见包括SQL、PL/pgSQL、Python等。基本结构如下：

CREATE FUNCTION function_name (params) RETURNS return_type AS $$
-- 函数体
$$ LANGUAGE language_name;

CREATE FUNCTION function_name (params) RETURNS return_type AS $$

-- 函数体

$$ LANGUAGE language_name;

2.2 示例：计算商品价格的函数
创建一个根据成本价和利润率计算销售价的函数

CREATE FUNCTION calculate_price(cost NUMERIC, margin NUMERIC)
RETURNS NUMERIC AS $$
BEGIN
    RETURN cost * (1 + margin / 100);
END;
$$ LANGUAGE plpgsql;

CREATE FUNCTION calculate_price(cost NUMERIC, margin NUMERIC)

RETURNS NUMERIC AS $$

BEGIN

RETURN cost * (1 + margin / 100);

END;

$$ LANGUAGE plpgsql;

2.3 函数的应用场景
- 业务逻辑封装：将复杂计算转化为可复用的函数模块

SELECT calculate_price(100, 20); -- 返回 120

1	SELECT calculate_price(100, 20); -- 返回 120

- 触发器集成：结合

BEFORE INSERT

1	BEFORE INSERT

触发器实现数据校验

CREATE TRIGGER check_stock BEFORE INSERT ON products
FOR EACH ROW EXECUTE FUNCTION validate_inventory();

1 2	CREATE TRIGGER check_stock BEFORE INSERT ON products FOR EACH ROW EXECUTE FUNCTION validate_inventory();

3. 高级扩展技巧
3.1 类型继承与多态
PostgreSQL支持类型继承，可构建层次化数据模型：

CREATE TYPE vehicle AS (
    wheels INTEGER
);
CREATE TYPE car AS (  
    brand TEXT,
    model TEXT
) INHERITS (vehicle);

CREATE TYPE vehicle AS (

wheels INTEGER

);

CREATE TYPE car AS (

brand TEXT,

model TEXT

) INHERITS (vehicle);

3.2 函数重载
通过不同参数列表实现同名函数的多态调用：

CREATE FUNCTION add(a NUMERIC, b NUMERIC)
RETURNS NUMERIC AS $$ BEGIN RETURN a + b; END $$ LANGUAGE plpgsql;

CREATE FUNCTION add(a TEXT, b TEXT)
RETURNS TEXT AS $$ BEGIN RETURN a || b; END $$ LANGUAGE plpgsql;

CREATE FUNCTION add(a NUMERIC, b NUMERIC)

RETURNS NUMERIC AS $$ BEGIN RETURN a + b; END $$ LANGUAGE plpgsql;

CREATE FUNCTION add(a TEXT, b TEXT)

RETURNS TEXT AS $$ BEGIN RETURN a || b; END $$ LANGUAGE plpgsql;

4. 实际开发建议
- 类型设计原则：优先使用现有类型避免重复定义，仅在必要时创建复合类型
- 函数性能优化：避免在函数中执行复杂查询，尽量通过索引和计划缓存提升效率
- 安全性考量：对敏感操作的函数应设置合理的权限控制

通过合理运用自定义类型与函数，开发者可以显著增强PostgreSQL数据库的灵活性与可维护性。建议结合具体业务场景进行设计，并持续关注PostgreSQL官方文档中的最新特性更新。

更新网址：https://feilong.org/custom-type-function-creation
最初发布：20260821 08:49:24 feilong.org 于广州

加入收藏夹，查看更方便。

旧文：流处理与事件捕获：使用pg_regress进行数据同步

飛龍博客