-
Notifications
You must be signed in to change notification settings - Fork 9
Expand file tree
/
Copy pathModelCustomizer.cs
More file actions
100 lines (88 loc) · 4.36 KB
/
ModelCustomizer.cs
File metadata and controls
100 lines (88 loc) · 4.36 KB
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
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
#if EFCORE
using Microsoft.EntityFrameworkCore;
using Microsoft.EntityFrameworkCore.Infrastructure;
using Microsoft.EntityFrameworkCore.Metadata.Builders;
using System;
using System.Linq;
namespace BlazarTech.QueryableValues
{
internal sealed class ModelCustomizer<TPreviousModelCustomizer> : IModelCustomizer
where TPreviousModelCustomizer : IModelCustomizer
{
private readonly TPreviousModelCustomizer _previousModelCustomizer;
public ModelCustomizer(TPreviousModelCustomizer previousModelCustomizer)
{
_previousModelCustomizer = previousModelCustomizer;
}
/// <summary>
/// Used to satisfice model validation and remove the SqlServerEventId.DecimalTypeDefaultWarning warning from the logs.
/// </summary>
private static void SetDefaultPrecision<T>(PropertyBuilder<T> property)
{
#if EFCORE3
property.HasColumnType("decimal(18, 6)");
#else
property.HasPrecision(18, 6);
#endif
}
private static void SetupEntity<TEntity>(ModelBuilder modelBuilder, Action<EntityTypeBuilder<TEntity>>? buildAction = null)
where TEntity : QueryableValuesEntity
{
modelBuilder
.Entity<TEntity>(entity =>
{
buildAction?.Invoke(entity);
// By mapping to a fake view, we stop EF from including these entities during
// SQL generation in migrations and by the Create and Drop apis in DbContext.Database.
entity
.ToView(GetViewName(typeof(TEntity)))
.HasKey(i => i.X);
static string GetViewName(Type type)
{
if (type.IsGenericType)
{
var typeName = type.FullName?.Substring(0, type.FullName.IndexOf('`')) ?? FallbackName(type);
var args = string.Join(".", type.GetGenericArguments().Select(t => t.Name));
return $"{typeName}.{args}";
}
return type.FullName ?? FallbackName(type);
static string FallbackName(Type type) => $"BlazarTech.QueryableValues.{type.Name}";
}
});
}
public void Customize(ModelBuilder modelBuilder, DbContext context)
{
SetupEntity<ComplexQueryableValuesEntity>(modelBuilder, entity =>
{
SetDefaultPrecision(entity.Property(p => p.M));
SetDefaultPrecision(entity.Property(p => p.M1));
SetDefaultPrecision(entity.Property(p => p.M2));
SetDefaultPrecision(entity.Property(p => p.M3));
SetDefaultPrecision(entity.Property(p => p.M4));
SetDefaultPrecision(entity.Property(p => p.M5));
SetDefaultPrecision(entity.Property(p => p.M6));
SetDefaultPrecision(entity.Property(p => p.M7));
SetDefaultPrecision(entity.Property(p => p.M8));
SetDefaultPrecision(entity.Property(p => p.M9));
});
SetupEntity<SimpleQueryableValuesEntity<byte>>(modelBuilder);
SetupEntity<SimpleQueryableValuesEntity<short>>(modelBuilder);
SetupEntity<SimpleQueryableValuesEntity<int>>(modelBuilder);
SetupEntity<SimpleQueryableValuesEntity<long>>(modelBuilder);
SetupEntity<SimpleQueryableValuesEntity<decimal>>(modelBuilder, entity => SetDefaultPrecision(entity.Property(p => p.V)));
SetupEntity<SimpleQueryableValuesEntity<float>>(modelBuilder);
SetupEntity<SimpleQueryableValuesEntity<double>>(modelBuilder);
SetupEntity<SimpleQueryableValuesEntity<DateTime>>(modelBuilder);
SetupEntity<SimpleQueryableValuesEntity<DateTimeOffset>>(modelBuilder);
SetupEntity<SimpleQueryableValuesEntity<Guid>>(modelBuilder);
SetupEntity<SimpleQueryableValuesEntity<char>>(modelBuilder);
SetupEntity<SimpleQueryableValuesEntity<string>>(modelBuilder);
#if EFCORE8
SetupEntity<SimpleQueryableValuesEntity<DateOnly>>(modelBuilder);
SetupEntity<SimpleQueryableValuesEntity<TimeOnly>>(modelBuilder);
#endif
_previousModelCustomizer.Customize(modelBuilder, context);
}
}
}
#endif