Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

fix SqlHierarchyId serialization and deserialization #55

Merged
merged 2 commits into from
Dec 11, 2020
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
34 changes: 34 additions & 0 deletions src/Microsoft.SqlServer.Types.Tests/DatabaseUtil.cs
Original file line number Diff line number Diff line change
@@ -0,0 +1,34 @@
using System;
using System.Collections.Generic;
using System.IO;
using System.Text;

namespace Microsoft.SqlServer.Types.Tests
{
static class DatabaseUtil
{
internal static void CreateSqlDatabase(string filename)
{
string databaseName = System.IO.Path.GetFileNameWithoutExtension(filename);
if (File.Exists(filename))
File.Delete(filename);
if (File.Exists(filename.Replace(".mdf", "_log.ldf")))
File.Delete(filename.Replace(".mdf", "_log.ldf"));
using (var connection = new System.Data.SqlClient.SqlConnection(
@"Data Source=(localdb)\mssqllocaldb;Initial Catalog=master; Integrated Security=true;"))
{
connection.Open();
using (var command = connection.CreateCommand())
{
command.CommandText =
String.Format("CREATE DATABASE {0} ON PRIMARY (NAME={0}, FILENAME='{1}')", databaseName, filename);
command.ExecuteNonQuery();

command.CommandText =
String.Format("EXEC sp_detach_db '{0}', 'true'", databaseName);
command.ExecuteNonQuery();
}
}
}
}
}
26 changes: 2 additions & 24 deletions src/Microsoft.SqlServer.Types.Tests/Geometry/DBTests.cs
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 +33,7 @@ public static void Init()
if(path == null)
{
path = Path.Combine(new FileInfo(typeof(DBTests).Assembly.Location).Directory.FullName, "UnitTestData.mdf");
CreateSqlDatabase(path);
DatabaseUtil.CreateSqlDatabase(path);
using (var conn = new System.Data.SqlClient.SqlConnection(connstr + path))
{
conn.Open();
Expand All @@ -58,29 +58,7 @@ public DBTests()
conn.Open();
}

private static void CreateSqlDatabase(string filename)
{
string databaseName = System.IO.Path.GetFileNameWithoutExtension(filename);
if (File.Exists(filename))
File.Delete(filename);
if (File.Exists(filename.Replace(".mdf","_log.ldf")))
File.Delete(filename.Replace(".mdf", "_log.ldf"));
using (var connection = new System.Data.SqlClient.SqlConnection(
@"Data Source=(localdb)\mssqllocaldb;Initial Catalog=master; Integrated Security=true;"))
{
connection.Open();
using (var command = connection.CreateCommand())
{
command.CommandText =
String.Format("CREATE DATABASE {0} ON PRIMARY (NAME={0}, FILENAME='{1}')", databaseName, filename);
command.ExecuteNonQuery();

command.CommandText =
String.Format("EXEC sp_detach_db '{0}', 'true'", databaseName);
command.ExecuteNonQuery();
}
}
}

public void Dispose()
{
conn.Close();
Expand Down
Original file line number Diff line number Diff line change
@@ -1,7 +1,11 @@
using Microsoft.SqlServer.Types;
using Microsoft.SqlServer.Types.SqlHierarchy;
using Microsoft.VisualStudio.TestTools.UnitTesting;
using System;
using System.Collections;
using System.Data.SqlClient;
using System.IO;
using System.Text;

namespace Microsoft.SqlServer.Types.Tests.HierarchyId
{
Expand All @@ -11,7 +15,7 @@ namespace Microsoft.SqlServer.Types.Tests.HierarchyId
[TestClass]
[TestCategory("Deserialize")]
[TestCategory("SqlHierarchyId")]
public class DeserializeFromBinary
public class DeserializeFromBinaryTests
{

[TestMethod]
Expand Down Expand Up @@ -40,13 +44,13 @@ public void TestSqlHiarchy2()
// The first three fields are the same as in the first example.That is, the first two bits(01) are the L1 field, the second two bits(01) are the O1 field, and the fifth bit(1) is the F1 field.This encodes the / 1 / portion of the logical representation.
// The next 5 bits(00111) are the L2 field, so the next integer is between - 8 and - 1.The following 3 bits(111) are the O2 field, representing the offset 7 from the beginning of this range.Thus, the L2 and O2 fields together encode the integer - 1.The next bit(0) is the F2 field.Because it is 0(zero), this level is fake, and 1 has to be subtracted from the integer yielded by the L2 and O2 fields. Therefore, the L2, O2, and F2 fields together represent -2 in the logical representation of this node.
// The next 3 bits(110) are the L3 field, so the next integer is between 16 and 79.The subsequent 8 bits(00001010) are the L4 field. Removing the anti - ambiguity bits from there(the third bit(0) and the fifth bit(1)) leaves 000010, which is the binary representation of 2.Thus, the integer encoded by the L3 and O3 fields is 16 + 2, which is 18.The next bit(1) is the F3 field, representing the slash(/) after the 18 in the logical representation.The final 6 bits(000000) are the W field, padding the physical representation to the nearest byte.
byte[] bytes = { 0x59,0xFB,0x05,0x40 }; //01011001 11111011 00000101 01000000
byte[] bytes = { 0x59, 0xFB, 0x05, 0x40 }; //01011001 11111011 00000101 01000000
var hid = new Microsoft.SqlServer.Types.SqlHierarchyId();
using (var r = new BinaryReader(new MemoryStream(bytes)))
{
hid.Read(r);
}
Assert.AreEqual("/1/-2.18/", hid.ToString());
}
}
}
}
189 changes: 189 additions & 0 deletions src/Microsoft.SqlServer.Types.Tests/HierarchyId/HierarchyDbTests.cs
Original file line number Diff line number Diff line change
@@ -0,0 +1,189 @@
using Microsoft.SqlServer.Types.SqlHierarchy;
using Microsoft.VisualStudio.TestTools.UnitTesting;
using System;
using System.Collections.Generic;
using System.Data.SqlClient;
using System.IO;
using System.Text;

