EF 4.1 Code First - 将枚举包装器映射为复杂类型

时间:2011-04-14 17:58:11

标签: .net entity-framework ef-code-first entity-framework-4.1

我正在尝试使用EF 4.1构建枚举问题的通用解决方案。我的解决方案基本上是How to fake enums in ef 4的通用版本。枚举包装类在其余代码中运行得非常好,并允许代码如下:

EnumWrapper<Color> c = Color.Red;

这是枚举包装类:

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 }

POCO:

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);
    }
}

我也尝试将它映射到ChickenSandwich的EntityTypeConfiguration,如下所示:

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

如果我把它留给ColorMapping并且在ChickenSandwichMapping上没有显式映射,它就不会把它放在数据库中。如果我将它映射到x.CheeseColor.Value方式,我会得到可怕的:

  

System.InvalidOperationException:The   配置属性'CheeseColor'是   不是实体的已申报财产   'ChickenSandwich'。验证它有   未被明确排除在外   模型,它是一个有效的原语   属性..


修改

我无法获得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 };
    }
}

我不得不在ChickenSandwich类上使用ColorWrapper。它或多或少地透明地工作。然后必须将它添加到我的映射类构造函数中以获取我想要的列名:

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

4 个答案:

答案 0 :(得分:30)

在EF 4中映射enums的方法要简单得多:只需在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,但这让我头疼不已。此方法效果很好,您可以调整解决方案以使用包装器作为“映射”属性,在这种情况下为CheeseColor

答案 1 :(得分:7)

我通过简单地将它抽象化并移动来获得Nathan通用枚举包装类:

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在名为ChildSortType_Value的数据库中创建了一个“ComplexType like field”

答案 2 :(得分:4)

这可能是枚举的最佳选择,但另一个想法是只使用常量而不是枚举:

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; }
}

答案 3 :(得分:0)

根据HenrikStenbæk的回答。 分配工作正常

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;
        }