130 lines
5.1 KiB
C#
130 lines
5.1 KiB
C#
using Microsoft.EntityFrameworkCore;
|
||
using Microsoft.EntityFrameworkCore.Metadata.Builders;
|
||
using System;
|
||
using System.Collections.Generic;
|
||
using System.Linq;
|
||
using System.Reflection;
|
||
|
||
namespace VOL.Entity.MappingConfiguration
|
||
{
|
||
|
||
public interface IEntityMappingConfiguration
|
||
{
|
||
void Map(ModelBuilder b);
|
||
}
|
||
|
||
public interface IEntityMappingConfiguration<T> : IEntityMappingConfiguration where T : class
|
||
{
|
||
void Map(EntityTypeBuilder<T> builder);
|
||
}
|
||
|
||
public abstract class EntityMappingConfiguration<T> : IEntityMappingConfiguration<T> where T : class
|
||
{
|
||
public abstract void Map(EntityTypeBuilder<T> b);
|
||
|
||
public void Map(ModelBuilder b)
|
||
{
|
||
Map(b.Entity<T>());
|
||
}
|
||
}
|
||
|
||
public static class ModelBuilderExtenions
|
||
{
|
||
private static IEnumerable<Type> GetMappingTypes(this Assembly assembly, Type mappingInterface)
|
||
{
|
||
return assembly.GetTypes().Where(x => !x.IsAbstract && x.GetInterfaces().Any(y => y.GetTypeInfo().IsGenericType && y.GetGenericTypeDefinition() == mappingInterface));
|
||
}
|
||
|
||
public static void AddEntityConfigurationsFromAssembly(this ModelBuilder modelBuilder, Assembly assembly)
|
||
{
|
||
var mappingTypes = assembly.GetMappingTypes(typeof(IEntityMappingConfiguration<>));
|
||
foreach (var config in mappingTypes.Select(Activator.CreateInstance).Cast<IEntityMappingConfiguration>())
|
||
{
|
||
config.Map(modelBuilder);
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
//字符串默认长度
|
||
//public class StringDefaultLengthConvention : IPropertyAddedConvention
|
||
//{
|
||
// public InternalPropertyBuilder Apply(InternalPropertyBuilder propertyBuilder)
|
||
// {
|
||
// if (propertyBuilder.Metadata.ClrType == typeof(string))
|
||
// propertyBuilder.HasMaxLength(32, ConfigurationSource.Convention);
|
||
// return propertyBuilder;
|
||
// }
|
||
//}
|
||
////attribute方式设置decimal精度
|
||
//public class DecimalPrecisionAttributeConvention : PropertyAttributeConvention<DecimalPrecisionAttribute>
|
||
//{
|
||
// public override InternalPropertyBuilder Apply(InternalPropertyBuilder propertyBuilder, DecimalPrecisionAttribute attribute, MemberInfo clrMember)
|
||
// {
|
||
// if (propertyBuilder.Metadata.ClrType == typeof(decimal))
|
||
// propertyBuilder.HasPrecision(attribute.Precision, attribute.Scale);
|
||
// return propertyBuilder;
|
||
// }
|
||
|
||
// /// <summary>
|
||
// /// decimal类型设置精度
|
||
// /// </summary>
|
||
// /// <param name="propertyBuilder"></param>
|
||
// /// <param name="precision">精度</param>
|
||
// /// <param name="scale">小数位数</param>
|
||
// public static PropertyBuilder<TProperty> HasPrecision<TProperty>(this PropertyBuilder<TProperty> propertyBuilder, int precision = 18, int scale = 4)
|
||
// {
|
||
// //fluntapi方式设置精度
|
||
// ((IInfrastructure<InternalPropertyBuilder>)propertyBuilder).Instance.HasPrecision(precision, scale);
|
||
|
||
// return propertyBuilder;
|
||
// }
|
||
|
||
// public static InternalPropertyBuilder HasPrecision(this InternalPropertyBuilder propertyBuilder, int precision, int scale)
|
||
// {
|
||
// propertyBuilder.Relational(ConfigurationSource.Explicit).HasColumnType($"decimal({precision},{scale})");
|
||
|
||
// return propertyBuilder;
|
||
// }
|
||
|
||
////servie,DI注入替换.
|
||
//services.AddSingleton<IModelCustomizer, MyRelationalModelCustomizer>();
|
||
//services.AddSingleton<ICoreConventionSetBuilder, MyCoreConventionSetBuilder>();
|
||
|
||
////实现entity的自动发现和mapper设置
|
||
//public class MyRelationalModelCustomizer : RelationalModelCustomizer
|
||
//{
|
||
// public MyRelationalModelCustomizer(ModelCustomizerDependencies dependencies)
|
||
// : base(dependencies) { }
|
||
|
||
// public override void Customize(ModelBuilder modelBuilder, DbContext dbContext)
|
||
// {
|
||
// base.Customize(modelBuilder, dbContext);
|
||
// var sp = (IInfrastructure<IServiceProvider>)dbContext;
|
||
// var dbOptions = sp.Instance.GetServices<DbContextOptions>();
|
||
// foreach (var item in dbOptions)
|
||
// {
|
||
// if (item.ContextType == dbContext.GetType())
|
||
// ConfigureDbContextEntityService.Configure(modelBuilder, item, dbContext);
|
||
// }
|
||
// }
|
||
//}
|
||
|
||
//public class MyCoreConventionSetBuilder : CoreConventionSetBuilder
|
||
//{
|
||
// public MyCoreConventionSetBuilder(CoreConventionSetBuilderDependencies dependencies) : base(dependencies) { }
|
||
// public override ConventionSet CreateConventionSet()
|
||
// {
|
||
// var conventionSet = base.CreateConventionSet();
|
||
// //默认字符串长度,而不是nvarchar(max).
|
||
// //为什么要insert(0,obj),则是因为这个默认规则要最优先处理,如果后续有规则的话就直接覆盖了。
|
||
// //propertyBuilder.HasMaxLength(32, ConfigurationSource.Convention);
|
||
// //理论上我指定了规则的级别为.Convention,应该和顺序就没有关系了。but,还没有完成测试,所以我也不知道
|
||
// conventionSet.PropertyAddedConventions.Insert(0, new StringDefaultLengthConvention());
|
||
// //decimal设置精度
|
||
// conventionSet.PropertyAddedConventions.Add(new DecimalPrecisionAttributeConvention());
|
||
// return conventionSet;
|
||
// }
|
||
//}
|
||
|