123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773 |
- #region Copyright notice and license
- // Protocol Buffers - Google's data interchange format
- // Copyright 2015 Google Inc. All rights reserved.
- // https://developers.google.com/protocol-buffers/
- //
- // Redistribution and use in source and binary forms, with or without
- // modification, are permitted provided that the following conditions are
- // met:
- //
- // * Redistributions of source code must retain the above copyright
- // notice, this list of conditions and the following disclaimer.
- // * Redistributions in binary form must reproduce the above
- // copyright notice, this list of conditions and the following disclaimer
- // in the documentation and/or other materials provided with the
- // distribution.
- // * Neither the name of Google Inc. nor the names of its
- // contributors may be used to endorse or promote products derived from
- // this software without specific prior written permission.
- //
- // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
- // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
- // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
- // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
- // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
- // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
- // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
- // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
- // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
- // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
- // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
- #endregion
- using Google.Protobuf.Collections;
- using Google.Protobuf.Compatibility;
- using Google.Protobuf.WellKnownTypes;
- using System;
- using System.Collections.Generic;
- namespace Google.Protobuf
- {
- /// <summary>
- /// Factory methods for <see cref="FieldCodec{T}"/>.
- /// </summary>
- public static class FieldCodec
- {
- // TODO: Avoid the "dual hit" of lambda expressions: create open delegates instead. (At least test...)
- /// <summary>
- /// Retrieves a codec suitable for a string field with the given tag.
- /// </summary>
- /// <param name="tag">The tag.</param>
- /// <returns>A codec for the given tag.</returns>
- public static FieldCodec<string> ForString(uint tag)
- {
- return FieldCodec.ForString(tag, "");
- }
- /// <summary>
- /// Retrieves a codec suitable for a bytes field with the given tag.
- /// </summary>
- /// <param name="tag">The tag.</param>
- /// <returns>A codec for the given tag.</returns>
- public static FieldCodec<ByteString> ForBytes(uint tag)
- {
- return FieldCodec.ForBytes(tag, ByteString.Empty);
- }
- /// <summary>
- /// Retrieves a codec suitable for a bool field with the given tag.
- /// </summary>
- /// <param name="tag">The tag.</param>
- /// <returns>A codec for the given tag.</returns>
- public static FieldCodec<bool> ForBool(uint tag)
- {
- return FieldCodec.ForBool(tag, false);
- }
- /// <summary>
- /// Retrieves a codec suitable for an int32 field with the given tag.
- /// </summary>
- /// <param name="tag">The tag.</param>
- /// <returns>A codec for the given tag.</returns>
- public static FieldCodec<int> ForInt32(uint tag)
- {
- return FieldCodec.ForInt32(tag, 0);
- }
- /// <summary>
- /// Retrieves a codec suitable for an sint32 field with the given tag.
- /// </summary>
- /// <param name="tag">The tag.</param>
- /// <returns>A codec for the given tag.</returns>
- public static FieldCodec<int> ForSInt32(uint tag)
- {
- return FieldCodec.ForSInt32(tag, 0);
- }
- /// <summary>
- /// Retrieves a codec suitable for a fixed32 field with the given tag.
- /// </summary>
- /// <param name="tag">The tag.</param>
- /// <returns>A codec for the given tag.</returns>
- public static FieldCodec<uint> ForFixed32(uint tag)
- {
- return FieldCodec.ForFixed32(tag, 0);
- }
- /// <summary>
- /// Retrieves a codec suitable for an sfixed32 field with the given tag.
- /// </summary>
- /// <param name="tag">The tag.</param>
- /// <returns>A codec for the given tag.</returns>
- public static FieldCodec<int> ForSFixed32(uint tag)
- {
- return FieldCodec.ForSFixed32(tag, 0);
- }
- /// <summary>
- /// Retrieves a codec suitable for a uint32 field with the given tag.
- /// </summary>
- /// <param name="tag">The tag.</param>
- /// <returns>A codec for the given tag.</returns>
- public static FieldCodec<uint> ForUInt32(uint tag)
- {
- return FieldCodec.ForUInt32(tag, 0);
- }
- /// <summary>
- /// Retrieves a codec suitable for an int64 field with the given tag.
- /// </summary>
- /// <param name="tag">The tag.</param>
- /// <returns>A codec for the given tag.</returns>
- public static FieldCodec<long> ForInt64(uint tag)
- {
- return FieldCodec.ForInt64(tag, 0);
- }
- /// <summary>
- /// Retrieves a codec suitable for an sint64 field with the given tag.
- /// </summary>
- /// <param name="tag">The tag.</param>
- /// <returns>A codec for the given tag.</returns>
- public static FieldCodec<long> ForSInt64(uint tag)
- {
- return FieldCodec.ForSInt64(tag, 0);
- }
- /// <summary>
- /// Retrieves a codec suitable for a fixed64 field with the given tag.
- /// </summary>
- /// <param name="tag">The tag.</param>
- /// <returns>A codec for the given tag.</returns>
- public static FieldCodec<ulong> ForFixed64(uint tag)
- {
- return FieldCodec.ForFixed64(tag, 0);
- }
- /// <summary>
- /// Retrieves a codec suitable for an sfixed64 field with the given tag.
- /// </summary>
- /// <param name="tag">The tag.</param>
- /// <returns>A codec for the given tag.</returns>
- public static FieldCodec<long> ForSFixed64(uint tag)
- {
- return FieldCodec.ForSFixed64(tag, 0);
- }
- /// <summary>
- /// Retrieves a codec suitable for a uint64 field with the given tag.
- /// </summary>
- /// <param name="tag">The tag.</param>
- /// <returns>A codec for the given tag.</returns>
- public static FieldCodec<ulong> ForUInt64(uint tag)
- {
- return FieldCodec.ForUInt64(tag, 0);
- }
- /// <summary>
- /// Retrieves a codec suitable for a float field with the given tag.
- /// </summary>
- /// <param name="tag">The tag.</param>
- /// <returns>A codec for the given tag.</returns>
- public static FieldCodec<float> ForFloat(uint tag)
- {
- return FieldCodec.ForFloat(tag, 0);
- }
- /// <summary>
- /// Retrieves a codec suitable for a double field with the given tag.
- /// </summary>
- /// <param name="tag">The tag.</param>
- /// <returns>A codec for the given tag.</returns>
- public static FieldCodec<double> ForDouble(uint tag)
- {
- return FieldCodec.ForDouble(tag, 0);
- }
- // Enums are tricky. We can probably use expression trees to build these delegates automatically,
- // but it's easy to generate the code for it.
- /// <summary>
- /// Retrieves a codec suitable for an enum field with the given tag.
- /// </summary>
- /// <param name="tag">The tag.</param>
- /// <param name="toInt32">A conversion function from <see cref="Int32"/> to the enum type.</param>
- /// <param name="fromInt32">A conversion function from the enum type to <see cref="Int32"/>.</param>
- /// <returns>A codec for the given tag.</returns>
- public static FieldCodec<T> ForEnum<T>(uint tag, Func<T, int> toInt32, Func<int, T> fromInt32)
- {
- return FieldCodec.ForEnum(tag, toInt32, fromInt32, default(T));
- }
- /// <summary>
- /// Retrieves a codec suitable for a string field with the given tag.
- /// </summary>
- /// <param name="tag">The tag.</param>
- /// <param name="defaultValue">The default value.</param>
- /// <returns>A codec for the given tag.</returns>
- public static FieldCodec<string> ForString(uint tag, string defaultValue)
- {
- return new FieldCodec<string>(input => input.ReadString(), (output, value) => output.WriteString(value), CodedOutputStream.ComputeStringSize, tag);
- }
- /// <summary>
- /// Retrieves a codec suitable for a bytes field with the given tag.
- /// </summary>
- /// <param name="tag">The tag.</param>
- /// <param name="defaultValue">The default value.</param>
- /// <returns>A codec for the given tag.</returns>
- public static FieldCodec<ByteString> ForBytes(uint tag, ByteString defaultValue)
- {
- return new FieldCodec<ByteString>(input => input.ReadBytes(), (output, value) => output.WriteBytes(value), CodedOutputStream.ComputeBytesSize, tag);
- }
- /// <summary>
- /// Retrieves a codec suitable for a bool field with the given tag.
- /// </summary>
- /// <param name="tag">The tag.</param>
- /// <param name="defaultValue">The default value.</param>
- /// <returns>A codec for the given tag.</returns>
- public static FieldCodec<bool> ForBool(uint tag, bool defaultValue)
- {
- return new FieldCodec<bool>(input => input.ReadBool(), (output, value) => output.WriteBool(value), CodedOutputStream.BoolSize, tag);
- }
- /// <summary>
- /// Retrieves a codec suitable for an int32 field with the given tag.
- /// </summary>
- /// <param name="tag">The tag.</param>
- /// <param name="defaultValue">The default value.</param>
- /// <returns>A codec for the given tag.</returns>
- public static FieldCodec<int> ForInt32(uint tag, int defaultValue)
- {
- return new FieldCodec<int>(input => input.ReadInt32(), (output, value) => output.WriteInt32(value), CodedOutputStream.ComputeInt32Size, tag);
- }
- /// <summary>
- /// Retrieves a codec suitable for an sint32 field with the given tag.
- /// </summary>
- /// <param name="tag">The tag.</param>
- /// <param name="defaultValue">The default value.</param>
- /// <returns>A codec for the given tag.</returns>
- public static FieldCodec<int> ForSInt32(uint tag, int defaultValue)
- {
- return new FieldCodec<int>(input => input.ReadSInt32(), (output, value) => output.WriteSInt32(value), CodedOutputStream.ComputeSInt32Size, tag);
- }
- /// <summary>
- /// Retrieves a codec suitable for a fixed32 field with the given tag.
- /// </summary>
- /// <param name="tag">The tag.</param>
- /// <param name="defaultValue">The default value.</param>
- /// <returns>A codec for the given tag.</returns>
- public static FieldCodec<uint> ForFixed32(uint tag, uint defaultValue)
- {
- return new FieldCodec<uint>(input => input.ReadFixed32(), (output, value) => output.WriteFixed32(value), 4, tag);
- }
- /// <summary>
- /// Retrieves a codec suitable for an sfixed32 field with the given tag.
- /// </summary>
- /// <param name="tag">The tag.</param>
- /// <param name="defaultValue">The default value.</param>
- /// <returns>A codec for the given tag.</returns>
- public static FieldCodec<int> ForSFixed32(uint tag, int defaultValue)
- {
- return new FieldCodec<int>(input => input.ReadSFixed32(), (output, value) => output.WriteSFixed32(value), 4, tag);
- }
- /// <summary>
- /// Retrieves a codec suitable for a uint32 field with the given tag.
- /// </summary>
- /// <param name="tag">The tag.</param>
- /// <param name="defaultValue">The default value.</param>
- /// <returns>A codec for the given tag.</returns>
- public static FieldCodec<uint> ForUInt32(uint tag, uint defaultValue)
- {
- return new FieldCodec<uint>(input => input.ReadUInt32(), (output, value) => output.WriteUInt32(value), CodedOutputStream.ComputeUInt32Size, tag);
- }
- /// <summary>
- /// Retrieves a codec suitable for an int64 field with the given tag.
- /// </summary>
- /// <param name="tag">The tag.</param>
- /// <param name="defaultValue">The default value.</param>
- /// <returns>A codec for the given tag.</returns>
- public static FieldCodec<long> ForInt64(uint tag, long defaultValue)
- {
- return new FieldCodec<long>(input => input.ReadInt64(), (output, value) => output.WriteInt64(value), CodedOutputStream.ComputeInt64Size, tag);
- }
- /// <summary>
- /// Retrieves a codec suitable for an sint64 field with the given tag.
- /// </summary>
- /// <param name="tag">The tag.</param>
- /// <param name="defaultValue">The default value.</param>
- /// <returns>A codec for the given tag.</returns>
- public static FieldCodec<long> ForSInt64(uint tag, long defaultValue)
- {
- return new FieldCodec<long>(input => input.ReadSInt64(), (output, value) => output.WriteSInt64(value), CodedOutputStream.ComputeSInt64Size, tag);
- }
- /// <summary>
- /// Retrieves a codec suitable for a fixed64 field with the given tag.
- /// </summary>
- /// <param name="tag">The tag.</param>
- /// <param name="defaultValue">The default value.</param>
- /// <returns>A codec for the given tag.</returns>
- public static FieldCodec<ulong> ForFixed64(uint tag, ulong defaultValue)
- {
- return new FieldCodec<ulong>(input => input.ReadFixed64(), (output, value) => output.WriteFixed64(value), 8, tag);
- }
- /// <summary>
- /// Retrieves a codec suitable for an sfixed64 field with the given tag.
- /// </summary>
- /// <param name="tag">The tag.</param>
- /// <param name="defaultValue">The default value.</param>
- /// <returns>A codec for the given tag.</returns>
- public static FieldCodec<long> ForSFixed64(uint tag, long defaultValue)
- {
- return new FieldCodec<long>(input => input.ReadSFixed64(), (output, value) => output.WriteSFixed64(value), 8, tag);
- }
- /// <summary>
- /// Retrieves a codec suitable for a uint64 field with the given tag.
- /// </summary>
- /// <param name="tag">The tag.</param>
- /// <param name="defaultValue">The default value.</param>
- /// <returns>A codec for the given tag.</returns>
- public static FieldCodec<ulong> ForUInt64(uint tag, ulong defaultValue)
- {
- return new FieldCodec<ulong>(input => input.ReadUInt64(), (output, value) => output.WriteUInt64(value), CodedOutputStream.ComputeUInt64Size, tag);
- }
- /// <summary>
- /// Retrieves a codec suitable for a float field with the given tag.
- /// </summary>
- /// <param name="tag">The tag.</param>
- /// <param name="defaultValue">The default value.</param>
- /// <returns>A codec for the given tag.</returns>
- public static FieldCodec<float> ForFloat(uint tag, float defaultValue)
- {
- return new FieldCodec<float>(input => input.ReadFloat(), (output, value) => output.WriteFloat(value), CodedOutputStream.FloatSize, tag);
- }
- /// <summary>
- /// Retrieves a codec suitable for a double field with the given tag.
- /// </summary>
- /// <param name="tag">The tag.</param>
- /// <param name="defaultValue">The default value.</param>
- /// <returns>A codec for the given tag.</returns>
- public static FieldCodec<double> ForDouble(uint tag, double defaultValue)
- {
- return new FieldCodec<double>(input => input.ReadDouble(), (output, value) => output.WriteDouble(value), CodedOutputStream.DoubleSize, tag);
- }
- // Enums are tricky. We can probably use expression trees to build these delegates automatically,
- // but it's easy to generate the code for it.
- /// <summary>
- /// Retrieves a codec suitable for an enum field with the given tag.
- /// </summary>
- /// <param name="tag">The tag.</param>
- /// <param name="toInt32">A conversion function from <see cref="Int32"/> to the enum type.</param>
- /// <param name="fromInt32">A conversion function from the enum type to <see cref="Int32"/>.</param>
- /// <param name="defaultValue">The default value.</param>
- /// <returns>A codec for the given tag.</returns>
- public static FieldCodec<T> ForEnum<T>(uint tag, Func<T, int> toInt32, Func<int, T> fromInt32, T defaultValue)
- {
- return new FieldCodec<T>(input => fromInt32(
- input.ReadEnum()),
- (output, value) => output.WriteEnum(toInt32(value)),
- value => CodedOutputStream.ComputeEnumSize(toInt32(value)), tag);
- }
- /// <summary>
- /// Retrieves a codec suitable for a message field with the given tag.
- /// </summary>
- /// <param name="tag">The tag.</param>
- /// <param name="parser">A parser to use for the message type.</param>
- /// <returns>A codec for the given tag.</returns>
- public static FieldCodec<T> ForMessage<T>(uint tag, MessageParser<T> parser) where T : class, IMessage<T>
- {
- return new FieldCodec<T>(input => { T message = parser.CreateTemplate(); input.ReadMessage(message); return message; },
- (output, value) => output.WriteMessage(value), (CodedInputStream i, ref T v) =>
- {
- if (v == null)
- {
- v = parser.CreateTemplate();
- }
- i.ReadMessage(v);
- },
- (ref T v, T v2) =>
- {
- if (v2 == null)
- {
- return false;
- }
- else if (v == null)
- {
- v = v2.Clone();
- }
- else
- {
- v.MergeFrom(v2);
- }
- return true;
- }, message => CodedOutputStream.ComputeMessageSize(message), tag);
- }
- /// <summary>
- /// Retrieves a codec suitable for a group field with the given tag.
- /// </summary>
- /// <param name="startTag">The start group tag.</param>
- /// <param name="endTag">The end group tag.</param>
- /// <param name="parser">A parser to use for the group message type.</param>
- /// <returns>A codec for given tag</returns>
- public static FieldCodec<T> ForGroup<T>(uint startTag, uint endTag, MessageParser<T> parser) where T : class, IMessage<T>
- {
- return new FieldCodec<T>(input => { T message = parser.CreateTemplate(); input.ReadGroup(message); return message; },
- (output, value) => output.WriteGroup(value), (CodedInputStream i, ref T v) => {
- if (v == null)
- {
- v = parser.CreateTemplate();
- }
- i.ReadGroup(v);
- },
- (ref T v, T v2) =>
- {
- if (v2 == null)
- {
- return v == null;
- }
- else if (v == null)
- {
- v = v2.Clone();
- }
- else
- {
- v.MergeFrom(v2);
- }
- return true;
- }, message => CodedOutputStream.ComputeGroupSize(message), startTag, endTag);
- }
- /// <summary>
- /// Creates a codec for a wrapper type of a class - which must be string or ByteString.
- /// </summary>
- public static FieldCodec<T> ForClassWrapper<T>(uint tag) where T : class
- {
- var nestedCodec = WrapperCodecs.GetCodec<T>();
- return new FieldCodec<T>(
- input => WrapperCodecs.Read<T>(input, nestedCodec),
- (output, value) => WrapperCodecs.Write<T>(output, value, nestedCodec),
- (CodedInputStream i, ref T v) => v = WrapperCodecs.Read<T>(i, nestedCodec),
- (ref T v, T v2) => { v = v2; return v == null; },
- value => WrapperCodecs.CalculateSize<T>(value, nestedCodec),
- tag, 0,
- null); // Default value for the wrapper
- }
- /// <summary>
- /// Creates a codec for a wrapper type of a struct - which must be Int32, Int64, UInt32, UInt64,
- /// Bool, Single or Double.
- /// </summary>
- public static FieldCodec<T?> ForStructWrapper<T>(uint tag) where T : struct
- {
- var nestedCodec = WrapperCodecs.GetCodec<T>();
- return new FieldCodec<T?>(
- input => WrapperCodecs.Read<T>(input, nestedCodec),
- (output, value) => WrapperCodecs.Write<T>(output, value.Value, nestedCodec),
- (CodedInputStream i, ref T? v) => v = WrapperCodecs.Read<T>(i, nestedCodec),
- (ref T? v, T? v2) => { if (v2.HasValue) { v = v2; } return v.HasValue; },
- value => value == null ? 0 : WrapperCodecs.CalculateSize<T>(value.Value, nestedCodec),
- tag, 0,
- null); // Default value for the wrapper
- }
- /// <summary>
- /// Helper code to create codecs for wrapper types.
- /// </summary>
- /// <remarks>
- /// Somewhat ugly with all the static methods, but the conversions involved to/from nullable types make it
- /// slightly tricky to improve. So long as we keep the public API (ForClassWrapper, ForStructWrapper) in place,
- /// we can refactor later if we come up with something cleaner.
- /// </remarks>
- private static class WrapperCodecs
- {
- private static readonly Dictionary<System.Type, object> Codecs = new Dictionary<System.Type, object>
- {
- { typeof(bool), ForBool(WireFormat.MakeTag(WrappersReflection.WrapperValueFieldNumber, WireFormat.WireType.Varint)) },
- { typeof(int), ForInt32(WireFormat.MakeTag(WrappersReflection.WrapperValueFieldNumber, WireFormat.WireType.Varint)) },
- { typeof(long), ForInt64(WireFormat.MakeTag(WrappersReflection.WrapperValueFieldNumber, WireFormat.WireType.Varint)) },
- { typeof(uint), ForUInt32(WireFormat.MakeTag(WrappersReflection.WrapperValueFieldNumber, WireFormat.WireType.Varint)) },
- { typeof(ulong), ForUInt64(WireFormat.MakeTag(WrappersReflection.WrapperValueFieldNumber, WireFormat.WireType.Varint)) },
- { typeof(float), ForFloat(WireFormat.MakeTag(WrappersReflection.WrapperValueFieldNumber, WireFormat.WireType.Fixed32)) },
- { typeof(double), ForDouble(WireFormat.MakeTag(WrappersReflection.WrapperValueFieldNumber, WireFormat.WireType.Fixed64)) },
- { typeof(string), ForString(WireFormat.MakeTag(WrappersReflection.WrapperValueFieldNumber, WireFormat.WireType.LengthDelimited)) },
- { typeof(ByteString), ForBytes(WireFormat.MakeTag(WrappersReflection.WrapperValueFieldNumber, WireFormat.WireType.LengthDelimited)) }
- };
- /// <summary>
- /// Returns a field codec which effectively wraps a value of type T in a message.
- ///
- /// </summary>
- internal static FieldCodec<T> GetCodec<T>()
- {
- object value;
- if (!Codecs.TryGetValue(typeof(T), out value))
- {
- throw new InvalidOperationException("Invalid type argument requested for wrapper codec: " + typeof(T));
- }
- return (FieldCodec<T>) value;
- }
- internal static T Read<T>(CodedInputStream input, FieldCodec<T> codec)
- {
- int length = input.ReadLength();
- int oldLimit = input.PushLimit(length);
- uint tag;
- T value = codec.DefaultValue;
- while ((tag = input.ReadTag()) != 0)
- {
- if (tag == codec.Tag)
- {
- value = codec.Read(input);
- }
- else
- {
- input.SkipLastField();
- }
- }
- input.CheckReadEndOfStreamTag();
- input.PopLimit(oldLimit);
- return value;
- }
- internal static void Write<T>(CodedOutputStream output, T value, FieldCodec<T> codec)
- {
- output.WriteLength(codec.CalculateSizeWithTag(value));
- codec.WriteTagAndValue(output, value);
- }
- internal static int CalculateSize<T>(T value, FieldCodec<T> codec)
- {
- int fieldLength = codec.CalculateSizeWithTag(value);
- return CodedOutputStream.ComputeLengthSize(fieldLength) + fieldLength;
- }
- }
- }
- /// <summary>
- /// <para>
- /// An encode/decode pair for a single field. This effectively encapsulates
- /// all the information needed to read or write the field value from/to a coded
- /// stream.
- /// </para>
- /// <para>
- /// This class is public and has to be as it is used by generated code, but its public
- /// API is very limited - just what the generated code needs to call directly.
- /// </para>
- /// </summary>
- /// <remarks>
- /// This never writes default values to the stream, and does not address "packedness"
- /// in repeated fields itself, other than to know whether or not the field *should* be packed.
- /// </remarks>
- public sealed class FieldCodec<T>
- {
- private static readonly EqualityComparer<T> EqualityComparer = ProtobufEqualityComparers.GetEqualityComparer<T>();
- private static readonly T DefaultDefault;
- // Only non-nullable value types support packing. This is the simplest way of detecting that.
- private static readonly bool TypeSupportsPacking = default(T) != null;
- /// <summary>
- /// Merges an input stream into a value
- /// </summary>
- internal delegate void InputMerger(CodedInputStream input, ref T value);
- /// <summary>
- /// Merges a value into a reference to another value, returning a boolean if the value was set
- /// </summary>
- internal delegate bool ValuesMerger(ref T value, T other);
- static FieldCodec()
- {
- if (typeof(T) == typeof(string))
- {
- DefaultDefault = (T)(object)"";
- }
- else if (typeof(T) == typeof(ByteString))
- {
- DefaultDefault = (T)(object)ByteString.Empty;
- }
- // Otherwise it's the default value of the CLR type
- }
- internal static bool IsPackedRepeatedField(uint tag) =>
- TypeSupportsPacking && WireFormat.GetTagWireType(tag) == WireFormat.WireType.LengthDelimited;
- internal bool PackedRepeatedField { get; }
- /// <summary>
- /// Returns a delegate to write a value (unconditionally) to a coded output stream.
- /// </summary>
- internal Action<CodedOutputStream, T> ValueWriter { get; }
- /// <summary>
- /// Returns the size calculator for just a value.
- /// </summary>
- internal Func<T, int> ValueSizeCalculator { get; }
- /// <summary>
- /// Returns a delegate to read a value from a coded input stream. It is assumed that
- /// the stream is already positioned on the appropriate tag.
- /// </summary>
- internal Func<CodedInputStream, T> ValueReader { get; }
- /// <summary>
- /// Returns a delegate to merge a value from a coded input stream.
- /// It is assumed that the stream is already positioned on the appropriate tag
- /// </summary>
- internal InputMerger ValueMerger { get; }
- /// <summary>
- /// Returns a delegate to merge two values together.
- /// </summary>
- internal ValuesMerger FieldMerger { get; }
- /// <summary>
- /// Returns the fixed size for an entry, or 0 if sizes vary.
- /// </summary>
- internal int FixedSize { get; }
- /// <summary>
- /// Gets the tag of the codec.
- /// </summary>
- /// <value>
- /// The tag of the codec.
- /// </value>
- internal uint Tag { get; }
- /// <summary>
- /// Gets the end tag of the codec or 0 if there is no end tag
- /// </summary>
- /// <value>
- /// The end tag of the codec.
- /// </value>
- internal uint EndTag { get; }
- /// <summary>
- /// Default value for this codec. Usually the same for every instance of the same type, but
- /// for string/ByteString wrapper fields the codec's default value is null, whereas for
- /// other string/ByteString fields it's "" or ByteString.Empty.
- /// </summary>
- /// <value>
- /// The default value of the codec's type.
- /// </value>
- internal T DefaultValue { get; }
- private readonly int tagSize;
- internal FieldCodec(
- Func<CodedInputStream, T> reader,
- Action<CodedOutputStream, T> writer,
- int fixedSize,
- uint tag) : this(reader, writer, _ => fixedSize, tag)
- {
- FixedSize = fixedSize;
- }
- internal FieldCodec(
- Func<CodedInputStream, T> reader,
- Action<CodedOutputStream, T> writer,
- Func<T, int> sizeCalculator,
- uint tag,
- uint endTag = 0) : this(reader, writer, (CodedInputStream i, ref T v) => v = reader(i), (ref T v, T v2) => { v = v2; return true; }, sizeCalculator, tag, endTag, DefaultDefault)
- {
- }
- internal FieldCodec(
- Func<CodedInputStream, T> reader,
- Action<CodedOutputStream, T> writer,
- InputMerger inputMerger,
- ValuesMerger valuesMerger,
- Func<T, int> sizeCalculator,
- uint tag,
- uint endTag = 0) : this(reader, writer, inputMerger, valuesMerger, sizeCalculator, tag, endTag, DefaultDefault)
- {
- }
- internal FieldCodec(
- Func<CodedInputStream, T> reader,
- Action<CodedOutputStream, T> writer,
- InputMerger inputMerger,
- ValuesMerger valuesMerger,
- Func<T, int> sizeCalculator,
- uint tag,
- uint endTag,
- T defaultValue)
- {
- ValueReader = reader;
- ValueWriter = writer;
- ValueMerger = inputMerger;
- FieldMerger = valuesMerger;
- ValueSizeCalculator = sizeCalculator;
- FixedSize = 0;
- Tag = tag;
- DefaultValue = defaultValue;
- tagSize = CodedOutputStream.ComputeRawVarint32Size(tag);
- if (endTag != 0)
- tagSize += CodedOutputStream.ComputeRawVarint32Size(endTag);
- // Detect packed-ness once, so we can check for it within RepeatedField<T>.
- PackedRepeatedField = IsPackedRepeatedField(tag);
- }
- /// <summary>
- /// Write a tag and the given value, *if* the value is not the default.
- /// </summary>
- public void WriteTagAndValue(CodedOutputStream output, T value)
- {
- if (!IsDefault(value))
- {
- output.WriteTag(Tag);
- ValueWriter(output, value);
- if (EndTag != 0)
- {
- output.WriteTag(EndTag);
- }
- }
- }
- /// <summary>
- /// Reads a value of the codec type from the given <see cref="CodedInputStream"/>.
- /// </summary>
- /// <param name="input">The input stream to read from.</param>
- /// <returns>The value read from the stream.</returns>
- public T Read(CodedInputStream input) => ValueReader(input);
- /// <summary>
- /// Calculates the size required to write the given value, with a tag,
- /// if the value is not the default.
- /// </summary>
- public int CalculateSizeWithTag(T value) => IsDefault(value) ? 0 : ValueSizeCalculator(value) + tagSize;
- private bool IsDefault(T value) => EqualityComparer.Equals(value, DefaultValue);
- }
- }
|