blob: 4fd78732ada6dedffedfd07cde57c0be324c0957 [file] [log] [blame]
// Copyright 2017 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 CHROME_BROWSER_PREDICTORS_GLOWPLUG_KEY_VALUE_TABLE_H_
#define CHROME_BROWSER_PREDICTORS_GLOWPLUG_KEY_VALUE_TABLE_H_
#include <map>
#include <string>
#include <vector>
#include "base/strings/stringprintf.h"
#include "sql/statement.h"
namespace google {
namespace protobuf {
class MessageLite;
}
}
namespace predictors {
namespace internal {
void BindDataToStatement(const std::string& key,
const google::protobuf::MessageLite& data,
sql::Statement* statement);
std::string GetSelectAllSql(const std::string& table_name);
std::string GetReplaceSql(const std::string& table_name);
std::string GetDeleteSql(const std::string& table_name);
std::string GetDeleteAllSql(const std::string& table_name);
} // namespace internal
// The backend class helps perform database operations on a single table. The
// table name is passed as a constructor argument. The table schema is fixed: it
// always consists of two columns, TEXT type "key" and BLOB type "proto". The
// class doesn't manage the creation and the deletion of the table.
//
// All the functions except of the constructor must be called on a DB sequence
// of the ResourcePrefetchPredictorTables.
// The preferred way to call the methods of this class is passing the method to
// ResourcePrefetchPredictorTables::ScheduleDBTask().
//
// Example:
// tables_->ScheduleDBTask(
// FROM_HERE,
// base::BindOnce(&GlowplugKeyValueTable<PrefetchData>::UpdateData,
// base::Unretained(table_), key, data));
template <typename T>
class GlowplugKeyValueTable {
public:
explicit GlowplugKeyValueTable(const std::string& table_name);
// Virtual for testing.
virtual ~GlowplugKeyValueTable() {}
virtual void GetAllData(std::map<std::string, T>* data_map,
sql::Connection* db) const;
virtual void UpdateData(const std::string& key,
const T& data,
sql::Connection* db);
virtual void DeleteData(const std::vector<std::string>& keys,
sql::Connection* db);
virtual void DeleteAllData(sql::Connection* db);
private:
const std::string table_name_;
DISALLOW_COPY_AND_ASSIGN(GlowplugKeyValueTable);
};
template <typename T>
GlowplugKeyValueTable<T>::GlowplugKeyValueTable(const std::string& table_name)
: table_name_(table_name) {}
template <typename T>
void GlowplugKeyValueTable<T>::GetAllData(std::map<std::string, T>* data_map,
sql::Connection* db) const {
sql::Statement reader(db->GetUniqueStatement(
::predictors::internal::GetSelectAllSql(table_name_).c_str()));
while (reader.Step()) {
auto it = data_map->emplace(reader.ColumnString(0), T()).first;
int size = reader.ColumnByteLength(1);
const void* blob = reader.ColumnBlob(1);
DCHECK(blob);
it->second.ParseFromArray(blob, size);
}
}
template <typename T>
void GlowplugKeyValueTable<T>::UpdateData(const std::string& key,
const T& data,
sql::Connection* db) {
sql::Statement inserter(db->GetUniqueStatement(
::predictors::internal::GetReplaceSql(table_name_).c_str()));
::predictors::internal::BindDataToStatement(key, data, &inserter);
inserter.Run();
}
template <typename T>
void GlowplugKeyValueTable<T>::DeleteData(const std::vector<std::string>& keys,
sql::Connection* db) {
sql::Statement deleter(db->GetUniqueStatement(
::predictors::internal::GetDeleteSql(table_name_).c_str()));
for (const auto& key : keys) {
deleter.BindString(0, key);
deleter.Run();
deleter.Reset(true);
}
}
template <typename T>
void GlowplugKeyValueTable<T>::DeleteAllData(sql::Connection* db) {
sql::Statement deleter(db->GetUniqueStatement(
::predictors::internal::GetDeleteAllSql(table_name_).c_str()));
deleter.Run();
}
} // namespace predictors
#endif // CHROME_BROWSER_PREDICTORS_GLOWPLUG_KEY_VALUE_TABLE_H_