Inheritance diagram for BitTorrent::DBResumeDataStorage::Worker:

Collaboration diagram for BitTorrent::DBResumeDataStorage::Worker:

Public Member Functions
	Worker (const QString &dbPath, const QString &dbConnectionName)

void	openDatabase () const

void	closeDatabase () const

void	store (const TorrentID &id, const LoadTorrentParams &resumeData) const

void	remove (const TorrentID &id) const

void	storeQueue (const QVector< TorrentID > &queue) const

Private Attributes
const QString	m_path

const QString	m_connectionName

Detailed Description

Definition at line 171 of file dbresumedatastorage.cpp.

Constructor & Destructor Documentation

◆ Worker()

BitTorrent::DBResumeDataStorage::Worker::Worker	(	const QString &	dbPath,
		const QString &	dbConnectionName
	)

Definition at line 488 of file dbresumedatastorage.cpp.

     : m_path {dbPath}
     , m_connectionName {dbConnectionName}
 {
 }

Member Function Documentation

◆ closeDatabase()

void BitTorrent::DBResumeDataStorage::Worker::closeDatabase ( ) const

Definition at line 502 of file dbresumedatastorage.cpp.

 {
     QSqlDatabase::removeDatabase(m_connectionName);
 }

◆ openDatabase()

void BitTorrent::DBResumeDataStorage::Worker::openDatabase ( ) const

Definition at line 494 of file dbresumedatastorage.cpp.

 {
     auto db = QSqlDatabase::addDatabase(QLatin1String("QSQLITE"), m_connectionName);
     db.setDatabaseName(m_path);
     if (!db.open())
         throw RuntimeError(db.lastError().text());
 }

Referenced by BitTorrent::DBResumeDataStorage::DBResumeDataStorage().

Here is the caller graph for this function:

◆ remove()

void BitTorrent::DBResumeDataStorage::Worker::remove ( const TorrentID & id ) const

Definition at line 624 of file dbresumedatastorage.cpp.

 {
     const auto deleteTorrentStatement = QString::fromLatin1("DELETE FROM %1 WHERE %2 = %3;")
             .arg(quoted(DB_TABLE_TORRENTS), quoted(DB_COLUMN_TORRENT_ID.name), DB_COLUMN_TORRENT_ID.placeholder);
  
     auto db = QSqlDatabase::database(m_connectionName);
     QSqlQuery query {db};
  
     try
     {
         if (!query.prepare(deleteTorrentStatement))
             throw RuntimeError(query.lastError().text());
  
         query.bindValue(DB_COLUMN_TORRENT_ID.placeholder, id.toString());
         if (!query.exec())
             throw RuntimeError(query.lastError().text());
     }
     catch (const RuntimeError &err)
     {
         LogMsg(tr("Couldn't delete resume data of torrent '%1'. Error: %2")
             .arg(id.toString(), err.message()), Log::CRITICAL);
     }
 }

References Log::CRITICAL, anonymous_namespace{dbresumedatastorage.cpp}::DB_COLUMN_TORRENT_ID, anonymous_namespace{dbresumedatastorage.cpp}::DB_TABLE_TORRENTS, LogMsg(), Exception::message(), anonymous_namespace{dbresumedatastorage.cpp}::Column::name, anonymous_namespace{dbresumedatastorage.cpp}::Column::placeholder, anonymous_namespace{dbresumedatastorage.cpp}::quoted(), and anonymous_namespace{session.cpp}::toString().

Here is the call graph for this function:

◆ store()

void BitTorrent::DBResumeDataStorage::Worker::store	(	const TorrentID &	id,
		const LoadTorrentParams &	resumeData
	)		const

