#pragma once
#include "ExceptionOr.h"
#include <wtf/Condition.h>
#include <wtf/Lock.h>
#include <wtf/Vector.h>
namespace WebCore {
class Database;
class SQLTransaction;
class DatabaseTaskSynchronizer {
WTF_MAKE_NONCOPYABLE(DatabaseTaskSynchronizer);
public:
DatabaseTaskSynchronizer();
void waitForTaskCompletion();
void taskCompleted();
#ifndef NDEBUG
bool hasCheckedForTermination() const { return m_hasCheckedForTermination; }
void setHasCheckedForTermination() { m_hasCheckedForTermination = true; }
#endif
private:
bool m_taskCompleted { false };
Lock m_synchronousMutex;
Condition m_synchronousCondition;
#ifndef NDEBUG
bool m_hasCheckedForTermination { false };
#endif
};
class DatabaseTask {
WTF_MAKE_FAST_ALLOCATED;
public:
virtual ~DatabaseTask();
void performTask();
Database& database() const { return m_database; }
#if !ASSERT_DISABLED
bool hasSynchronizer() const { return m_synchronizer; }
bool hasCheckedForTermination() const { return m_synchronizer->hasCheckedForTermination(); }
#endif
protected:
DatabaseTask(Database&, DatabaseTaskSynchronizer*);
private:
virtual void doPerformTask() = 0;
Database& m_database;
DatabaseTaskSynchronizer* m_synchronizer;
#if !LOG_DISABLED
virtual const char* debugTaskName() const = 0;
#endif
#if !ASSERT_DISABLED
bool m_complete { false };
#endif
};
class DatabaseOpenTask final : public DatabaseTask {
public:
DatabaseOpenTask(Database&, bool setVersionInNewDatabase, DatabaseTaskSynchronizer&, ExceptionOr<void>& result);
private:
void doPerformTask() final;
#if !LOG_DISABLED
const char* debugTaskName() const final;
#endif
bool m_setVersionInNewDatabase;
ExceptionOr<void>& m_result;
};
class DatabaseCloseTask final : public DatabaseTask {
public:
DatabaseCloseTask(Database&, DatabaseTaskSynchronizer&);
private:
void doPerformTask() final;
#if !LOG_DISABLED
const char* debugTaskName() const final;
#endif
};
class DatabaseTransactionTask final : public DatabaseTask {
public:
explicit DatabaseTransactionTask(RefPtr<SQLTransaction>&&);
virtual ~DatabaseTransactionTask();
SQLTransaction* transaction() const { return m_transaction.get(); }
private:
void doPerformTask() final;
#if !LOG_DISABLED
const char* debugTaskName() const final;
#endif
RefPtr<SQLTransaction> m_transaction;
bool m_didPerformTask;
};
class DatabaseTableNamesTask final : public DatabaseTask {
public:
DatabaseTableNamesTask(Database&, DatabaseTaskSynchronizer&, Vector<String>& result);
private:
void doPerformTask() final;
#if !LOG_DISABLED
const char* debugTaskName() const override;
#endif
Vector<String>& m_result;
};
}