ckaczor/sqltoolsservice
1
0
Fork 0
mirror of https://github.com/ckaczor/sqltoolsservice.git synced 2026-02-16 10:58:30 -05:00
Code Issues Packages Projects Releases Wiki Activity

fixed the bug with editing more than one row should fail (#613)

Browse Source 
* fixed the bug with editing more than one row should fail
This commit is contained in:
Leila Lali
2018-05-02 13:51:07 -07:00
committed by GitHub
parent c8263d8330
commit 5baa3bc18a
2 changed files with 22 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
src/Microsoft.SqlTools.ServiceLayer/EditData/UpdateManagement/RowUpdate.cs
View File

@@ -3,11 +3,13 @@
// Licensed under the MIT license. See LICENSE file in the project root for full license information. // Licensed under the MIT license. See LICENSE file in the project root for full license information.
// //
using System;
using System.Collections.Concurrent; using System.Collections.Concurrent;
using System.Collections.Generic; using System.Collections.Generic;
using System.Data; using System.Data;
using System.Data.Common; using System.Data.Common;
using System.Data.SqlClient; using System.Data.SqlClient;
using System.Globalization;
using System.Linq; using System.Linq;
using System.Text; using System.Text;
using System.Threading.Tasks; using System.Threading.Tasks;
@@ -30,6 +32,14 @@ namespace Microsoft.SqlTools.ServiceLayer.EditData.UpdateManagement
private const string UpdateScript = "UPDATE {0} SET {1} {2}"; private const string UpdateScript = "UPDATE {0} SET {1} {2}";
private const string UpdateScriptMemOptimized = "UPDATE {0} WITH (SNAPSHOT) SET {1} {2}"; private const string UpdateScriptMemOptimized = "UPDATE {0} WITH (SNAPSHOT) SET {1} {2}";
private const string SelectStatement = "SELECT {0} FROM {1}"; private const string SelectStatement = "SELECT {0} FROM {1}";
private string validateUpdateOnlyOneRow = "DECLARE @numberOfRows int = 0;" + Environment.NewLine +
"Select @numberOfRows = count(*) FROM {0} {1} " + Environment.NewLine +
"IF (@numberOfRows > 1) " + Environment.NewLine +
"Begin" + Environment.NewLine +
" DECLARE @error NVARCHAR(100) = N'The row value(s) updated do not make the row unique or they alter multiple rows(' + CAST(@numberOfRows as varchar(10)) + ' rows)';" + Environment.NewLine +
" RAISERROR (@error, 16, 1) " + Environment.NewLine +
"End" + Environment.NewLine +
"ELSE BEGIN" + Environment.NewLine;
internal readonly ConcurrentDictionary<int, CellUpdate> cellUpdates; internal readonly ConcurrentDictionary<int, CellUpdate> cellUpdates;
private readonly IList<DbCellValue> associatedRow; private readonly IList<DbCellValue> associatedRow;
@@ -121,14 +131,21 @@ namespace Microsoft.SqlTools.ServiceLayer.EditData.UpdateManagement
tempTableName, tempTableName,
whereClause.CommandText); whereClause.CommandText);
string validateScript = string.Format(CultureInfo.InvariantCulture, validateUpdateOnlyOneRow,
AssociatedObjectMetadata.EscapedMultipartName,
whereClause.CommandText);
// Step 3) Build the select statement // Step 3) Build the select statement
string selectStatement = string.Format(SelectStatement, string.Join(", ", selectColumns), tempTableName); string selectStatement = string.Format(SelectStatement, string.Join(", ", selectColumns), tempTableName);
// Step 4) Put it all together into a results object // Step 4) Put it all together into a results object
StringBuilder query = new StringBuilder(); StringBuilder query = new StringBuilder();
query.AppendLine(declareStatement); query.AppendLine(declareStatement);
query.AppendLine(validateScript);
query.AppendLine(updateStatement); query.AppendLine(updateStatement);
query.Append(selectStatement); query.AppendLine(selectStatement);
query.Append("END");
// Build the command // Build the command
DbCommand command = connection.CreateCommand(); DbCommand command = connection.CreateCommand();

6
test/Microsoft.SqlTools.ServiceLayer.UnitTests/EditData/RowUpdateTests.cs
View File

@@ -271,7 +271,7 @@ namespace Microsoft.SqlTools.ServiceLayer.UnitTests.EditData
// ... Validate the command's makeup // ... Validate the command's makeup
// Break the query into parts // Break the query into parts
string[] splitSql = cmd.CommandText.Split(Environment.NewLine); string[] splitSql = cmd.CommandText.Split(Environment.NewLine);
Assert.Equal(3, splitSql.Length); Assert.True(splitSql.Length >= 3);
// Check the declare statement first // Check the declare statement first
Regex declareRegex = new Regex(@"^DECLARE @(.+) TABLE \((.+)\)$"); Regex declareRegex = new Regex(@"^DECLARE @(.+) TABLE \((.+)\)$");
@@ -291,7 +291,7 @@ namespace Microsoft.SqlTools.ServiceLayer.UnitTests.EditData
? @"^UPDATE (.+) WITH \(SNAPSHOT\) SET (.+) OUTPUT (.+) INTO @(.+) WHERE .+$" ? @"^UPDATE (.+) WITH \(SNAPSHOT\) SET (.+) OUTPUT (.+) INTO @(.+) WHERE .+$"
: @"^UPDATE (.+) SET (.+) OUTPUT (.+) INTO @(.+) WHERE .+$"; : @"^UPDATE (.+) SET (.+) OUTPUT (.+) INTO @(.+) WHERE .+$";
Regex updateRegex = new Regex(regex); Regex updateRegex = new Regex(regex);
Match updateMatch = updateRegex.Match(splitSql[1]); Match updateMatch = updateRegex.Match(splitSql[10]);
Assert.True(updateMatch.Success); Assert.True(updateMatch.Success);
// Table name matches // Table name matches
@@ -313,7 +313,7 @@ namespace Microsoft.SqlTools.ServiceLayer.UnitTests.EditData
// Check the select statement last // Check the select statement last
Regex selectRegex = new Regex(@"^SELECT (.+) FROM @(.+)$"); Regex selectRegex = new Regex(@"^SELECT (.+) FROM @(.+)$");
Match selectMatch = selectRegex.Match(splitSql[2]); Match selectMatch = selectRegex.Match(splitSql[11]);
Assert.True(selectMatch.Success); Assert.True(selectMatch.Success);
// Correct number of columns in select statement // Correct number of columns in select statement