欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  php教程

TP 3.2.3 简单解决pgsql无法返回自增主键问题

程序员文章站 2022-05-29 17:48:29
...
tp中pgsql向数据库插入数据之后,无法正常返回自增主键,可以考虑使用下面的方法简单处理下
对于pgsql,在使用 add() 方法插入数据之后,返回的结果总是 1,而不是新插入数据的自增 ID。

-----------------------------------------------------------------------------------------------

原因

ThinkPHP 3.2.3 中,所有的数据库都是采用 PDO 方式连接数据库的。PDO 中返回插入数据的自增 ID 的方法是下面的这个方法:string PDO::lastInsertId ([ string $name = NULL ] )参数 name 表示应该返回 ID 的那个序列对象的名称,是可选的。

但是这个函数的具体行为是基于具体的底层驱动实现的。在不同的 PDO 驱动之间,此方法可能不会返回一个有意义或一致的结果,因为底层数据库可能不支持自增字段或序列的概念。

-----------------------------------------------------------------------------------------------

对于 pgsql 来说:PDO_PGSQL() 要求为 name 参数指定序列对象的名称。

而这正是导致问题的所在:
在 ThinkPHP 的 DB 类 Driver.class.php 中,其获取自增 ID 的方式是使用无参数的 lastInsertId() 方法。增加数据后,没有参数时,pgsql 返回的就是 1。

-----------------------------------------------------------------------------------------------
-----------------------------------------------------------------------------------------------

解决

对于 pgsql,如果要返回正确的自增 ID,必然就是需要在调用 lastInsertId() 方法时添加对应的序列对象的名称(sequence name)。

所以,首先,我们需要在 pgsql 的对应驱动中重写 DB 驱动的`getLastInsID()`方法,给其传入参数:# 位置:ThinkPHP/Library/Think/Db/Driver/Pgsql.class.php
/**
* 用于获取最后插入的ID
* @access public
* @return integer
*/
public function getLastInsID($sequenceName) {
return $this->_linkID->lastInsertId($sequenceName);
}
那么,这个参数应该是什么呢?

一般情况下,pgsql 数据表中设置了自增主键的话,那么自增主键的命名方式为:tableNmae_columnName_seq其中:

* tableName 表示当前表的名称(如果不是 public 命名空间,则需要添加命名空间);
* columnName 表示设置主键的列的名称;
* seq 就是默认设置的,一般不需要修改。

所以,参数不是一个确定的字符串:表名不确定、主键列名不确定:

* 对于表名,我们在用 M() 或者 D() 方法的时候,一般都会传入表名,那么可以考虑在 Model 类中进行拼接参数;
* 对于主键列名,如果做一个约定:数据表中,主键列名均为`id`,那我们就可以绕过去这个问题了。如果不能遵守这个约定,那就只能考虑在 Model 类中增加一个方法来传入对应的参数了。

现在来实现上述的方法,找到并修改 Model 类中的`getLastInsID()`方法:# 位置:ThinkPHP/Library/Think/Model.class.php
/**
* 返回最后插入的ID
* @access public
* @return string
*/
public function getLastInsID() {
$col = null;
if (strtolower(C('DB_TYPE')) == 'pgsql')
$col = strtolower($this->trueTableName).'_id_seq';

return $this->db->getLastInsID($col);
}
这里使用`$this->trueTableName`来获取表名,就可以获取到包含前缀,甚至数据库名的真实表名了,能避免出错。

这样修改之后,基本上就能正常了。但是很快就会发现,有时候还是会出错,提示`relation xx does not exist`。一般这是在访问没有设置主键的表的时候出现的错误。因为使用 pgsql 驱动的`lastInsertId()`方法获取没有主键的表的时候,这个序列对象不存在,于是报错。

知道原因之后,就可以考虑使用`try...catch`语句来重写一下 pgsql 的`lastInsertId()`方法:# 位置:ThinkPHP/Library/Think/Db/Driver/Pgsql.class.php
/**
* 用于获取最后插入的ID
* @access public
* @return integer
*/
public function getLastInsID($sequenceName) {
try {
return $this->_linkID->lastInsertId($sequenceName);
} catch (\PDOException $e) {
return $this->lastInsID;
}
}
MD文档:https://github.com/Lin07ux/notes/blob/master/PHP/ThinkPHP/Thinkphp%20%E8%BF%9E%E6%8E%A5%20pgsql%20%E9%97%AE%E9%A2%98%E5%8F%8A%E8%A7%A3%E5%86%B3.md#add-之后无法返回正确的自增-id

AD:真正免费,域名+虚机+企业邮箱=0元