-
Notifications
You must be signed in to change notification settings - Fork 3.2k
/
CustomPartitionKeyIdGenerator.cs
85 lines (68 loc) · 2.5 KB
/
CustomPartitionKeyIdGenerator.cs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
using System.Collections;
namespace Microsoft.EntityFrameworkCore.TestUtilities;
#nullable disable
public class CustomPartitionKeyIdGenerator<T> : ValueGenerator<T>
{
public override bool GeneratesTemporaryValues
=> false;
public override T Next(EntityEntry entry)
=> (T)NextValue(entry);
protected override object NextValue(EntityEntry entry)
{
var builder = new StringBuilder();
var entityType = entry.Metadata;
var primaryKey = entityType.FindPrimaryKey();
var discriminator = entityType.GetDiscriminatorValue();
if (discriminator != null
&& !primaryKey.Properties.Contains(entityType.FindDiscriminatorProperty()))
{
AppendString(builder, discriminator);
builder.Append("-");
}
var partitionKey = entityType.GetPartitionKeyPropertyName();
foreach (var property in primaryKey.Properties)
{
if (property.Name == partitionKey
|| property.GetJsonPropertyName() == StoreKeyConvention.IdPropertyJsonName)
{
continue;
}
var value = entry.Property(property).CurrentValue;
var converter = property.GetTypeMapping().Converter;
if (converter != null)
{
value = converter.ConvertToProvider(value);
}
// We don't allow the Id to be zero for our custom generator.
if (value is 0)
{
return default;
}
AppendString(builder, value);
builder.Append("-");
}
builder.Remove(builder.Length - 1, 1);
return builder.ToString();
}
private static void AppendString(StringBuilder builder, object propertyValue)
{
switch (propertyValue)
{
case string stringValue:
builder.Append(stringValue.Replace("-", "/-"));
return;
case IEnumerable enumerable:
foreach (var item in enumerable)
{
builder.Append(item.ToString().Replace("-", "/-"));
builder.Append("|");
}
return;
default:
builder.Append(propertyValue == null ? "null" : propertyValue.ToString().Replace("-", "/-"));
return;
}
}
}