protected static CloudQueueMessage CreateJobQueueMessage(TestMsg message)
{
DataContractJsonSerializer serializer = new DataContractJsonSerializer(message.GetType());
MemoryStream memoryStream = new MemoryStream();
serializer.WriteObject(memoryStream, message);
return(new CloudQueueMessage(System.Text.Encoding.Default.GetString(memoryStream.GetBuffer())));
}
public async Task Start(string[] args)
{
// Parse the arguments.
if (!ParseArguments(args, out uint levelOfConcurrency))
{
// If invalid arguments were provided, exit the role.
return;
}
// Wait for a message to appear on the job queue.
while (true)
{
Console.WriteLine("Awaiting a job to perform.");
CloudQueueMessage rawMessage = await jobQueue_.GetMessageAsync();
while (rawMessage == null)
{
await Task.Delay(TimeSpan.FromSeconds(1));
rawMessage = await jobQueue_.GetMessageAsync();
}
TestMsg message = null;
try
{
message = ConvertQueueMessage(rawMessage);
}
catch (Exception ex)
{
// Upon failure to deserialize, log and throw the invalid message away.
Console.WriteLine($"[ERROR] Failed to deserialize message. Details: {ex.Message}");
continue;
}
finally
{
await jobQueue_.DeleteMessageAsync(rawMessage);
}
Console.WriteLine("Message received.");
CloudBlobContainer container = blobClient_.GetContainerReference(message.Operation.ContainerName);
CloudBlockBlob blob = container.GetBlockBlobReference(message.Operation.BlobName);
TestOperation operation = message.Operation;
if (operation is PutBlobOperation || operation is GetBlobOperation)
{
// Define the task to be performed.
Func <Task <TimeSpan> > executeOperation = null;
if (operation is PutBlobOperation)
{
var detailedOperation = operation as PutBlobOperation;
executeOperation = () => { return(UploadBlocksAsync(blobClient_, container, detailedOperation.BlobName, detailedOperation.BlockSizeBytes, detailedOperation.NumBlocks, detailedOperation.StartingBlockId, levelOfConcurrency)); };
}
else if (operation is GetBlobOperation)
{
var detailedOperation = operation as GetBlobOperation;
executeOperation = () => { return(GetBlocksAsync(blobClient_, container, detailedOperation.BlobName, detailedOperation.StartIndex, detailedOperation.LengthBytes, detailedOperation.ChunkSizeBytes, levelOfConcurrency)); };
}
// Perform the task and report corresponding metrics.
DateTimeOffset startTime = DateTimeOffset.UtcNow;
StatusMsg statusMessage = null;
try
{
TimeSpan duration = await executeOperation();
statusMessage = new StatusMsg(operation.ID, true, "NA", startTime, duration);
}
catch (Exception ex)
{
statusMessage = new StatusMsg(operation.ID, false, ex.Message, startTime, DateTimeOffset.UtcNow - startTime);
}
finally
{
DataContractJsonSerializer serializer = new DataContractJsonSerializer(typeof(StatusMsg));
MemoryStream memoryStream = new MemoryStream();
serializer.WriteObject(memoryStream, statusMessage);
string messageStr = System.Text.Encoding.Default.GetString(memoryStream.GetBuffer());
messageStr = messageStr.Substring(0, (int)memoryStream.Length);
await statusQueue_.AddMessageAsync(new CloudQueueMessage(messageStr));
}
}
else
{
Console.WriteLine($"Operation for {message.GetType().FullName} currently unsupported.");
}
}
}
