python - SQLAlchemy:条件自动增量

Question

我想创建一个平面论坛，其中线程没有单独的表，并且具有帖子的复合主键。

因此，帖子有两个字段形成自然键:thread_id 和 post_number，其中进一步是它们所属线程的 ID，后者是它们的 ID线程中的位置。如果您不相信，请检查该行下方。

我的问题是我不知道如何告诉 SQLAlchemy

when committing the addition of new Post instances with thread_id tid, look up how many posts with thread_id tid exist, and autoincrement from that number on.

<小时/>

为什么我认为该架构是个好主意？因为它自然且高效:

class Post(Base):
    number    = Column(Integer, primary_key=True, autoincrement=False, nullable=False)
    thread_id = Column(Integer, primary_key=True, autoincrement=False, nullable=False)
    title     = Column(Text) #nullable for not-first posts
    text      = Column(Text, nullable=False)
    ...
PAGESIZE = 10
#test
tid = 5
page = 4

整个线程(查询):

thread5 = session.query(Post).filter_by(thread_id=5)

主题标题:

title = thread5.filter_by(number=0).one().title

话题页面

page4 = thread5.filter(
    Post.number >= (page    * PAGESIZE),
    Post.number < ((page+1) * PAGESIZE)).all()
#or
page4 = thread5.offset(page * PAGESIZE).limit(PAGESIZE).all()

页数:

ceil(thread5.count() / PAGESIZE)

Answer 1

您可以使用an SQL expression as a default value来做到这一点(参见 default 参数)。给它一个像这样的可调用:

from sqlalchemy.sql import func

def maxnumber_for_threadid(context):
    return post_table.select([func.max(post_table.c.number)]).where(post_table.c.thread_id==context.current_parameters['thread_id'])

我不确定您是否可以从默认可调用函数返回 sql 表达式 - 您可能必须实际执行此查询并在回调中返回标量值。 (光标应该可以从 context 参数中获得。)

但是，我强烈建议您按照 @kindall 的说法进行操作，并为 number 使用另一个自动递增序列。柱子。即使没有 SQLAlchemy，您想要做的事情实际上也很难正确完成。例如，如果您使用的是 MVCC 数据库，则需要引入特殊的行级锁定，以便匹配 thread_id 的行数。当您运行事务时不会改变。如何完成此操作取决于数据库。例如使用 MySQL InnoDB，您需要执行以下操作:

BEGIN TRANSACTION;
SELECT MAX(number)+1 FROM posts WHERE thread_id=? FOR UPDATE;
INSERT INTO posts (thread_id, number) VALUES (?, ?); -- number is from previous query
COMMIT;

如果您没有使用FOR UPDATE ，那么可以想象，尝试同时将新帖子插入同一线程的另一个连接可能会获得与 number 相同的值。 .

因此，由于需要额外的查询和锁定，后插入实际上相当慢(相对而言)，而不是高性能。

所有这些都可以通过使用单独的序列来解决，而不用担心帖子编号仅在 thread_id 内递增。