Mercurial > hg > orthanc
view OrthancServer/Plugins/Samples/DelayedDeletion/PendingDeletionsDatabase.cpp @ 5272:a45e8b6115f6
fix
author | Sebastien Jodogne <s.jodogne@gmail.com> |
---|---|
date | Thu, 13 Apr 2023 21:17:55 +0200 |
parents | 99751c5a7cfe |
children | 48b8dae6dc77 |
line wrap: on
line source
/** * 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) 2021-2023 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)"); } 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; }