Home | History | Annotate | Download | only in android 
      1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
      2 // Use of this source code is governed by a BSD-style license that can be
      3 // found in the LICENSE file.
      4 
      5 #include "chrome/browser/history/android/android_urls_database.h"
      6 
      7 #include "base/logging.h"
      8 
      9 namespace history {
     10 
     11 AndroidURLsDatabase::AndroidURLsDatabase() {
     12 }
     13 
     14 AndroidURLsDatabase::~AndroidURLsDatabase() {
     15 }
     16 
     17 bool AndroidURLsDatabase::CreateAndroidURLsTable() {
     18   const char* name = "android_urls";
     19   if (!GetDB().DoesTableExist(name)) {
     20     std::string sql;
     21     sql.append("CREATE TABLE ");
     22     sql.append(name);
     23     sql.append("("
     24                "id INTEGER PRIMARY KEY,"
     25                "raw_url LONGVARCHAR,"              // Passed in raw url.
     26                "url_id INTEGER NOT NULL"           // url id in urls table.
     27                ")");
     28     if (!GetDB().Execute(sql.c_str())) {
     29       LOG(ERROR) << GetDB().GetErrorMessage();
     30       return false;
     31     }
     32 
     33     if (!GetDB().Execute("CREATE INDEX android_urls_raw_url_idx"
     34                          " ON android_urls(raw_url)")) {
     35       LOG(ERROR) << GetDB().GetErrorMessage();
     36       return false;
     37     }
     38 
     39     if (!GetDB().Execute("CREATE INDEX android_urls_url_id_idx"
     40                          " ON android_urls(url_id)")) {
     41       LOG(ERROR) << GetDB().GetErrorMessage();
     42       return false;
     43     }
     44   }
     45   return true;
     46 }
     47 
     48 AndroidURLID AndroidURLsDatabase::AddAndroidURLRow(const std::string& raw_url,
     49                                                    URLID url_id) {
     50   if (GetAndroidURLRow(url_id, NULL)) {
     51     LOG(ERROR) << "url_id already exist";
     52     return 0;
     53   }
     54 
     55   sql::Statement statement(GetDB().GetCachedStatement(SQL_FROM_HERE,
     56       "INSERT INTO android_urls (raw_url, url_id) VALUES (?, ?)"));
     57 
     58   statement.BindString(0, raw_url);
     59   statement.BindInt64(1, static_cast<int64>(url_id));
     60 
     61   if (!statement.Run()) {
     62     LOG(ERROR) << GetDB().GetErrorMessage();
     63     return 0;
     64   }
     65   return GetDB().GetLastInsertRowId();
     66 }
     67 
     68 bool AndroidURLsDatabase::GetAndroidURLRow(URLID url_id, AndroidURLRow* row) {
     69   sql::Statement statement(GetDB().GetCachedStatement(SQL_FROM_HERE,
     70       "SELECT id, raw_url, url_id FROM android_urls WHERE url_id = ?"));
     71 
     72   statement.BindInt64(0, url_id);
     73 
     74   if (!statement.Step())
     75     return false;
     76   if (row) {
     77     row->id = statement.ColumnInt64(0);
     78     row->raw_url = statement.ColumnString(1);
     79     row->url_id = statement.ColumnInt64(2);
     80   }
     81   return true;
     82 }
     83 
     84 bool AndroidURLsDatabase::DeleteAndroidURLRows(
     85     const std::vector<URLID>& url_ids) {
     86   if (url_ids.empty())
     87     return true;
     88 
     89   std::string sql;
     90   sql.append("DELETE FROM android_urls ");
     91   std::ostringstream oss;
     92   bool has_id = false;
     93   for (std::vector<URLID>::const_iterator i = url_ids.begin();
     94        i != url_ids.end(); ++i) {
     95     if (has_id)
     96       oss << ", ";
     97     else
     98       has_id = true;
     99     oss << *i;
    100   }
    101 
    102   if (has_id) {
    103     sql.append(" WHERE url_id in ( ");
    104     sql.append(oss.str());
    105     sql.append(" )");
    106   }
    107 
    108   if (!GetDB().Execute(sql.c_str())) {
    109     LOG(ERROR) << GetDB().GetErrorMessage();
    110     return false;
    111   }
    112   return true;
    113 }
    114 
    115 bool AndroidURLsDatabase::DeleteUnusedAndroidURLs() {
    116   return GetDB().Execute("DELETE FROM android_urls WHERE url_id NOT IN ("
    117                          "SELECT id FROM urls)");
    118 }
    119 
    120 bool AndroidURLsDatabase::UpdateAndroidURLRow(AndroidURLID id,
    121                                               const std::string& raw_url,
    122                                               URLID url_id) {
    123   sql::Statement statement(GetDB().GetCachedStatement(SQL_FROM_HERE,
    124       "UPDATE android_urls SET raw_url = ?, url_id = ? WHERE id = ?"));
    125 
    126   statement.BindString(0, raw_url);
    127   statement.BindInt64(1, url_id);
    128   statement.BindInt64(2, id);
    129 
    130   if (!statement.is_valid()) {
    131     LOG(ERROR) << GetDB().GetErrorMessage();
    132     return false;
    133   }
    134 
    135   if (!statement.Run()) {
    136     LOG(ERROR) << GetDB().GetErrorMessage();
    137     return false;
    138   }
    139 
    140   return true;
    141 }
    142 
    143 bool AndroidURLsDatabase::ClearAndroidURLRows() {
    144   // The android_urls table might not exist if the Android content provider is
    145   // never used, especially in the unit tests. See http://b/6385692.
    146   if (GetDB().DoesTableExist("android_urls"))
    147     return GetDB().Execute("DELETE FROM android_urls");
    148 
    149   return true;
    150 }
    151 
    152 bool AndroidURLsDatabase::MigrateToVersion22() {
    153   if (!GetDB().DoesTableExist("android_urls"))
    154     return true;
    155 
    156   if (!GetDB().Execute("ALTER TABLE android_urls RENAME TO android_urls_tmp"))
    157     return false;
    158 
    159   if (!GetDB().Execute("DROP INDEX android_urls_raw_url_idx"))
    160     return false;
    161 
    162   if (!GetDB().Execute("DROP INDEX android_urls_url_id_idx"))
    163     return false;
    164 
    165   if (!CreateAndroidURLsTable())
    166     return false;
    167 
    168   if (!GetDB().Execute(
    169       "INSERT INTO android_urls (id, raw_url, url_id) "
    170       "SELECT id, raw_url, url_id FROM android_urls_tmp"))
    171     return false;
    172 
    173   if (!GetDB().Execute("DROP TABLE android_urls_tmp"))
    174     return false;
    175 
    176   return true;
    177 }
    178 
    179 }  // namespace history
    180