From e8fa48a257ef304070207eb64d3e87fc6fa47fc8 Mon Sep 17 00:00:00 2001
From: Jakub Vrana <jakub@vrana.cz>
Date: Wed, 26 Feb 2025 22:11:41 +0100
Subject: [PATCH] MS SQL: Set identity_insert in CSV import

---
 adminer/drivers/mssql.inc.php | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/adminer/drivers/mssql.inc.php b/adminer/drivers/mssql.inc.php
index 1d170cf6..d559b4ef 100644
--- a/adminer/drivers/mssql.inc.php
+++ b/adminer/drivers/mssql.inc.php
@@ -187,15 +187,14 @@ if (isset($_GET["mssql"])) {
 
 		function insertUpdate($table, $rows, $primary) {
 			$fields = fields($table);
+			queries("SET IDENTITY_INSERT " . table($table) . " ON");
 			foreach ($rows as $set) {
 				$update = array();
-				$insert = array();
 				$where = array();
 				foreach ($set as $key => $val) {
 					$name = idf_unescape($key);
 					if (!$fields[$name]["auto_increment"]) {
 						$update[] = "$key = $val";
-						$insert[$key] = $val;
 					}
 					if (isset($primary[$name])) {
 						$where[] = "$key = $val";
@@ -204,11 +203,12 @@ if (isset($_GET["mssql"])) {
 				//! can use only one query for all rows
 				if (!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($insert)) . ") VALUES (" . implode(", ", $insert) . ");" // ; is mandatory
+					. " WHEN NOT MATCHED THEN INSERT (" . implode(", ", array_keys($set)) . ") VALUES (" . implode(", ", $set) . ");" // ; is mandatory
 				)) {
 					return false;
 				}
 			}
+			queries("SET IDENTITY_INSERT " . table($table) . " OFF");
 			return true;
 		}