Definition at line 507 of file dbresumedatastorage.cpp.

 {
     // We need to adjust native libtorrent resume data
     lt::add_torrent_params p = resumeData.ltAddTorrentParams;
     p.save_path = Profile::instance()->toPortablePath(QString::fromStdString(p.save_path)).toStdString();
     if (resumeData.stopped)
     {
         p.flags |= lt::torrent_flags::paused;
         p.flags &= ~lt::torrent_flags::auto_managed;
     }
     else
     {
         // Torrent can be actually "running" but temporarily "paused" to perform some
         // service jobs behind the scenes so we need to restore it as "running"
         if (resumeData.operatingMode == BitTorrent::TorrentOperatingMode::AutoManaged)
         {
             p.flags |= lt::torrent_flags::auto_managed;
         }
         else
         {
             p.flags &= ~lt::torrent_flags::paused;
             p.flags &= ~lt::torrent_flags::auto_managed;
         }
     }
  
     QVector<Column> columns {
         DB_COLUMN_TORRENT_ID,
         DB_COLUMN_NAME,
         DB_COLUMN_CATEGORY,
         DB_COLUMN_TAGS,
         DB_COLUMN_TARGET_SAVE_PATH,
         DB_COLUMN_CONTENT_LAYOUT,
         DB_COLUMN_RATIO_LIMIT,
         DB_COLUMN_SEEDING_TIME_LIMIT,
         DB_COLUMN_HAS_OUTER_PIECES_PRIORITY,
         DB_COLUMN_HAS_SEED_STATUS,
         DB_COLUMN_OPERATING_MODE,
         DB_COLUMN_STOPPED,
         DB_COLUMN_RESUMEDATA
     };
  
     lt::entry data = lt::write_resume_data(p);
  
     // metadata is stored in separate column
     QByteArray bencodedMetadata;
     if (p.ti)
     {
         lt::entry::dictionary_type &dataDict = data.dict();
         lt::entry metadata {lt::entry::dictionary_t};
         lt::entry::dictionary_type &metadataDict = metadata.dict();
         metadataDict.insert(dataDict.extract("info"));
         metadataDict.insert(dataDict.extract("creation date"));
         metadataDict.insert(dataDict.extract("created by"));
         metadataDict.insert(dataDict.extract("comment"));
  
         try
         {
             bencodedMetadata.reserve(512 * 1024);
             lt::bencode(std::back_inserter(bencodedMetadata), metadata);
         }
         catch (const std::exception &err)
         {
             LogMsg(tr("Couldn't save torrent metadata. Error: %1.")
                    .arg(QString::fromLocal8Bit(err.what())), Log::CRITICAL);
             return;
         }
  
         columns.append(DB_COLUMN_METADATA);
     }
  
     QByteArray bencodedResumeData;
     bencodedResumeData.reserve(256 * 1024);
     lt::bencode(std::back_inserter(bencodedResumeData), data);
  
     const QString insertTorrentStatement = makeInsertStatement(DB_TABLE_TORRENTS, columns)
             + makeOnConflictUpdateStatement(DB_COLUMN_TORRENT_ID, columns);
     auto db = QSqlDatabase::database(m_connectionName);
     QSqlQuery query {db};
  
     try
     {
         if (!query.prepare(insertTorrentStatement))
             throw RuntimeError(query.lastError().text());
  
         query.bindValue(DB_COLUMN_TORRENT_ID.placeholder, id.toString());
         query.bindValue(DB_COLUMN_NAME.placeholder, resumeData.name);
         query.bindValue(DB_COLUMN_CATEGORY.placeholder, resumeData.category);
         query.bindValue(DB_COLUMN_TAGS.placeholder, (resumeData.tags.isEmpty()
             ? QVariant(QVariant::String) : resumeData.tags.join(QLatin1String(","))));
         query.bindValue(DB_COLUMN_CONTENT_LAYOUT.placeholder, Utils::String::fromEnum(resumeData.contentLayout));
         query.bindValue(DB_COLUMN_RATIO_LIMIT.placeholder, static_cast<int>(resumeData.ratioLimit * 1000));
         query.bindValue(DB_COLUMN_SEEDING_TIME_LIMIT.placeholder, resumeData.seedingTimeLimit);
         query.bindValue(DB_COLUMN_HAS_OUTER_PIECES_PRIORITY.placeholder, resumeData.firstLastPiecePriority);
         query.bindValue(DB_COLUMN_HAS_SEED_STATUS.placeholder, resumeData.hasSeedStatus);
         query.bindValue(DB_COLUMN_OPERATING_MODE.placeholder, Utils::String::fromEnum(resumeData.operatingMode));
         query.bindValue(DB_COLUMN_STOPPED.placeholder, resumeData.stopped);
  
         if (!resumeData.useAutoTMM)
         {
             query.bindValue(DB_COLUMN_TARGET_SAVE_PATH.placeholder, Profile::instance()->toPortablePath(resumeData.savePath));
             query.bindValue(DB_COLUMN_DOWNLOAD_PATH.placeholder, Profile::instance()->toPortablePath(resumeData.downloadPath));
         }
  
         query.bindValue(DB_COLUMN_RESUMEDATA.placeholder, bencodedResumeData);
         if (!bencodedMetadata.isEmpty())
             query.bindValue(DB_COLUMN_METADATA.placeholder, bencodedMetadata);
  
         if (!query.exec())
             throw RuntimeError(query.lastError().text());
     }
     catch (const RuntimeError &err)
     {
         LogMsg(tr("Couldn't store resume data for torrent '%1'. Error: %2")
             .arg(id.toString(), err.message()), Log::CRITICAL);
     }
 }

