fix(lib/yukari.js): add another condition branch with no insertId whe…

…n insert
XadillaX · Mar 9, 2016 · f82f624921eac3a13bffbf5a2dcd562196299163 · f82f624
1 parent 8e26822
commit f82f624921eac3a13bffbf5a2dcd562196299163
Unified Split

Showing with 7 additions and 2 deletions.

+7 −2 lib/yukari.js
diff --git a/lib/yukari.js b/lib/yukari.js
@@ -318,8 +318,8 @@ Yukari.prototype.insert = function(callback) {
            } else {
                // 有多个主键
-  
-                for(var i = 0; i < self.$model.primaryKeys.length; i++) {
-  
-                    where[pk[0].name] = (self[pk[i].name] !== undefined) ?
+  
+                for(var i = 0; i < pk.length; i++) {
+  
+                    where[pk[i].name] = (self[pk[i].name] !== undefined) ?
                        self[pk[i].name] : self[pk[i].column];
                }
@@ -328,6 +328,11 @@ Yukari.prototype.insert = function(callback) {
                    self[ai.name] = row.insertId;
                }
            }
+  
+        } else {
+  
+            for(var i = 0; i < pk.length; i++) {
+  
+                where[pk[i].name] = (self[pk[i].name] !== undefined) ?
+  
+                    self[pk[i].name] : self[pk[i].column];
+  
+            }
        }
        self.$model.where(where).findOne(function(err, row) {