From 714557378324f656bcf08cff37d6fad34f3ed646 Mon Sep 17 00:00:00 2001 From: Mike Bayer Date: Mon, 21 Mar 2022 17:47:34 -0400 Subject: [PATCH] use begin() for settting up first PK Change-Id: I227bbb46fbcbae1f60d3f5bb4dd2b9f41ca3dd0c (cherry picked from commit 257f67b180cadf4ca0df0d1facf27d55c98df676) --- examples/sharding/separate_databases.py | 2 +- examples/sharding/separate_tables.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/sharding/separate_databases.py b/examples/sharding/separate_databases.py index accbfd79b4..9818656c3c 100644 --- a/examples/sharding/separate_databases.py +++ b/examples/sharding/separate_databases.py @@ -56,7 +56,7 @@ ids = Table("ids", Base.metadata, Column("nextid", Integer, nullable=False)) def id_generator(ctx): # in reality, might want to use a separate transaction for this. - with db1.connect() as conn: + with db1.begin() as conn: nextid = conn.scalar(ids.select().with_for_update()) conn.execute(ids.update().values({ids.c.nextid: ids.c.nextid + 1})) return nextid diff --git a/examples/sharding/separate_tables.py b/examples/sharding/separate_tables.py index 4314e22337..0f6e2ffd83 100644 --- a/examples/sharding/separate_tables.py +++ b/examples/sharding/separate_tables.py @@ -70,7 +70,7 @@ ids = Table("ids", Base.metadata, Column("nextid", Integer, nullable=False)) def id_generator(ctx): # in reality, might want to use a separate transaction for this. - with engine.connect() as conn: + with engine.begin() as conn: nextid = conn.scalar(ids.select().with_for_update()) conn.execute(ids.update().values({ids.c.nextid: ids.c.nextid + 1})) return nextid -- 2.47.2