69ae4b99fdbab2f65e1c55a96208fe87e86e2ba0
[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, String&& databaseDirectory)
36     : m_server(server)
37     , m_database(RegistrationDatabase::create(*this, WTFMove(databaseDirectory)))
38     , m_databasePushTimer(*this, &RegistrationStore::pushChangesToDatabase)
39 {
40 }
41
42 RegistrationStore::~RegistrationStore()
43 {
44 }
45
46 void RegistrationStore::scheduleDatabasePushIfNecessary()
47 {
48     if (m_databasePushTimer.isActive())
49         return;
50     m_databasePushTimer.startOneShot(0_s);
51 }
52
53 void RegistrationStore::pushChangesToDatabase(WTF::CompletionHandler<void()>&& completionHandler)
54 {
55     Vector<ServiceWorkerContextData> changesToPush;
56     changesToPush.reserveInitialCapacity(m_updatedRegistrations.size());
57     for (auto& value : m_updatedRegistrations.values())
58         changesToPush.uncheckedAppend(WTFMove(value));
59
60     m_updatedRegistrations.clear();
61     m_database->pushChanges(WTFMove(changesToPush), WTFMove(completionHandler));
62 }
63
64 void RegistrationStore::clearAll(WTF::CompletionHandler<void()>&& completionHandler)
65 {
66     m_updatedRegistrations.clear();
67     m_databasePushTimer.stop();
68     m_database->clearAll(WTFMove(completionHandler));
69 }
70
71 void RegistrationStore::flushChanges(WTF::CompletionHandler<void()>&& completionHandler)
72 {
73     if (m_databasePushTimer.isActive()) {
74         pushChangesToDatabase(WTFMove(completionHandler));
75         m_databasePushTimer.stop();
76         return;
77     }
78     completionHandler();
79 }
80
81 void RegistrationStore::updateRegistration(const ServiceWorkerContextData& data)
82 {
83     m_updatedRegistrations.set(data.registration.key, data);
84     scheduleDatabasePushIfNecessary();
85 }
86
87 void RegistrationStore::removeRegistration(SWServerRegistration& registration)
88 {
89     ServiceWorkerContextData contextData;
90     contextData.registration.key = registration.key();
91     m_updatedRegistrations.set(registration.key(), WTFMove(contextData));
92     scheduleDatabasePushIfNecessary();
93 }
94
95 void RegistrationStore::addRegistrationFromDatabase(ServiceWorkerContextData&& context)
96 {
97     m_server.addRegistrationFromStore(WTFMove(context));
98 }
99
100 void RegistrationStore::databaseFailedToOpen()
101 {
102     m_server.registrationStoreDatabaseFailedToOpen();
103 }
104
105 void RegistrationStore::databaseOpenedAndRecordsImported()
106 {
107     m_server.registrationStoreImportComplete();
108 }
109
110 } // namespace WebCore
111
112 #endif // ENABLE(SERVICE_WORKER)