我们有一个项目,我们需要迁移最多10GB的二进制文件。步骤是1)按消息大小读取文件2)进行一些处理3)将原始消息或处理过的消息写回新的二进制文件。
对于10GB文件,处理后它变为14GB。目前需要近2个小时。
我想知道我是否可以做一些IO技巧来缩短时间。
using (FileStream fsInput =new FileStream(inputfilename, FileMode.Open, FileAccess.Read, FileShare.ReadWrite))
{
using (FileStream fsOutput = File.Create(outputfilename))
{
long total = fsInput.Length;
long progress = 0;
unsafe
{
int hdrSize = sizeof(FullMessageHeader);
byte[] headerBuffer = new byte[hdrSize];
while (fsInput.Position < fsInput.Length)
{
progress += fsInput.Read(headerBuffer, 0, hdrSize);
int msgSize = 0;
fixed (byte* hdr = headerBuffer)
{
msgSize = *(int*)(hdr + MessageHeaderOffsets.Size);
}
byte[] msg = new byte[msgSize];
Buffer.BlockCopy(headerBuffer, 0, msg, 0, headerBuffer.Length);
fsInput.Position -= hdrSize;
progress += fsInput.Read(msg, 0, msgSize);
fixed (byte* ptr = msg)
{
byte[] ba = ProcessMessage(ptr);
if (ba.Length == 0)
{
fsOutput.Write(msg, 0, msg.Length);
}
else
{
fsOutput.Write(ba, 0, ba.Length);
}
}
}
}
}
}
最后,正如Georg所建议的,以及上面代码的一些清理,我能够将时间从2小时减少到10分钟,这不是最佳但可接受的。
Dictionary<int, byte[]> byteDictionary = new Dictionary<int, byte[]>();
using (FileStream fsInput =new FileStream(inputfilename, FileMode.Open, FileAccess.Read, FileShare.ReadWrite))
{
using (FileStream fsOutput = File.Create(outputfilename))
{
long total = fsInput.Length;
long progress = 0;
unsafe
{
int hdrSize = sizeof(FullMessageHeader);
byte[] headerBuffer = new byte[hdrSize];
while (fsInput.Position < fsInput.Length)
{
progress += fsInput.Read(headerBuffer, 0, hdrSize);
int msgSize = 0;
fixed (byte* hdr = headerBuffer)
{
msgSize = *(int*)(hdr + MessageHeaderOffsets.Size);
}
byte[] msg = byteDictionary.ContainsKey(msgSize)
? byteDictionary[msgSize]
: new byte[msgSize];
if (!byteDictionary.ContainsKey(msgSize))
{
byteDictionary[msgSize] = msg;
}
//byte[] msg = new byte[msgSize];
//Buffer.BlockCopy(headerBuffer, 0, msg, 0, headerBuffer.Length);
fsInput.Position -= hdrSize;
progress += fsInput.Read(msg, 0, msgSize);
fixed (byte* ptr = msg)
{
//fsOutput.Write(msg,0,msg.Length);
byte[] ba = ProcessMessage(ptr);
if (ba.Length == 0)
{
fsOutput.Write(msg, 0, msg.Length);
}
else
{
fsOutput.Write(ba, 0, ba.Length);
}
}
}
}
}
}