source: box/trunk/bin/bbackupd/BackupClientDeleteList.cpp @ 2983

// --------------------------------------------------------------------------
//
// File
//              Name:    BackupClientDeleteList.cpp
//              Purpose: List of pending deletes for backup
//              Created: 10/11/03
//
// --------------------------------------------------------------------------

#include "Box.h"

#include <algorithm>

#include "BackupClientDeleteList.h"
#include "BackupClientContext.h"
#include "autogen_BackupProtocol.h"

#include "MemLeakFindOn.h"


// --------------------------------------------------------------------------
//
// Function
//              Name:    BackupClientDeleteList::BackupClientDeleteList()
//              Purpose: Constructor
//              Created: 10/11/03
//
// --------------------------------------------------------------------------
BackupClientDeleteList::BackupClientDeleteList()
{
}

// --------------------------------------------------------------------------
//
// Function
//              Name:    BackupClientDeleteList::~BackupClientDeleteList()
//              Purpose: Destructor
//              Created: 10/11/03
//
// --------------------------------------------------------------------------
BackupClientDeleteList::~BackupClientDeleteList()
{
}

BackupClientDeleteList::FileToDelete::FileToDelete(int64_t DirectoryID,
        const BackupStoreFilename& rFilename,
        const std::string& rLocalPath)
: mDirectoryID(DirectoryID),
  mFilename(rFilename),
  mLocalPath(rLocalPath)
{ }

BackupClientDeleteList::DirToDelete::DirToDelete(int64_t ObjectID,
        const std::string& rLocalPath)
: mObjectID(ObjectID),
  mLocalPath(rLocalPath)
{ }

// --------------------------------------------------------------------------
//
// Function
//              Name:    BackupClientDeleteList::AddDirectoryDelete(int64_t,
//                       const BackupStoreFilename&)
//              Purpose: Add a directory to the list of directories to be deleted.
//              Created: 10/11/03
//
// --------------------------------------------------------------------------
void BackupClientDeleteList::AddDirectoryDelete(int64_t ObjectID,
        const std::string& rLocalPath)
{
        // Only add the delete to the list if it's not in the "no delete" set
        if(mDirectoryNoDeleteList.find(ObjectID) ==
                mDirectoryNoDeleteList.end())
        {
                // Not in the list, so should delete it
                mDirectoryList.push_back(DirToDelete(ObjectID, rLocalPath));
        }
}


// --------------------------------------------------------------------------
//
// Function
//              Name:    BackupClientDeleteList::AddFileDelete(int64_t,
//                       const BackupStoreFilename &)
//              Purpose: 
//              Created: 10/11/03
//
// --------------------------------------------------------------------------
void BackupClientDeleteList::AddFileDelete(int64_t DirectoryID,
        const BackupStoreFilename &rFilename, const std::string& rLocalPath)
{
        // Try to find it in the no delete list
        std::vector<std::pair<int64_t, BackupStoreFilename> >::iterator
                delEntry(mFileNoDeleteList.begin());
        while(delEntry != mFileNoDeleteList.end())
        {
                if((delEntry)->first == DirectoryID 
                        && (delEntry)->second == rFilename)
                {
                        // Found!
                        break;
                }
                ++delEntry;
        }
        
        // Only add it to the delete list if it wasn't in the no delete list
        if(delEntry == mFileNoDeleteList.end())
        {
                mFileList.push_back(FileToDelete(DirectoryID, rFilename,
                        rLocalPath));
        }
}


        
// --------------------------------------------------------------------------
//
// Function
//              Name:    BackupClientDeleteList::PerformDeletions(BackupClientContext &rContext)
//              Purpose: Perform all the pending deletes
//              Created: 10/11/03
//
// --------------------------------------------------------------------------
void BackupClientDeleteList::PerformDeletions(BackupClientContext &rContext)
{
        // Anything to do?
        if(mDirectoryList.empty() && mFileList.empty())
        {
                // Nothing!
                return;
        }
        
        // Get a connection
        BackupProtocolClient &connection(rContext.GetConnection());
        
        // Do the deletes
        for(std::vector<DirToDelete>::iterator i(mDirectoryList.begin());
                i != mDirectoryList.end(); ++i)
        {
                connection.QueryDeleteDirectory(i->mObjectID);
                rContext.GetProgressNotifier().NotifyDirectoryDeleted(
                        i->mObjectID, i->mLocalPath);
        }
        
        // Clear the directory list
        mDirectoryList.clear();
        
        // Delete the files
        for(std::vector<FileToDelete>::iterator i(mFileList.begin());
                i != mFileList.end(); ++i)
        {
                connection.QueryDeleteFile(i->mDirectoryID, i->mFilename);
                rContext.GetProgressNotifier().NotifyFileDeleted(
                        i->mDirectoryID, i->mLocalPath);
        }
}


// --------------------------------------------------------------------------
//
// Function
//              Name:    BackupClientDeleteList::StopDirectoryDeletion(int64_t)
//              Purpose: Stop a directory being deleted
//              Created: 19/11/03
//
// --------------------------------------------------------------------------
void BackupClientDeleteList::StopDirectoryDeletion(int64_t ObjectID)
{
        // First of all, is it in the delete vector?
        std::vector<DirToDelete>::iterator delEntry(mDirectoryList.begin());
        for(; delEntry != mDirectoryList.end(); delEntry++)
        {
                if(delEntry->mObjectID == ObjectID)
                {
                        // Found!
                        break;
                }
        }
        if(delEntry != mDirectoryList.end())
        {
                // erase this entry
                mDirectoryList.erase(delEntry);
        }
        else
        {
                // Haven't been asked to delete it yet, put it in the
                // no delete list
                mDirectoryNoDeleteList.insert(ObjectID);
        }
}


// --------------------------------------------------------------------------
//
// Function
//              Name:    BackupClientDeleteList::StopFileDeletion(int64_t, const BackupStoreFilename &)
//              Purpose: Stop a file from being deleted
//              Created: 19/11/03
//
// --------------------------------------------------------------------------
void BackupClientDeleteList::StopFileDeletion(int64_t DirectoryID,
        const BackupStoreFilename &rFilename)
{
        // Find this in the delete list
        std::vector<FileToDelete>::iterator delEntry(mFileList.begin());
        while(delEntry != mFileList.end())
        {
                if(delEntry->mDirectoryID == DirectoryID
                        && delEntry->mFilename == rFilename)
                {
                        // Found!
                        break;
                }
                ++delEntry;
        }
        
        if(delEntry != mFileList.end())
        {
                // erase this entry
                mFileList.erase(delEntry);
        }
        else
        {
                // Haven't been asked to delete it yet, put it in the no delete list
                mFileNoDeleteList.push_back(std::pair<int64_t, BackupStoreFilename>(DirectoryID, rFilename));
        }
}