PowerToys/Wox.Infrastructure/Storage/BinaryStorage.cs

using System;
using System.IO;
using System.Reflection;
using System.Runtime.Serialization.Formatters;
using System.Runtime.Serialization.Formatters.Binary;
using System.Threading;
using Wox.Infrastructure.Logger;

namespace Wox.Infrastructure.Storage
{
    /// <summary>
    /// Stroage object using binary data
    /// Normally, it has better performance, but not readable
    /// You MUST mark implement class as Serializable
    /// </summary>
    [Serializable]
    public abstract class BinaryStorage<T> : BaseStorage<T> where T : class, IStorage, new()
    {
        private static object syncObject = new object();
        protected override string FileSuffix
        {
            get { return ".dat"; }
        }

        protected override void LoadInternal()
        {
            //http://stackoverflow.com/questions/2120055/binaryformatter-deserialize-gives-serializationexception
            AppDomain.CurrentDomain.AssemblyResolve += CurrentDomain_AssemblyResolve;
            try
            {
                using (FileStream fileStream = new FileStream(FilePath, FileMode.Open, FileAccess.Read, FileShare.ReadWrite))
                {
                    if (fileStream.Length > 0)
                    {
                        BinaryFormatter binaryFormatter = new BinaryFormatter
                        {
                            AssemblyFormat = FormatterAssemblyStyle.Simple
                        };
                        serializedObject = binaryFormatter.Deserialize(fileStream) as T;
                        if (serializedObject == null)
                        {
                            serializedObject = LoadDefault();
#if (DEBUG)
                            {
                                throw new System.Exception("deserialize failed");
                            }
#endif
                        }
                    }
                    else
                    {
                        serializedObject = LoadDefault();
                    }
                }
            }
            catch (System.Exception e)
            {
                Log.Error(e);
                serializedObject = LoadDefault();
#if (DEBUG)
                {
                    throw;
                }
#endif
            }
            finally
            {
                AppDomain.CurrentDomain.AssemblyResolve -= CurrentDomain_AssemblyResolve;
            }
        }

        private Assembly CurrentDomain_AssemblyResolve(object sender, ResolveEventArgs args)
        {
            Assembly ayResult = null;
            string sShortAssemblyName = args.Name.Split(',')[0];
            Assembly[] ayAssemblies = AppDomain.CurrentDomain.GetAssemblies();
            foreach (Assembly ayAssembly in ayAssemblies)
            {
                if (sShortAssemblyName == ayAssembly.FullName.Split(',')[0])
                {
                    ayResult = ayAssembly;
                    break;
                }
            }
            return ayResult;
        }

        protected override void SaveInternal()
        {
            ThreadPool.QueueUserWorkItem(o =>
            {
                lock (syncObject)
                {
                    try
                    {
                        FileStream fileStream = new FileStream(FilePath, FileMode.Create);
                        BinaryFormatter binaryFormatter = new BinaryFormatter
                        {
                            AssemblyFormat = FormatterAssemblyStyle.Simple
                        };
                        binaryFormatter.Serialize(fileStream, serializedObject);
                        fileStream.Close();
                    }
                    catch (System.Exception e)
                    {
                        Log.Error(e);
#if (DEBUG)
                        {
                            throw;
                        }
#endif
                    }
                }
            });
        }
    }
}