Ticket #696: SC_Query.patch

SC_Query.php

@@ -504 +504 @@
      * @param string $table_name テーブル名
      * @return integer
      */
-    function get_auto_increment($table_name){
-        // ロックする
-        $this->query("LOCK TABLES $table_name WRITE");
-
-        // 次のIncrementを取得
-        $arrRet = $this->getAll("SHOW TABLE STATUS LIKE ?", array($table_name));
-        $auto_inc_no = $arrRet[0]["Auto_increment"];
-
-        // 値をカウントアップしておく
-        $this->conn->query("ALTER TABLE $table_name AUTO_INCREMENT=?" , $auto_inc_no + 1);
-
-        // 解除する
-        $this->query('UNLOCK TABLES');
-
-        return $auto_inc_no;
+    function get_auto_increment($table_name,$old = true){
+        if($old){
+            // ロックする
+            $this->query("LOCK TABLES $table_name WRITE");
+            // 次のIncrementを取得
+            $arrRet = $this->getAll("SHOW TABLE STATUS LIKE ?", array($table_name));
+            $auto_inc_no = $arrRet[0]["Auto_increment"];
+            // 値をカウントアップしておく
+            $this->conn->query("ALTER TABLE $table_name AUTO_INCREMENT=?" , $auto_inc_no + 1);
+            // 解除する
+            $this->query('UNLOCK TABLES');
+            return $auto_inc_no;
+        }else{
+            // ロックする
+            $this->query("LOCK TABLES $table_name WRITE");
+            // 次のIncrementを取得
+            $arrRet = $this->getAll("SHOW TABLE STATUS LIKE ?", array($table_name));
+            $auto_inc_no = $arrRet[0]["Auto_increment"];
+            $next_ain = $auto_inc_no +1;
+            $arrNotNull = $this->getAll("SHOW COLUMNS FROM {$table_name} WHERE `NULL` = 'NO' AND `Default` IS NULL");
+            $data = array();
+            $pkfield = "";
+            foreach ( $arrNotNull as $field ){
+                $value = null;
+    
+                switch(true){
+                    case ($field['Key'] == "PRI" && preg_match('/auto_increment/',$field['Extra']) ):
+                        //primary key & auto_increment
+                        $value = (string)$auto_inc_no;
+                        $pkfield = $field['Field'];
+                        break;
+                    case preg_match('/^[datetime|date|time]/',$field['Type']):
+                        $value = date("Y-m-d H:i:s");
+                        break;
+                    default :
+                        $value = "0";
+                }
+                $data[$field['Field']]    = $value;
+            }
+            if($pkfield != ""){
+                // 値をカウントアップさせる
+                $this->insert($table_name,$data);
+                $this->delete($table_name,$pkfield." = ?",array($auto_inc_no));
+            }else{
+                // 値をカウントアップしておく
+                $this->conn->query("ALTER TABLE $table_name AUTO_INCREMENT=?" , $auto_inc_no + 1);
+            }
+    
+            // 解除する
+            $this->query('UNLOCK TABLES');
+    
+            return $auto_inc_no;
+          }
     }
 }
-
-?>