CSV import for MS SQL

vrana · Aug 3, 2010 · 52e8c03 · 52e8c03
1 parent dba6a90
commit 52e8c03
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 1 deletion.
diff --git a/adminer/drivers/mssql.inc.php b/adminer/drivers/mssql.inc.php
@@ -476,6 +476,22 @@ function insert_into($table, $set) {
 		return queries("INSERT INTO " . table($table) . ($set ? " (" . implode(", ", array_keys($set)) . ")\nVALUES (" . implode(", ", $set) . ")" : "DEFAULT VALUES"));
 	}
 
+	function insert_update($table, $set, $primary) {
+		$update = array();
+		$where = array();
+		foreach ($set as $key => $val) {
+			$update[] = "$key = $val";
+			if (isset($primary[idf_unescape($key)])) {
+				$where[] = "$key = $val";
+			}
+		}
+		// can use only one query for all rows with different API
+		return queries("MERGE " . table($table) . " USING (VALUES(" . implode(", ", $set) . ")) AS source (c" . implode(", c", range(1, count($set))) . ") ON " . implode(" AND ", $where) //! source, c1 - possible conflict
+			. " WHEN MATCHED THEN UPDATE SET " . implode(", ", $update)
+			. " WHEN NOT MATCHED THEN INSERT (" . implode(", ", array_keys($set)) . ") VALUES (" . implode(", ", $set) . ");" // ; is mandatory
+		);
+	}
+
 	function last_id() {
 		global $connection;
 		return $connection->result("SELECT SCOPE_IDENTITY()"); // @@IDENTITY can return trigger INSERT

diff --git a/adminer/drivers/pgsql.inc.php b/adminer/drivers/pgsql.inc.php
@@ -473,7 +473,7 @@ function insert_update($table, $set, $primary) {
 		$where = array();
 		foreach ($set as $key => $val) {
 			$update[] = "$key = $val";
-			if (isset($primary[$key])) {
+			if (isset($primary[idf_unescape($key)])) {
 				$where[] = "$key = $val";
 			}
 		}