EnumSchema.cs

Go to the documentation of this file.

using System;
using System.Collections.Generic;
using System.Text;
using Newtonsoft.Json.Linq;

namespace Avro
{
    public class EnumSchema : NamedSchema
    {
        public IList<string> Symbols { get; private set;  }

        private readonly IDictionary<string, int> symbolMap;

        public int Count { get { return Symbols.Count; } }

        internal static EnumSchema NewInstance(JToken jtok, PropertyMap props, SchemaNames names, string encspace)
        {
            SchemaName name = NamedSchema.GetName(jtok, encspace);
            var aliases = NamedSchema.GetAliases(jtok, name.Space, name.EncSpace);

            JArray jsymbols = jtok["symbols"] as JArray;
            if (null == jsymbols)
                throw new SchemaParseException("Enum has no symbols: " + name);

            List<string> symbols = new List<string>();
            IDictionary<string, int> symbolMap = new Dictionary<string, int>();
            int i = 0;
            foreach (JValue jsymbol in jsymbols)
            {
                string s = (string)jsymbol.Value;
                if (symbolMap.ContainsKey(s))
                    throw new SchemaParseException("Duplicate symbol: " + s);

                symbolMap[s] = i++;
                symbols.Add(s);
            }
            return new EnumSchema(name, aliases, symbols, symbolMap, props, names);
        }

        private EnumSchema(SchemaName name, IList<SchemaName> aliases, List<string> symbols,
                            IDictionary<String, int> symbolMap, PropertyMap props, SchemaNames names)
                            : base(Type.Enumeration, name, aliases, props, names)
        {
            if (null == name.Name) throw new SchemaParseException("name cannot be null for enum schema.");
            this.Symbols = symbols;
            this.symbolMap = symbolMap;
        }

        protected internal override void WriteJsonFields(Newtonsoft.Json.JsonTextWriter writer, 
                                                            SchemaNames names, string encspace)
        {
            base.WriteJsonFields(writer, names, encspace);
            writer.WritePropertyName("symbols");
            writer.WriteStartArray();
            foreach (string s in this.Symbols)
                writer.WriteValue(s);
            writer.WriteEndArray();
        }

        public int Ordinal(string symbol)
        {
            int result;
            if (symbolMap.TryGetValue(symbol, out result)) return result;
            throw new AvroException("No such symbol: " + symbol);
        }

        public string this[int index]
        {
            get
            {
                if (index < Symbols.Count) return Symbols[index];
                throw new AvroException("Enumeration out of range. Must be less than " + Symbols.Count + ", but is " + index);
            }
        }

        public bool Contains(string symbol)
        {
            return symbolMap.ContainsKey(symbol);
        }

        public IEnumerator<string> GetEnumerator()
        {
            return Symbols.GetEnumerator();
        }

        public override bool Equals(object obj)
        {
            if (obj == this) return true;
            if (obj != null && obj is EnumSchema)
            {
                EnumSchema that = obj as EnumSchema;
                if (SchemaName.Equals(that.SchemaName) && Count == that.Count)
                {
                    for (int i = 0; i < Count; i++) if (!Symbols[i].Equals(that.Symbols[i])) return false;
                    return areEqual(that.Props, this.Props);
                }
            }
            return false;
        }

        public override int GetHashCode()
        {
            int result = SchemaName.GetHashCode() + getHashCode(Props);
            foreach (string s in Symbols) result += 23 * s.GetHashCode();
            return result;
        }

        public override bool CanRead(Schema writerSchema)
        {
            if (writerSchema.Tag != Tag) return false;

            EnumSchema that = writerSchema as EnumSchema;
            if (!that.SchemaName.Equals(SchemaName))
                if (!InAliases(that.SchemaName)) return false;

            // we defer checking of symbols. Writer may have a symbol missing from the reader, 
            // but if writer never used the missing symbol, then reader should still be able to read the data

            return true;
        }
    }
}