tdlight/tddb/td/db/detail/RawSqliteDb.cpp
Arseny Smirnov 03b674ab1c SqliteDb: use proper destroy instead of unlink
GitOrigin-RevId: 73d2d992c362c7b0c6d6f28adef56c416ec8c158
2018-06-07 22:56:03 +03:00

47 lines
1.2 KiB
C++

//
// Copyright Aliaksei Levin (levlam@telegram.org), Arseny Smirnov (arseny30@gmail.com) 2014-2018
//
// Distributed under the Boost Software License, Version 1.0. (See accompanying
// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
//
#include "td/db/detail/RawSqliteDb.h"
#include "sqlite/sqlite3.h"
#include "td/utils/common.h"
#include "td/utils/logging.h"
#include "td/utils/port/path.h"
#include "td/utils/port/Stat.h"
namespace td {
namespace detail {
Status RawSqliteDb::last_error(sqlite3 *db) {
return Status::Error(Slice(sqlite3_errmsg(db)));
}
Status RawSqliteDb::destroy(Slice path) {
Status error;
with_db_path(path, [&](auto path) {
unlink(path).ignore();
if (!stat(path).is_error()) {
error = Status::Error(PSLICE() << "Failed to delete " << tag("path", path));
}
});
return error;
}
Status RawSqliteDb::last_error() {
//If database was corrupted, try to delete it.
auto code = sqlite3_errcode(db_);
if (code == SQLITE_CORRUPT) {
destroy(path_).ignore();
}
return last_error(db_);
}
RawSqliteDb::~RawSqliteDb() {
auto rc = sqlite3_close(db_);
LOG_IF(FATAL, rc != SQLITE_OK) << last_error(db_);
}
} // namespace detail
} // namespace td