namespace Microsoft.SqlServer.Types.Tests.HierarchyId
{
[TestCategory("Database")]
[TestCategory("SqlHierarchyId")]
[TestClass]
public class HierarchyDbTests : IDisposable
{
const string DataSource = @"Data Source=(localdb)\mssqllocaldb;Integrated Security=True;AttachDbFileName=";


public static string DropTables = @"IF OBJECT_ID('dbo.TreeNode', 'U') IS NOT NULL DROP TABLE TreeNode;";

public static string CreateTables = @"
-- hierarchy_tests
CREATE TABLE [dbo].[TreeNode]([Id] [int] IDENTITY(1,1) NOT NULL, [Route] [hierarchyid] NOT NULL);
";

private readonly SqlConnection _connection;
private static string _path = null!;
private static readonly object lockObj = new object();
static HierarchyDbTests()
dotMorten marked this conversation as resolved.
Show resolved Hide resolved
{
Init();
}

private static void Init()
{
lock (lockObj)
{
if (_path == null)
{
_path = Path.Combine(new FileInfo(typeof(HierarchyDbTests).Assembly.Location).Directory.FullName, "HierarchyUnitTestData.mdf");
DatabaseUtil.CreateSqlDatabase(_path);
using (var conn = new SqlConnection(DataSource + _path))
{
conn.Open();
var cmd = conn.CreateCommand();
cmd.CommandText = DropTables;
cmd.ExecuteNonQuery();
dotMorten marked this conversation as resolved.
Show resolved Hide resolved
cmd.CommandText = CreateTables;
cmd.ExecuteNonQuery();
conn.Close();
}
}
}
}

private static string ConnectionString => DataSource + _path;

public HierarchyDbTests()
{
Init();
dotMorten marked this conversation as resolved.
Show resolved Hide resolved
_connection = new SqlConnection(ConnectionString);
_connection.Open();
dotMorten marked this conversation as resolved.
Show resolved Hide resolved
}

public void Dispose()
{
_connection.Close();
_connection.Dispose();
}
dotMorten marked this conversation as resolved.
Show resolved Hide resolved




[DataTestMethod]
[DataRow("/-4294971464/")]
[DataRow("/4294972495/")]
[DataRow("/3.2725686107/")]
[DataRow("/0/")]
[DataRow("/1/")]
[DataRow("/1.0.2/")]
[DataRow("/1.1.2/")]
[DataRow("/1.2.2/")]
[DataRow("/1.3.2/")]
[DataRow("/3.0/")]
public void SerializeDeserialize(string route)
{
var parsed = SqlHierarchyId.Parse(route);
var ms = new MemoryStream();
parsed.Write(new BinaryWriter(ms));
ms.Position = 0;
var dumMem = Dump(ms);
ms.Position = 0;
var roundTrip = new Microsoft.SqlServer.Types.SqlHierarchyId();
roundTrip.Read(new BinaryReader(ms));
if (parsed != roundTrip)
Assert.AreEqual(parsed, roundTrip); //breakpoint here

var id = new SqlCommand($"INSERT INTO [dbo].[TreeNode] (Route) output INSERTED.ID VALUES ('{route}') ", _connection).ExecuteScalar();

using (var reader = new SqlCommand($"SELECT Route FROM [dbo].[TreeNode] WHERE ID = " + id, _connection).ExecuteReader())
{
while (reader.Read())
{
var sqlRoundTrip = new Microsoft.SqlServer.Types.SqlHierarchyId();
var dumSql = Dump(reader.GetStream(0));
Assert.AreEqual(dumMem, dumSql);
sqlRoundTrip.Read(new BinaryReader(reader.GetStream(0)));
if (parsed != sqlRoundTrip)
Assert.AreEqual(parsed, sqlRoundTrip); //breakpoint here
}
}
}

[DataTestMethod]
[DynamicData(nameof(GetData), DynamicDataSourceType.Method)]
public void SerializeDeserializeRandom(string route)
{
SerializeDeserialize(route);
}

private const int CountOfGeneratedCases = 1000;
public static IEnumerable<object[]> GetData()
{
Random r = new Random();
for (var i = 0; i < CountOfGeneratedCases; i++)
{
yield return new object[] { RandomHierarhyId(r)};
}
}

public static string RandomHierarhyId(Random random)
{
StringBuilder sb = new StringBuilder();
sb.Append("/");
var levels = random.Next(4);
for (int i = 0; i < levels; i++)
{
var subLevels = random.Next(1, 4);
for (int j = 0; j < subLevels; j++)
{
var pattern = KnownPatterns.RandomPattern(random);
sb.Append(random.NextLong(pattern.MinValue, pattern.MaxValue + 1).ToString());
if (j < subLevels - 1)
sb.Append(".");
}
sb.Append("/");
}

return sb.ToString();
}

static string Dump(Stream ms)
{
return new BitReader(new BinaryReader(ms)).ToString();
}
}

public static class RandomExtensionMethods
{
/// <summary>
/// Returns a random long from min (inclusive) to max (exclusive)
/// </summary>
/// <param name="random">The given random instance</param>
/// <param name="min">The inclusive minimum bound</param>
/// <param name="max">The exclusive maximum bound. Must be greater than min</param>
public static long NextLong(this Random random, long min, long max)
{
if (max <= min)
throw new ArgumentOutOfRangeException("max", "max must be > min!");

//Working with ulong so that modulo works correctly with values > long.MaxValue
ulong uRange = (ulong)(max - min);

//Prevent a modolo bias; see https://stackoverflow.com/a/10984975/238419
//for more information.
//In the worst case, the expected number of calls is 2 (though usually it's
//much closer to 1) so this loop doesn't really hurt performance at all.
ulong ulongRand;
do
{
byte[] buf = new byte[8];
random.NextBytes(buf);
ulongRand = (ulong)BitConverter.ToInt64(buf, 0);
} while (ulongRand > ulong.MaxValue - ((ulong.MaxValue % uRange) + 1) % uRange);

return (long)(ulongRand % uRange) + min;
}
}
}
Original file line number Diff line number Diff line change
@@ -1,14 +1,17 @@
<Project Sdk="MSBuild.Sdk.Extras/2.0.54">

