PostgreSQL 高级功能（五）

原创

已于 2024-06-25 19:30:03 修改 · 1.2k 阅读

10 ·

CC 4.0 BY-SA版权

文章标签：

#postgresql #大数据 #数据库

于 2024-06-25 18:30:53 首次发布

1. 存储过程与函数

1.1 创建存储过程

存储过程是一组预编译的SQL语句，可以简化复杂的操作。以下是一个简单的存储过程示例：

CREATE OR REPLACE FUNCTION add_user(username VARCHAR, email VARCHAR) 
RETURNS VOID AS $$
BEGIN
    INSERT INTO users (username, email) VALUES (username, email);
END;
$$ LANGUAGE plpgsql;

调用存储过程：

SELECT add_user('john_doe', '[email protected]');

1.2 创建函数

函数与存储过程类似，但函数可以返回值。以下是一个计算两数之和的函数示例：

CREATE OR REPLACE FUNCTION add_numbers(a INTEGER, b INTEGER) 
RETURNS INTEGER AS $$
BEGIN
    RETURN a + b;
END;
$$ LANGUAGE plpgsql;

调用函数：

SELECT add_numbers(10, 20);

2. 触发器

触发器是在特定事件发生时自动执行的程序，如插入、更新或删除操作。

2.1 创建触发器

以下示例展示了在 users 表上创建触发器，当插入新用户时，自动记录时间戳到 audit_log 表：

CREATE TABLE audit_log (
    id SERIAL PRIMARY KEY,
    username VARCHAR(50),
    action VARCHAR(50),
    action_time TIMESTAMP DEFAULT CURRENT_TIMESTAMP
);

CREATE OR REPLACE FUNCTION log_user_action() 
RETURNS TRIGGER AS $$
BEGIN
    INSERT INTO audit_log (username, action) VALUES (NEW.username, 'INSERT');
    RETURN NEW;
END;
$$ LANGUAGE plpgsql;

CREATE TRIGGER after_user_insert
AFTER INSERT ON users
FOR EACH ROW
EXECUTE FU