X-Git-Url: https://git.quassel-irc.org/?p=quassel.git;a=blobdiff_plain;f=src%2Fcore%2Fsqlitestorage.cpp;h=cbcacf7149959875a2f529b9ef3b22bb50d44376;hp=613fa32df2c8a6deca9920af1feda166e99aff7d;hb=674ad8ee4369cbc9516eb6d6629a616e9310c71f;hpb=4fef2ef49965b4e89a25e8cdc33100d9717d1aa8 diff --git a/src/core/sqlitestorage.cpp b/src/core/sqlitestorage.cpp index 613fa32d..cbcacf71 100644 --- a/src/core/sqlitestorage.cpp +++ b/src/core/sqlitestorage.cpp @@ -27,6 +27,8 @@ #include "util.h" #include "logger.h" +int SqliteStorage::_maxRetryCount = 150; // yes this is a large number... only other way to "handle" this is bailing out... + SqliteStorage::SqliteStorage(QObject *parent) : AbstractSqlStorage(parent) { @@ -68,14 +70,14 @@ UserId SqliteStorage::addUser(const QString &user, const QString &password) { query.prepare(queryString("insert_quasseluser")); query.bindValue(":username", user); query.bindValue(":password", cryptedPassword(password)); - query.exec(); + safeExec(query); if(query.lastError().isValid() && query.lastError().number() == 19) { // user already exists - sadly 19 seems to be the general constraint violation error... return 0; } query.prepare(queryString("select_userid")); query.bindValue(":username", user); - query.exec(); + safeExec(query); query.first(); UserId uid = query.value(0).toInt(); emit userAdded(uid, user); @@ -87,7 +89,7 @@ void SqliteStorage::updateUser(UserId user, const QString &password) { query.prepare(queryString("update_userpassword")); query.bindValue(":userid", user.toInt()); query.bindValue(":password", cryptedPassword(password)); - query.exec(); + safeExec(query); } void SqliteStorage::renameUser(UserId user, const QString &newName) { @@ -95,7 +97,7 @@ void SqliteStorage::renameUser(UserId user, const QString &newName) { query.prepare(queryString("update_username")); query.bindValue(":userid", user.toInt()); query.bindValue(":username", newName); - query.exec(); + safeExec(query); emit userRenamed(user, newName); } @@ -104,7 +106,7 @@ UserId SqliteStorage::validateUser(const QString &user, const QString &password) query.prepare(queryString("select_authuser")); query.bindValue(":username", user); query.bindValue(":password", cryptedPassword(password)); - query.exec(); + safeExec(query); if(query.first()) { return query.value(0).toInt(); @@ -116,7 +118,7 @@ UserId SqliteStorage::validateUser(const QString &user, const QString &password) UserId SqliteStorage::internalUser() { QSqlQuery query(logDb()); query.prepare(queryString("select_internaluser")); - query.exec(); + safeExec(query); if(query.first()) { return query.value(0).toInt(); @@ -129,19 +131,19 @@ void SqliteStorage::delUser(UserId user) { QSqlQuery query(logDb()); query.prepare(queryString("delete_backlog_by_uid")); query.bindValue(":userid", user.toInt()); - query.exec(); - + safeExec(query); + query.prepare(queryString("delete_buffers_by_uid")); query.bindValue(":userid", user.toInt()); - query.exec(); - + safeExec(query); + query.prepare(queryString("delete_networks_by_uid")); query.bindValue(":userid", user.toInt()); - query.exec(); - + safeExec(query); + query.prepare(queryString("delete_quasseluser")); query.bindValue(":userid", user.toInt()); - query.exec(); + safeExec(query); // I hate the lack of foreign keys and on delete cascade... :( emit userRemoved(user); } @@ -157,7 +159,7 @@ void SqliteStorage::setUserSetting(UserId userId, const QString &settingName, co query.bindValue(":userid", userId.toInt()); query.bindValue(":settingname", settingName); query.bindValue(":settingvalue", rawData); - query.exec(); + safeExec(query); if(query.lastError().isValid()) { QSqlQuery updateQuery(logDb()); @@ -165,9 +167,9 @@ void SqliteStorage::setUserSetting(UserId userId, const QString &settingName, co updateQuery.bindValue(":userid", userId.toInt()); updateQuery.bindValue(":settingname", settingName); updateQuery.bindValue(":settingvalue", rawData); - updateQuery.exec(); + safeExec(updateQuery); } - + } QVariant SqliteStorage::getUserSetting(UserId userId, const QString &settingName, const QVariant &defaultData) { @@ -175,7 +177,7 @@ QVariant SqliteStorage::getUserSetting(UserId userId, const QString &settingName query.prepare(queryString("select_user_setting")); query.bindValue(":userid", userId.toInt()); query.bindValue(":settingname", settingName); - query.exec(); + safeExec(query); if(query.first()) { QVariant data; @@ -189,14 +191,198 @@ QVariant SqliteStorage::getUserSetting(UserId userId, const QString &settingName } } +IdentityId SqliteStorage::createIdentity(UserId user, CoreIdentity &identity) { + IdentityId identityId; + + QSqlQuery query(logDb()); + query.prepare(queryString("insert_identity")); + query.bindValue(":userid", user.toInt()); + query.bindValue(":identityname", identity.identityName()); + query.bindValue(":realname", identity.realName()); + query.bindValue(":awaynick", identity.awayNick()); + query.bindValue(":awaynickenabled", identity.awayNickEnabled() ? 1 : 0); + query.bindValue(":awayreason", identity.awayReason()); + query.bindValue(":awayreasonenabled", identity.awayReasonEnabled() ? 1 : 0); + query.bindValue(":autoawayenabled", identity.awayReasonEnabled() ? 1 : 0); + query.bindValue(":autoawaytime", identity.autoAwayTime()); + query.bindValue(":autoawayreason", identity.autoAwayReason()); + query.bindValue(":autoawayreasonenabled", identity.autoAwayReasonEnabled() ? 1 : 0); + query.bindValue(":detachawayenabled", identity.detachAwayEnabled() ? 1 : 0); + query.bindValue(":detachawayreason", identity.detachAwayReason()); + query.bindValue(":detachawayreasonenabled", identity.detachAwayReasonEnabled() ? 1 : 0); + query.bindValue(":ident", identity.ident()); + query.bindValue(":kickreason", identity.kickReason()); + query.bindValue(":partreason", identity.partReason()); + query.bindValue(":quitreason", identity.quitReason()); +#ifdef HAVE_SSL + query.bindValue(":sslcert", identity.sslCert().toPem()); + query.bindValue(":sslkey", identity.sslKey().toPem()); +#else + query.bindValue(":sslcert", QByteArray()); + query.bindValue(":sslkey", QByteArray()); +#endif + + safeExec(query); + + identityId = query.lastInsertId().toInt(); + if(!identityId.isValid()) { + watchQuery(query); + } else { + QSqlQuery deleteNickQuery(logDb()); + deleteNickQuery.prepare(queryString("delete_nicks")); + deleteNickQuery.bindValue(":identityid", identityId.toInt()); + safeExec(deleteNickQuery); + + QSqlQuery insertNickQuery(logDb()); + insertNickQuery.prepare(queryString("insert_nick")); + foreach(QString nick, identity.nicks()) { + insertNickQuery.bindValue(":identityid", identityId.toInt()); + insertNickQuery.bindValue(":nick", nick); + safeExec(insertNickQuery); + } + } + identity.setId(identityId); + return identityId; +} + +bool SqliteStorage::updateIdentity(UserId user, const CoreIdentity &identity) { + QSqlQuery checkQuery(logDb()); + checkQuery.prepare(queryString("select_checkidentity")); + checkQuery.bindValue(":identityid", identity.id().toInt()); + checkQuery.bindValue(":userid", user.toInt()); + safeExec(checkQuery); + + if(!checkQuery.first() || checkQuery.value(0).toInt() != 1) // there should be exactly one identity for the given id and user + return false; + + QSqlQuery query(logDb()); + query.prepare(queryString("update_identity")); + query.bindValue(":identityname", identity.identityName()); + query.bindValue(":realname", identity.realName()); + query.bindValue(":awaynick", identity.awayNick()); + query.bindValue(":awaynickenabled", identity.awayNickEnabled() ? 1 : 0); + query.bindValue(":awayreason", identity.awayReason()); + query.bindValue(":awayreasonenabled", identity.awayReasonEnabled() ? 1 : 0); + query.bindValue(":autoawayenabled", identity.awayReasonEnabled() ? 1 : 0); + query.bindValue(":autoawaytime", identity.autoAwayTime()); + query.bindValue(":autoawayreason", identity.autoAwayReason()); + query.bindValue(":autoawayreasonenabled", identity.autoAwayReasonEnabled() ? 1 : 0); + query.bindValue(":detachawayenabled", identity.detachAwayEnabled() ? 1 : 0); + query.bindValue(":detachawayreason", identity.detachAwayReason()); + query.bindValue(":detachawayreasonenabled", identity.detachAwayReasonEnabled() ? 1 : 0); + query.bindValue(":ident", identity.ident()); + query.bindValue(":kickreason", identity.kickReason()); + query.bindValue(":partreason", identity.partReason()); + query.bindValue(":quitreason", identity.quitReason()); +#ifdef HAVE_SSL + query.bindValue(":sslcert", identity.sslCert().toPem()); + query.bindValue(":sslkey", identity.sslKey().toPem()); +#else + query.bindValue(":sslcert", QByteArray()); + query.bindValue(":sslkey", QByteArray()); +#endif + query.bindValue(":identityid", identity.id().toInt()); + safeExec(query); + watchQuery(query); + + QSqlQuery deleteNickQuery(logDb()); + deleteNickQuery.prepare(queryString("delete_nicks")); + deleteNickQuery.bindValue(":identityid", identity.id().toInt()); + safeExec(deleteNickQuery); + watchQuery(deleteNickQuery); + + QSqlQuery insertNickQuery(logDb()); + insertNickQuery.prepare(queryString("insert_nick")); + foreach(QString nick, identity.nicks()) { + insertNickQuery.bindValue(":identityid", identity.id().toInt()); + insertNickQuery.bindValue(":nick", nick); + safeExec(insertNickQuery); + watchQuery(insertNickQuery); + } + + return true; +} + +void SqliteStorage::removeIdentity(UserId user, IdentityId identityId) { + QSqlQuery checkQuery(logDb()); + checkQuery.prepare(queryString("select_checkidentity")); + checkQuery.bindValue(":identityid", identityId.toInt()); + checkQuery.bindValue(":userid", user.toInt()); + safeExec(checkQuery); + + if(!checkQuery.first() || checkQuery.value(0).toInt() != 1) // there should be exactly one identity for the given id and user + return; + + QSqlQuery deleteNickQuery(logDb()); + deleteNickQuery.prepare(queryString("delete_nicks")); + deleteNickQuery.bindValue(":identityid", identityId.toInt()); + safeExec(deleteNickQuery); + + QSqlQuery deleteIdentityQuery(logDb()); + deleteIdentityQuery.prepare(queryString("delete_identity")); + deleteIdentityQuery.bindValue(":identityid", identityId.toInt()); + deleteIdentityQuery.bindValue(":userid", user.toInt()); + safeExec(deleteIdentityQuery); +} + +QList SqliteStorage::identities(UserId user) { + QList identities; + + QSqlQuery query(logDb()); + query.prepare(queryString("select_identities")); + query.bindValue(":userid", user.toInt()); + + QSqlQuery nickQuery(logDb()); + nickQuery.prepare(queryString("select_nicks")); + + safeExec(query); + + while(query.next()) { + CoreIdentity identity(IdentityId(query.value(0).toInt())); + + identity.setIdentityName(query.value(1).toString()); + identity.setRealName(query.value(2).toString()); + identity.setAwayNick(query.value(3).toString()); + identity.setAwayNickEnabled(!!query.value(4).toInt()); + identity.setAwayReason(query.value(5).toString()); + identity.setAwayReasonEnabled(!!query.value(6).toInt()); + identity.setAutoAwayEnabled(!!query.value(7).toInt()); + identity.setAutoAwayTime(query.value(8).toInt()); + identity.setAutoAwayReason(query.value(9).toString()); + identity.setAutoAwayReasonEnabled(!!query.value(10).toInt()); + identity.setDetachAwayEnabled(!!query.value(11).toInt()); + identity.setDetachAwayReason(query.value(12).toString()); + identity.setDetachAwayReasonEnabled(!!query.value(13).toInt()); + identity.setIdent(query.value(14).toString()); + identity.setKickReason(query.value(15).toString()); + identity.setPartReason(query.value(16).toString()); + identity.setQuitReason(query.value(17).toString()); +#ifdef HAVE_SSL + identity.setSslCert(query.value(18).toByteArray()); + identity.setSslKey(query.value(19).toByteArray()); +#endif + + nickQuery.bindValue(":identityid", identity.id().toInt()); + QList nicks; + safeExec(nickQuery); + watchQuery(nickQuery); + while(nickQuery.next()) { + nicks << nickQuery.value(0).toString(); + } + identity.setNicks(nicks); + identities << identity; + } + return identities; +} + NetworkId SqliteStorage::createNetwork(UserId user, const NetworkInfo &info) { NetworkId networkId; QSqlQuery query(logDb()); query.prepare(queryString("insert_network")); query.bindValue(":userid", user.toInt()); query.bindValue(":networkname", info.networkName); - query.exec(); - + safeExec(query); + networkId = getNetworkId(user, info.networkName); if(!networkId.isValid()) { watchQuery(query); @@ -211,7 +397,7 @@ NetworkId SqliteStorage::createNetwork(UserId user, const NetworkInfo &info) { bool SqliteStorage::updateNetwork(UserId user, const NetworkInfo &info) { if(!isValidNetwork(user, info.networkId)) return false; - + QSqlQuery updateQuery(logDb()); updateQuery.prepare(queryString("update_network")); updateQuery.bindValue(":networkname", info.networkName); @@ -231,33 +417,39 @@ bool SqliteStorage::updateNetwork(UserId user, const NetworkInfo &info) { updateQuery.bindValue(":unlimitedconnectretries", info.unlimitedReconnectRetries ? 1 : 0); updateQuery.bindValue(":rejoinchannels", info.rejoinChannels ? 1 : 0); updateQuery.bindValue(":networkid", info.networkId.toInt()); - updateQuery.exec(); + safeExec(updateQuery); if(!watchQuery(updateQuery)) return false; QSqlQuery dropServersQuery(logDb()); dropServersQuery.prepare("DELETE FROM ircserver WHERE networkid = :networkid"); dropServersQuery.bindValue(":networkid", info.networkId.toInt()); - dropServersQuery.exec(); + safeExec(dropServersQuery); if(!watchQuery(dropServersQuery)) return false; QSqlQuery insertServersQuery(logDb()); insertServersQuery.prepare(queryString("insert_server")); - foreach(QVariant server_, info.serverList) { - QVariantMap server = server_.toMap(); - insertServersQuery.bindValue(":hostname", server["Host"]); - insertServersQuery.bindValue(":port", server["Port"].toInt()); - insertServersQuery.bindValue(":password", server["Password"]); - insertServersQuery.bindValue(":ssl", server["UseSSL"].toBool() ? 1 : 0); + foreach(Network::Server server, info.serverList) { insertServersQuery.bindValue(":userid", user.toInt()); insertServersQuery.bindValue(":networkid", info.networkId.toInt()); - - insertServersQuery.exec(); + insertServersQuery.bindValue(":hostname", server.host); + insertServersQuery.bindValue(":port", server.port); + insertServersQuery.bindValue(":password", server.password); + insertServersQuery.bindValue(":ssl", server.useSsl ? 1 : 0); + insertServersQuery.bindValue(":sslversion", server.sslVersion); + insertServersQuery.bindValue(":useproxy", server.useProxy ? 1 : 0); + insertServersQuery.bindValue(":proxytype", server.proxyType); + insertServersQuery.bindValue(":proxyhost", server.proxyHost); + insertServersQuery.bindValue(":proxyport", server.proxyPort); + insertServersQuery.bindValue(":proxyuser", server.proxyUser); + insertServersQuery.bindValue(":proxypass", server.proxyPass); + + safeExec(insertServersQuery); if(!watchQuery(insertServersQuery)) return false; } - + return true; } @@ -273,41 +465,41 @@ bool SqliteStorage::removeNetwork(UserId user, const NetworkId &networkId) { withTransaction = false; } } - + QSqlQuery deleteBacklogQuery(logDb()); deleteBacklogQuery.prepare(queryString("delete_backlog_for_network")); deleteBacklogQuery.bindValue(":networkid", networkId.toInt()); - deleteBacklogQuery.exec(); + safeExec(deleteBacklogQuery); if(!watchQuery(deleteBacklogQuery)) { if(withTransaction) logDb().rollback(); return false; } - + QSqlQuery deleteBuffersQuery(logDb()); deleteBuffersQuery.prepare(queryString("delete_buffers_for_network")); deleteBuffersQuery.bindValue(":networkid", networkId.toInt()); - deleteBuffersQuery.exec(); + safeExec(deleteBuffersQuery); if(!watchQuery(deleteBuffersQuery)) { if(withTransaction) logDb().rollback(); return false; } - + QSqlQuery deleteServersQuery(logDb()); deleteServersQuery.prepare(queryString("delete_ircservers_for_network")); deleteServersQuery.bindValue(":networkid", networkId.toInt()); - deleteServersQuery.exec(); + safeExec(deleteServersQuery); if(!watchQuery(deleteServersQuery)) { if(withTransaction) logDb().rollback(); return false; } - + QSqlQuery deleteNetworkQuery(logDb()); deleteNetworkQuery.prepare(queryString("delete_network")); deleteNetworkQuery.bindValue(":networkid", networkId.toInt()); - deleteNetworkQuery.exec(); + safeExec(deleteNetworkQuery); if(!watchQuery(deleteNetworkQuery)) { if(withTransaction) logDb().rollback(); @@ -324,11 +516,11 @@ QList SqliteStorage::networks(UserId user) { QSqlQuery networksQuery(logDb()); networksQuery.prepare(queryString("select_networks_for_user")); networksQuery.bindValue(":userid", user.toInt()); - + QSqlQuery serversQuery(logDb()); serversQuery.prepare(queryString("select_servers_for_network")); - networksQuery.exec(); + safeExec(networksQuery); if(!watchQuery(networksQuery)) return nets; @@ -352,17 +544,24 @@ QList SqliteStorage::networks(UserId user) { net.rejoinChannels = networksQuery.value(15).toInt() == 1 ? true : false; serversQuery.bindValue(":networkid", net.networkId.toInt()); - serversQuery.exec(); + safeExec(serversQuery); if(!watchQuery(serversQuery)) return nets; - QVariantList servers; + Network::ServerList servers; while(serversQuery.next()) { - QVariantMap server; - server["Host"] = serversQuery.value(0).toString(); - server["Port"] = serversQuery.value(1).toInt(); - server["Password"] = serversQuery.value(2).toString(); - server["UseSSL"] = serversQuery.value(3).toInt() == 1 ? true : false; + Network::Server server; + server.host = serversQuery.value(0).toString(); + server.port = serversQuery.value(1).toUInt(); + server.password = serversQuery.value(2).toString(); + server.useSsl = serversQuery.value(3).toInt() == 1 ? true : false; + server.sslVersion = serversQuery.value(4).toInt(); + server.useProxy = serversQuery.value(5).toInt() == 1 ? true : false; + server.proxyType = serversQuery.value(6).toInt(); + server.proxyHost = serversQuery.value(7).toString(); + server.proxyPort = serversQuery.value(8).toUInt(); + server.proxyUser = serversQuery.value(9).toString(); + server.proxyPass = serversQuery.value(10).toString(); servers << server; } net.serverList = servers; @@ -376,12 +575,12 @@ bool SqliteStorage::isValidNetwork(UserId user, const NetworkId &networkId) { query.prepare(queryString("select_networkExists")); query.bindValue(":userid", user.toInt()); query.bindValue(":networkid", networkId.toInt()); - query.exec(); + safeExec(query); watchQuery(query); // there should not occur any errors if(!query.first()) return false; - + Q_ASSERT(!query.next()); return true; } @@ -391,7 +590,7 @@ bool SqliteStorage::isValidBuffer(const UserId &user, const BufferId &bufferId) query.prepare(queryString("select_bufferExists")); query.bindValue(":userid", user.toInt()); query.bindValue(":bufferid", bufferId.toInt()); - query.exec(); + safeExec(query); watchQuery(query); if(!query.first()) @@ -407,8 +606,8 @@ NetworkId SqliteStorage::getNetworkId(UserId user, const QString &network) { "WHERE userid = :userid AND networkname = :networkname"); query.bindValue(":userid", user.toInt()); query.bindValue(":networkname", network); - query.exec(); - + safeExec(query); + if(query.first()) return query.value(0).toInt(); else @@ -420,13 +619,13 @@ QList SqliteStorage::connectedNetworks(UserId user) { QSqlQuery query(logDb()); query.prepare(queryString("select_connected_networks")); query.bindValue(":userid", user.toInt()); - query.exec(); + safeExec(query); watchQuery(query); while(query.next()) { connectedNets << query.value(0).toInt(); } - + return connectedNets; } @@ -436,7 +635,7 @@ void SqliteStorage::setNetworkConnected(UserId user, const NetworkId &networkId, query.bindValue(":userid", user.toInt()); query.bindValue(":networkid", networkId.toInt()); query.bindValue(":connected", isConnected ? 1 : 0); - query.exec(); + safeExec(query); watchQuery(query); } @@ -446,13 +645,13 @@ QHash SqliteStorage::persistentChannels(UserId user, const Net query.prepare(queryString("select_persistent_channels")); query.bindValue(":userid", user.toInt()); query.bindValue(":networkid", networkId.toInt()); - query.exec(); + safeExec(query); watchQuery(query); while(query.next()) { persistentChans[query.value(0).toString()] = query.value(1).toString(); } - + return persistentChans; } @@ -463,7 +662,7 @@ void SqliteStorage::setChannelPersistent(UserId user, const NetworkId &networkId query.bindValue(":networkId", networkId.toInt()); query.bindValue(":buffercname", channel.toLower()); query.bindValue(":joined", isJoined ? 1 : 0); - query.exec(); + safeExec(query); watchQuery(query); } @@ -474,33 +673,38 @@ void SqliteStorage::setPersistentChannelKey(UserId user, const NetworkId &networ query.bindValue(":networkId", networkId.toInt()); query.bindValue(":buffercname", channel.toLower()); query.bindValue(":key", key); - query.exec(); + safeExec(query); watchQuery(query); } void SqliteStorage::createBuffer(UserId user, const NetworkId &networkId, BufferInfo::Type type, const QString &buffer) { - QSqlQuery &query = cachedQuery("insert_buffer"); + QSqlQuery query(logDb()); + query.prepare(queryString("insert_buffer")); query.bindValue(":userid", user.toInt()); query.bindValue(":networkid", networkId.toInt()); query.bindValue(":buffertype", (int)type); query.bindValue(":buffername", buffer); query.bindValue(":buffercname", buffer.toLower()); - query.exec(); + safeExec(query); watchQuery(query); } -BufferInfo SqliteStorage::getBufferInfo(UserId user, const NetworkId &networkId, BufferInfo::Type type, const QString &buffer) { - QSqlQuery &query = cachedQuery("select_bufferByName"); +BufferInfo SqliteStorage::bufferInfo(UserId user, const NetworkId &networkId, BufferInfo::Type type, const QString &buffer, bool create) { + QSqlQuery query(logDb()); + query.prepare(queryString("select_bufferByName")); query.bindValue(":networkid", networkId.toInt()); query.bindValue(":userid", user.toInt()); query.bindValue(":buffercname", buffer.toLower()); - query.exec(); + safeExec(query); if(!query.first()) { + if(!create) + return BufferInfo(); + createBuffer(user, networkId, type, buffer); - query.exec(); + safeExec(query); if(!query.first()) { watchQuery(query); qWarning() << "unable to create BufferInfo for:" << user << networkId << buffer; @@ -527,7 +731,7 @@ BufferInfo SqliteStorage::getBufferInfo(UserId user, const BufferId &bufferId) { query.prepare(queryString("select_buffer_by_id")); query.bindValue(":userid", user.toInt()); query.bindValue(":bufferid", bufferId.toInt()); - query.exec(); + safeExec(query); if(!watchQuery(query)) return BufferInfo(); @@ -545,8 +749,8 @@ QList SqliteStorage::requestBuffers(UserId user) { QSqlQuery query(logDb()); query.prepare(queryString("select_buffers")); query.bindValue(":userid", user.toInt()); - - query.exec(); + + safeExec(query); watchQuery(query); while(query.next()) { bufferlist << BufferInfo(query.value(0).toInt(), query.value(1).toInt(), (BufferInfo::Type)query.value(2).toInt(), query.value(3).toInt(), query.value(4).toString()); @@ -561,7 +765,7 @@ QList SqliteStorage::requestBufferIdsForNetwork(UserId user, NetworkId query.bindValue(":networkid", networkId.toInt()); query.bindValue(":userid", user.toInt()); - query.exec(); + safeExec(query); watchQuery(query); while(query.next()) { bufferList << BufferId(query.value(0).toInt()); @@ -576,67 +780,69 @@ bool SqliteStorage::removeBuffer(const UserId &user, const BufferId &bufferId) { QSqlQuery delBacklogQuery(logDb()); delBacklogQuery.prepare(queryString("delete_backlog_for_buffer")); delBacklogQuery.bindValue(":bufferid", bufferId.toInt()); - delBacklogQuery.exec(); + safeExec(delBacklogQuery); if(!watchQuery(delBacklogQuery)) return false; QSqlQuery delBufferQuery(logDb()); delBufferQuery.prepare(queryString("delete_buffer_for_bufferid")); delBufferQuery.bindValue(":bufferid", bufferId.toInt()); - delBufferQuery.exec(); + safeExec(delBufferQuery); if(!watchQuery(delBufferQuery)) return false; return true; } -BufferId SqliteStorage::renameBuffer(const UserId &user, const NetworkId &networkId, const QString &newName, const QString &oldName) { - // check if such a buffer exists... - QSqlQuery existsQuery(logDb()); - existsQuery.prepare(queryString("select_bufferByName")); - existsQuery.bindValue(":networkid", networkId.toInt()); - existsQuery.bindValue(":userid", user.toInt()); - existsQuery.bindValue(":buffercname", oldName.toLower()); - existsQuery.exec(); - if(!watchQuery(existsQuery)) +bool SqliteStorage::renameBuffer(const UserId &user, const BufferId &bufferId, const QString &newName) { + if(!isValidBuffer(user, bufferId)) return false; - if(!existsQuery.first()) + QSqlQuery query(logDb()); + query.prepare(queryString("update_buffer_name")); + query.bindValue(":buffername", newName); + query.bindValue(":buffercname", newName.toLower()); + query.bindValue(":bufferid", bufferId.toInt()); + safeExec(query); + if(query.lastError().isValid()) { + // unexepcted error occured (19 == constraint violation) + if(query.lastError().number() != 19) + watchQuery(query); return false; + } - const int bufferid = existsQuery.value(0).toInt(); - - Q_ASSERT(!existsQuery.next()); + return true; +} - // ... and if the new name is still free. - existsQuery.bindValue(":networkid", networkId.toInt()); - existsQuery.bindValue(":userid", user.toInt()); - existsQuery.bindValue(":buffercname", newName.toLower()); - existsQuery.exec(); - if(!watchQuery(existsQuery)) +bool SqliteStorage::mergeBuffersPermanently(const UserId &user, const BufferId &bufferId1, const BufferId &bufferId2) { + if(!isValidBuffer(user, bufferId1) || !isValidBuffer(user, bufferId2)) return false; - if(existsQuery.first()) + QSqlQuery query(logDb()); + query.prepare(queryString("update_backlog_bufferid")); + query.bindValue(":oldbufferid", bufferId2.toInt()); + query.bindValue(":newbufferid", bufferId1.toInt()); + safeExec(query); + if(!watchQuery(query)) return false; - QSqlQuery renameBufferQuery(logDb()); - renameBufferQuery.prepare(queryString("update_buffer_name")); - renameBufferQuery.bindValue(":buffername", newName); - renameBufferQuery.bindValue(":buffercname", newName.toLower()); - renameBufferQuery.bindValue(":bufferid", bufferid); - renameBufferQuery.exec(); - if(watchQuery(existsQuery)) - return BufferId(bufferid); - else - return BufferId(); + QSqlQuery delBufferQuery(logDb()); + delBufferQuery.prepare(queryString("delete_buffer_for_bufferid")); + delBufferQuery.bindValue(":bufferid", bufferId2.toInt()); + safeExec(delBufferQuery); + watchQuery(delBufferQuery); + + return true; } void SqliteStorage::setBufferLastSeenMsg(UserId user, const BufferId &bufferId, const MsgId &msgId) { - QSqlQuery &query = cachedQuery("update_buffer_lastseen"); + QSqlQuery query(logDb()); + query.prepare(queryString("update_buffer_lastseen")); + query.bindValue(":userid", user.toInt()); query.bindValue(":bufferid", bufferId.toInt()); query.bindValue(":lastseenmsgid", msgId.toInt()); - query.exec(); + safeExec(query); watchQuery(query); } @@ -645,7 +851,7 @@ QHash SqliteStorage::bufferLastSeenMsgIds(UserId user) { QSqlQuery query(logDb()); query.prepare(queryString("select_buffer_lastseen_messages")); query.bindValue(":userid", user.toInt()); - query.exec(); + safeExec(query); if(!watchQuery(query)) return lastSeenHash; @@ -656,23 +862,25 @@ QHash SqliteStorage::bufferLastSeenMsgIds(UserId user) { } MsgId SqliteStorage::logMessage(Message msg) { - QSqlQuery &logMessageQuery = cachedQuery("insert_message"); + QSqlQuery logMessageQuery(logDb()); + logMessageQuery.prepare(queryString("insert_message")); + logMessageQuery.bindValue(":time", msg.timestamp().toTime_t()); logMessageQuery.bindValue(":bufferid", msg.bufferInfo().bufferId().toInt()); logMessageQuery.bindValue(":type", msg.type()); logMessageQuery.bindValue(":flags", (int)msg.flags()); logMessageQuery.bindValue(":sender", msg.sender()); logMessageQuery.bindValue(":message", msg.contents()); - logMessageQuery.exec(); - + safeExec(logMessageQuery); + if(logMessageQuery.lastError().isValid()) { // constraint violation - must be NOT NULL constraint - probably the sender is missing... if(logMessageQuery.lastError().number() == 19) { - QSqlQuery &addSenderQuery = cachedQuery("insert_sender"); + QSqlQuery addSenderQuery(logDb()); + addSenderQuery.prepare(queryString("insert_sender")); addSenderQuery.bindValue(":sender", msg.sender()); - addSenderQuery.exec(); - watchQuery(addSenderQuery); - logMessageQuery.exec(); + safeExec(addSenderQuery); + safeExec(logMessageQuery); if(!watchQuery(logMessageQuery)) return 0; } else { @@ -685,110 +893,71 @@ MsgId SqliteStorage::logMessage(Message msg) { return msgId; } -QList SqliteStorage::requestMsgs(UserId user, BufferId bufferId, int lastmsgs, int offset) { +QList SqliteStorage::requestMsgs(UserId user, BufferId bufferId, MsgId first, MsgId last, int limit) { QList messagelist; BufferInfo bufferInfo = getBufferInfo(user, bufferId); if(!bufferInfo.isValid()) return messagelist; - if(offset == -1) { - offset = 0; + QSqlQuery query(logDb()); + if(last == -1) { + query.prepare(queryString("select_messagesNew")); } else { - // we have to determine the real offset first - QSqlQuery &offsetQuery = cachedQuery("select_messagesOffset"); - offsetQuery.bindValue(":bufferid", bufferId.toInt()); - offsetQuery.bindValue(":messageid", offset); - offsetQuery.exec(); - offsetQuery.first(); - offset = offsetQuery.value(0).toInt(); + query.prepare(queryString("select_messages")); + query.bindValue(":lastmsg", last.toInt()); } - // now let's select the messages - QSqlQuery &msgQuery = cachedQuery("select_messages"); - msgQuery.bindValue(":bufferid", bufferId.toInt()); - msgQuery.bindValue(":limit", lastmsgs); - msgQuery.bindValue(":offset", offset); - msgQuery.exec(); - - watchQuery(msgQuery); - - while(msgQuery.next()) { - Message msg(QDateTime::fromTime_t(msgQuery.value(1).toInt()), - bufferInfo, - (Message::Type)msgQuery.value(2).toUInt(), - msgQuery.value(5).toString(), - msgQuery.value(4).toString(), - (Message::Flags)msgQuery.value(3).toUInt()); - msg.setMsgId(msgQuery.value(0).toInt()); - messagelist << msg; - } - return messagelist; -} - - -QList SqliteStorage::requestMsgs(UserId user, BufferId bufferId, QDateTime since, int offset) { - QList messagelist; + query.bindValue(":bufferid", bufferId.toInt()); + query.bindValue(":firstmsg", first.toInt()); + query.bindValue(":limit", limit); + safeExec(query); - BufferInfo bufferInfo = getBufferInfo(user, bufferId); - if(!bufferInfo.isValid()) - return messagelist; + watchQuery(query); - // we have to determine the real offset first - QSqlQuery &offsetQuery = cachedQuery("select_messagesSinceOffset"); - offsetQuery.bindValue(":bufferid", bufferId.toInt()); - offsetQuery.bindValue(":since", since.toTime_t()); - offsetQuery.exec(); - offsetQuery.first(); - offset = offsetQuery.value(0).toInt(); - - // now let's select the messages - QSqlQuery &msgQuery = cachedQuery("select_messagesSince"); - msgQuery.bindValue(":bufferid", bufferId.toInt()); - msgQuery.bindValue(":since", since.toTime_t()); - msgQuery.bindValue(":offset", offset); - msgQuery.exec(); - - watchQuery(msgQuery); - - while(msgQuery.next()) { - Message msg(QDateTime::fromTime_t(msgQuery.value(1).toInt()), + while(query.next()) { + Message msg(QDateTime::fromTime_t(query.value(1).toInt()), bufferInfo, - (Message::Type)msgQuery.value(2).toUInt(), - msgQuery.value(5).toString(), - msgQuery.value(4).toString(), - (Message::Flags)msgQuery.value(3).toUInt()); - msg.setMsgId(msgQuery.value(0).toInt()); + (Message::Type)query.value(2).toUInt(), + query.value(5).toString(), + query.value(4).toString(), + (Message::Flags)query.value(3).toUInt()); + msg.setMsgId(query.value(0).toInt()); messagelist << msg; } - return messagelist; } - -QList SqliteStorage::requestMsgRange(UserId user, BufferId bufferId, int first, int last) { +QList SqliteStorage::requestAllMsgs(UserId user, MsgId first, MsgId last, int limit) { QList messagelist; - BufferInfo bufferInfo = getBufferInfo(user, bufferId); - if(!bufferInfo.isValid()) - return messagelist; + QHash bufferInfoHash; + foreach(BufferInfo bufferInfo, requestBuffers(user)) { + bufferInfoHash[bufferInfo.bufferId()] = bufferInfo; + } - QSqlQuery &rangeQuery = cachedQuery("select_messageRange"); - rangeQuery.bindValue(":bufferid", bufferId.toInt()); - rangeQuery.bindValue(":firstmsg", first); - rangeQuery.bindValue(":lastmsg", last); - rangeQuery.exec(); + QSqlQuery query(logDb()); + if(last == -1) { + query.prepare(queryString("select_messagesAllNew")); + } else { + query.prepare(queryString("select_messagesAll")); + query.bindValue(":lastmsg", last.toInt()); + } + query.bindValue(":userid", user.toInt()); + query.bindValue(":firstmsg", first.toInt()); + query.bindValue(":limit", limit); + safeExec(query); - watchQuery(rangeQuery); - - while(rangeQuery.next()) { - Message msg(QDateTime::fromTime_t(rangeQuery.value(1).toInt()), - bufferInfo, - (Message::Type)rangeQuery.value(2).toUInt(), - rangeQuery.value(5).toString(), - rangeQuery.value(4).toString(), - (Message::Flags)rangeQuery.value(3).toUInt()); - msg.setMsgId(rangeQuery.value(0).toInt()); + watchQuery(query); + + while(query.next()) { + Message msg(QDateTime::fromTime_t(query.value(2).toInt()), + bufferInfoHash[query.value(1).toInt()], + (Message::Type)query.value(3).toUInt(), + query.value(6).toString(), + query.value(5).toString(), + (Message::Flags)query.value(4).toUInt()); + msg.setMsgId(query.value(0).toInt()); messagelist << msg; } @@ -796,5 +965,21 @@ QList SqliteStorage::requestMsgRange(UserId user, BufferId bufferId, in } QString SqliteStorage::backlogFile() { - return quasselDir().absolutePath() + "/quassel-storage.sqlite"; + return quasselDir().absolutePath() + "/quassel-storage.sqlite"; +} + +bool SqliteStorage::safeExec(QSqlQuery &query, int retryCount) { + query.exec(); + + if(!query.lastError().isValid()) + return true; + + switch(query.lastError().number()) { + case 5: // SQLITE_BUSY 5 /* The database file is locked */ + case 6: // SQLITE_LOCKED 6 /* A table in the database is locked */ + if(retryCount < _maxRetryCount) + return safeExec(query, retryCount + 1); + default: + return false; + } }