Chromium Code Reviews| Index: components/precache/core/precache_referrer_host_table.h |
| diff --git a/components/precache/core/precache_referrer_host_table.h b/components/precache/core/precache_referrer_host_table.h |
| new file mode 100644 |
| index 0000000000000000000000000000000000000000..98b36ac9a9aeb18f86cc517b33638455108a0021 |
| --- /dev/null |
| +++ b/components/precache/core/precache_referrer_host_table.h |
| @@ -0,0 +1,85 @@ |
| +// Copyright 2016 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 COMPONENTS_PRECACHE_CORE_PRECACHE_REFERRER_HOST_TABLE_H_ |
| +#define COMPONENTS_PRECACHE_CORE_PRECACHE_REFERRER_HOST_TABLE_H_ |
| + |
| +#include <stdint.h> |
| + |
| +#include <map> |
| +#include <string> |
| + |
| +#include "base/macros.h" |
| +#include "base/time/time.h" |
| + |
| +namespace sql { |
| +class Connection; |
| +} |
| + |
| +namespace precache { |
| + |
| +struct PrecacheReferrerHostEntry { |
| + static const int64_t kInvalidId; |
| + |
| + PrecacheReferrerHostEntry(int64_t id, |
| + const std::string& referrer_host, |
| + int64_t manifest_id, |
| + const base::Time& time) |
| + : id(id), |
| + referrer_host(referrer_host), |
| + manifest_id(manifest_id), |
| + time(time) {} |
| + |
| + // Comparison for testing. |
| + bool operator==(const PrecacheReferrerHostEntry entry) const { |
|
sclittle
2016/08/15 20:13:10
Make this take in a const PrecacheReferrerHostEntr
Raj
2016/08/16 17:54:37
Done.
|
| + return id == entry.id && referrer_host == entry.referrer_host && |
|
sclittle
2016/08/15 20:13:10
nit: Move this definition to the .cc file.
Raj
2016/08/16 17:54:37
Done.
|
| + manifest_id == entry.manifest_id && time == entry.time; |
| + ; |
|
sclittle
2016/08/15 20:13:10
nit: remove this line
Raj
2016/08/16 17:54:37
Done.
|
| + } |
| + |
| + int64_t id; |
| + std::string referrer_host; |
| + int64_t manifest_id; |
| + base::Time time; |
| +}; |
| + |
| +class PrecacheReferrerHostTable { |
| + public: |
| + PrecacheReferrerHostTable(); |
| + ~PrecacheReferrerHostTable(); |
| + |
| + // Initialize the precache referrer host table for use with the specified |
| + // database connection. The caller keeps ownership of |db|, and |db| must not |
| + // be NULL. Init must be called before any other methods. |
| + bool Init(sql::Connection* db); |
| + |
| + // Returns the referrer host information about |referrer_host|. |
| + PrecacheReferrerHostEntry GetReferrerHost(const std::string& referrer_host); |
| + |
| + // Updates the referrer host information about |referrer_host|. |
| + int64_t UpdateReferrerHost(const std::string& referrer_host, |
| + int64_t manifest_id, |
| + const base::Time& time); |
| + |
| + // Deletes entries that were created before the time of |delete_end|. |
| + void DeleteAllEntriesBefore(const base::Time& delete_end); |
| + |
| + // Delete all entries. |
| + void DeleteAll(); |
| + |
| + // Used by tests to get the contents of the table. |
| + std::map<std::string, PrecacheReferrerHostEntry> GetAllDataForTesting(); |
| + |
| + private: |
| + bool CreateTableIfNonExistent(); |
| + |
| + // Not owned by |this|. |
| + sql::Connection* db_; |
| + |
| + DISALLOW_COPY_AND_ASSIGN(PrecacheReferrerHostTable); |
| +}; |
| + |
| +} // namespace precache |
| + |
| +#endif // COMPONENTS_PRECACHE_CORE_PRECACHE_REFERRER_HOST_TABLE_H_ |