大家好,我是三叔,许久不见,这期给大家介绍一下笔者在开发中遇到的业务处理:pgsql 创建触发器生成每日递增序列,并且第二天重置,根据不同的用户进行不同的控制。
1.创建生成递增序列的 table 表
-- 创建user_sequences表,用于保存生成的序列号
CREATE TABLE user_sequences
(
id serial,
user_name varchar(255) NULL, -- 区分不同用户(可根据不同业务组装key)
increment_value int4 NULL, -- 递增序列号
last_increment_date timestamp NULL, -- 触发器参数时间
description varchar(255) NULL, -- 描述
tenant_id varchar(32) NULL, -- 租户ID
create_user varchar(32) NULL, -- 创建人ID
create_time timestamp(6) NULL DEFAULT CURRENT_TIMESTAMP, -- 创建时间
update_user varchar(32) NULL, -- 最后修改人ID
update_time timestamp NULL DEFAULT CURRENT_TIMESTAMP, -- 更新时间
CONSTRAINT user_sequences_pk PRIMARY KEY (id)
);
2.创建一个触发器函数
要创建一个触发器,你需要:
1、使用 create function 语句创建一个触发器函数
2、使用 create trigger 语句将触发器函数绑定到数据表
3、创建触发器函数语法:
触发器函数与普通函数类似,不同之处在于触发器不需要任何参数,并且触发器函数需要返回trigger类型值
-- 创建触发器函数
CREATE OR REPLACE FUNCTION insert_increment_value_if_not_exists()
RETURNS TRIGGER AS $$
BEGIN
-- 检查 user_sequences 表中是否存在对应的 user_name
IF NOT EXISTS (
SELECT 1 FROM user_sequences WHERE user_name = NEW.user_name
) THEN
-- 在 user_sequences 表中插入新的记录
RETURN NEW;
else
UPDATE user_sequences
SET increment_value = increment_value + 1
WHERE user_name = NEW.user_name
;
END IF;
RETURN null;
END;
$$ LANGUAGE plpgsql;
上述部分语法解释
NEW:该变量为 INSERT/UPDATE 操作触发的行级触发器中存储新的数据行
3.创建触发器
-- 这里介绍一下创建触发器语法(如下简单语法):
create trigger trigger_name
{before | after | instead of} event1 or event2...
on table_name
{for each row | for each statement}
execute procedure trigger_function;
event1: 事件可以是:insert, update, delete, truncate
before: 表示定义触发器在事件之前
after: 表示定义触发器在事件之后
intead of: 表示仅用于视图的insert, update, delete
for each row:表示行级触发器
for each statement:表示语级触发器
实际业务逻辑如下:笔者在插入表数据的时候触发执行器执行begin -》end 的 sql 逻辑,这样,我就能在每天对不同的用户,生成一个从1开始的递增序列,每次相同用户同一天访问一次,序列递增加1:
CREATE TRIGGER insert_increment_value_trigger
BEFORE INSERT ON user_sequences
FOR EACH ROW
EXECUTE FUNCTION insert_increment_value_if_not_exists();
4.业务处理逻辑
大家可以看到,在创建触发器的时候,有这么一句语法:BEFORE INSERT ON user_sequences,这句话是什么意思呢?
根据上面的触发器语法介绍可以知道,在每次执行 insert 语句之前,会执行触发器里面的语法。
@Service
public class xxxCommon {
@Autowired
private TableMapper mapper;
@Transactional(rollbackFor = Exception.class)
public String getSequencesNum(String userName) {
// 需要注意的是这里的userName需要保证唯一,可以拼接时间yyyyMMdd,这样可以保证每天获取新的递增序列
// 第二种方法,就是在笔者的基础上,添加时间的判断条件,判断上一次修改或生成数据的时间是否不等于当前时间,如果是的,则重新从1开始递增
UserSequencesVO vo = new UserSequencesVO();
assembleParam(vo);
vo.setIncrementValue(1);
vo.setUserName(userName);
// 触发器触发条件,insert 语句进行触发,这里我在xml里面写了一条新增的sql语句,从而触发触发器
mapper.insertTriger(vo);
// 这里获取数据库返回的序列号,
UserSequencesVO sequencesVO = UserSequencesMapper.selectByUserName(userName);
// 根据自己的业务组装返回值
return sequencesVO.getIncrementValue();
}
private void assembleParam(UserSequencesVO vo) {
vo.setDescription("获取当前key的递增序列");
vo.setTenantId(userUtil.getTenantId());
vo.setLastIncrementDate(LocalDateTime.now());
vo.setCreateTime(LocalDateTime.now());
vo.setCreateUser(userUtil.getId());
vo.setTenantId(userUtil.getTenantId());
vo.setUpdateTime(LocalDateTime.now());
vo.setUpdateUser(userUtil.getTenantId());
}
}
业务处理
在拿到递增的序列号之后,就可以对序列号进行一系列的组装操作,这样,根据不同的用户,互不影响,获取从 1 开始递增的序列号,并且第二天重置
结果如下:
笔者成功获取到每天操作文档的次数:
最后:如何删除触发器
DROP TRIGGER [ IF EXISTS ] name ON table_name [ CASCADE | RESTRICT ]
参数说明:
IF EXISTS:如果该触发器不存在
name:要移除的刚刚定义的触发器的名称
table_name:定义了该触发器的表的名称
CASCADE:自动删除依赖于该触发器的对象,然后删除所有 依赖于那些对象的对象
RESTRICT:如果有任何对象依赖于该触发器,则拒绝删除它,这是默认值
-- 示例
DROP TRIGGER insert_increment_value_trigger ON user_sequences;