| // Copyright (c) 2012 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 "base/logging.h" |
| #include "base/strings/string_util.h" |
| #include "base/strings/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() |
| : ref_(new Connection::StatementRef(NULL, NULL, false)), |
| stepped_(false), |
| succeeded_(false) { |
| } |
| |
| Statement::Statement(scoped_refptr<Connection::StatementRef> ref) |
| : ref_(ref), |
| stepped_(false), |
| succeeded_(false) { |
| } |
| |
| Statement::~Statement() { |
| // Free the resources associated with this statement. We assume there's only |
| // one statement active for a given sqlite3_stmt at any time, so this won't |
| // mess with anything. |
| Reset(true); |
| } |
| |
| void Statement::Assign(scoped_refptr<Connection::StatementRef> ref) { |
| Reset(true); |
| ref_ = ref; |
| } |
| |
| void Statement::Clear() { |
| Assign(new Connection::StatementRef(NULL, NULL, false)); |
| succeeded_ = false; |
| } |
| |
| bool Statement::CheckValid() const { |
| // Allow operations to fail silently if a statement was invalidated |
| // because the database was closed by an error handler. |
| DLOG_IF(FATAL, !ref_->was_valid()) |
| << "Cannot call mutating statements on an invalid statement."; |
| return is_valid(); |
| } |
| |
| int Statement::StepInternal(bool timer_flag) { |
| ref_->AssertIOAllowed(); |
| if (!CheckValid()) |
| return SQLITE_ERROR; |
| |
| const bool was_stepped = stepped_; |
| stepped_ = true; |
| int ret = SQLITE_ERROR; |
| if (!ref_->connection()) { |
| ret = sqlite3_step(ref_->stmt()); |
| } else { |
| if (!timer_flag) { |
| ret = sqlite3_step(ref_->stmt()); |
| } else { |
| const base::TimeTicks before = ref_->connection()->Now(); |
| ret = sqlite3_step(ref_->stmt()); |
| const base::TimeTicks after = ref_->connection()->Now(); |
| const bool read_only = !!sqlite3_stmt_readonly(ref_->stmt()); |
| ref_->connection()->RecordTimeAndChanges(after - before, read_only); |
| } |
| |
| if (!was_stepped) |
| ref_->connection()->RecordOneEvent(Connection::EVENT_STATEMENT_RUN); |
| |
| if (ret == SQLITE_ROW) |
| ref_->connection()->RecordOneEvent(Connection::EVENT_STATEMENT_ROWS); |
| } |
| return CheckError(ret); |
| } |
| |
| bool Statement::Run() { |
| DCHECK(!stepped_); |
| return StepInternal(true) == SQLITE_DONE; |
| } |
| |
| bool Statement::RunWithoutTimers() { |
| DCHECK(!stepped_); |
| return StepInternal(false) == SQLITE_DONE; |
| } |
| |
| bool Statement::Step() { |
| return StepInternal(true) == SQLITE_ROW; |
| } |
| |
| void Statement::Reset(bool clear_bound_vars) { |
| ref_->AssertIOAllowed(); |
| if (is_valid()) { |
| if (clear_bound_vars) |
| sqlite3_clear_bindings(ref_->stmt()); |
| |
| // StepInternal() cannot track success because statements may be reset |
| // before reaching SQLITE_DONE. Don't call CheckError() because |
| // sqlite3_reset() returns the last step error, which StepInternal() already |
| // checked. |
| const int rc =sqlite3_reset(ref_->stmt()); |
| if (rc == SQLITE_OK && ref_->connection()) |
| ref_->connection()->RecordOneEvent(Connection::EVENT_STATEMENT_SUCCESS); |
| } |
| |
| // Potentially release dirty cache pages if an autocommit statement made |
| // changes. |
| if (ref_->connection()) |
| ref_->connection()->ReleaseCacheMemoryIfNeeded(false); |
| |
| succeeded_ = false; |
| stepped_ = false; |
| } |
| |
| bool Statement::Succeeded() const { |
| if (!is_valid()) |
| return false; |
| |
| return succeeded_; |
| } |
| |
| bool Statement::BindNull(int col) { |
| DCHECK(!stepped_); |
| if (!is_valid()) |
| return false; |
| |
| return CheckOk(sqlite3_bind_null(ref_->stmt(), col + 1)); |
| } |
| |
| bool Statement::BindBool(int col, bool val) { |
| return BindInt(col, val ? 1 : 0); |
| } |
| |
| bool Statement::BindInt(int col, int val) { |
| DCHECK(!stepped_); |
| if (!is_valid()) |
| return false; |
| |
| return CheckOk(sqlite3_bind_int(ref_->stmt(), col + 1, val)); |
| } |
| |
| bool Statement::BindInt64(int col, int64_t val) { |
| DCHECK(!stepped_); |
| if (!is_valid()) |
| return false; |
| |
| return CheckOk(sqlite3_bind_int64(ref_->stmt(), col + 1, val)); |
| } |
| |
| bool Statement::BindDouble(int col, double val) { |
| DCHECK(!stepped_); |
| if (!is_valid()) |
| return false; |
| |
| return CheckOk(sqlite3_bind_double(ref_->stmt(), col + 1, val)); |
| } |
| |
| bool Statement::BindCString(int col, const char* val) { |
| DCHECK(!stepped_); |
| if (!is_valid()) |
| return false; |
| |
| return CheckOk( |
| sqlite3_bind_text(ref_->stmt(), col + 1, val, -1, SQLITE_TRANSIENT)); |
| } |
| |
| bool Statement::BindString(int col, const std::string& val) { |
| DCHECK(!stepped_); |
| if (!is_valid()) |
| return false; |
| |
| return CheckOk(sqlite3_bind_text(ref_->stmt(), |
| col + 1, |
| val.data(), |
| val.size(), |
| SQLITE_TRANSIENT)); |
| } |
| |
| bool Statement::BindString16(int col, const base::string16& value) { |
| return BindString(col, base::UTF16ToUTF8(value)); |
| } |
| |
| bool Statement::BindBlob(int col, const void* val, int val_len) { |
| DCHECK(!stepped_); |
| if (!is_valid()) |
| return false; |
| |
| return CheckOk( |
| sqlite3_bind_blob(ref_->stmt(), col + 1, val, val_len, SQLITE_TRANSIENT)); |
| } |
| |
| int Statement::ColumnCount() const { |
| if (!is_valid()) |
| return 0; |
| |
| return sqlite3_column_count(ref_->stmt()); |
| } |
| |
| ColType Statement::ColumnType(int col) const { |
| // Verify that our enum matches sqlite's values. |
| static_assert(COLUMN_TYPE_INTEGER == SQLITE_INTEGER, "integer no match"); |
| static_assert(COLUMN_TYPE_FLOAT == SQLITE_FLOAT, "float no match"); |
| static_assert(COLUMN_TYPE_TEXT == SQLITE_TEXT, "integer no match"); |
| static_assert(COLUMN_TYPE_BLOB == SQLITE_BLOB, "blob no match"); |
| static_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 = base::ToLowerASCII( |
| sqlite3_column_decltype(ref_->stmt(), col)); |
| |
| 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); |
| } |
| |
| int64_t Statement::ColumnInt64(int col) const { |
| if (!CheckValid()) |
| return 0; |
| |
| return sqlite3_column_int64(ref_->stmt(), col); |
| } |
| |
| double Statement::ColumnDouble(int col) const { |
| if (!CheckValid()) |
| return 0; |
| |
| return sqlite3_column_double(ref_->stmt(), col); |
| } |
| |
| std::string Statement::ColumnString(int col) const { |
| if (!CheckValid()) |
| return std::string(); |
| |
| const char* str = reinterpret_cast<const char*>( |
| sqlite3_column_text(ref_->stmt(), col)); |
| int len = sqlite3_column_bytes(ref_->stmt(), col); |
| |
| std::string result; |
| if (str && len > 0) |
| result.assign(str, len); |
| return result; |
| } |
| |
| base::string16 Statement::ColumnString16(int col) const { |
| if (!CheckValid()) |
| return base::string16(); |
| |
| std::string s = ColumnString(col); |
| return !s.empty() ? base::UTF8ToUTF16(s) : base::string16(); |
| } |
| |
| int Statement::ColumnByteLength(int col) const { |
| if (!CheckValid()) |
| return 0; |
| |
| return sqlite3_column_bytes(ref_->stmt(), col); |
| } |
| |
| const void* Statement::ColumnBlob(int col) const { |
| if (!CheckValid()) |
| return NULL; |
| |
| return sqlite3_column_blob(ref_->stmt(), col); |
| } |
| |
| bool Statement::ColumnBlobAsString(int col, std::string* blob) { |
| if (!CheckValid()) |
| return false; |
| |
| const void* p = ColumnBlob(col); |
| size_t len = ColumnByteLength(col); |
| blob->resize(len); |
| if (blob->size() != len) { |
| return false; |
| } |
| blob->assign(reinterpret_cast<const char*>(p), len); |
| return true; |
| } |
| |
| bool Statement::ColumnBlobAsString16(int col, base::string16* val) const { |
| if (!CheckValid()) |
| return false; |
| |
| const void* data = ColumnBlob(col); |
| size_t len = ColumnByteLength(col) / sizeof(base::char16); |
| val->resize(len); |
| if (val->size() != len) |
| return false; |
| val->assign(reinterpret_cast<const base::char16*>(data), len); |
| return true; |
| } |
| |
| bool Statement::ColumnBlobAsVector(int col, std::vector<char>* val) const { |
| val->clear(); |
| |
| 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); |
| memcpy(&(*val)[0], data, len); |
| } |
| return true; |
| } |
| |
| bool Statement::ColumnBlobAsVector( |
| int col, |
| std::vector<unsigned char>* val) const { |
| return ColumnBlobAsVector(col, reinterpret_cast< std::vector<char>* >(val)); |
| } |
| |
| 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_ && ref_.get() && ref_->connection()) |
| return ref_->connection()->OnSqliteError(err, this, NULL); |
| return err; |
| } |
| |
| } // namespace sql |