3684f2274f1bb2fc792612792452a72ceaeaad5e
[WebKit-https.git] / Source / WebCore / Modules / indexeddb / server / IDBServer.h
1 /*
2  * Copyright (C) 2015 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 #ifndef IDBServer_h
27 #define IDBServer_h
28
29 #if ENABLE(INDEXED_DATABASE)
30
31 #include "CrossThreadTask.h"
32 #include "IDBConnectionToClient.h"
33 #include "IDBDatabaseIdentifier.h"
34 #include "UniqueIDBDatabase.h"
35 #include "UniqueIDBDatabaseConnection.h"
36 #include <wtf/HashMap.h>
37 #include <wtf/Lock.h>
38 #include <wtf/MessageQueue.h>
39 #include <wtf/Ref.h>
40 #include <wtf/RefCounted.h>
41 #include <wtf/RefPtr.h>
42
43 namespace WebCore {
44
45 class CrossThreadTask;
46 class IDBRequestData;
47
48 namespace IDBServer {
49
50 class IDBServer : public RefCounted<IDBServer> {
51 public:
52     static Ref<IDBServer> create();
53
54     void registerConnection(IDBConnectionToClient&);
55     void unregisterConnection(IDBConnectionToClient&);
56
57     // Operations requested by the client.
58     void openDatabase(const IDBRequestData&);
59     void deleteDatabase(const IDBRequestData&);
60     void abortTransaction(const IDBResourceIdentifier&);
61     void commitTransaction(const IDBResourceIdentifier&);
62     void createObjectStore(const IDBRequestData&, const IDBObjectStoreInfo&);
63     void deleteObjectStore(const IDBRequestData&, const String& objectStoreName);
64     void clearObjectStore(const IDBRequestData&, uint64_t objectStoreIdentifier);
65     void createIndex(const IDBRequestData&, const IDBIndexInfo&);
66     void putOrAdd(const IDBRequestData&, const IDBKeyData&, const ThreadSafeDataBuffer& valueData, IndexedDB::ObjectStoreOverwriteMode);
67     void getRecord(const IDBRequestData&, const IDBKeyRangeData&);
68     void getCount(const IDBRequestData&, const IDBKeyRangeData&);
69     void deleteRecord(const IDBRequestData&, const IDBKeyRangeData&);
70     void establishTransaction(uint64_t databaseConnectionIdentifier, const IDBTransactionInfo&);
71     void databaseConnectionClosed(uint64_t databaseConnectionIdentifier);
72
73     void postDatabaseTask(std::unique_ptr<CrossThreadTask>&&);
74     void postDatabaseTaskReply(std::unique_ptr<CrossThreadTask>&&);
75
76     void registerDatabaseConnection(UniqueIDBDatabaseConnection&);
77     void unregisterDatabaseConnection(UniqueIDBDatabaseConnection&);
78     void registerTransaction(UniqueIDBDatabaseTransaction&);
79     void unregisterTransaction(UniqueIDBDatabaseTransaction&);
80
81     std::unique_ptr<IDBBackingStore> createBackingStore(const IDBDatabaseIdentifier&);
82
83 private:
84     IDBServer();
85
86     UniqueIDBDatabase& getOrCreateUniqueIDBDatabase(const IDBDatabaseIdentifier&);
87
88     static void databaseThreadEntry(void*);
89     void databaseRunLoop();
90     void handleTaskRepliesOnMainThread();
91
92     HashMap<uint64_t, RefPtr<IDBConnectionToClient>> m_connectionMap;
93     HashMap<IDBDatabaseIdentifier, RefPtr<UniqueIDBDatabase>> m_uniqueIDBDatabaseMap;
94
95     ThreadIdentifier m_threadID { 0 };
96     Lock m_databaseThreadCreationLock;
97     Lock m_mainThreadReplyLock;
98     bool m_mainThreadReplyScheduled { false };
99
100     MessageQueue<CrossThreadTask> m_databaseQueue;
101     MessageQueue<CrossThreadTask> m_databaseReplyQueue;
102
103     HashMap<uint64_t, UniqueIDBDatabaseConnection*> m_databaseConnections;
104     HashMap<IDBResourceIdentifier, UniqueIDBDatabaseTransaction*> m_transactions;
105 };
106
107 } // namespace IDBServer
108 } // namespace WebCore
109
110 #endif // ENABLE(INDEXED_DATABASE)
111 #endif // IDBServer_h