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(CompletionHandler<void()>&& completionHandler)
{
if (m_isSuspended) {
m_needsPushingChanges = true;
return;
}
m_database->pushChanges(m_updatedRegistrations, WTFMove(completionHandler));
m_updatedRegistrations.clear();
}
void RegistrationStore::clearAll(CompletionHandler<void()>&& completionHandler)
{
m_needsPushingChanges = false;
m_updatedRegistrations.clear();
m_databasePushTimer.stop();
m_database->clearAll(WTFMove(completionHandler));
}
void RegistrationStore::flushChanges(CompletionHandler<void()>&& completionHandler)
{
if (m_databasePushTimer.isActive()) {
pushChangesToDatabase(WTFMove(completionHandler));
m_databasePushTimer.stop();
return;
}
completionHandler();
}
void RegistrationStore::startSuspension(CompletionHandler<void()>&& completionHandler)
{
m_isSuspended = true;
closeDatabase(WTFMove(completionHandler));
}
void RegistrationStore::closeDatabase(CompletionHandler<void()>&& completionHandler)
{
m_database->close(WTFMove(completionHandler));
}
void RegistrationStore::endSuspension()
{
m_isSuspended = false;
if (m_needsPushingChanges)
scheduleDatabasePushIfNecessary();
}
void RegistrationStore::updateRegistration(const ServiceWorkerContextData& data)
{
ASSERT(isMainThread());
ASSERT(!data.registration.key.isEmpty());
if (data.registration.key.isEmpty())
return;
m_updatedRegistrations.set(data.registration.key, data);
scheduleDatabasePushIfNecessary();
}
void RegistrationStore::removeRegistration(const ServiceWorkerRegistrationKey& key)
{
ASSERT(isMainThread());
ASSERT(!key.isEmpty());
if (key.isEmpty())
return;
m_updatedRegistrations.set(key, WTF::nullopt);
scheduleDatabasePushIfNecessary();
}
void RegistrationStore::addRegistrationFromDatabase(ServiceWorkerContextData&& data)
{
ASSERT(!data.registration.key.isEmpty());
if (data.registration.key.isEmpty())
return;
m_server.addRegistrationFromStore(WTFMove(data));
}
void RegistrationStore::databaseFailedToOpen()
{
m_server.registrationStoreDatabaseFailedToOpen();
}
void RegistrationStore::databaseOpenedAndRecordsImported()
{
m_server.registrationStoreImportComplete();
}
}
#endif // ENABLE(SERVICE_WORKER)