2011-02-28 Jeremy Orlow <jorlow@chromium.org>
[WebKit-https.git] / Source / WebCore / storage / IDBCursorBackendImpl.cpp
1 /*
2  * Copyright (C) 2010 Google 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  *
8  * 1.  Redistributions of source code must retain the above copyright
9  *     notice, this list of conditions and the following disclaimer.
10  * 2.  Redistributions in binary form must reproduce the above copyright
11  *     notice, this list of conditions and the following disclaimer in the
12  *     documentation and/or other materials provided with the distribution.
13  *
14  * THIS SOFTWARE IS PROVIDED BY APPLE AND ITS CONTRIBUTORS "AS IS" AND ANY
15  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
16  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
17  * DISCLAIMED. IN NO EVENT SHALL APPLE OR ITS CONTRIBUTORS BE LIABLE FOR ANY
18  * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
19  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
20  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
21  * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
22  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
23  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
24  */
25
26 #include "config.h"
27 #include "IDBCursorBackendImpl.h"
28
29 #if ENABLE(INDEXED_DATABASE)
30
31 #include "CrossThreadTask.h"
32 #include "IDBBackingStore.h"
33 #include "IDBCallbacks.h"
34 #include "IDBDatabaseBackendImpl.h"
35 #include "IDBDatabaseError.h"
36 #include "IDBDatabaseException.h"
37 #include "IDBIndexBackendImpl.h"
38 #include "IDBKeyRange.h"
39 #include "IDBObjectStoreBackendImpl.h"
40 #include "IDBRequest.h"
41 #include "IDBTransactionBackendInterface.h"
42 #include "SQLiteDatabase.h"
43 #include "SQLiteStatement.h"
44 #include "SerializedScriptValue.h"
45
46 namespace WebCore {
47
48 IDBCursorBackendImpl::IDBCursorBackendImpl(IDBBackingStore* backingStore, PassRefPtr<IDBKeyRange> keyRange, IDBCursor::Direction direction, PassOwnPtr<SQLiteStatement> query, CursorType cursorType, IDBTransactionBackendInterface* transaction, IDBObjectStoreBackendInterface* objectStore)
49     : m_backingStore(backingStore)
50     , m_keyRange(keyRange)
51     , m_direction(direction)
52     , m_query(query)
53     , m_cursorType(cursorType)
54     , m_transaction(transaction)
55     , m_objectStore(objectStore)
56 {
57     loadCurrentRow();
58 }
59
60 IDBCursorBackendImpl::~IDBCursorBackendImpl()
61 {
62 }
63
64 unsigned short IDBCursorBackendImpl::direction() const
65 {
66     return m_direction;
67 }
68
69 PassRefPtr<IDBKey> IDBCursorBackendImpl::key() const
70 {
71     return m_currentKey;
72 }
73
74 PassRefPtr<IDBKey> IDBCursorBackendImpl::primaryKey() const
75 {
76     return m_currentPrimaryKey;
77 }
78
79 PassRefPtr<SerializedScriptValue> IDBCursorBackendImpl::value() const
80 {
81     ASSERT(m_cursorType != IndexKeyCursor);
82     return m_currentValue;
83 }
84
85 void IDBCursorBackendImpl::update(PassRefPtr<SerializedScriptValue> value, PassRefPtr<IDBCallbacks> callbacks, ExceptionCode& ec)
86 {
87     if (!m_query || m_currentId == InvalidId || m_cursorType == IndexKeyCursor) {
88         ec = IDBDatabaseException::NOT_ALLOWED_ERR;
89         return;
90     }
91
92     m_objectStore->put(value, m_currentPrimaryKey, IDBObjectStoreBackendInterface::CursorUpdate, callbacks, m_transaction.get(), ec);
93 }
94
95 void IDBCursorBackendImpl::continueFunction(PassRefPtr<IDBKey> prpKey, PassRefPtr<IDBCallbacks> prpCallbacks, ExceptionCode& ec)
96 {
97     RefPtr<IDBCursorBackendImpl> cursor = this;
98     RefPtr<IDBKey> key = prpKey;
99     RefPtr<IDBCallbacks> callbacks = prpCallbacks;
100     if (!m_transaction->scheduleTask(createCallbackTask(&IDBCursorBackendImpl::continueFunctionInternal, cursor, key, callbacks)))
101         ec = IDBDatabaseException::NOT_ALLOWED_ERR;
102 }
103
104 bool IDBCursorBackendImpl::currentRowExists()
105 {
106     String sql = m_cursorType == ObjectStoreCursor ? "SELECT id FROM ObjectStoreData WHERE id = ?" : "SELECT id FROM IndexData WHERE id = ?";
107     SQLiteStatement statement(m_backingStore->db(), sql);
108
109     bool ok = statement.prepare() == SQLResultOk;
110     ASSERT_UNUSED(ok, ok);
111
112     statement.bindInt64(1, m_currentId);
113     return statement.step() == SQLResultRow;
114 }
115
116 // IMPORTANT: If this ever 1) fires an 'error' event and 2) it's possible to fire another event afterwards,
117 //            IDBRequest::hasPendingActivity() will need to be modified to handle this!!!
118 void IDBCursorBackendImpl::continueFunctionInternal(ScriptExecutionContext*, PassRefPtr<IDBCursorBackendImpl> prpCursor, PassRefPtr<IDBKey> prpKey, PassRefPtr<IDBCallbacks> callbacks)
119 {
120     RefPtr<IDBCursorBackendImpl> cursor = prpCursor;
121     RefPtr<IDBKey> key = prpKey;
122     while (true) {
123         if (!cursor->m_query || cursor->m_query->step() != SQLResultRow) {
124             cursor->m_query = 0;
125             cursor->m_currentId = InvalidId;
126             cursor->m_currentKey = 0;
127             cursor->m_currentPrimaryKey = 0;
128             cursor->m_currentValue = 0;
129             callbacks->onSuccess(SerializedScriptValue::nullValue());
130             return;
131         }
132
133         RefPtr<IDBKey> oldKey = cursor->m_currentKey;
134         cursor->loadCurrentRow();
135
136         // Skip if this entry has been deleted from the object store.
137         if (!cursor->currentRowExists())
138             continue;
139
140         // If a key was supplied, we must loop until we find that key (or hit the end).
141         if (key && !key->isEqual(cursor->m_currentKey.get()))
142             continue;
143
144         // If we don't have a uniqueness constraint, we can stop now.
145         if (cursor->m_direction == IDBCursor::NEXT || cursor->m_direction == IDBCursor::PREV)
146             break;
147         if (!cursor->m_currentKey->isEqual(oldKey.get()))
148             break;
149     }
150
151     callbacks->onSuccess(cursor.get());
152 }
153
154 void IDBCursorBackendImpl::deleteFunction(PassRefPtr<IDBCallbacks> prpCallbacks, ExceptionCode& ec)
155 {
156     if (!m_query || m_currentId == InvalidId || m_cursorType == IndexKeyCursor) {
157         ec = IDBDatabaseException::NOT_ALLOWED_ERR;
158         return;
159     }
160
161     m_objectStore->deleteFunction(m_currentPrimaryKey, prpCallbacks, m_transaction.get(), ec);
162 }
163
164
165 void IDBCursorBackendImpl::loadCurrentRow()
166 {
167     // The column numbers depend on the query in IDBObjectStoreBackendImpl::openCursorInternal and/or IDBIndexBackendImpl::openCursorInternal.
168     m_currentId = m_query->getColumnInt64(0);
169     m_currentKey = IDBKey::fromQuery(*m_query, 1);
170     m_currentValue = SerializedScriptValue::createFromWire(m_query->getColumnBlobAsString(4));
171     m_currentPrimaryKey = IDBKey::fromQuery(*m_query, 5);
172 }
173
174 SQLiteDatabase& IDBCursorBackendImpl::database() const
175 {
176     return m_backingStore->db();
177 }
178
179 } // namespace WebCore
180
181 #endif // ENABLE(INDEXED_DATABASE)