static void Main(string[] args)
{
bool runForever = true;
string userInput = "";
string filename = "";
string key = "";
long contentLength = 0;
Stream stream = null;
List <Chunk> chunks;
List <string> keys;
ObjectMetadata md;
int numObjects;
int numChunks;
long logicalBytes;
long physicalBytes;
decimal dedupeRatioX;
decimal dedupeRatioPercent;
Initialize();
while (runForever)
{
Console.Write("Command [? for help] > ");
userInput = Console.ReadLine();
if (String.IsNullOrEmpty(userInput))
{
continue;
}
switch (userInput)
{
case "?":
Console.WriteLine("Available commands:");
Console.WriteLine(" q quit");
Console.WriteLine(" cls clear the screen");
Console.WriteLine(" store store an object");
Console.WriteLine(" retrieve retrieve an object");
Console.WriteLine(" delete delete an object");
Console.WriteLine(" metadata retrieve object metadata");
Console.WriteLine(" list list objects in the index");
Console.WriteLine(" exists check if object exists in the index");
Console.WriteLine(" stats list index stats");
Console.WriteLine("");
break;
case "q":
case "Q":
runForever = false;
break;
case "cls":
Console.Clear();
break;
case "store":
filename = InputString("Input filename:", null, false);
key = InputString("Object key:", null, false);
contentLength = GetContentLength(filename);
using (FileStream fs = new FileStream(filename, FileMode.Open))
{
if (_Dedupe.StoreObject(key, contentLength, fs, out chunks))
{
if (chunks != null && chunks.Count > 0)
{
Console.WriteLine("Success: " + chunks.Count + " chunks");
}
else
{
Console.WriteLine("Success (no chunks)");
}
}
else
{
Console.WriteLine("Failed");
}
}
break;
case "retrieve":
key = InputString("Object key:", null, false);
filename = InputString("Output filename:", null, false);
if (_Dedupe.RetrieveObject(key, out contentLength, out stream))
{
if (contentLength > 0)
{
using (FileStream fs = new FileStream(filename, FileMode.OpenOrCreate))
{
int bytesRead = 0;
long bytesRemaining = contentLength;
byte[] readBuffer = new byte[65536];
while (bytesRemaining > 0)
{
bytesRead = stream.Read(readBuffer, 0, readBuffer.Length);
if (bytesRead > 0)
{
fs.Write(readBuffer, 0, bytesRead);
bytesRemaining -= bytesRead;
}
}
}
Console.WriteLine("Success");
}
else
{
Console.WriteLine("Success, (no data)");
}
}
else
{
Console.WriteLine("Failed");
}
break;
case "delete":
key = InputString("Object key:", null, false);
if (_Dedupe.DeleteObject(key))
{
Console.WriteLine("Success");
}
else
{
Console.WriteLine("Failed");
}
break;
case "metadata":
key = InputString("Object key:", null, false);
if (_Dedupe.RetrieveObjectMetadata(key, true, out md))
{
Console.WriteLine("Success");
Console.WriteLine(md.ToString());
}
else
{
Console.WriteLine("Failed");
}
break;
case "list":
_Dedupe.ListObjects(out keys);
if (keys != null && keys.Count > 0)
{
Console.WriteLine("Objects: ");
foreach (string curr in keys)
{
Console.WriteLine(" " + curr);
}
Console.WriteLine(keys.Count + " objects listed");
}
break;
case "exists":
key = InputString("Object name:", null, false);
if (_Dedupe.ObjectExists(key))
{
Console.WriteLine("Object exists");
}
else
{
Console.WriteLine("Object does not exist");
}
break;
case "stats":
if (_Dedupe.IndexStats(out numObjects, out numChunks, out logicalBytes, out physicalBytes, out dedupeRatioX, out dedupeRatioPercent))
{
Console.WriteLine("Statistics:");
Console.WriteLine(" Number of objects : " + numObjects);
Console.WriteLine(" Number of chunks : " + numChunks);
Console.WriteLine(" Logical bytes : " + logicalBytes + " bytes");
Console.WriteLine(" Physical bytes : " + physicalBytes + " bytes");
Console.WriteLine(" Dedupe ratio : " + DecimalToString(dedupeRatioX) + "X, " + DecimalToString(dedupeRatioPercent) + "%");
Console.WriteLine("");
}
else
{
Console.WriteLine("Failed");
}
break;
default:
break;
}
}
}
static void Main(string[] args)
{
try
{
#region Parse-Arguments
if (args == null || args.Length < 2)
{
Usage("No arguments specified");
return;
}
IndexFile = args[0];
Command = args[1];
for (int i = 2; i < args.Length; i++)
{
if (String.IsNullOrEmpty(args[i]))
{
continue;
}
if (args[i].StartsWith("--chunks=") && args[i].Length > 9)
{
ChunkDirectory = args[i].Substring(9);
if (!ChunkDirectory.EndsWith("\\"))
{
ChunkDirectory += "\\";
}
if (!Directory.Exists(ChunkDirectory))
{
Directory.CreateDirectory(ChunkDirectory);
}
}
else if (args[i].StartsWith("--key=") && args[i].Length > 6)
{
ObjectKey = args[i].Substring(6);
}
else if (String.Compare(args[i], "--debug") == 0)
{
DebugDedupe = true;
}
else if (String.Compare(args[i], "--debugsql") == 0)
{
DebugSql = true;
}
else if (args[i].StartsWith("--params=") && args[i].Length > 9)
{
CreateParams = args[i].Substring(9);
if (new Regex(@"^\d+,\d+,\d+,\d+$").IsMatch(CreateParams))
{
string[] currParams = CreateParams.Split(',');
if (currParams.Length != 4)
{
Usage("Value for 'params' is invalid");
return;
}
if (!Int32.TryParse(currParams[0], out MinChunkSize) ||
!Int32.TryParse(currParams[1], out MaxChunkSize) ||
!Int32.TryParse(currParams[2], out ShiftCount) ||
!Int32.TryParse(currParams[3], out BoundaryCheckBytes)
)
{
Usage("Value for 'params' is not of the form int,int,int,int");
return;
}
}
else
{
Usage("Value for 'params' is not of the form int,int,int,int");
return;
}
}
else
{
Usage("Unknown argument: " + args[i]);
return;
}
}
#endregion
#region Verify-Values
List <string> validCommands = new List <string>()
{
"create", "stats", "store", "retrieve", "delete", "list", "exists", "metadata"
};
if (!validCommands.Contains(Command))
{
Usage("Invalid command: " + Command);
return;
}
#endregion
#region Enumerate
if (DebugDedupe)
{
Console.WriteLine("Command : " + Command);
Console.WriteLine("Index File : " + IndexFile);
Console.WriteLine("Chunk Directory : " + ChunkDirectory);
Console.WriteLine("Object Key : " + ObjectKey);
if (MinChunkSize > 0)
{
Console.WriteLine("Min Chunk Size : " + MinChunkSize);
}
if (MaxChunkSize > 0)
{
Console.WriteLine("Max Chunk Size : " + MaxChunkSize);
}
if (ShiftCount > 0)
{
Console.WriteLine("Shift Count : " + ShiftCount);
}
if (BoundaryCheckBytes > 0)
{
Console.WriteLine("Boundary Bytes : " + BoundaryCheckBytes);
}
}
#endregion
#region Create
if (String.Compare(Command, "create") == 0)
{
Dedupe = new DedupeLibrary(IndexFile, MinChunkSize, MaxChunkSize, ShiftCount, BoundaryCheckBytes, WriteChunk, ReadChunk, DeleteChunk, DebugDedupe, DebugSql);
if (DebugDedupe)
{
Console.WriteLine("Successfully wrote new index: " + IndexFile);
}
return;
}
#endregion
#region Initialize-Index
if (!File.Exists(IndexFile))
{
Console.WriteLine("*** Index file " + IndexFile + " not found");
}
Dedupe = new DedupeLibrary(IndexFile, WriteChunk, ReadChunk, DeleteChunk, DebugDedupe, DebugSql);
#endregion
#region Process-by-Command
switch (Command)
{
case "stats":
if (Dedupe.IndexStats(out NumObjects, out NumChunks, out LogicalBytes, out PhysicalBytes, out DedupeRatioX, out DedupeRatioPercent))
{
Console.WriteLine("Statistics:");
Console.WriteLine(" Number of objects : " + NumObjects);
Console.WriteLine(" Number of chunks : " + NumChunks);
Console.WriteLine(" Logical bytes : " + LogicalBytes + " bytes");
Console.WriteLine(" Physical bytes : " + PhysicalBytes + " bytes");
Console.WriteLine(" Dedupe ratio : " + DecimalToString(DedupeRatioX) + "X, " + DecimalToString(DedupeRatioPercent) + "%");
return;
}
else
{
Console.WriteLine("Failed");
}
return;
case "retrieve":
if (String.IsNullOrEmpty(ObjectKey))
{
Usage("Object key must be supplied");
}
else
{
if (!Dedupe.RetrieveObject(ObjectKey, out ResponseData))
{
Console.WriteLine("Failed");
}
else
{
WriteConsoleData(ResponseData);
}
}
return;
case "store":
if (String.IsNullOrEmpty(ObjectKey))
{
Usage("Object key must be supplied");
}
else
{
if (Dedupe.ObjectExists(ObjectKey))
{
Console.WriteLine("Already exists");
}
else
{
ReadConsoleData();
if (!Dedupe.StoreObject(ObjectKey, RequestData, out Chunks))
{
Console.WriteLine("Failed");
}
else
{
Console.WriteLine("Success");
}
}
}
return;
case "delete":
if (String.IsNullOrEmpty(ObjectKey))
{
Usage("Object key must be supplied");
}
else
{
if (!Dedupe.DeleteObject(ObjectKey))
{
Console.WriteLine("Failed");
}
else
{
Console.WriteLine("Success");
}
}
return;
case "metadata":
if (String.IsNullOrEmpty(ObjectKey))
{
Usage("Object key must be supplied");
}
else
{
if (!Dedupe.RetrieveObjectMetadata(ObjectKey, out Metadata))
{
Console.WriteLine("Failed");
}
else
{
Console.WriteLine(Metadata.ToString());
}
}
return;
case "list":
Dedupe.ListObjects(out Keys);
if (Keys == null || Keys.Count < 1)
{
Console.WriteLine("No objects");
}
else
{
Console.WriteLine("Objects:");
foreach (string curr in Keys)
{
Console.WriteLine(" " + curr);
}
Console.WriteLine(Keys.Count + " objects in index");
}
return;
case "exists":
if (String.IsNullOrEmpty(ObjectKey))
{
Usage("Object key must be supplied");
return;
}
else
{
Console.WriteLine(Dedupe.ObjectExists(ObjectKey));
}
return;
default:
Usage("Unknown command: " + Command);
return;
}
#endregion
}
catch (Exception e)
{
ExceptionConsole("Dedupe", "Outer exception", e);
}
}