bankdb.cpp

来自「C++高级编程这本书所附的源代码」· C++ 代码 · 共 57 行

#include "BankDB.h"
#include <utility>
using namespace std;

bool BankDB::addAccount(const BankAccount& acct)
{
  // Declare a variable to store the return from insert()
  pair<map<int, BankAccount>::iterator, bool> res;
  // do the actual insert, using the account number as the key
  res = mAccounts.insert(make_pair(acct.getAcctNum(), acct));

  // Return the bool field of the pair specifying seccess or failure
  return (res.second);
}

void BankDB::deleteAccount(int acctNum)
{
  mAccounts.erase(acctNum);
}

BankAccount& BankDB::findAccount(int acctNum) throw(out_of_range)
{
  // Finding an element via its key can be done with find()
  map<int, BankAccount>::iterator it = mAccounts.find(acctNum);
  if (it == mAccounts.end()) {
    throw (out_of_range("No account with that number."));
  }
  // Remember that iterators into maps refer to pairs of key/value
  return (it->second);
}

BankAccount& BankDB::findAccount(const string& name) throw(out_of_range)
{
  //
  // Finding an element by a non-key attribute requires a linear
  // search through the elements.
  //
  for (map<int, BankAccount>::iterator it = mAccounts.begin();
       it != mAccounts.end(); ++it) {
    if (it->second.getClientName() == name) {
      // found it!
      return (it->second);
    }
  }
  throw (out_of_range("No account with that name."));
}

void BankDB::mergeDatabase(BankDB& db)
{
  // Just insert copies of all the accounts in the old db
  // to the new one.
  mAccounts.insert(db.mAccounts.begin(), db.mAccounts.end());

  // Now delete all the accounts in the old one.
  db.mAccounts.clear();
}