blob: fe66ac2c824c16e21b474cfaff629016f8cba26b [file] [log] [blame]
// Copyright (c) 2011 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef STORAGE_COMMON_DATABASE_DATABASE_CONNECTIONS_H_
#define STORAGE_COMMON_DATABASE_DATABASE_CONNECTIONS_H_
#include <stdint.h>
#include <map>
#include <string>
#include <vector>
#include "base/memory/ref_counted.h"
#include "base/strings/string16.h"
#include "base/synchronization/lock.h"
#include "base/time/time.h"
#include "storage/common/storage_common_export.h"
namespace base {
class SingleThreadTaskRunner;
class WaitableEvent;
}
namespace storage {
class STORAGE_COMMON_EXPORT DatabaseConnections {
public:
DatabaseConnections();
~DatabaseConnections();
bool IsEmpty() const;
bool IsDatabaseOpened(const std::string& origin_identifier,
const base::string16& database_name) const;
bool IsOriginUsed(const std::string& origin_identifier) const;
// Returns true if this is the first connection.
bool AddConnection(const std::string& origin_identifier,
const base::string16& database_name);
// Returns true if the last connection was removed.
bool RemoveConnection(const std::string& origin_identifier,
const base::string16& database_name);
void RemoveAllConnections();
void RemoveConnections(
const DatabaseConnections& connections,
std::vector<std::pair<std::string, base::string16> >* closed_dbs);
// Database sizes can be kept only if IsDatabaseOpened returns true.
int64_t GetOpenDatabaseSize(const std::string& origin_identifier,
const base::string16& database_name) const;
void SetOpenDatabaseSize(const std::string& origin_identifier,
const base::string16& database_name,
int64_t size);
// Returns a list of the connections, <origin_id, name>.
void ListConnections(
std::vector<std::pair<std::string, base::string16> > *list) const;
private:
// Mapping from name to <openCount, size>
typedef std::map<base::string16, std::pair<int, int64_t>> DBConnections;
typedef std::map<std::string, DBConnections> OriginConnections;
mutable OriginConnections connections_; // mutable for GetOpenDatabaseSize
// Returns true if the last connection was removed.
bool RemoveConnectionsHelper(const std::string& origin_identifier,
const base::string16& database_name,
int num_connections);
};
// A wrapper class that provides thread-safety and the
// ability to wait until all connections have closed.
// Intended for use in renderer processes.
class STORAGE_COMMON_EXPORT DatabaseConnectionsWrapper
: public base::RefCountedThreadSafe<DatabaseConnectionsWrapper> {
public:
DatabaseConnectionsWrapper();
bool HasOpenConnections();
void AddOpenConnection(const std::string& origin_identifier,
const base::string16& database_name);
void RemoveOpenConnection(const std::string& origin_identifier,
const base::string16& database_name);
// Returns true if all databases are closed.
bool WaitForAllDatabasesToClose(base::TimeDelta timeout);
private:
~DatabaseConnectionsWrapper();
friend class base::RefCountedThreadSafe<DatabaseConnectionsWrapper>;
base::Lock open_connections_lock_;
DatabaseConnections open_connections_;
base::WaitableEvent* waiting_to_close_event_ = nullptr;
};
} // namespace storage
#endif // STORAGE_COMMON_DATABASE_DATABASE_CONNECTIONS_H_