1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
|
// 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 <map>
#include <string>
#include <vector>
#include "base/memory/ref_counted.h"
#include "base/strings/string16.h"
#include "base/synchronization/lock.h"
#include "storage/common/storage_common_export.h"
namespace base {
class SingleThreadTaskRunner;
}
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 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 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> > 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();
// The Wait and Has methods should only be called on the
// main thread (the thread on which the wrapper is constructed).
void WaitForAllDatabasesToClose();
bool HasOpenConnections();
// Add and Remove may be called on any thread.
void AddOpenConnection(const std::string& origin_identifier,
const base::string16& database_name);
void RemoveOpenConnection(const std::string& origin_identifier,
const base::string16& database_name);
private:
~DatabaseConnectionsWrapper();
friend class base::RefCountedThreadSafe<DatabaseConnectionsWrapper>;
bool waiting_for_dbs_to_close_;
base::Lock open_connections_lock_;
DatabaseConnections open_connections_;
scoped_refptr<base::SingleThreadTaskRunner> main_thread_;
};
} // namespace storage
#endif // STORAGE_COMMON_DATABASE_DATABASE_CONNECTIONS_H_
|
