FreeSql/FreeSql.Extensions.EFCoreModelBuilder/CodeFirstExtensions.cs

using Microsoft.EntityFrameworkCore;
using System;
using System.Linq;

namespace FreeSql.Extensions.EFCoreModelBuilder {

	public static class CodeFirstExtensions {

		public static void ConfigEntity(this ICodeFirst codeFirst, ModelBuilder modelBuilder) {

			foreach (var type in modelBuilder.Model.GetEntityTypes()) {

				codeFirst.ConfigEntity(type.ClrType, a => {

					//表名
					var relationalTableName = type.FindAnnotation("Relational:TableName");
					if (relationalTableName != null) {
						a.Name(relationalTableName.Value?.ToString() ?? type.ClrType.Name);
					}

					foreach (var prop in type.GetProperties()) {

						var freeProp = a.Property(prop.Name);

						//列名
						var relationalColumnName = prop.FindAnnotation("Relational:ColumnName");
						if (relationalColumnName != null) {

							freeProp.Name(relationalColumnName.Value?.ToString() ?? prop.Name);
						}

						//主键
						freeProp.IsPrimary(prop.IsPrimaryKey());

						//自增
						freeProp.IsIdentity(prop.GetAnnotations().Where(z => 
							z.Name == "SqlServer:ValueGenerationStrategy" && z.Value.ToString().Contains("IdentityColumn") //sqlserver 自增
							|| z.Value.ToString().Contains("IdentityColumn") //其他数据库实现未经测试
						).Any());

						//可空
						freeProp.IsNullable(prop.AfterSaveBehavior != Microsoft.EntityFrameworkCore.Metadata.PropertySaveBehavior.Throw);

						//类型
						var relationalColumnType = prop.FindAnnotation("Relational:ColumnType");
						if (relationalColumnType != null) {

							var dbType = relationalColumnType.ToString();

							if (!string.IsNullOrEmpty(dbType)) {

								var maxLength = prop.FindAnnotation("MaxLength");
								if (maxLength != null)
									dbType += $"({maxLength})";

								freeProp.DbType(dbType);
							}
						}
					}
				});
			}
		}
	}
}