public void TestMultiThread()
{
const string data = "ABAB ABAB ABAB ABAB ABAB ABAB";
byte[] bytes = Encoding.UTF8.GetBytes(data);
using (var input = new MemoryStream(bytes))
{
using (var output = new MemoryStream())
{
var compress = new ParallelFileCompress(input, output);
compress.Run(5, 2, CancellationToken.None);
// seek compressed to beginning
output.Seek(0, SeekOrigin.Begin);
// reset original stream
input.Seek(0, SeekOrigin.Begin);
input.SetLength(0);
var decompressor = new ParallelFileDecompress(output, input);
decompressor.Run(1, CancellationToken.None);
input.Seek(0, SeekOrigin.Begin);
var decompressedBytes = input.ToArray();
CollectionAssert.AreEqual(bytes, decompressedBytes);
}
}
}
static int Main(string[] args)
{
if (args.Length != 3)
{
Console.WriteLine("Invalid number arguments... Expecting 3 arguments.\n" + Usage);
return(FAILURE);
}
CompressionMode compressionMode;
if (args[0].Equals("compress", StringComparison.OrdinalIgnoreCase))
{
compressionMode = CompressionMode.Compress;
}
else if (args[0].Equals("decompress", StringComparison.OrdinalIgnoreCase))
{
compressionMode = CompressionMode.Decompress;
}
else
{
Console.WriteLine("First argument did not have valid value...\n" + Usage);
return(FAILURE);
}
string inputPath = args[1];
string outputPath = args[2];
using (CancellationTokenSource tokenSource = new CancellationTokenSource())
{
Console.CancelKeyPress += (sender, e) =>
{
e.Cancel = true;
tokenSource.Cancel();
};
var token = tokenSource.Token;
ParallelFileIO parallelWorker;
try
{
if (compressionMode == CompressionMode.Compress)
{
parallelWorker = new ParallelFileCompress(inputPath, outputPath);
}
else if (compressionMode == CompressionMode.Decompress)
{
parallelWorker = new ParallelFileDecompress2(inputPath, outputPath);
//parallelWorker = new ParallelFileDecompress(inputPath, outputPath);
}
else
{
throw new ApplicationException("Uknown compression mode.");
}
parallelWorker.Run(token);
if (token.IsCancellationRequested)
{
Console.WriteLine("Operation canceled by user.");
return(FAILURE);
}
}
catch (FileNotFoundException ex)
{
// FileNotFoundException should provide nice user-readable exception message.
Console.WriteLine(ex.Message);
return(FAILURE);
}
catch (DirectoryNotFoundException ex)
{
Console.WriteLine("Directory not found... " + ex.Message);
return(FAILURE);
}
catch (UnauthorizedAccessException ex)
{
Console.WriteLine($"Program does not have required permissions. {ex.Message}");
return(FAILURE);
}
catch (OperationCanceledException ex) when(ex.CancellationToken == token)
{
Console.WriteLine("Operation canceled by user.");
return(FAILURE);
}
catch (PathTooLongException ex)
{
Console.WriteLine($"One of the given paths is too long..." + ex.Message);
return(FAILURE);
}
catch (IOException ex)
{
if (!string.IsNullOrEmpty(ex.Message))
{
Console.WriteLine(ex.Message);
}
else
{
Console.WriteLine($"IOException occured. HResult: {ex.HResult}");
}
return(FAILURE);
}
catch (Exception ex)
{
// the exception message is localized, so the user should work out what went wrong
Console.WriteLine($"Generic exception occured: " + ex.Message);
return(FAILURE);
}
}
return(SUCCESS);
}
