RegistrationStore.cpp [plain text]
#include "config.h"
#include "RegistrationStore.h"
#if ENABLE(SERVICE_WORKER)
#include "SWServerRegistration.h"
namespace WebCore {
RegistrationStore::RegistrationStore(SWServer& server, String&& databaseDirectory)
: m_server(server)
, m_database(RegistrationDatabase::create(*this, WTFMove(databaseDirectory)))
, m_databasePushTimer(*this, &RegistrationStore::pushChangesToDatabase)
{
}
RegistrationStore::~RegistrationStore()
{
}
void RegistrationStore::scheduleDatabasePushIfNecessary()
{
if (m_databasePushTimer.isActive())
return;
m_databasePushTimer.startOneShot(0_s);
}
void RegistrationStore::pushChangesToDatabase(WTF::CompletionHandler<void()>&& completionHandler)
{
Vector<ServiceWorkerContextData> changesToPush;
changesToPush.reserveInitialCapacity(m_updatedRegistrations.size());
for (auto& value : m_updatedRegistrations.values())
changesToPush.uncheckedAppend(WTFMove(value));
m_updatedRegistrations.clear();
m_database->pushChanges(WTFMove(changesToPush), WTFMove(completionHandler));
}
void RegistrationStore::clearAll(WTF::CompletionHandler<void()>&& completionHandler)
{
m_updatedRegistrations.clear();
m_databasePushTimer.stop();
m_database->clearAll(WTFMove(completionHandler));
}
void RegistrationStore::flushChanges(WTF::CompletionHandler<void()>&& completionHandler)
{
if (m_databasePushTimer.isActive()) {
pushChangesToDatabase(WTFMove(completionHandler));
m_databasePushTimer.stop();
return;
}
completionHandler();
}
void RegistrationStore::updateRegistration(const ServiceWorkerContextData& data)
{
m_updatedRegistrations.set(data.registration.key, data);
scheduleDatabasePushIfNecessary();
}
void RegistrationStore::removeRegistration(SWServerRegistration& registration)
{
ServiceWorkerContextData contextData;
contextData.registration.key = registration.key();
m_updatedRegistrations.set(registration.key(), WTFMove(contextData));
scheduleDatabasePushIfNecessary();
}
void RegistrationStore::addRegistrationFromDatabase(ServiceWorkerContextData&& context)
{
m_server.addRegistrationFromStore(WTFMove(context));
}
void RegistrationStore::databaseFailedToOpen()
{
m_server.registrationStoreDatabaseFailedToOpen();
}
void RegistrationStore::databaseOpenedAndRecordsImported()
{
m_server.registrationStoreImportComplete();
}
}
#endif // ENABLE(SERVICE_WORKER)