mirror of
https://github.com/facebook/sapling.git
synced 2024-10-09 00:14:35 +03:00
fbdb46f5cb
Reviewed By: chadaustin Differential Revision: D17872966 fbshipit-source-id: cd60a364a2146f0dadbeca693b1d4a5d7c97ff63
111 lines
2.7 KiB
C++
111 lines
2.7 KiB
C++
/*
|
|
* Copyright (c) Facebook, Inc. and its affiliates.
|
|
*
|
|
* This software may be used and distributed according to the terms of the
|
|
* GNU General Public License version 2.
|
|
*/
|
|
|
|
#include "eden/fs/sqlite/Sqlite.h"
|
|
using folly::StringPiece;
|
|
using folly::Synchronized;
|
|
using folly::to;
|
|
using std::string;
|
|
|
|
namespace facebook {
|
|
namespace eden {
|
|
|
|
// Given a sqlite result code, if the result was not successful
|
|
// (SQLITE_OK), format an error message and throw an exception.
|
|
void checkSqliteResult(sqlite3* db, int result) {
|
|
if (result == SQLITE_OK) {
|
|
return;
|
|
}
|
|
// Sometimes the db instance holds more useful context
|
|
if (db) {
|
|
throw std::runtime_error(to<string>(
|
|
"sqlite error: ",
|
|
result,
|
|
": ",
|
|
sqlite3_errstr(result),
|
|
" ",
|
|
sqlite3_errmsg(db)));
|
|
}
|
|
// otherwise resort to a simpler number->string mapping
|
|
throw std::runtime_error(
|
|
to<string>("sqlite error: ", result, ": ", sqlite3_errstr(result)));
|
|
}
|
|
|
|
SqliteDatabase::SqliteDatabase(AbsolutePathPiece path) {
|
|
sqlite3* db = nullptr;
|
|
auto result = sqlite3_open(path.copy().c_str(), &db);
|
|
if (result != SQLITE_OK) {
|
|
// On most error conditions sqlite3_open() does allocate the DB object,
|
|
// and it needs to be closed afterwards if it is non-null.
|
|
sqlite3_close(db);
|
|
checkSqliteResult(nullptr, result);
|
|
}
|
|
db_ = db;
|
|
}
|
|
|
|
void SqliteDatabase::close() {
|
|
auto db = db_.wlock();
|
|
if (*db) {
|
|
sqlite3_close(*db);
|
|
*db = nullptr;
|
|
}
|
|
}
|
|
|
|
SqliteDatabase::~SqliteDatabase() {
|
|
close();
|
|
}
|
|
|
|
Synchronized<sqlite3*>::LockedPtr SqliteDatabase::lock() {
|
|
return db_.wlock();
|
|
}
|
|
|
|
SqliteStatement::SqliteStatement(
|
|
folly::Synchronized<sqlite3*>::LockedPtr& db,
|
|
folly::StringPiece query)
|
|
: db_{*db} {
|
|
checkSqliteResult(
|
|
db_,
|
|
sqlite3_prepare_v2(db_, query.data(), query.size(), &stmt_, nullptr));
|
|
}
|
|
|
|
bool SqliteStatement::step() {
|
|
auto result = sqlite3_step(stmt_);
|
|
switch (result) {
|
|
case SQLITE_ROW:
|
|
return true;
|
|
case SQLITE_DONE:
|
|
sqlite3_reset(stmt_);
|
|
return false;
|
|
default:
|
|
checkSqliteResult(db_, result);
|
|
folly::assume_unreachable();
|
|
}
|
|
}
|
|
|
|
void SqliteStatement::bind(
|
|
size_t paramNo,
|
|
folly::StringPiece blob,
|
|
void (*bindType)(void*)) {
|
|
checkSqliteResult(
|
|
db_,
|
|
sqlite3_bind_blob64(
|
|
stmt_, paramNo, blob.data(), sqlite3_uint64(blob.size()), bindType));
|
|
}
|
|
|
|
StringPiece SqliteStatement::columnBlob(size_t colNo) const {
|
|
return StringPiece(
|
|
reinterpret_cast<const char*>(sqlite3_column_blob(stmt_, colNo)),
|
|
sqlite3_column_bytes(stmt_, colNo));
|
|
}
|
|
|
|
SqliteStatement::~SqliteStatement() {
|
|
sqlite3_finalize(stmt_);
|
|
}
|
|
|
|
} // namespace eden
|
|
} // namespace facebook
|