49 lines
1.6 KiB
C#
49 lines
1.6 KiB
C#
|
using System;
|
|||
|
|
using System.Linq;
|
|||
|
|
using System.Reflection;
|
|||
|
|
using Marr.Data.Mapping;
|
|||
|
|
|
|||
|
|
namespace NzbDrone.Core.Datastore
|
|||
|
|
{
|
|||
|
|
public static class MappingExtensions
|
|||
|
|
{
|
|||
|
|
public static ColumnMapBuilder<T> RegisterModel<T>(this FluentMappings.MappingsFluentEntity<T> mapBuilder, string tableName) where T : ModelBase
|
|||
|
|
{
|
|||
|
|
return mapBuilder.Table.MapTable(tableName)
|
|||
|
|
.Columns
|
|||
|
|
.AutoMapPropertiesWhere(IsMappableProperty)
|
|||
|
|
.For(c => c.Id)
|
|||
|
|
.SetPrimaryKey()
|
|||
|
|
.SetReturnValue()
|
|||
|
|
.SetAutoIncrement();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public static bool IsMappableProperty(MemberInfo memberInfo)
|
|||
|
|
{
|
|||
|
|
var propertyInfo = memberInfo as PropertyInfo;
|
|||
|
|
|
|||
|
|
if (propertyInfo == null) return false;
|
|||
|
|
|
|||
|
|
if (propertyInfo.PropertyType.GetInterfaces().Any(i => i == typeof(IEmbeddedDocument)))
|
|||
|
|
{
|
|||
|
|
return true;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return propertyInfo.CanRead && propertyInfo.CanWrite && IsSimpleType(propertyInfo.PropertyType);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public static bool IsSimpleType(Type type)
|
|||
|
|
{
|
|||
|
|
if (type.IsGenericType && type.GetGenericTypeDefinition() == typeof(Nullable<>))
|
|||
|
|
{
|
|||
|
|
type = type.GetGenericArguments()[0];
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return type.IsPrimitive
|
|||
|
|
|| type.IsEnum
|
|||
|
|
|| type == typeof(string)
|
|||
|
|
|| type == typeof(DateTime)
|
|||
|
|
|| type == typeof(Decimal);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|