NetSerializer - A Fast, Simple Serializer for .NET

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using NetSerializer;
using System.IO;
using System.Diagnostics;

namespace Test
{
	interface INetTest
	{
		string Framework { get; }
		void Prepare(int numMessages);
		MessageBase[] Test(MessageBase[] msgs);
	}

	interface IMemStreamTest
	{
		string Framework { get; }
		void Prepare(int numMessages);
		long Serialize(MessageBase[] msgs);
		MessageBase[] Deserialize();
	}

	static class Program
	{
		static bool s_runProtoBufTests = true;
		static bool s_quickRun = false;

		static void Main(string[] args)
		{
			System.Threading.Thread.CurrentThread.Priority = System.Threading.ThreadPriority.Highest;

			var types = typeof(MessageBase).Assembly.GetTypes().Where(t => t.IsSubclassOf(typeof(MessageBase)))
				.Concat(new Type[] { typeof(SimpleClass), typeof(SimpleClass2) })
				.ToArray();

			Serializer.Initialize(types);

			Warmup();

			RunTests(typeof(U8Message), 2000000);
			RunTests(typeof(S16Message), 2000000);
			RunTests(typeof(S32Message), 2000000);
			RunTests(typeof(S64Message), 2000000);

			RunTests(typeof(PrimitivesMessage), 1000000);
			RunTests(typeof(DictionaryMessage), 5000);

			RunTests(typeof(ComplexMessage), 1000000);

			RunTests(typeof(StringMessage), 200000);

			RunTests(typeof(StructMessage), 1000000);

			RunTests(typeof(BoxedPrimitivesMessage), 1000000);

			RunTests(typeof(ByteArrayMessage), 5000);
			RunTests(typeof(IntArrayMessage), 800);

			//Console.WriteLine("Press enter to quit");
			//Console.ReadLine();
		}

		static void Warmup()
		{
			var msgs = new MessageBase[] { new S16Message(), new ComplexMessage(), new IntArrayMessage() };

			IMemStreamTest t;

			t = new MemStreamTest();
			t.Prepare(msgs.Length);
			t.Serialize(msgs);
			t.Deserialize();

			if (s_runProtoBufTests)
			{
				t = new PBMemStreamTest();
				t.Prepare(msgs.Length);
				t.Serialize(msgs);
				t.Deserialize();
			}
		}

		static void RunTests(Type msgType, int numMessages)
		{
			if (s_quickRun)
				numMessages = 50;

			Console.WriteLine("== {0} {1} ==", numMessages, msgType.Name);

			bool protobufCompatible = msgType.GetCustomAttributes(typeof(ProtoBuf.ProtoContractAttribute), false).Any();

			var msgs = MessageBase.CreateMessages(msgType, numMessages);

			GC.Collect();
			GC.WaitForPendingFinalizers();
			GC.Collect();

			Test(new MemStreamTest(), msgs);
			if (s_runProtoBufTests && protobufCompatible)
				Test(new PBMemStreamTest(), msgs);

			Test(new NetTest(), msgs);
			if (s_runProtoBufTests && protobufCompatible)
				Test(new PBNetTest(), msgs);
		}

		static void Test(IMemStreamTest test, MessageBase[] msgs)
		{
			test.Prepare(msgs.Length);

			/* Serialize part */
			{
				GC.Collect();
				GC.WaitForPendingFinalizers();
				GC.Collect();

				var c0 = GC.CollectionCount(0);
				var c1 = GC.CollectionCount(1);
				var c2 = GC.CollectionCount(2);

				var sw = Stopwatch.StartNew();

				long size = test.Serialize(msgs);

				sw.Stop();

				c0 = GC.CollectionCount(0) - c0;
				c1 = GC.CollectionCount(1) - c1;
				c2 = GC.CollectionCount(2) - c2;

				Console.WriteLine("{0,-13} | {1,-21} | {2,10} | {3,3} {4,3} {5,3} | {6,10} |",
					test.Framework, "MemStream Serialize", sw.ElapsedMilliseconds, c0, c1, c2, size);
			}

			/* Deerialize part */

			{
				GC.Collect();
				GC.WaitForPendingFinalizers();
				GC.Collect();

				var c0 = GC.CollectionCount(0);
				var c1 = GC.CollectionCount(1);
				var c2 = GC.CollectionCount(2);

				var sw = Stopwatch.StartNew();

				var received = test.Deserialize();

				sw.Stop();

				c0 = GC.CollectionCount(0) - c0;
				c1 = GC.CollectionCount(1) - c1;
				c2 = GC.CollectionCount(2) - c2;

				Console.WriteLine("{0,-13} | {1,-21} | {2,10} | {3,3} {4,3} {5,3} | {6,10} |",
					test.Framework, "MemStream Deserialize", sw.ElapsedMilliseconds, c0, c1, c2, "");

				for (int i = 0; i < msgs.Length; ++i)
				{
					var msg1 = msgs[i];
					var msg2 = received[i];

					msg1.Compare(msg2);
				}
			}
		}

		static void Test(INetTest test, MessageBase[] msgs)
		{
			test.Prepare(msgs.Length);

			GC.Collect();
			GC.WaitForPendingFinalizers();
			GC.Collect();

			var c0 = GC.CollectionCount(0);
			var c1 = GC.CollectionCount(1);
			var c2 = GC.CollectionCount(2);

			var sw = Stopwatch.StartNew();

			var received = test.Test(msgs);

			sw.Stop();

			c0 = GC.CollectionCount(0) - c0;
			c1 = GC.CollectionCount(1) - c1;
			c2 = GC.CollectionCount(2) - c2;

			Console.WriteLine("{0,-13} | {1,-21} | {2,10} | {3,3} {4,3} {5,3} | {6,10} |",
				test.Framework, "NetTest", sw.ElapsedMilliseconds, c0, c1, c2, "");

			for (int i = 0; i < msgs.Length; ++i)
			{
				var msg1 = msgs[i];
				var msg2 = received[i];

				msg1.Compare(msg2);
			}
		}
	}
}