postgresql在插入数据后怎么获取自增id

要获取数据库自动分配的ID(通常是主键),可以使用数据库连接对象(在这里是 Connection)的相应方法来获取插入的最后一个自增ID。

不同的数据库管理系统有不同的方法来实现这一点。以下是两个常见的数据库管理系统的示例:

对于 PostgreSQL

如果你使用的是 PostgreSQL 数据库,可以使用 RETURNING 子句来返回插入的行的信息,包括自动生成的ID。以下是一个示例:

sqlStr = """
    INSERT INTO charts (name, file_name, scale,)
    VALUES (%(name)s, %(fileName)s, %(scale)s)
    RETURNING id;
"""
Cursor.execute(sqlStr, chartsInfo)
inserted_id = Cursor.fetchone()[0]  # 获取自增ID
Connection.commit()
print('插入成功!新记录的ID为:', inserted_id)

这个示例中,我们在 SQL 查询中使用 RETURNING id; 来返回插入的新记录的ID,然后使用 self._Cursor.fetchone()[0] 获取这个ID。

对于 MySQL

如果使用的是 MySQL 数据库,可以使用 LAST_INSERT_ID() 函数来获取最后插入的自增ID。以下是一个示例:

sqlStr = """
    INSERT INTO charts (name, file_name, scale)
    VALUES (%(name)s, %(fileName)s, %(scale)s);
"""
Cursor.execute(sqlStr, chartsInfo)
Connection.commit()
inserted_id = Cursor.lastrowid  # 获取自增ID
print('插入成功!新记录的ID为:', inserted_id)

在这个示例中,我们插入了数据,然后使用 Cursor.lastrowid 获取最后插入的自增ID。

具体的方法取决于所使用的数据库管理系统,因此请根据数据库类型选择适当的方法。

在这两个示例中,演示了如何在 PostgreSQL 和 MySQL 中获取自增ID。

posted @ 2023-09-03 16:27  槑孒  阅读(1170)  评论(0编辑  收藏  举报