Create a class to represent a DOM Storage Database.

sql/statement.cc

 // 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.
 
 #include "sql/statement.h"
 
+#include <algorithm>
+
 #include "base/logging.h"
 #include "base/utf_string_conversions.h"
 #include "third_party/sqlite/sqlite3.h"
 
 namespace sql {
 
 // This empty constructor initializes our reference with an empty one so that
 // we don't have to NULL-check the ref_ to see if the statement is valid: we
 // only have to check the ref's validity bit.
 Statement::Statement()
@@ skipping 131 matching lines @@
   // Verify that our enum matches sqlite's values.
   COMPILE_ASSERT(COLUMN_TYPE_INTEGER == SQLITE_INTEGER, integer_no_match);
   COMPILE_ASSERT(COLUMN_TYPE_FLOAT == SQLITE_FLOAT, float_no_match);
   COMPILE_ASSERT(COLUMN_TYPE_TEXT == SQLITE_TEXT, integer_no_match);
   COMPILE_ASSERT(COLUMN_TYPE_BLOB == SQLITE_BLOB, blob_no_match);
   COMPILE_ASSERT(COLUMN_TYPE_NULL == SQLITE_NULL, null_no_match);
 
   return static_cast<ColType>(sqlite3_column_type(ref_->stmt(), col));
 }
 
+ColType Statement::DeclaredColumnType(int col) const {
+  std::string column_type(sqlite3_column_decltype(ref_->stmt(), col));
+  std::transform(column_type.begin(), column_type.end(), column_type.begin(),
+                 ::tolower);
+
+  if (column_type == "integer")
+    return COLUMN_TYPE_INTEGER;
+  else if (column_type == "float")
+    return COLUMN_TYPE_FLOAT;
+  else if (column_type == "text")
+    return COLUMN_TYPE_TEXT;
+  else if (column_type == "blob")
+    return COLUMN_TYPE_BLOB;
+
+  return COLUMN_TYPE_NULL;
+}
+
 bool Statement::ColumnBool(int col) const {
   return !!ColumnInt(col);
 }
 
 int Statement::ColumnInt(int col) const {
   if (!CheckValid())
     return 0;
 
   return sqlite3_column_int(ref_->stmt(), col);
 }
@@ skipping 76 matching lines @@
   }
   return true;
 }
 
 bool Statement::ColumnBlobAsVector(
     int col,
     std::vector<unsigned char>* val) const {
   return ColumnBlobAsVector(col, reinterpret_cast< std::vector<char>* >(val));
 }
 
+bool Statement::ColumnBlobAsString16(int col, string16* val) const {

[michaeln, 2012/02/01 21:03:11]

it be good to more closely match the impl of ColumnBlobAsString

[benm (inactive), 2012/02/02 12:14:49]

Done.

+  if (!CheckValid())
+    return false;
+
+  const void* data = sqlite3_column_blob(ref_->stmt(), col);
+  int len = sqlite3_column_bytes(ref_->stmt(), col);
+  if (data && len > 0) {
+    val->resize(len);

[michaeln, 2012/02/01 21:03:11]

len/2 since resize(arg) is in characters whereas len is in bytes?

[benm (inactive), 2012/02/02 12:14:49]

Done.

+    val->assign(reinterpret_cast<const char16*>(data));
+  }
+  return true;
+}
+
 const char* Statement::GetSQLStatement() {
   return sqlite3_sql(ref_->stmt());
 }
 
 bool Statement::CheckOk(int err) const {
   // Binding to a non-existent variable is evidence of a serious error.
   // TODO(gbillock,shess): make this invalidate the statement so it
   // can't wreak havoc.
   if (err == SQLITE_RANGE)
     DLOG(FATAL) << "Bind value out of range";
   return err == SQLITE_OK;
 }
 
 int Statement::CheckError(int err) {
   // Please don't add DCHECKs here, OnSqliteError() already has them.
   succeeded_ = (err == SQLITE_OK || err == SQLITE_ROW || err == SQLITE_DONE);
   if (!succeeded_ && is_valid())
     return ref_->connection()->OnSqliteError(err, this);
   return err;
 }
 
 }  // namespace sql