Here is the call graph for this function:

◆ storeQueue()

void BitTorrent::DBResumeDataStorage::Worker::storeQueue ( const QVector< TorrentID > & queue ) const

Definition at line 648 of file dbresumedatastorage.cpp.

 {
     const auto updateQueuePosStatement = QString::fromLatin1("UPDATE %1 SET %2 = %3 WHERE %4 = %5;")
             .arg(quoted(DB_TABLE_TORRENTS), quoted(DB_COLUMN_QUEUE_POSITION.name), DB_COLUMN_QUEUE_POSITION.placeholder
                  , quoted(DB_COLUMN_TORRENT_ID.name), DB_COLUMN_TORRENT_ID.placeholder);
  
     auto db = QSqlDatabase::database(m_connectionName);
  
     try
     {
         if (!db.transaction())
             throw RuntimeError(db.lastError().text());
  
         QSqlQuery query {db};
  
         try
         {
             if (!query.prepare(updateQueuePosStatement))
                 throw RuntimeError(query.lastError().text());
  
             int pos = 0;
             for (const TorrentID &torrentID : queue)
             {
                 query.bindValue(DB_COLUMN_TORRENT_ID.placeholder, torrentID.toString());
                 query.bindValue(DB_COLUMN_QUEUE_POSITION.placeholder, pos++);
                 if (!query.exec())
                     throw RuntimeError(query.lastError().text());
             }
  
             if (!db.commit())
                 throw RuntimeError(db.lastError().text());
         }
         catch (const RuntimeError &)
         {
             db.rollback();
             throw;
         }
     }
     catch (const RuntimeError &err)
     {
         LogMsg(tr("Couldn't store torrents queue positions. Error: %1")
             .arg(err.message()), Log::CRITICAL);
     }
 }

References Log::CRITICAL, anonymous_namespace{dbresumedatastorage.cpp}::DB_COLUMN_QUEUE_POSITION, anonymous_namespace{dbresumedatastorage.cpp}::DB_COLUMN_TORRENT_ID, anonymous_namespace{dbresumedatastorage.cpp}::DB_TABLE_TORRENTS, LogMsg(), Exception::message(), anonymous_namespace{dbresumedatastorage.cpp}::Column::name, anonymous_namespace{dbresumedatastorage.cpp}::Column::placeholder, and anonymous_namespace{dbresumedatastorage.cpp}::quoted().

Here is the call graph for this function:

Member Data Documentation

◆ m_connectionName

const QString BitTorrent::DBResumeDataStorage::Worker::m_connectionName

private

Definition at line 187 of file dbresumedatastorage.cpp.

◆ m_path

const QString BitTorrent::DBResumeDataStorage::Worker::m_path

private

Definition at line 186 of file dbresumedatastorage.cpp.

The documentation for this class was generated from the following file:

src/base/bittorrent/dbresumedatastorage.cpp

Public Member Functions

Private Attributes

Detailed Description

Constructor & Destructor Documentation

◆ Worker()

Member Function Documentation

◆ closeDatabase()

◆ openDatabase()

◆ remove()

◆ store()

◆ storeQueue()

Member Data Documentation

◆ m_connectionName

◆ m_path