<PropertyGroup>
<TargetFrameworks>netcoreapp2.1;net461</TargetFrameworks>
<IsPackable>false</IsPackable>
<nullable>enable</nullable>
<LangVersion>8.0</LangVersion>
</PropertyGroup>

<ItemGroup>
<Compile Include="..\Microsoft.SqlServer.Types\SqlHierarchy\BitPattern.cs" Link="HierarchyId\BitPattern.cs" />
<Compile Include="..\Microsoft.SqlServer.Types\SqlHierarchy\BitReader.cs" Link="HierarchyId\BitReader.cs" />
<Compile Include="..\Microsoft.SqlServer.Types\SqlHierarchy\BitWriter.cs" Link="HierarchyId\BitWriter.cs" />
<Compile Include="..\Microsoft.SqlServer.Types\SqlHierarchy\KnownPatterns.cs" Link="HierarchyId\KnownPatterns.cs" />
</ItemGroup>

<ItemGroup>
Expand Down
9 changes: 3 additions & 6 deletions src/Microsoft.SqlServer.Types/SqlHierarchy/BitPattern.cs
Original file line number Diff line number Diff line change
Expand Up @@ -46,15 +46,12 @@ internal bool ContainsValue(long value)

public override string ToString() => Pattern;

public ulong EncodeValue(long val, bool isLast)
public ulong EncodeValue(long val)
{
ulong expand = Expand(PatternMask, val - MinValue);

ulong value = PatternOnes | expand | 1;

if (!isLast)
value++;

return value;
}

Expand All @@ -69,12 +66,12 @@ private ulong Expand(ulong mask, long value)
return Expand(mask >> 1, value) << 1;
}

internal int Decode(ulong encodedValue, out bool isLast)
internal long Decode(ulong encodedValue, out bool isLast)
{
var decodedValue = Compress(encodedValue, PatternMask);

isLast = (encodedValue & 0x1) == 0x1;
return (int)((isLast ? decodedValue : decodedValue - 1) + MinValue);
return ((isLast ? decodedValue : decodedValue - 1) + MinValue);
}

private long Compress(ulong value, ulong mask)
Expand Down
8 changes: 8 additions & 0 deletions src/Microsoft.SqlServer.Types/SqlHierarchy/KnownPatterns.cs
Original file line number Diff line number Diff line change
Expand Up @@ -27,6 +27,14 @@ internal static class KnownPatterns
new BitPattern(-4294971464, -4169, "000101xxxxxxxxxxxxxxxxxxx0xxxxxx0xxx0x1xxxT"),
};

internal static BitPattern RandomPattern(Random r)
{
var index = r.Next(PositivePatterns.Length + NegativePatterns.Length);

return index < PositivePatterns.Length ? PositivePatterns[index] :
NegativePatterns[index - PositivePatterns.Length];
}

internal static BitPattern GetPatternByValue(long value)
{
if (value >= 0)
Expand Down
Loading