Merge pull request #16 from coco543/master

实现insert之后返回主键AI的ID
walkor · Jan 30, 2015 · 67c0f6b · 67c0f6b
2 parents 5ebfcbb + c212c76
commit 67c0f6b
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/applications/Demo/Lib/DbConnection.php b/applications/Demo/Lib/DbConnection.php
@@ -1672,9 +1672,14 @@ public function query($query = '',$params = null, $fetchmode = \PDO::FETCH_ASSOC
  if ($statement === 'select' || $statement === 'show') {
  return $this->sQuery->fetchAll($fetchmode);
  }
- elseif ( $statement === 'insert' || $statement === 'update' || $statement === 'delete' ) {
+ elseif ( $statement === 'update' || $statement === 'delete' ) {
  return $this->sQuery->rowCount();
  }
+ elseif( $statement === 'insert' ){
+ if( $this->sQuery->rowCount() > 0 ){
+ return $this->lastInsertId();
+ }
+ }
  else {
  return NULL;
  }