mirror of
https://github.com/facebook/sapling.git
synced 2024-10-09 08:18:15 +03:00
eaeaf8f23c
Summary: Adds a SQLite storage implementation and makes it the default engine for integration tests; this requires fewer resources to run and the integration tests thus run faster and more reliably. In the future we may add a configuration option to remember the storage engine that was used as it is currently not "safe" to switch from one to the other because the hgproxyhash data cannot be recreated without re-importing a revision. Reviewed By: simpkins Differential Revision: D6919456 fbshipit-source-id: 3afbfafb190cca0e3c797cd9b7cd051768575a8c
107 lines
2.6 KiB
C++
107 lines
2.6 KiB
C++
/*
|
|
* Copyright (c) 2018-present, Facebook, Inc.
|
|
* All rights reserved.
|
|
*
|
|
* This source code is licensed under the BSD-style license found in the
|
|
* LICENSE file in the root directory of this source tree. An additional grant
|
|
* of patent rights can be found in the PATENTS file in the same directory.
|
|
*
|
|
*/
|
|
#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;
|
|
checkSqliteResult(nullptr, sqlite3_open(path.copy().c_str(), &db));
|
|
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
|