Storing Enums as strings in MongoDB

后端 未结 9 1176
盖世英雄少女心
盖世英雄少女心 2020-12-08 03:50

Is there a way to store Enums as string names rather than ordinal values?

Example:

Imagine I\'ve got this enum:

public enum Gender
{
    Fema         


        
相关标签:
9条回答
  • 2020-12-08 04:10

    The MongoDB .NET Driver lets you apply conventions to determine how certain mappings between CLR types and database elements are handled.

    If you want this to apply to all your enums, you only have to set up conventions once per AppDomain (usually when starting your application), as opposed to adding attributes to all your types or manually map every type:

    // Set up MongoDB conventions
    var pack = new ConventionPack
    {
        new EnumRepresentationConvention(BsonType.String)
    };
    
    ConventionRegistry.Register("EnumStringConvention", pack, t => true);
    
    0 讨论(0)
  • 2020-12-08 04:10

    If you are using .NET Core 3.1 and above, use the latest ultra-fast Json Serializer/Deserializer from Microsoft, System.Text.Json (https://www.nuget.org/packages/System.Text.Json).

    See the metrics comparison at https://medium.com/@samichkhachkhi/system-text-json-vs-newtonsoft-json-d01935068143

    using System;
    using MongoDB.Bson;
    using MongoDB.Bson.Serialization.Attributes;
    using System.Text.Json.Serialization;;
    
    public class Person
    {
        [JsonConverter(typeof(JsonStringEnumConverter))]  // System.Text.Json.Serialization
        [BsonRepresentation(BsonType.String)]         // MongoDB.Bson.Serialization.Attributes
        public Gender Gender { get; set; }
    }
    
    0 讨论(0)
  • 2020-12-08 04:12

    With driver 2.x I solved using a specific serializer:

    BsonClassMap.RegisterClassMap<Person>(cm =>
                {
                    cm.AutoMap();
                    cm.MapMember(c => c.Gender).SetSerializer(new EnumSerializer<Gender>(BsonType.String));
                });
    
    0 讨论(0)
  • 2020-12-08 04:19

    Use MemberSerializationOptionsConvention to define a convention on how an enum will be saved.

    new MemberSerializationOptionsConvention(typeof(Gender), new RepresentationSerializationOptions(BsonType.String))
    
    0 讨论(0)
  • 2020-12-08 04:20

    The answers posted here work well for TEnum and TEnum[], however won't work with Dictionary<TEnum, object>. You could achieve this when initializing serializer using code, however I wanted to do this through attributes. I've created a flexible DictionarySerializer that can be configured with a serializer for the key and value.

    public class DictionarySerializer<TDictionary, KeySerializer, ValueSerializer> : DictionarySerializerBase<TDictionary>
        where TDictionary : class, IDictionary, new()
        where KeySerializer : IBsonSerializer, new()
        where ValueSerializer : IBsonSerializer, new()
    {
        public DictionarySerializer() : base(DictionaryRepresentation.Document, new KeySerializer(), new ValueSerializer())
        {
        }
    
        protected override TDictionary CreateInstance()
        {
            return new TDictionary();
        }
    }
    
    public class EnumStringSerializer<TEnum> : EnumSerializer<TEnum>
        where TEnum : struct
    {
        public EnumStringSerializer() : base(BsonType.String) { }
    }
    

    Usage like this, where both key and value are enum types, but could be any combination of serializers:

        [BsonSerializer(typeof(DictionarySerializer<
            Dictionary<FeatureToggleTypeEnum, LicenseFeatureStateEnum>, 
            EnumStringSerializer<FeatureToggleTypeEnum>,
            EnumStringSerializer<LicenseFeatureStateEnum>>))]
        public Dictionary<FeatureToggleTypeEnum, LicenseFeatureStateEnum> FeatureSettings { get; set; }
    
    0 讨论(0)
  • 2020-12-08 04:21

    I have found that just applying Ricardo Rodriguez' answer is not sufficient in some cases to properly serialize enum values to string into MongoDb:

    // Set up MongoDB conventions
    var pack = new ConventionPack
    {
        new EnumRepresentationConvention(BsonType.String)
    };
    
    ConventionRegistry.Register("EnumStringConvention", pack, t => true);
    

    If your data structure involves enum values being boxed into objects, the MongoDb serialization will not use the set EnumRepresentationConvention to serialize it.

    Indeed, if you look at the implementation of MongoDb driver's ObjectSerializer, it will resolve the TypeCode of the boxed value (Int32 for enum values), and use that type to store your enum value in the database. So boxed enum values end up being serialized as int values. They will remain as int values when being deserialized as well.

    To change this, it's possible to write a custom ObjectSerializer that will enforce the set EnumRepresentationConvention if the boxed value is an enum. Something like this:

    public class ObjectSerializer : MongoDB.Bson.Serialization.Serializers.ObjectSerializer
    {
         public override void Serialize(BsonSerializationContext context, BsonSerializationArgs args, object value)
        {
            var bsonWriter = context.Writer;
            if (value != null && value.GetType().IsEnum)
            {
                var conventions = ConventionRegistry.Lookup(value.GetType());
                var enumRepresentationConvention = (EnumRepresentationConvention) conventions.Conventions.FirstOrDefault(convention => convention is EnumRepresentationConvention);
                if (enumRepresentationConvention != null)
                {
                    switch (enumRepresentationConvention.Representation)
                    {
                        case BsonType.String:
                            value = value.ToString();
                            bsonWriter.WriteString(value.ToString());
                            return;
                    }
                }
            }
    
            base.Serialize(context, args, value);
        }
    }
    

    and then set the custom serializer as the one to use for serializing objects:

    BsonSerializer.RegisterSerializer(typeof(object), new ObjectSerializer());
    

    Doing this will ensure boxed enum values will be stored as strings just like the unboxed ones.

    Keep in mind however that when deserializing your document, the boxed value will remain a string. It will not be converted back to the original enum value. If you need to convert the string back to the original enum value, a discrimination field will likely have to be added in your document so the serializer can know what is the enum type to desrialize into.

    One way to do it would be to store a bson document instead of just a string, into which the discrimination field (_t) and a value field (_v) would be used to store the enum type and its string value.

    0 讨论(0)
提交回复
热议问题