OriginUsageRecord.cpp [plain text]
#include "config.h"
#include "OriginUsageRecord.h"
#include "FileSystem.h"
#include <limits>
namespace WebCore {
unsigned long long OriginUsageRecord::unknownDiskUsage()
{
return std::numeric_limits<unsigned long long>::max();
}
OriginUsageRecord::OriginUsageRecord()
: m_diskUsage(unknownDiskUsage())
{
}
void OriginUsageRecord::addDatabase(const String& identifier, const String& fullPath)
{
ASSERT(!m_databaseMap.contains(identifier));
m_databaseMap.set(identifier, DatabaseEntry(fullPath, unknownDiskUsage()));
m_unknownSet.add(identifier);
m_diskUsage = unknownDiskUsage();
}
void OriginUsageRecord::removeDatabase(const String& identifier)
{
ASSERT(m_databaseMap.contains(identifier));
m_diskUsage = unknownDiskUsage();
m_databaseMap.remove(identifier);
m_unknownSet.remove(identifier);
}
void OriginUsageRecord::markDatabase(const String& identifier)
{
m_unknownSet.add(identifier);
m_diskUsage = unknownDiskUsage();
}
unsigned long long OriginUsageRecord::diskUsage()
{
if (m_diskUsage != unknownDiskUsage())
return m_diskUsage;
HashSet<String>::iterator iUnknown = m_unknownSet.begin();
HashSet<String>::iterator endUnknown = m_unknownSet.end();
for (; iUnknown != endUnknown; ++iUnknown) {
String path = m_databaseMap.get(*iUnknown).filename;
ASSERT(!path.isEmpty());
long long size;
if (getFileSize(path, size))
m_databaseMap.set(*iUnknown, DatabaseEntry(path, size));
else {
m_databaseMap.set(*iUnknown, DatabaseEntry(path, 0));
}
}
m_unknownSet.clear();
HashMap<String, DatabaseEntry>::iterator iDatabase = m_databaseMap.begin();
HashMap<String, DatabaseEntry>::iterator endDatabase = m_databaseMap.end();
m_diskUsage = 0;
for (; iDatabase != endDatabase; ++iDatabase) {
m_diskUsage += iDatabase->second.size;
}
return m_diskUsage;
}
}