mirror of
https://github.com/microsoft/PowerToys.git
synced 2024-12-14 11:39:16 +08:00
118 lines
3.8 KiB
C#
118 lines
3.8 KiB
C#
using System;
|
|
using System.IO;
|
|
using System.Reflection;
|
|
using System.Runtime.Serialization;
|
|
using System.Runtime.Serialization.Formatters;
|
|
using System.Runtime.Serialization.Formatters.Binary;
|
|
using Wox.Infrastructure.Logger;
|
|
|
|
namespace Wox.Infrastructure.Storage
|
|
{
|
|
/// <summary>
|
|
/// Stroage object using binary data
|
|
/// Normally, it has better performance, but not readable
|
|
/// </summary>
|
|
public class BinaryStorage<T>
|
|
{
|
|
public BinaryStorage(string filename)
|
|
{
|
|
const string directoryName = "Cache";
|
|
var directoryPath = Path.Combine(Constant.DataDirectory, directoryName);
|
|
Helper.ValidateDirectory(directoryPath);
|
|
|
|
const string fileSuffix = ".cache";
|
|
FilePath = Path.Combine(directoryPath, $"{filename}{fileSuffix}");
|
|
}
|
|
|
|
public string FilePath { get; }
|
|
|
|
public T TryLoad(T defaultData)
|
|
{
|
|
if (File.Exists(FilePath))
|
|
{
|
|
using (var stream = new FileStream(FilePath, FileMode.Open))
|
|
{
|
|
if (stream.Length > 0)
|
|
{
|
|
var d = Deserialize(stream, defaultData);
|
|
return d;
|
|
}
|
|
else
|
|
{
|
|
Log.Error($"|BinaryStorage.TryLoad|Zero length cache file <{FilePath}>");
|
|
Save(defaultData);
|
|
return defaultData;
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
Log.Info("|BinaryStorage.TryLoad|Cache file not exist, load default data");
|
|
Save(defaultData);
|
|
return defaultData;
|
|
}
|
|
}
|
|
|
|
private T Deserialize(FileStream stream, T defaultData)
|
|
{
|
|
//http://stackoverflow.com/questions/2120055/binaryformatter-deserialize-gives-serializationexception
|
|
AppDomain.CurrentDomain.AssemblyResolve += CurrentDomain_AssemblyResolve;
|
|
BinaryFormatter binaryFormatter = new BinaryFormatter
|
|
{
|
|
AssemblyFormat = FormatterAssemblyStyle.Simple
|
|
};
|
|
|
|
try
|
|
{
|
|
var t = ((T)binaryFormatter.Deserialize(stream)).NonNull();
|
|
return t;
|
|
}
|
|
catch (System.Exception e)
|
|
{
|
|
Log.Exception($"|BinaryStorage.Deserialize|Deserialize error for file <{FilePath}>", e);
|
|
return defaultData;
|
|
}
|
|
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;
|
|
}
|
|
|
|
public void Save(T data)
|
|
{
|
|
using (var stream = new FileStream(FilePath, FileMode.Create))
|
|
{
|
|
BinaryFormatter binaryFormatter = new BinaryFormatter
|
|
{
|
|
AssemblyFormat = FormatterAssemblyStyle.Simple
|
|
};
|
|
|
|
try
|
|
{
|
|
binaryFormatter.Serialize(stream, data);
|
|
}
|
|
catch (SerializationException e)
|
|
{
|
|
Log.Exception($"|BinaryStorage.Save|serialize error for file <{FilePath}>", e);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|