改进model类的updateFields属性处理

top-think · Feb 20, 2016 · 352a4e6 · 352a4e6
1 parent 9b5b4e0
commit 352a4e6
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/ThinkPHP/Library/Think/Model.class.php b/ThinkPHP/Library/Think/Model.class.php
@@ -1101,6 +1101,15 @@ public function create($data = '', $type = '')
             $fields = $this->insertFields;
         } elseif (self::MODEL_UPDATE == $type && isset($this->updateFields)) {
             $fields = $this->updateFields;
+            $pk     = $this->getPk();
+            if (is_string($pk)) {
+                array_push($fields, $pk);
+            }
+            if (is_array($pk)) {
+                foreach ($pk as $pkTemp) {
+                    array_push($fields, $pkTemp);
+                }
+            }
         }
         if (isset($fields)) {
             if (is_string($fields)) {