Flask SQLAlchemy如何设置约束防止插入重叠整数区间

你用的是PostgreSQL数据库，最佳实现是靠数据库层面的排他约束（Exclusion Constraint），从根源上避免并发插入导致的重叠问题，比纯应用层校验可靠得多。

前置操作：安装btree_gist扩展 #

排他约束依赖GiST索引实现区间重叠判断，需要先安装PostgreSQL的btree_gist扩展，你可以直接在Alembic迁移文件中执行以下SQL完成安装：

CREATE EXTENSION IF NOT EXISTS btree_gist;

调整模型配置 #

不需要修改你原有字段结构，只要在模型中新增约束配置即可：

from sqlalchemy import ExclusionConstraint, CheckConstraint

class IntervalModel(BaseModel):
    """Interval table"""

    __tablename__ = 'interval'

    start = db.Column(db.Integer, index=True, nullable=False)
    end = db.Column(db.Integer, index=True, nullable=False)
    desc = db.Column(db.String(256), index=True, nullable=True)

    __table_args__ = (
        # 先保证区间合法性：起始值必须小于结束值
        CheckConstraint('"start" < "end"', name='check_start_lt_end'),
        # 核心排他约束：禁止任意两个区间重叠
        # 此处用'[]'表示区间左右闭包，匹配你示例中[2,8]和[9,20]相邻不冲突的规则
        ExclusionConstraint(
            [('int4range("start", "end", \'[]\')', '&&')],
            name='exclude_overlapping_intervals'
        ),
        # 可选配置：interval是PostgreSQL内置关键字，开启表名加引号避免语法冲突
        # {'quote': True}
    )

补充说明 #

1. 如果你是给已有数据的表加约束，需要先清理表中已有的重叠区间，否则约束创建会失败
2. 你也可以在应用层提前做重叠校验，提前给用户返回友好提示，避免直接抛出数据库错误，应用层校验逻辑参考：

def check_overlap(new_start, new_end):
    # 重叠判断逻辑：现有区间起始 < 新区间结束，且现有区间结束 > 新区间起始
    return IntervalModel.query.filter(
        IntervalModel.start < new_end,
        IntervalModel.end > new_start
    ).first() is not None

注意：应用层校验只能作为辅助，并发场景下必须靠数据库约束才能保证数据正确性

3. 插入重叠区间时，数据库会抛出IntegrityError异常，你可以捕获该异常后返回对应的业务错误提示。

内容的提问来源于stack exchange，提问作者user2156115