142 lines
3.6 KiB
C++
142 lines
3.6 KiB
C++
/**
|
|
* Orthanc - A Lightweight, RESTful DICOM Store
|
|
* Copyright (C) 2012-2016 Sebastien Jodogne, Medical Physics
|
|
* Department, University Hospital of Liege, Belgium
|
|
* Copyright (C) 2017-2023 Osimis S.A., Belgium
|
|
* Copyright (C) 2024-2025 Orthanc Team SRL, Belgium
|
|
* Copyright (C) 2021-2025 Sebastien Jodogne, ICTEAM UCLouvain, Belgium
|
|
*
|
|
* This program is free software: you can redistribute it and/or
|
|
* modify it under the terms of the GNU General Public License as
|
|
* published by the Free Software Foundation, either version 3 of the
|
|
* License, or (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful, but
|
|
* WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
* General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
**/
|
|
|
|
|
|
#include "PendingDeletionsDatabase.h"
|
|
|
|
#include "../../../../OrthancFramework/Sources/SQLite/Statement.h"
|
|
#include "../../../../OrthancFramework/Sources/SQLite/Transaction.h"
|
|
|
|
void PendingDeletionsDatabase::Setup()
|
|
{
|
|
// Performance tuning of SQLite with PRAGMAs
|
|
// http://www.sqlite.org/pragma.html
|
|
db_.Execute("PRAGMA SYNCHRONOUS=NORMAL;");
|
|
db_.Execute("PRAGMA JOURNAL_MODE=WAL;");
|
|
db_.Execute("PRAGMA LOCKING_MODE=EXCLUSIVE;");
|
|
db_.Execute("PRAGMA WAL_AUTOCHECKPOINT=1000;");
|
|
|
|
{
|
|
Orthanc::SQLite::Transaction t(db_);
|
|
t.Begin();
|
|
|
|
if (!db_.DoesTableExist("Pending"))
|
|
{
|
|
db_.Execute("CREATE TABLE Pending(uuid TEXT, type INTEGER)");
|
|
|
|
// New in v 1.12.7+
|
|
// add an index on uuid to speed up the DELETE FROM Pending WHERE uuid=?
|
|
// With this patch, we observed a 100 fold performance
|
|
// improvement when the Pending table contains 1-2 millions files.
|
|
db_.Execute("CREATE INDEX PendingIndex ON Pending(uuid)");
|
|
}
|
|
|
|
t.Commit();
|
|
}
|
|
}
|
|
|
|
|
|
PendingDeletionsDatabase::PendingDeletionsDatabase(const std::string& path)
|
|
{
|
|
db_.Open(path);
|
|
Setup();
|
|
}
|
|
|
|
|
|
void PendingDeletionsDatabase::Enqueue(const std::string& uuid,
|
|
Orthanc::FileContentType type)
|
|
{
|
|
boost::mutex::scoped_lock lock(mutex_);
|
|
|
|
Orthanc::SQLite::Transaction t(db_);
|
|
t.Begin();
|
|
|
|
{
|
|
Orthanc::SQLite::Statement s(db_, SQLITE_FROM_HERE, "INSERT INTO Pending VALUES(?, ?)");
|
|
s.BindString(0, uuid);
|
|
s.BindInt(1, type);
|
|
s.Run();
|
|
}
|
|
|
|
t.Commit();
|
|
}
|
|
|
|
|
|
bool PendingDeletionsDatabase::Dequeue(std::string& uuid,
|
|
Orthanc::FileContentType& type)
|
|
{
|
|
bool ok = false;
|
|
|
|
boost::mutex::scoped_lock lock(mutex_);
|
|
|
|
Orthanc::SQLite::Transaction t(db_);
|
|
t.Begin();
|
|
|
|
{
|
|
Orthanc::SQLite::Statement s(db_, SQLITE_FROM_HERE, "SELECT uuid, type FROM Pending LIMIT 1");
|
|
|
|
if (s.Step())
|
|
{
|
|
uuid = s.ColumnString(0);
|
|
type = static_cast<Orthanc::FileContentType>(s.ColumnInt(1));
|
|
|
|
Orthanc::SQLite::Statement s(db_, SQLITE_FROM_HERE, "DELETE FROM Pending WHERE uuid=?");
|
|
s.BindString(0, uuid);
|
|
s.Run();
|
|
|
|
ok = true;
|
|
}
|
|
}
|
|
|
|
t.Commit();
|
|
|
|
return ok;
|
|
}
|
|
|
|
|
|
unsigned int PendingDeletionsDatabase::GetSize()
|
|
{
|
|
boost::mutex::scoped_lock lock(mutex_);
|
|
|
|
unsigned int value = 0;
|
|
|
|
Orthanc::SQLite::Transaction t(db_);
|
|
t.Begin();
|
|
|
|
{
|
|
Orthanc::SQLite::Statement s(db_, SQLITE_FROM_HERE, "SELECT COUNT(*) FROM Pending");
|
|
|
|
if (s.Step())
|
|
{
|
|
int tmp = s.ColumnInt(0);
|
|
if (tmp > 0)
|
|
{
|
|
value = static_cast<unsigned int>(tmp);
|
|
}
|
|
}
|
|
}
|
|
|
|
t.Commit();
|
|
|
|
return value;
|
|
}
|