File indexing completed on 2025-01-05 04:46:58

 /***************************************************************************
  *   SPDX-FileCopyrightText: 2006 Andreas Gungl <a.gungl@gmx.de>           *
  *                                                                         *
  *   SPDX-License-Identifier: LGPL-2.0-or-later                            *
  ***************************************************************************/
 
 #include "entity.h"
 #include "countquerybuilder.h"
 #include "datastore.h"
 
 #include <QSqlDatabase>
 #include <QVariant>
 
 using namespace Akonadi::Server;
 
 Entity::Entity()
     : m_id(-1)
 {
 }
 
 Entity::Entity(qint64 id)
     : m_id(id)
 {
 }
 
 Entity::~Entity()
 {
 }
 
 DataStore *Entity::dataStore()
 {
     return DataStore::self();
 }
 
 qint64 Entity::id() const
 {
     return m_id;
 }
 
 void Entity::setId(qint64 id)
 {
     m_id = id;
 }
 
 bool Entity::isValid() const
 {
     return m_id != -1;
 }
 
 int Entity::countImpl(DataStore *store, const QString &tableName, const QString &column, const QVariant &value)
 {
     QSqlDatabase db = store->database();
     if (!db.isOpen()) {
         return -1;
     }
 
     CountQueryBuilder builder(store, tableName);
     builder.addValueCondition(column, Query::Equals, value);
 
     if (!builder.exec()) {
         qCWarning(AKONADISERVER_LOG) << "Error counting records in table" << tableName;
         return -1;
     }
 
     return builder.result();
 }
 
 bool Entity::removeImpl(DataStore *store, const QString &tableName, const QString &column, const QVariant &value)
 {
     QSqlDatabase db = store->database();
     if (!db.isOpen()) {
         return false;
     }
 
     QueryBuilder builder(store, tableName, QueryBuilder::Delete);
     builder.addValueCondition(column, Query::Equals, value);
 
     if (!builder.exec()) {
         qCWarning(AKONADISERVER_LOG) << "Error during deleting records from table" << tableName;
         return false;
     }
     return true;
 }
 
 bool Entity::relatesToImpl(DataStore *store, const QString &tableName, const QString &leftColumn, const QString &rightColumn, qint64 leftId, qint64 rightId)
 {
     QSqlDatabase db = store->database();
     if (!db.isOpen()) {
         return false;
     }
 
     CountQueryBuilder builder(store, tableName);
     builder.addValueCondition(leftColumn, Query::Equals, leftId);
     builder.addValueCondition(rightColumn, Query::Equals, rightId);
 
     if (!builder.exec()) {
         qCWarning(AKONADISERVER_LOG) << "Error during counting records in table" << tableName;
         return false;
     }
 
     return builder.result() > 0;
 }
 
 bool Entity::addToRelationImpl(DataStore *store, const QString &tableName, const QString &leftColumn, const QString &rightColumn, qint64 leftId, qint64 rightId)
 {
     QSqlDatabase db = store->database();
     if (!db.isOpen()) {
         return false;
     }
 
     QueryBuilder qb(store, tableName, QueryBuilder::Insert);
     qb.setColumnValue(leftColumn, leftId);
     qb.setColumnValue(rightColumn, rightId);
     qb.setIdentificationColumn(QString());
 
     if (!qb.exec()) {
         qCWarning(AKONADISERVER_LOG) << "Error during adding a record to table" << tableName;
         return false;
     }
 
     return true;
 }
 
 bool Entity::removeFromRelationImpl(DataStore *store,
                                     const QString &tableName,
                                     const QString &leftColumn,
                                     const QString &rightColumn,
                                     qint64 leftId,
                                     qint64 rightId)
 {
     QSqlDatabase db = store->database();
     if (!db.isOpen()) {
         return false;
     }
 
     QueryBuilder builder(store, tableName, QueryBuilder::Delete);
     builder.addValueCondition(leftColumn, Query::Equals, leftId);
     builder.addValueCondition(rightColumn, Query::Equals, rightId);
 
     if (!builder.exec()) {
         qCWarning(AKONADISERVER_LOG) << "Error during removing a record from relation table" << tableName;
         return false;
     }
 
     return true;
 }
 
 bool Entity::clearRelationImpl(DataStore *store, const QString &tableName, const QString &leftColumn, const QString &rightColumn, qint64 id, RelationSide side)
 {
     QSqlDatabase db = store->database();
     if (!db.isOpen()) {
         return false;
     }
 
     QueryBuilder builder(store, tableName, QueryBuilder::Delete);
     switch (side) {
     case Left:
         builder.addValueCondition(leftColumn, Query::Equals, id);
         break;
     case Right:
         builder.addValueCondition(rightColumn, Query::Equals, id);
         break;
     default:
         qFatal("Invalid enum value");
     }
     if (!builder.exec()) {
         qCWarning(AKONADISERVER_LOG) << "Error during clearing relation table" << tableName << "for ID" << id;
         return false;
     }
 
     return true;
 }