CodeIgniter2.1 数据库添加支持postgres包含RETURNING查询的取值

mark35

postgreSQL支持 INSERT|UPDATE|DELETE查询时返回指定栏目值，比如

1. 
   
2. INSERT INTO user (id, name) VALUES (DEFAULT, 'new user') RETURNING id;
   
3. INSERT INTO user (id, name) VALUES (DEFAULT, 'new user') RETURNING *;
   

复制代码

对以上任意一条SQL查询的结果对象进行取值就可以获得需要的值，

1. 
   
2. $last_inset_id = $this->db->query($sql)->row()->id;
   
3. or
   
4. $res = $this->db->query($sql)->row_array();
   
5. $last_insert_id = $res ? $res['id'] ? 0;
   

复制代码

但CI2默认数据库驱动如果判断是写入的语句那么返回的是bool值而不是db结果集对象。修改如下：
system/databaseDB_driver.php
添加函数

1. 
   
2.     // --------------------------------------------------------------------
   
3.    
   
4.     /**
   
5.      * Determines if a query is a "write" type with returning colums values (postgres)
   
6.      * @author    [email]waiting@xiaozhong.biz[/email]
   
7.      *
   
8.      * @access    public
   
9.      * @param    string    An SQL query string
   
10.      * @return    boolean
    
11.      */
    
12.     function is_write_return_type($sql)
    
13.     {
    
14.         if ( ! preg_match('/^\s*"?(INSERT|UPDATE|DELETE)\s+?/i', $sql))
    
15.         {
    
16.             return FALSE;
    
17.         }
    
18.         return (bool) preg_match('/\sRETURNING\b.+?/i', $sql);
    
19.     }
    

复制代码

然后此文件中约351行处代码找到

1. 
   
2.         if ($this->is_write_type($sql) === TRUE)
   
3.         {
   
4.             // If caching is enabled we'll auto-cleanup any
   
5.             // existing files related to this particular URI
   
6.             if ($this->cache_on == TRUE AND $this->cache_autodel == TRUE AND $this->_cache_init())
   
7.             {
   
8.                 $this->CACHE->delete();
   
9.             }
   
10.             return TRUE;
    
11.         }
    

复制代码

修改为

1. 
   
2.         if ($this->is_write_type($sql) === TRUE)
   
3.         {
   
4.             // If caching is enabled we'll auto-cleanup any
   
5.             // existing files related to this particular URI
   
6.             if ($this->cache_on == TRUE AND $this->cache_autodel == TRUE AND $this->_cache_init())
   
7.             {
   
8.                 $this->CACHE->delete();
   
9.             }
   
10. 
    
11.             // waiting ADD
    
12.             if ( ! $this->is_write_return_type($sql))
    
13.             {
    
14.                 return TRUE;
    
15.             } // ADD END
    
16.         }
    
17. 
    

复制代码

注：通过 RETURNING PK 方式返回值这才是正确的、严谨的获得 last insert id的方法。CI定义的 $this-db->insert_id() 方法是根据垃圾mysql的方式处理得来的


首发 http://xiaozhong.biz/thread-312-1-1.html