postgresql 触发器如何生成递增序列号,从1开始,并且每天重置

大家好,我是三叔,许久不见,这期给大家介绍一下笔者在开发中遇到的业务处理: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 开始递增的序列号,并且第二天重置

结果如下:

笔者成功获取到每天操作文档的次数:
postgresql 触发器如何生成递增序列号,从1开始,并且每天重置_第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;

你可能感兴趣的:(Postgresql,postgresql,数据库,java,spring,boot,mysql,递增序列)