From eb8f880803ee6577263d53c75442b55fe399a8a6 Mon Sep 17 00:00:00 2001
From: Chris Wilson <chris+github@qwirx.com>
Date: Fri, 7 Feb 2014 22:06:23 +0000
Subject: Add an AccountUsage2 command to backup protocol.

Allows us to get information about the account that's not available
using the old AccountUsage command. Currently only used in tests.
---
 lib/backupstore/BackupCommands.cpp | 46 ++++++++++++++++++++++++++++++++++++++
 1 file changed, 46 insertions(+)

(limited to 'lib/backupstore/BackupCommands.cpp')

diff --git a/lib/backupstore/BackupCommands.cpp b/lib/backupstore/BackupCommands.cpp
index d2326869..471ed56a 100644
--- a/lib/backupstore/BackupCommands.cpp
+++ b/lib/backupstore/BackupCommands.cpp
@@ -953,3 +953,49 @@ std::auto_ptr<BackupProtocolMessage> BackupProtocolGetIsAlive::DoCommand(BackupP
 	//
 	return std::auto_ptr<BackupProtocolMessage>(new BackupProtocolIsAlive());
 }
+
+// --------------------------------------------------------------------------
+//
+// Function
+//		Name:    BackupProtocolGetAccountUsage2::DoCommand(BackupProtocolReplyable &, BackupStoreContext &)
+//		Purpose: Return the amount of disc space used
+//		Created: 26/12/13
+//
+// --------------------------------------------------------------------------
+std::auto_ptr<BackupProtocolMessage> BackupProtocolGetAccountUsage2::DoCommand(
+	BackupProtocolReplyable &rProtocol, BackupStoreContext &rContext) const
+{
+	CHECK_PHASE(Phase_Commands)
+
+	// Get store info from context
+	const BackupStoreInfo &info(rContext.GetBackupStoreInfo());
+	
+	// Find block size
+	RaidFileController &rcontroller(RaidFileController::GetController());
+	RaidFileDiscSet &rdiscSet(rcontroller.GetDiscSet(info.GetDiscSetNumber()));
+	
+	// Return info
+	BackupProtocolAccountUsage2* usage = new BackupProtocolAccountUsage2();
+	std::auto_ptr<BackupProtocolMessage> reply(usage);
+	#define COPY(name) usage->Set ## name (info.Get ## name ())
+	COPY(AccountName);
+	usage->SetAccountEnabled(info.IsAccountEnabled());
+	COPY(ClientStoreMarker);
+	usage->SetBlockSize(rdiscSet.GetBlockSize());
+	COPY(LastObjectIDUsed);
+	COPY(BlocksUsed);
+	COPY(BlocksInCurrentFiles);
+	COPY(BlocksInOldFiles);
+	COPY(BlocksInDeletedFiles);
+	COPY(BlocksInDirectories);
+	COPY(BlocksSoftLimit);
+	COPY(BlocksHardLimit);
+	COPY(NumCurrentFiles);
+	COPY(NumOldFiles);
+	COPY(NumDeletedFiles);
+	COPY(NumDirectories);
+	#undef COPY
+
+	return reply;
+}
+
-- 
cgit v1.2.3