EF 4.1 Code First - отображать оболочку перечисления как сложный тип - PullRequest
15 голосов
/ 14 апреля 2011

Я пытаюсь найти общее решение проблемы перечислений с EF 4.1. Мое решение в основном является общей версией Как подделать перечисления в ef 4 . Класс-оболочка enum прекрасно работает в остальной части кода и допускает такой код:

EnumWrapper<Color> c = Color.Red;

Вот класс-обертка enum:

public class EnumWrapper<TEnum> where TEnum : struct, IConvertible
{
    public EnumWrapper()
    {
        if (!typeof(TEnum).IsEnum)
            throw new ArgumentException("Not an enum");
    }

    public TEnum Enum { get; set; }

    public int Value
    {
        get { return Convert.ToInt32(Enum); }
        set { Enum = (TEnum)(object)value; }
    }

    public static implicit operator TEnum(EnumWrapper<TEnum> w)
    {
        if (w == null) return default(TEnum);
        else return w.Enum;
    }

    public static implicit operator EnumWrapper<TEnum>(TEnum e)
    {
        return new EnumWrapper<TEnum>() { Enum = e };
    }

    public static implicit operator int(EnumWrapper<TEnum> w)
    {
        if (w == null)
            return Convert.ToInt32(default(TEnum));
        else
            return w.Value;
    }
}

перечисление:

public enum Color { red = 1, green = 2, blue = 3 }

ПОКО:

public class ChickenSandwich 
{
    public ChickenSandwich() {
        CheeseColor = new EnumWrapper<Color>();
    }

    public int ID { get; set; }
    public string Name { get; set; }
    public EnumWrapper<Color> CheeseColor { get; set; }
}

Отображение:

public class ColorMapping : ComplexTypeConfiguration<EnumWrapper<Color>> 
{
    public ColorMapping() {
        Ignore(x => x.Enum);
        Property(x => x.Value);
    }
}

Я также попытался сопоставить его с конфигурацией EntityType в ChickenSandwich следующим образом:

Property(x => x.CheeseColor.Value).HasColumnName("CheeseColor");

Если я оставлю это до ColorMapping и не сделаю явного отображения на ChickenSandwichMapping, оно просто не поместит его в базу данных. Если я сопоставлю это способом x.CheeseColor.Value, я получу ужас:

System.InvalidOperationException: настроенное свойство 'CheeseColor' не объявленная собственность на юридическом лице 'Куриный сендвич'. Убедитесь, что он имеет не был явно исключен из модель и что это действительный примитив свойство ..


Редактировать

Мне не удалось заставить работать универсальную версию оболочки enum, поэтому я перешел к написанию отдельных оболочек. Это не совсем то, что я хотел, потому что он нарушает принцип DRY , но он позволяет запрашивать столбец как перечисление.

[ComplexType]
public class ColorWrapper
{
    [NotMapped]
    public Color Enum { get; set; }

    public int Value
    {
        get { return (int)Enum; }
        set { Enum = (Color)value; }
    }

    public static implicit operator Color(ColorWrapper w)
    {
        if (w == null) return default(Color);

        return w.Enum;
    }

    public static implicit operator ColorWrapper(Color c)
    {
        return new ColorWrapper { Enum = c };
    }
}

Мне пришлось использовать ColorWrapper в классе ChickenSandwich. Работает более-менее прозрачно. Затем пришлось добавить это в мой конструктор класса отображения, чтобы получить имя столбца, который я хотел:

Property(x => x.CheeseColor.Value).HasColumnName("CheeseColorId");

Ответы [ 4 ]

30 голосов
/ 15 апреля 2011

Существует намного более простой способ отобразить enums в EF 4: просто создайте свойство int в своем классе ChickenSandwich, чтобы представить значение int для перечисления. Это свойство, которое EF должен отобразить, а затем иметь свойство «мини-оболочки», позволяющее вам использовать enum

public class ChickenSandwich 
{   
    public int ID { get; set; }
    public string Name { get; set; }

    // This property will be mapped
    public int CheeseColorValue { get; set; }

    public Color CheseColor
    {
        get { return (Color) CheeseColorValue; }
        set { CheeseColorValue = (int) value; }
    }
}

Мне на самом деле не нужно использовать Fluent API или какой-либо другой вид атрибутов, чтобы это работало. При создании базы данных EF с радостью проигнорирует любой тип, который он не умеет отображать, но свойство int будет отображено.

Я также пытался составить карту enums на основе этой статьи, но это не вызвало у меня головной боли. Этот метод работает хорошо, и вы могли бы адаптировать свое решение для использования вашей обертки в качестве свойства "mapping", то есть CheeseColor в этом случае.

7 голосов
/ 12 сентября 2011

Я заставил Nathan generic enum wrapper class работать, просто сделав его абстрактным и переместив:

public static implicit operator EnumWrapper <TEnum> (TEnum e)

к производным классам, например так:

public class CategorySortWrapper : EnumWrapper<CategorySort>
{
    public static implicit operator CategorySortWrapper(CategorySort e)
    {
        return new CategorySortWrapper() { Enum = e };
    }
}

public abstract class EnumWrapper<TEnum> where TEnum : struct, IConvertible
{
    public EnumWrapper()
    {
        if (!typeof(TEnum).IsEnum)
            throw new ArgumentException("Not an enum");
    }

    public TEnum Enum { get; set; }

    public int Value
    {
        get { return Convert.ToInt32(Enum); }
        set { Enum = (TEnum)(object)value; }
    }

    public static implicit operator int(EnumWrapper<TEnum> w)
    {
        if (w == null)
            return Convert.ToInt32(default(TEnum));
        else
            return w.Value;
    }
}

в моем коде я простоиспользуя его вот так

public CategorySortWrapper ChildSortType { get; set; }

category.ChildSortType = CategorySort.AlphabeticOrder;

Я больше ничего не делал, а EF 4.1 создал «поле типа ComplexType» в базе данных ChildSortType_Value

4 голосов
/ 14 апреля 2011

Это может быть лучшим вариантом для перечислений, но другой идеей будет просто использовать константы вместо перечислений:

static void Main(string[] args)
{
    Console.WriteLine("There are {0} red chicken sandwiches.", 
        sandwiches.ChickenSandwiches
                  .Where(s => s.Color == Color.red)
                  .ToArray().Length);
}

public struct Color
{
    public const int red = 1;
    public const int green = 2;
}

public class ChickenSandwich
{
    public ChickenSandwich()
    {
    }

    public int ID { get; set; }
    public string Name { get; set; }
    public int Color { get; set; }
}

public class Sandwiches : DbContext
{
    public DbSet<ChickenSandwich> ChickenSandwiches { get; set; }
}
0 голосов
/ 24 мая 2012

основано на ответе Хенрика Стенбека. назначение работает нормально

category.ChildSortType = CategorySort.AlphabeticOrder

, но сравнение оболочки с ее перечислением не работает.

if(category.ChildSortType == CategorySort.AlphabeticOrder)
{

}

следующий оператор должен быть добавлен к абстрактному классу

public static implicit operator TEnum(EnumWrapper<TEnum> w)
        {
            if (w == null)
                return default(TEnum);
            else
                return w.EnumVal;
        }
...