Source/WebCore/workers/service/server/RegistrationStore.cpp

   1 /*
   2  * Copyright (C) 2017 Apple Inc. All rights reserved.
   3  *
   4  * Redistribution and use in source and binary forms, with or without
   5  * modification, are permitted provided that the following conditions
   6  * are met:
   7  * 1. Redistributions of source code must retain the above copyright
   8  *    notice, this list of conditions and the following disclaimer.
   9  * 2. Redistributions in binary form must reproduce the above copyright
  10  *    notice, this list of conditions and the following disclaimer in the
  11  *    documentation and/or other materials provided with the distribution.
  12  *
  13  * THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS''
  14  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
  15  * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  16  * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS
  17  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  18  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  19  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  20  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  21  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  22  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
  23  * THE POSSIBILITY OF SUCH DAMAGE.
  24  */
  25
  26 #include "config.h"
  27 #include "RegistrationStore.h"
  28
  29 #if ENABLE(SERVICE_WORKER)
  30
  31 #include "SWServerRegistration.h"
  32
  33 namespace WebCore {
  34
  35 RegistrationStore::RegistrationStore(SWServer& server, const String& databaseDirectory)
  36     : m_server(server)
  37     , m_database(*this, databaseDirectory)
  38     , m_databasePushTimer(*this, &RegistrationStore::pushChangesToDatabase)
  39 {
  40 }
  41
  42 RegistrationStore::~RegistrationStore()
  43 {
  44     ASSERT(m_database.isClosed());
  45 }
  46
  47 void RegistrationStore::scheduleDatabasePushIfNecessary()
  48 {
  49     if (m_databasePushTimer.isActive())
  50         return;
  51     m_databasePushTimer.startOneShot(0_s);
  52 }
  53
  54 void RegistrationStore::pushChangesToDatabase(WTF::CompletionHandler<void()>&& completionHandler)
  55 {
  56     Vector<ServiceWorkerContextData> changesToPush;
  57     changesToPush.reserveInitialCapacity(m_updatedRegistrations.size());
  58     for (auto& value : m_updatedRegistrations.values())
  59         changesToPush.uncheckedAppend(WTFMove(value));
  60
  61     m_updatedRegistrations.clear();
  62     m_database.pushChanges(WTFMove(changesToPush), WTFMove(completionHandler));
  63 }
  64
  65 void RegistrationStore::clearAll(WTF::CompletionHandler<void()>&& completionHandler)
  66 {
  67     m_updatedRegistrations.clear();
  68     m_databasePushTimer.stop();
  69     m_database.clearAll(WTFMove(completionHandler));
  70 }
  71
  72 void RegistrationStore::flushChanges(WTF::CompletionHandler<void()>&& completionHandler)
  73 {
  74     if (m_databasePushTimer.isActive()) {
  75         pushChangesToDatabase(WTFMove(completionHandler));
  76         m_databasePushTimer.stop();
  77         return;
  78     }
  79     completionHandler();
  80 }
  81
  82 void RegistrationStore::updateRegistration(const ServiceWorkerContextData& data)
  83 {
  84     m_updatedRegistrations.set(data.registration.key, data);
  85     scheduleDatabasePushIfNecessary();
  86 }
  87
  88 void RegistrationStore::removeRegistration(SWServerRegistration& registration)
  89 {
  90     ServiceWorkerContextData contextData;
  91     contextData.registration.key = registration.key();
  92     m_updatedRegistrations.set(registration.key(), WTFMove(contextData));
  93     scheduleDatabasePushIfNecessary();
  94 }
  95
  96 void RegistrationStore::addRegistrationFromDatabase(ServiceWorkerContextData&& context)
  97 {
  98     m_server.addRegistrationFromStore(WTFMove(context));
  99 }
 100
 101 void RegistrationStore::databaseFailedToOpen()
 102 {
 103     m_server.registrationStoreImportComplete();
 104 }
 105
 106 void RegistrationStore::databaseOpenedAndRecordsImported()
 107 {
 108     m_server.registrationStoreImportComplete();
 109 }
 110
 111 } // namespace WebCore
 112
 113 #endif // ENABLE(SERVICE_WORKER)