зеркало из https://github.com/mozilla/pjs.git
364 строки
9.0 KiB
C++
364 строки
9.0 KiB
C++
/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
|
|
/* vim: set ts=2 et sw=2 tw=80: */
|
|
/* ***** BEGIN LICENSE BLOCK *****
|
|
* Version: MPL 1.1/GPL 2.0/LGPL 2.1
|
|
*
|
|
* The contents of this file are subject to the Mozilla Public License Version
|
|
* 1.1 (the "License"); you may not use this file except in compliance with
|
|
* the License. You may obtain a copy of the License at
|
|
* http://www.mozilla.org/MPL/
|
|
*
|
|
* Software distributed under the License is distributed on an "AS IS" basis,
|
|
* WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
|
|
* for the specific language governing rights and limitations under the
|
|
* License.
|
|
*
|
|
* The Original Code is Indexed Database.
|
|
*
|
|
* The Initial Developer of the Original Code is
|
|
* The Mozilla Foundation.
|
|
* Portions created by the Initial Developer are Copyright (C) 2010
|
|
* the Initial Developer. All Rights Reserved.
|
|
*
|
|
* Contributor(s):
|
|
* Ben Turner <bent.mozilla@gmail.com>
|
|
*
|
|
* Alternatively, the contents of this file may be used under the terms of
|
|
* either the GNU General Public License Version 2 or later (the "GPL"), or
|
|
* the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
|
|
* in which case the provisions of the GPL or the LGPL are applicable instead
|
|
* of those above. If you wish to allow use of your version of this file only
|
|
* under the terms of either the GPL or the LGPL, and not to allow others to
|
|
* use your version of this file under the terms of the MPL, indicate your
|
|
* decision by deleting the provisions above and replace them with the notice
|
|
* and other provisions required by the GPL or the LGPL. If you do not delete
|
|
* the provisions above, a recipient may use your version of this file under
|
|
* the terms of any one of the MPL, the GPL or the LGPL.
|
|
*
|
|
* ***** END LICENSE BLOCK ***** */
|
|
|
|
#ifndef mozilla_dom_indexeddb_idbtransaction_h__
|
|
#define mozilla_dom_indexeddb_idbtransaction_h__
|
|
|
|
#include "mozilla/dom/indexedDB/IndexedDatabase.h"
|
|
|
|
#include "mozIStorageConnection.h"
|
|
#include "mozIStorageStatement.h"
|
|
#include "mozIStorageFunction.h"
|
|
#include "nsIIDBTransaction.h"
|
|
#include "nsIRunnable.h"
|
|
|
|
#include "nsAutoPtr.h"
|
|
#include "nsClassHashtable.h"
|
|
#include "nsHashKeys.h"
|
|
#include "nsInterfaceHashtable.h"
|
|
|
|
#include "mozilla/dom/indexedDB/IDBDatabase.h"
|
|
#include "mozilla/dom/indexedDB/IDBWrapperCache.h"
|
|
#include "mozilla/dom/indexedDB/FileInfo.h"
|
|
|
|
class nsIThread;
|
|
|
|
BEGIN_INDEXEDDB_NAMESPACE
|
|
|
|
class AsyncConnectionHelper;
|
|
class CommitHelper;
|
|
struct ObjectStoreInfo;
|
|
class TransactionThreadPool;
|
|
class UpdateRefcountFunction;
|
|
|
|
class IDBTransactionListener
|
|
{
|
|
public:
|
|
NS_IMETHOD_(nsrefcnt) AddRef() = 0;
|
|
NS_IMETHOD_(nsrefcnt) Release() = 0;
|
|
|
|
virtual nsresult NotifyTransactionComplete(IDBTransaction* aTransaction) = 0;
|
|
};
|
|
|
|
class IDBTransaction : public IDBWrapperCache,
|
|
public nsIIDBTransaction,
|
|
public nsIRunnable
|
|
{
|
|
friend class AsyncConnectionHelper;
|
|
friend class CommitHelper;
|
|
friend class ThreadObserver;
|
|
friend class TransactionThreadPool;
|
|
|
|
public:
|
|
NS_DECL_ISUPPORTS_INHERITED
|
|
NS_DECL_NSIIDBTRANSACTION
|
|
NS_DECL_NSIRUNNABLE
|
|
|
|
NS_DECL_CYCLE_COLLECTION_CLASS_INHERITED(IDBTransaction, IDBWrapperCache)
|
|
|
|
enum Mode
|
|
{
|
|
READ_ONLY = 0,
|
|
READ_WRITE,
|
|
VERSION_CHANGE
|
|
};
|
|
|
|
enum ReadyState
|
|
{
|
|
INITIAL = 0,
|
|
LOADING,
|
|
COMMITTING,
|
|
DONE
|
|
};
|
|
|
|
static already_AddRefed<IDBTransaction>
|
|
Create(IDBDatabase* aDatabase,
|
|
nsTArray<nsString>& aObjectStoreNames,
|
|
Mode aMode,
|
|
bool aDispatchDelayed);
|
|
|
|
// nsIDOMEventTarget
|
|
virtual nsresult PreHandleEvent(nsEventChainPreVisitor& aVisitor);
|
|
|
|
void OnNewRequest();
|
|
void OnRequestFinished();
|
|
|
|
void RemoveObjectStore(const nsAString& aName);
|
|
|
|
void SetTransactionListener(IDBTransactionListener* aListener);
|
|
|
|
bool StartSavepoint();
|
|
nsresult ReleaseSavepoint();
|
|
void RollbackSavepoint();
|
|
|
|
// Only meant to be called on mStorageThread!
|
|
nsresult GetOrCreateConnection(mozIStorageConnection** aConnection);
|
|
|
|
already_AddRefed<mozIStorageStatement>
|
|
GetCachedStatement(const nsACString& aQuery);
|
|
|
|
template<int N>
|
|
already_AddRefed<mozIStorageStatement>
|
|
GetCachedStatement(const char (&aQuery)[N])
|
|
{
|
|
return GetCachedStatement(NS_LITERAL_CSTRING(aQuery));
|
|
}
|
|
|
|
bool IsOpen() const;
|
|
|
|
bool IsWriteAllowed() const
|
|
{
|
|
return mMode == READ_WRITE || mMode == VERSION_CHANGE;
|
|
}
|
|
|
|
bool IsAborted() const
|
|
{
|
|
return mAborted;
|
|
}
|
|
|
|
// 'Get' prefix is to avoid name collisions with the enum
|
|
Mode GetMode()
|
|
{
|
|
return mMode;
|
|
}
|
|
|
|
IDBDatabase* Database()
|
|
{
|
|
NS_ASSERTION(mDatabase, "This should never be null!");
|
|
return mDatabase;
|
|
}
|
|
|
|
DatabaseInfo* DBInfo() const
|
|
{
|
|
return mDatabaseInfo;
|
|
}
|
|
|
|
already_AddRefed<IDBObjectStore>
|
|
GetOrCreateObjectStore(const nsAString& aName,
|
|
ObjectStoreInfo* aObjectStoreInfo);
|
|
|
|
void OnNewFileInfo(FileInfo* aFileInfo);
|
|
|
|
void ClearCreatedFileInfos();
|
|
|
|
private:
|
|
IDBTransaction();
|
|
~IDBTransaction();
|
|
|
|
nsresult CommitOrRollback();
|
|
|
|
nsRefPtr<IDBDatabase> mDatabase;
|
|
nsRefPtr<DatabaseInfo> mDatabaseInfo;
|
|
nsTArray<nsString> mObjectStoreNames;
|
|
ReadyState mReadyState;
|
|
Mode mMode;
|
|
PRUint32 mPendingRequests;
|
|
|
|
// Only touched on the main thread.
|
|
NS_DECL_EVENT_HANDLER(error)
|
|
NS_DECL_EVENT_HANDLER(complete)
|
|
NS_DECL_EVENT_HANDLER(abort)
|
|
|
|
nsInterfaceHashtable<nsCStringHashKey, mozIStorageStatement>
|
|
mCachedStatements;
|
|
|
|
nsRefPtr<IDBTransactionListener> mListener;
|
|
|
|
// Only touched on the database thread.
|
|
nsCOMPtr<mozIStorageConnection> mConnection;
|
|
|
|
// Only touched on the database thread.
|
|
PRUint32 mSavepointCount;
|
|
|
|
nsTArray<nsRefPtr<IDBObjectStore> > mCreatedObjectStores;
|
|
|
|
bool mAborted;
|
|
bool mCreating;
|
|
|
|
#ifdef DEBUG
|
|
bool mFiredCompleteOrAbort;
|
|
#endif
|
|
|
|
nsRefPtr<UpdateRefcountFunction> mUpdateFileRefcountFunction;
|
|
nsTArray<nsRefPtr<FileInfo> > mCreatedFileInfos;
|
|
};
|
|
|
|
class CommitHelper : public nsIRunnable
|
|
{
|
|
public:
|
|
NS_DECL_ISUPPORTS
|
|
NS_DECL_NSIRUNNABLE
|
|
|
|
CommitHelper(IDBTransaction* aTransaction,
|
|
IDBTransactionListener* aListener,
|
|
const nsTArray<nsRefPtr<IDBObjectStore> >& mUpdatedObjectStores);
|
|
~CommitHelper();
|
|
|
|
template<class T>
|
|
bool AddDoomedObject(nsCOMPtr<T>& aCOMPtr)
|
|
{
|
|
if (aCOMPtr) {
|
|
if (!mDoomedObjects.AppendElement(do_QueryInterface(aCOMPtr))) {
|
|
NS_ERROR("Out of memory!");
|
|
return false;
|
|
}
|
|
aCOMPtr = nsnull;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
private:
|
|
// Writes new autoincrement counts to database
|
|
nsresult WriteAutoIncrementCounts();
|
|
|
|
// Updates counts after a successful commit
|
|
void CommitAutoIncrementCounts();
|
|
|
|
// Reverts counts when a transaction is aborted
|
|
void RevertAutoIncrementCounts();
|
|
|
|
nsRefPtr<IDBTransaction> mTransaction;
|
|
nsRefPtr<IDBTransactionListener> mListener;
|
|
nsCOMPtr<mozIStorageConnection> mConnection;
|
|
nsRefPtr<UpdateRefcountFunction> mUpdateFileRefcountFunction;
|
|
nsAutoTArray<nsCOMPtr<nsISupports>, 10> mDoomedObjects;
|
|
nsAutoTArray<nsRefPtr<IDBObjectStore>, 10> mAutoIncrementObjectStores;
|
|
|
|
bool mAborted;
|
|
};
|
|
|
|
class UpdateRefcountFunction : public mozIStorageFunction
|
|
{
|
|
public:
|
|
NS_DECL_ISUPPORTS
|
|
NS_DECL_MOZISTORAGEFUNCTION
|
|
|
|
UpdateRefcountFunction(FileManager* aFileManager)
|
|
: mFileManager(aFileManager)
|
|
{ }
|
|
|
|
~UpdateRefcountFunction()
|
|
{ }
|
|
|
|
nsresult Init();
|
|
|
|
void ClearFileInfoEntries()
|
|
{
|
|
mFileInfoEntries.Clear();
|
|
}
|
|
|
|
nsresult UpdateDatabase(mozIStorageConnection* aConnection)
|
|
{
|
|
DatabaseUpdateFunction function(aConnection);
|
|
|
|
mFileInfoEntries.EnumerateRead(DatabaseUpdateCallback, &function);
|
|
|
|
return function.ErrorCode();
|
|
}
|
|
|
|
void UpdateFileInfos()
|
|
{
|
|
mFileInfoEntries.EnumerateRead(FileInfoUpdateCallback, nsnull);
|
|
}
|
|
|
|
private:
|
|
class FileInfoEntry
|
|
{
|
|
public:
|
|
FileInfoEntry(FileInfo* aFileInfo)
|
|
: mFileInfo(aFileInfo), mDelta(0)
|
|
{ }
|
|
|
|
~FileInfoEntry()
|
|
{ }
|
|
|
|
nsRefPtr<FileInfo> mFileInfo;
|
|
PRInt32 mDelta;
|
|
};
|
|
|
|
enum UpdateType {
|
|
eIncrement,
|
|
eDecrement
|
|
};
|
|
|
|
class DatabaseUpdateFunction
|
|
{
|
|
public:
|
|
DatabaseUpdateFunction(mozIStorageConnection* aConnection)
|
|
: mConnection(aConnection), mErrorCode(NS_OK)
|
|
{ }
|
|
|
|
bool Update(PRInt64 aId, PRInt32 aDelta);
|
|
nsresult ErrorCode()
|
|
{
|
|
return mErrorCode;
|
|
}
|
|
|
|
private:
|
|
nsresult UpdateInternal(PRInt64 aId, PRInt32 aDelta);
|
|
|
|
nsCOMPtr<mozIStorageConnection> mConnection;
|
|
nsCOMPtr<mozIStorageStatement> mUpdateStatement;
|
|
nsCOMPtr<mozIStorageStatement> mInsertStatement;
|
|
|
|
nsresult mErrorCode;
|
|
};
|
|
|
|
nsresult ProcessValue(mozIStorageValueArray* aValues,
|
|
PRInt32 aIndex,
|
|
UpdateType aUpdateType);
|
|
|
|
static PLDHashOperator
|
|
DatabaseUpdateCallback(const PRUint64& aKey,
|
|
FileInfoEntry* aValue,
|
|
void* aUserArg);
|
|
|
|
static PLDHashOperator
|
|
FileInfoUpdateCallback(const PRUint64& aKey,
|
|
FileInfoEntry* aValue,
|
|
void* aUserArg);
|
|
|
|
FileManager* mFileManager;
|
|
nsClassHashtable<nsUint64HashKey, FileInfoEntry> mFileInfoEntries;
|
|
};
|
|
|
|
END_INDEXEDDB_NAMESPACE
|
|
|
|
#endif // mozilla_dom_indexeddb_idbtransaction_h__
|