Delay some service worker operations until after the database import completes.
[WebKit-https.git] / 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)