mirror of
https://github.com/microsoft/PowerToys.git
synced 2024-12-15 20:19:17 +08:00
118 lines
3.9 KiB
C#
118 lines
3.9 KiB
C#
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
|
|
}
|
|
}
|
|
});
|
|
}
|
|
}
|
|
}
|