diff --git a/src/backend/datasources/MQTTClient.cpp b/src/backend/datasources/MQTTClient.cpp index 8a8f47c43..4c1d67b24 100644 --- a/src/backend/datasources/MQTTClient.cpp +++ b/src/backend/datasources/MQTTClient.cpp @@ -1,1132 +1,1140 @@ #ifdef HAVE_MQTT #include "backend/datasources/MQTTClient.h" #include "backend/datasources/MQTTSubscriptions.h" #include "backend/datasources/MQTTTopic.h" #include "backend/datasources/filters/AsciiFilter.h" #include "backend/datasources/filters/FITSFilter.h" #include "backend/datasources/filters/BinaryFilter.h" #include "backend/core/Project.h" #include "kdefrontend/spreadsheet/PlotDataDialog.h" #include "commonfrontend/spreadsheet/SpreadsheetView.h" #include "kdefrontend/datasources/MQTTErrorWidget.h" #include #include #include #include #include #include #include #include #include #include #include /*! \class MQTTClient \brief Represents data stored in a file. Reading and writing is done with the help of appropriate I/O-filters. \ingroup datasources */ MQTTClient::MQTTClient(const QString& name) : Folder(name), m_paused(false), m_prepared(false), m_keepLastValues(false), m_filter(nullptr), m_updateTimer(new QTimer(this)), m_willTimer(new QTimer(this)), m_client(new QMqttClient(this)), m_mqttTest(false), m_mqttRetain(false), m_mqttUseWill(false), m_mqttUseID(false), m_loaded(false), m_sampleRate(1), m_keepNvalues(0), m_updateInterval(1000), m_disconnectForWill(false), m_mqttUseAuthentication(false), m_subscriptionsLoaded(0), - m_subscriptionCount(0), + m_subscriptionCountToLoad(0), m_mqttFirstConnectEstablished(false) { //initActions(); qDebug()<<"MQTTClient constructor"; connect(m_updateTimer, &QTimer::timeout, this, &MQTTClient::read); m_willStatistics.fill(false, 15); connect(m_client, &QMqttClient::connected, this, &MQTTClient::onMqttConnect); connect(m_willTimer, &QTimer::timeout, this, &MQTTClient::setWillForMqtt); connect(m_client, &QMqttClient::errorChanged, this, &MQTTClient::mqttErrorChanged); //connect(this, &MQTTClient::mqttAllArrived, this, &MQTTClient::onAllArrived ); } MQTTClient::~MQTTClient() { emit clientAboutToBeDeleted(m_client->hostname()); //stop reading before deleting the objects pauseReading(); qDebug()<<"destructor"; if (m_filter) delete m_filter; qDebug()<<"delete timers"; delete m_updateTimer; delete m_willTimer; qDebug()<<"disocnnect"; m_client->disconnectFromHost(); qDebug()<<"delete client"; delete m_client; } /*! * depending on the update type, periodically or on data changes, starts the timer or activates the file watchers, respectively. */ void MQTTClient::ready() { if (m_updateType == TimeInterval) m_updateTimer->start(m_updateInterval); } /*void MQTTClient::initActions() { m_reloadAction = new QAction(QIcon::fromTheme("view-refresh"), i18n("Reload"), this); connect(m_reloadAction, &QAction::triggered, this, &MQTTClient::read); m_toggleLinkAction = new QAction(i18n("Link the file"), this); m_toggleLinkAction->setCheckable(true); connect(m_toggleLinkAction, &QAction::triggered, this, &MQTTClient::linkToggled); m_plotDataAction = new QAction(QIcon::fromTheme("office-chart-line"), i18n("Plot data"), this); connect(m_plotDataAction, &QAction::triggered, this, &MQTTClient::plotData); }*/ /*! * \brief Updates this data source at this moment */ void MQTTClient::updateNow() { qDebug()<<"Update now"; m_updateTimer->stop(); read(); if (m_updateType == TimeInterval && !m_paused) m_updateTimer->start(m_updateInterval); } /*! * \brief Continue reading from the live data source after it was paused. */ void MQTTClient::continueReading() { qDebug()<<"continue reading"; m_paused = false; if (m_updateType == TimeInterval) m_updateTimer->start(m_updateInterval); } /*! * \brief Pause the reading of the live data source. */ void MQTTClient::pauseReading() { qDebug()<<"pause reading"; m_paused = true; if (m_updateType == TimeInterval) m_updateTimer->stop(); } void MQTTClient::setFilter(AbstractFileFilter* f) { m_filter = f; } AbstractFileFilter* MQTTClient::filter() const { return m_filter; } /*! * \brief Sets the source's update interval to \c interval * \param interval */ void MQTTClient::setUpdateInterval(int interval) { qDebug()<<"Update interval " << interval; m_updateInterval = interval; if(!m_paused) m_updateTimer->start(m_updateInterval); } int MQTTClient::updateInterval() const { return m_updateInterval; } /*! * \brief Sets how many values we should store * \param keepnvalues */ void MQTTClient::setKeepNvalues(int keepnvalues) { qDebug()<<"Keep N values" << keepnvalues; m_keepNvalues = keepnvalues; } int MQTTClient::keepNvalues() const { return m_keepNvalues; } bool MQTTClient::isPaused() const { return m_paused; } /*! * \brief Sets the sample rate to samplerate * \param samplerate */ void MQTTClient::setSampleRate(int samplerate) { qDebug()<<"Sample rate: " << samplerate; m_sampleRate = samplerate; } int MQTTClient::sampleRate() const { return m_sampleRate; } /*! * \brief Sets the source's reading type to readingType * \param readingType */ void MQTTClient::setReadingType(ReadingType readingType) { qDebug()<<"Read Type : " << static_cast(readingType); m_readingType = readingType; } MQTTClient::ReadingType MQTTClient::readingType() const { return m_readingType; } /*! * \brief Sets the source's update type to updatetype and handles this change * \param updatetype */ void MQTTClient::setUpdateType(UpdateType updatetype) { qDebug()<<"Update Type : " << static_cast(updatetype); if (updatetype == NewData) { m_updateTimer->stop(); } m_updateType = updatetype; } MQTTClient::UpdateType MQTTClient::updateType() const { return m_updateType; } QIcon MQTTClient::icon() const { QIcon icon; icon = QIcon::fromTheme("labplot-MQTT"); return icon; } /*QMenu* MQTTClient::createContextMenu() { QMenu* menu = AbstractPart::createContextMenu(); QAction* firstAction = 0; // if we're populating the context menu for the project explorer, then //there're already actions available there. Skip the first title-action //and insert the action at the beginning of the menu. if (menu->actions().size()>1) firstAction = menu->actions().at(1); menu->insertAction(firstAction, m_plotDataAction); menu->insertSeparator(firstAction); //TODO: doesnt' always make sense... // if (!m_fileWatched) // menu->insertAction(firstAction, m_reloadAction); // // m_toggleWatchAction->setChecked(m_fileWatched); // menu->insertAction(firstAction, m_toggleWatchAction); // // m_toggleLinkAction->setChecked(m_fileLinked); // menu->insertAction(firstAction, m_toggleLinkAction); return menu; }*/ //############################################################################## //################################# SLOTS #################################### //############################################################################## /* * called periodically or on new data changes (file changed, new data in the socket, etc.) */ void MQTTClient::read() { if (m_filter == nullptr) return; //initialize the device (file, socket, serial port), when calling this function for the first time if (!m_prepared) { qDebug()<<"Read & Connect"; m_client->connectToHost(); qDebug()<<"connectTOHost called"; m_prepared = true; } if((m_client->state() == QMqttClient::ClientState::Connected) && m_mqttFirstConnectEstablished) { qDebug()<<"Read"; emit readFromTopics(); } } /*! Saves as XML. */ void MQTTClient::save(QXmlStreamWriter* writer) const { writer->writeStartElement("MQTTClient"); writeBasicAttributes(writer); writeCommentElement(writer); //general writer->writeStartElement("general"); writer->writeAttribute("subscriptionCount", QString::number(m_mqttSubscriptions.size())); writer->writeAttribute("updateType", QString::number(m_updateType)); writer->writeAttribute("readingType", QString::number(m_readingType)); writer->writeAttribute("keepValues", QString::number(m_keepNvalues)); if (m_updateType == TimeInterval) writer->writeAttribute("updateInterval", QString::number(m_updateInterval)); if (m_readingType != TillEnd) writer->writeAttribute("sampleRate", QString::number(m_sampleRate)); writer->writeAttribute("host", m_client->hostname()); writer->writeAttribute("port", QString::number(m_client->port())); writer->writeAttribute("username", m_client->username()); writer->writeAttribute("pasword", m_client->password()); writer->writeAttribute("clientId", m_client->clientId()); writer->writeAttribute("useRetain", QString::number(m_mqttRetain)); writer->writeAttribute("useWill", QString::number(m_mqttUseWill)); writer->writeAttribute("willTopic", m_willTopic); writer->writeAttribute("willOwnMessage", m_willOwnMessage); writer->writeAttribute("willQoS", QString::number(m_willQoS)); writer->writeAttribute("willRetain", QString::number(m_willRetain)); writer->writeAttribute("willMessageType", QString::number(static_cast(m_willMessageType))); writer->writeAttribute("willUpdateType", QString::number(static_cast(m_willUpdateType))); writer->writeAttribute("willTimeInterval", QString::number(m_willTimeInterval)); for( int i = 0; i < m_willStatistics.count(); ++i){ writer->writeAttribute("willStatistics"+QString::number(i), QString::number(m_willStatistics[i])); } writer->writeAttribute("useID", QString::number(m_mqttUseID)); writer->writeAttribute("useAuthentication", QString::number(m_mqttUseAuthentication)); writer->writeEndElement(); //filter m_filter->save(writer); //MQTTSubscriptions for(auto* sub : children(IncludeHidden)) sub->save(writer); writer->writeEndElement(); // "MQTTClient" } /*! Loads from XML. */ bool MQTTClient::load(XmlStreamReader* reader, bool preview) { qDebug()<<"Start loading MQTTClient"; if (!readBasicAttributes(reader)) return false; QString attributeWarning = i18n("Attribute '%1' missing or empty, default value is used"); QXmlStreamAttributes attribs; QString str; while (!reader->atEnd()) { reader->readNext(); if (reader->isEndElement() && reader->name() == "MQTTClient") break; if (!reader->isStartElement()) continue; if (reader->name() == "comment") { if (!readCommentElement(reader)) return false; } else if (reader->name() == "general") { qDebug()<<"MQTTClient general"; attribs = reader->attributes(); str = attribs.value("subscriptionCount").toString(); if(str.isEmpty()) reader->raiseWarning(attributeWarning.arg("'subscriptionCount'")); else - m_subscriptionCount = str.toInt(); + m_subscriptionCountToLoad = str.toInt(); str = attribs.value("keepValues").toString(); if(str.isEmpty()) reader->raiseWarning(attributeWarning.arg("'keepValues'")); else m_keepNvalues = str.toInt(); str = attribs.value("updateType").toString(); if(str.isEmpty()) reader->raiseWarning(attributeWarning.arg("'updateType'")); else m_updateType = static_cast(str.toInt()); str = attribs.value("readingType").toString(); if(str.isEmpty()) reader->raiseWarning(attributeWarning.arg("'readingType'")); else m_readingType = static_cast(str.toInt()); if (m_updateType == TimeInterval) { str = attribs.value("updateInterval").toString(); if(str.isEmpty()) reader->raiseWarning(attributeWarning.arg("'updateInterval'")); else m_updateInterval = str.toInt(); } if (m_readingType != TillEnd) { str = attribs.value("sampleRate").toString(); if(str.isEmpty()) reader->raiseWarning(attributeWarning.arg("'sampleRate'")); else m_sampleRate = str.toInt(); } str = attribs.value("host").toString(); if(str.isEmpty()) reader->raiseWarning(attributeWarning.arg("'host'")); else m_client->setHostname(str); str =attribs.value("port").toString(); if(str.isEmpty()) reader->raiseWarning(attributeWarning.arg("'port'")); else m_client->setPort(str.toUInt()); str = attribs.value("useAuthentication").toString(); if(str.isEmpty()) reader->raiseWarning(attributeWarning.arg("'useAuthentication'")); else m_mqttUseAuthentication = str.toInt(); if(m_mqttUseAuthentication) { str =attribs.value("username").toString(); if(!str.isEmpty()) m_client->setUsername(str); str =attribs.value("password").toString(); if(!str.isEmpty()) m_client->setPassword(str); } str = attribs.value("useID").toString(); if(str.isEmpty()) reader->raiseWarning(attributeWarning.arg("'useID'")); else m_mqttUseID = str.toInt(); if(m_mqttUseID) { str =attribs.value("clientId").toString(); if(!str.isEmpty()) m_client->setClientId(str); } str =attribs.value("useRetain").toString(); if(str.isEmpty()) reader->raiseWarning(attributeWarning.arg("'useRetain'")); else m_mqttRetain = str.toInt(); str =attribs.value("useWill").toString(); if(str.isEmpty()) reader->raiseWarning(attributeWarning.arg("'useWill'")); else m_mqttUseWill = str.toInt(); if(m_mqttUseWill) { str =attribs.value("willTopic").toString(); if(str.isEmpty()) reader->raiseWarning(attributeWarning.arg("'willTopic'")); else m_willTopic = str; str =attribs.value("willOwnMessage").toString(); if(str.isEmpty()) reader->raiseWarning(attributeWarning.arg("'willOwnMessage'")); else m_willOwnMessage = str; str =attribs.value("willQoS").toString(); if(str.isEmpty()) reader->raiseWarning(attributeWarning.arg("'willQoS'")); else m_willQoS = str.toUInt(); str =attribs.value("willRetain").toString(); if(str.isEmpty()) reader->raiseWarning(attributeWarning.arg("'willRetain'")); else m_willRetain = str.toInt(); str =attribs.value("willMessageType").toString(); if(str.isEmpty()) reader->raiseWarning(attributeWarning.arg("'willMessageType'")); else m_willMessageType = static_cast(str.toInt()); str =attribs.value("willUpdateType").toString(); if(str.isEmpty()) reader->raiseWarning(attributeWarning.arg("'willUpdateType'")); else m_willUpdateType = static_cast(str.toInt()); str =attribs.value("willTimeInterval").toString(); if(str.isEmpty()) reader->raiseWarning(attributeWarning.arg("'willTimeInterval'")); else m_willTimeInterval = str.toInt(); for( int i = 0; i < m_willStatistics.count(); ++i){ str =attribs.value("willStatistics"+QString::number(i)).toString(); if(str.isEmpty()) reader->raiseWarning(attributeWarning.arg("'willTimeInterval'")); else m_willStatistics[i] = str.toInt(); } } } else if (reader->name() == "asciiFilter") { qDebug()<<"load filter"; m_filter = new AsciiFilter(); if (!m_filter->load(reader)) return false; } else if(reader->name() == "MQTTSubscriptions") { qDebug()<<"Load MQTTSubscription"; MQTTSubscriptions* subscription = new MQTTSubscriptions(""); subscription->setMQTTClient(this); connect(subscription, &MQTTSubscriptions::loaded, this, &MQTTClient::subscriptionLoaded); if (!subscription->load(reader, preview)) { delete subscription; return false; } m_mqttSubscriptions.push_back(subscription); addChildFast(subscription); } else {// unknown element reader->raiseWarning(i18n("unknown element '%1'", reader->name().toString())); if (!reader->skipToEndElement()) return false; } } return !reader->hasError(); } void MQTTClient::setMqttClientHostPort(const QString& host, const quint16& port) { m_client->setHostname(host); m_client->setPort(port); } void MQTTClient::setMqttClientAuthentication(const QString& username, const QString& password) { m_client->setUsername(username); m_client->setPassword(password); } void MQTTClient::setMqttClientId(const QString &id){ m_client->setClientId(id); } void MQTTClient::addMqttSubscriptions(const QMqttTopicFilter& filter, const quint8& qos) { m_subscribedTopicNameQoS[filter] = qos; } void MQTTClient::onMqttConnect() { qDebug() << "on mqtt connect"; if(m_client->error() == QMqttClient::NoError) { if(!m_mqttFirstConnectEstablished) { qDebug()<<"connection made in MQTTClient"; QMapIterator i(m_subscribedTopicNameQoS); while(i.hasNext()) { i.next(); qDebug()<subscribe(i.key(), i.value()); if(temp) { qDebug()<topic()<<" "<qos(); if(!m_loaded) { m_subscriptions.push_back(temp->topic().filter()); qDebug()<<"New MQTTSubscription"; MQTTSubscriptions* newSubscription = new MQTTSubscriptions(temp->topic().filter()); newSubscription->setMQTTClient(this); qDebug()<<"Add child"; addChild(newSubscription); qDebug()<<"Add to vector"; m_mqttSubscriptions.push_back(newSubscription); } connect(temp, &QMqttSubscription::messageReceived, this, &MQTTClient::mqttSubscribtionMessageReceived); qDebug()<<"Added topic"; } } m_mqttFirstConnectEstablished = true; emit mqttSubscribed(); } else { qDebug() << "Resubscribing after will set"; QMapIterator i(m_subscribedTopicNameQoS); while(i.hasNext()) { i.next(); QMqttSubscription *temp = m_client->subscribe(i.key(), i.value()); if(temp) { qDebug()<topic()<<" "<qos(); connect(temp, &QMqttSubscription::messageReceived, this, &MQTTClient::mqttSubscribtionMessageReceived); } else qDebug()<<"Couldn't subscribe after will change"; } } } } void MQTTClient::mqttSubscribtionMessageReceived(const QMqttMessage& msg) { if(!msg.retain() || (msg.retain() && m_mqttRetain) ) { qDebug()<<"message received from "<subscriptionName(), msg.topic().name())) { m_mqttSubscriptions[i]->messageArrived(QString(msg.payload()), msg.topic().name()); break; } /*QString subscriptionName = m_mqttSubscriptions[i]->subscriptionName(); if(subscriptionName.contains('#') || subscriptionName.contains('+')) { if(subscriptionName.contains('#')) { if(msg.topic().name().startsWith(subscriptionName.left(subscriptionName.count() - 2)) ){ m_mqttSubscriptions[i]->messageArrived(QString(msg.payload()), msg.topic().name()); break; } } else if (subscriptionName.contains('+')) { int pos = subscriptionName.indexOf('+'); QString start = subscriptionName.left(pos); QString end = subscriptionName.right(subscriptionName.count() - pos); if(msg.topic().name().startsWith(start) && msg.topic().name().endsWith(end)) { m_mqttSubscriptions[i]->messageArrived(QString(msg.payload()), msg.topic().name()); break; } } } else if(subscriptionName == msg.topic().name()) { m_mqttSubscriptions[i]->messageArrived(QString(msg.payload()), msg.topic().name()); break; }*/ } if(msg.topic().name() == m_willTopic) m_willLastMessage = QString(msg.payload()); } } int MQTTClient::topicNumber() { return m_subscriptions.count(); } int MQTTClient::topicIndex(const QString& topic) { return m_subscriptions.indexOf(topic, 0); } void MQTTClient::setMqttWillUse(bool use) { m_mqttUseWill = use; if(use == false) m_willTimer->stop(); } bool MQTTClient::mqttWillUse() const{ return m_mqttUseWill; } void MQTTClient::setWillTopic(const QString& topic) { m_willTopic = topic; } QString MQTTClient::willTopic() const{ return m_willTopic; } void MQTTClient::setWillRetain(bool retain) { m_willRetain = retain; } bool MQTTClient::willRetain() const { return m_willRetain; } void MQTTClient::setWillQoS(quint8 QoS) { m_willQoS = QoS; } quint8 MQTTClient::willQoS() const { return m_willQoS; } void MQTTClient::setWillMessageType(WillMessageType messageType) { m_willMessageType = messageType; } MQTTClient::WillMessageType MQTTClient::willMessageType() const { return m_willMessageType; } void MQTTClient::setWillOwnMessage(const QString& ownMessage) { m_willOwnMessage = ownMessage; } QString MQTTClient::willOwnMessage() const { return m_willOwnMessage; } QVector MQTTClient::topicNames() const { return m_topicNames; } void MQTTClient::setWillForMqtt() { if(m_mqttUseWill && (m_client->state() == QMqttClient::ClientState::Connected) ) { if(!m_disconnectForWill) { qDebug() << "Disconnecting from host"; m_client->disconnectFromHost(); m_disconnectForWill = true; } setWillForMqtt(); } else if(m_mqttUseWill && (m_client->state() == QMqttClient::ClientState::Disconnected) && m_disconnectForWill) { m_client->setWillQoS(m_willQoS); qDebug()<<"Will QoS" << m_willQoS; m_client->setWillRetain(m_willRetain); qDebug()<<"Will retain" << m_willRetain; m_client->setWillTopic(m_willTopic); qDebug()<<"Will Topic" << m_willTopic; switch (m_willMessageType) { case WillMessageType::OwnMessage: m_client->setWillMessage(m_willOwnMessage.toUtf8()); qDebug()<<"Will own message" << m_willOwnMessage; break; case WillMessageType::Statistics: { qDebug()<<"Start will statistics"; QVector topics = children(AbstractAspect::Recursive); const AsciiFilter* asciiFilter = nullptr; const MQTTTopic* tempTopic = nullptr; qDebug()<<"Searching for topic"; for (int i = 0; i < topics.count(); ++i) { if(topics[i]->topicName() == m_willTopic) { asciiFilter = dynamic_cast(topics[i]->filter()); tempTopic = topics[i]; break; } } qDebug()<<"Check if topic found"; if(asciiFilter != nullptr && tempTopic != nullptr) { qDebug()<<"Checking column mode"; if((asciiFilter->mqttColumnMode() == AbstractColumn::ColumnMode::Integer) || (asciiFilter->mqttColumnMode() == AbstractColumn::ColumnMode::Numeric)) { m_client->setWillMessage(asciiFilter->mqttColumnStatistics(tempTopic).toUtf8()); qDebug() << "Will statistics message: "<< QString(m_client->willMessage()); } else { m_client->setWillMessage(QString("").toUtf8()); qDebug() << "Will statistics message: "<< QString(m_client->willMessage()); } } break; } case WillMessageType::LastMessage: m_client->setWillMessage(m_willLastMessage.toUtf8()); qDebug()<<"Will last message:\n" << m_willLastMessage; break; default: break; } m_disconnectForWill = false; m_client->connectToHost(); qDebug()<< "Reconnect to host"; } } MQTTClient::WillUpdateType MQTTClient::willUpdateType() const{ return m_willUpdateType; } void MQTTClient::setWillUpdateType(WillUpdateType updateType) { m_willUpdateType = updateType; } int MQTTClient::willTimeInterval() const{ return m_willTimeInterval; } void MQTTClient::setWillTimeInterval(int interval) { m_willTimeInterval = interval; } void MQTTClient::clearLastMessage() { m_willLastMessage.clear(); } void MQTTClient::addWillStatistics(WillStatistics statistic){ m_willStatistics[static_cast(statistic)] = true; } void MQTTClient::removeWillStatistics(WillStatistics statistic) { m_willStatistics[static_cast(statistic)] = false; } QVector MQTTClient::willStatistics() const{ return m_willStatistics; } void MQTTClient::startWillTimer() const{ if(m_willUpdateType == WillUpdateType::TimePeriod) m_willTimer->start(m_willTimeInterval); } void MQTTClient::stopWillTimer() const{ m_willTimer->stop(); } void MQTTClient::setMqttRetain(bool retain) { m_mqttRetain = retain; } bool MQTTClient::mqttRetain() const { return m_mqttRetain; } void MQTTClient::mqttErrorChanged(QMqttClient::ClientError clientError) { if(clientError != QMqttClient::ClientError::NoError) { MQTTErrorWidget* errorWidget = new MQTTErrorWidget(clientError, this); errorWidget->show(); } } QString MQTTClient::clientHostName() const{ return m_client->hostname(); } quint16 MQTTClient::clientPort() const { return m_client->port(); } QString MQTTClient::clientPassword() const{ return m_client->password(); } QString MQTTClient::clientUserName() const{ return m_client->username(); } QString MQTTClient::clientID () const{ return m_client->clientId(); } void MQTTClient::setMQTTUseID(bool use) { m_mqttUseID = use; } bool MQTTClient::mqttUseID() const { return m_mqttUseID; } void MQTTClient::setMQTTUseAuthentication(bool use) { m_mqttUseAuthentication = use; } bool MQTTClient::mqttUseAuthentication() const { return m_mqttUseAuthentication; } QVector MQTTClient::mqttSubscribtions() const { return m_subscriptions; } void MQTTClient::newMQTTSubscription(const QString& topic, quint8 QoS) { - QMqttTopicFilter filter {topic}; - QMqttSubscription* temp = m_client->subscribe(filter, QoS); + if(!m_subscriptions.contains(topic)) { + QMqttTopicFilter filter {topic}; + QMqttSubscription* temp = m_client->subscribe(filter, QoS); - if (temp) { - qDebug()<topic()<<" "<qos(); - m_subscriptions.push_back(temp->topic().filter()); - m_subscribedTopicNameQoS[temp->topic().filter()] = temp->qos(); + if (temp) { + qDebug()<topic()<<" "<qos(); + m_subscriptions.push_back(temp->topic().filter()); + m_subscribedTopicNameQoS[temp->topic().filter()] = temp->qos(); - qDebug()<<"New MQTTSubscription"; - MQTTSubscriptions* newSubscription = new MQTTSubscriptions(temp->topic().filter()); - newSubscription->setMQTTClient(this); + qDebug()<<"New MQTTSubscription"; + MQTTSubscriptions* newSubscription = new MQTTSubscriptions(temp->topic().filter()); + newSubscription->setMQTTClient(this); - qDebug()<<"Add child"; - addChild(newSubscription); + qDebug()<<"Add child"; + addChild(newSubscription); - qDebug()<<"Add to vector"; - m_mqttSubscriptions.push_back(newSubscription); + qDebug()<<"Add to vector"; + m_mqttSubscriptions.push_back(newSubscription); - qDebug()<<"Added topic"; + qDebug()<<"Added topic"; - qDebug()<<"Check for inferior subscriptions"; - bool found = false; - QVector inferiorSubscriptions; + qDebug()<<"Check for inferior subscriptions"; + bool found = false; + QVector inferiorSubscriptions; - for(int i = 0; i < m_mqttSubscriptions.size(); ++i) { - if(checkTopicContains(topic, m_mqttSubscriptions[i]->subscriptionName()) - && topic != m_mqttSubscriptions[i]->subscriptionName()) { - found = true; - inferiorSubscriptions.push_back(m_mqttSubscriptions[i]); + for(int i = 0; i < m_mqttSubscriptions.size(); ++i) { + if(checkTopicContains(topic, m_mqttSubscriptions[i]->subscriptionName()) + && topic != m_mqttSubscriptions[i]->subscriptionName()) { + found = true; + inferiorSubscriptions.push_back(m_mqttSubscriptions[i]); + } } - } - if(found) { - for(int sub = 0; sub < inferiorSubscriptions.size(); ++sub) { - qDebug()<<"Inferior subscription: "<subscriptionName(); - QVector topics = inferiorSubscriptions[sub]->topics(); - qDebug()<< topics.size(); + if(found) { + for(int sub = 0; sub < inferiorSubscriptions.size(); ++sub) { + qDebug()<<"Inferior subscription: "<subscriptionName(); + QVector topics = inferiorSubscriptions[sub]->topics(); + qDebug()<< topics.size(); - for(int i = 0; i < topics.size() ; ++i) { - qDebug()<topicName(); - topics[i]->reparent(newSubscription); - } + for(int i = 0; i < topics.size() ; ++i) { + qDebug()<topicName(); + topics[i]->reparent(newSubscription); + } - QMqttTopicFilter unsubscribeFilter {inferiorSubscriptions[sub]->subscriptionName()}; - m_client->unsubscribe(unsubscribeFilter); + QMqttTopicFilter unsubscribeFilter {inferiorSubscriptions[sub]->subscriptionName()}; + m_client->unsubscribe(unsubscribeFilter); - for (int j = 0; j < m_mqttSubscriptions.size(); ++j) { - if(m_mqttSubscriptions[j]->subscriptionName() == - inferiorSubscriptions[sub]->subscriptionName()) { - m_mqttSubscriptions.remove(j); + for (int j = 0; j < m_mqttSubscriptions.size(); ++j) { + if(m_mqttSubscriptions[j]->subscriptionName() == + inferiorSubscriptions[sub]->subscriptionName()) { + m_mqttSubscriptions.remove(j); + } } - } - m_subscriptions.removeAll(inferiorSubscriptions[sub]->subscriptionName()); + m_subscriptions.removeAll(inferiorSubscriptions[sub]->subscriptionName()); - removeChild(inferiorSubscriptions[sub]); + removeChild(inferiorSubscriptions[sub]); + } } - } - connect(temp, &QMqttSubscription::messageReceived, this, &MQTTClient::mqttSubscribtionMessageReceived); + connect(temp, &QMqttSubscription::messageReceived, this, &MQTTClient::mqttSubscribtionMessageReceived); + } } } void MQTTClient::removeMQTTSubscription(const QString &name) { - qDebug()<<"Start to remove subscription in MQTTClient: "<unsubscribe(filter); - qDebug()<<"QMqttClient's unsubscribe occured"; - - m_subscriptions.removeAll(name); - - for (int i = 0; i < m_mqttSubscriptions.size(); ++i) { - if(m_mqttSubscriptions[i]->subscriptionName() == name) { - qDebug()<<"Subscription name"<subscriptionName() << " "<name(); - MQTTSubscriptions* removeSubscription = m_mqttSubscriptions[i]; - m_mqttSubscriptions.remove(i); - QVector topics = removeSubscription->topics(); - for (int j = 0; j < topics.size(); ++j) { - qDebug()<<"Removing topic name: "<topicName(); - m_topicNames.removeAll(topics[j]->topicName()); + if(m_subscriptions.contains(name)) { + qDebug()<<"Start to remove subscription in MQTTClient: "<unsubscribe(filter); + qDebug()<<"QMqttClient's unsubscribe occured"; + + m_subscriptions.removeAll(name); + + for (int i = 0; i < m_mqttSubscriptions.size(); ++i) { + if(m_mqttSubscriptions[i]->subscriptionName() == name) { + qDebug()<<"Subscription name"<subscriptionName() << " "<name(); + MQTTSubscriptions* removeSubscription = m_mqttSubscriptions[i]; + m_mqttSubscriptions.remove(i); + QVector topics = removeSubscription->topics(); + for (int j = 0; j < topics.size(); ++j) { + qDebug()<<"Removing topic name: "<topicName(); + m_topicNames.removeAll(topics[j]->topicName()); + } + qDebug()<<"Removed from topic names and subscription names"; + removeChild(removeSubscription); + qDebug()<<"removed child"; + break; } - qDebug()<<"Removed from topic names and subscription names"; - removeChild(removeSubscription); - qDebug()<<"removed child"; - break; } - } - QMapIterator j(m_subscribedTopicNameQoS); - while(j.hasNext()) { - j.next(); - if(j.key().filter() == name) { - m_subscribedTopicNameQoS.remove(j.key()); - qDebug()<<"Removed from TopicNameQoS map "< j(m_subscribedTopicNameQoS); + while(j.hasNext()) { + j.next(); + if(j.key().filter() == name) { + m_subscribedTopicNameQoS.remove(j.key()); + qDebug()<<"Removed from TopicNameQoS map "<subscribe(filter, QoS); + if(!m_subscriptions.contains(topic)) { + QMqttTopicFilter filter {topic}; + QMqttSubscription* temp = m_client->subscribe(filter, QoS); - if (temp) { - qDebug()<topic()<<" "<qos(); - m_subscriptions.push_back(temp->topic().filter()); - m_subscribedTopicNameQoS[temp->topic().filter()] = temp->qos(); + if (temp) { + qDebug()<topic()<<" "<qos(); + m_subscriptions.push_back(temp->topic().filter()); + m_subscribedTopicNameQoS[temp->topic().filter()] = temp->qos(); - qDebug()<<"New MQTTSubscription"; - MQTTSubscriptions* newSubscription = new MQTTSubscriptions(temp->topic().filter()); - newSubscription->setMQTTClient(this); + qDebug()<<"New MQTTSubscription"; + MQTTSubscriptions* newSubscription = new MQTTSubscriptions(temp->topic().filter()); + newSubscription->setMQTTClient(this); - qDebug()<<"Add child"; - addChild(newSubscription); + qDebug()<<"Add child"; + addChild(newSubscription); - qDebug()<<"Add to vector"; - m_mqttSubscriptions.push_back(newSubscription); + qDebug()<<"Add to vector"; + m_mqttSubscriptions.push_back(newSubscription); - qDebug()<<"Added topic"; + qDebug()<<"Added topic"; - qDebug()<<"Check for superior subscription"; - bool found = false; - MQTTSubscriptions* superiorSubscription; + qDebug()<<"Check for superior subscription"; + bool found = false; + MQTTSubscriptions* superiorSubscription; - for(int i = 0; i < m_mqttSubscriptions.size(); ++i) { - if(checkTopicContains(m_mqttSubscriptions[i]->subscriptionName(), topic) - && topic != m_mqttSubscriptions[i]->subscriptionName()) { - found = true; - superiorSubscription = m_mqttSubscriptions[i]; - break; + for(int i = 0; i < m_mqttSubscriptions.size(); ++i) { + if(checkTopicContains(m_mqttSubscriptions[i]->subscriptionName(), topic) + && topic != m_mqttSubscriptions[i]->subscriptionName()) { + found = true; + superiorSubscription = m_mqttSubscriptions[i]; + break; + } } - } - if(found) { - qDebug()<<"Superior subscription: "<subscriptionName(); - QVector topics = superiorSubscription->topics(); - qDebug()<< topics.size(); + if(found) { + qDebug()<<"Superior subscription: "<subscriptionName(); + QVector topics = superiorSubscription->topics(); + qDebug()<< topics.size(); - QVector inferiorTopics; + QVector inferiorTopics; - for(int i = 0; i < topics.size(); ++i) { - if(checkTopicContains(topic, topics[i]->topicName())) { - inferiorTopics.push_back(topics[i]); + for(int i = 0; i < topics.size(); ++i) { + if(checkTopicContains(topic, topics[i]->topicName())) { + inferiorTopics.push_back(topics[i]); + } } - } - for(int i = 0; i < inferiorTopics.size() ; ++i) { - qDebug()<topicName(); - inferiorTopics[i]->reparent(newSubscription); + for(int i = 0; i < inferiorTopics.size() ; ++i) { + qDebug()<topicName(); + inferiorTopics[i]->reparent(newSubscription); + } } + connect(temp, &QMqttSubscription::messageReceived, this, &MQTTClient::mqttSubscribtionMessageReceived); } - connect(temp, &QMqttSubscription::messageReceived, this, &MQTTClient::mqttSubscribtionMessageReceived); } } void MQTTClient::reparentTopic(const QString& topic, const QString& parent) { - bool found = false; - MQTTSubscriptions* superiorSubscription; + if(m_subscriptions.contains(parent) && m_topicNames.contains(topic)) { + bool found = false; + MQTTSubscriptions* superiorSubscription; - for(int i = 0; i < m_mqttSubscriptions.size(); ++i) { - if(m_mqttSubscriptions[i]->subscriptionName() == parent) { - found = true; - superiorSubscription = m_mqttSubscriptions[i]; - break; + for(int i = 0; i < m_mqttSubscriptions.size(); ++i) { + if(m_mqttSubscriptions[i]->subscriptionName() == parent) { + found = true; + superiorSubscription = m_mqttSubscriptions[i]; + break; + } } - } - if(found) { - qDebug()<<"Superior subscription: "<subscriptionName(); - QVector topics = children(AbstractAspect::Recursive); - qDebug()<< topics.size(); - - for(int i = 0; i < topics.size(); ++i) { - qDebug()<topicName()<<" "<parentAspect()->name(); - if(topic == topics[i]->topicName()) { - qDebug()<topicName()<<" "<subscriptionName(); - topics[i]->reparent(superiorSubscription); - break; + if(found) { + qDebug()<<"Superior subscription: "<subscriptionName(); + QVector topics = children(AbstractAspect::Recursive); + qDebug()<< topics.size(); + + for(int i = 0; i < topics.size(); ++i) { + qDebug()<topicName()<<" "<parentAspect()->name(); + if(topic == topics[i]->topicName()) { + qDebug()<topicName()<<" "<subscriptionName(); + topics[i]->reparent(superiorSubscription); + break; + } } } + qDebug()<<"reparent done"; } - qDebug()<<"reparent done"; } void MQTTClient::subscriptionLoaded(const QString &name) { qDebug()< 0 && matchIndex < firstList.size() -1) { for(int j = matchIndex +1; j < firstList.size(); ++j) { if(firstList.at(j) != secondtList.at(j)) { differ = true; break; } } } else differ = true; if(!differ) { for(int i = 0; i < firstList.size(); ++i) { if(i != matchIndex) commonTopic.append(firstList.at(i)); else commonTopic.append("+"); if(i != firstList.size() - 1) commonTopic.append("/"); } } } } qDebug() << "Common topic: "< #include #include #include #include #include #include #include class QString; class AbstractFileFilter; class MQTTSubscriptions; class QAction; class MQTTClient : public Folder{ Q_OBJECT public: enum UpdateType { TimeInterval = 0, NewData }; enum ReadingType { ContinuousFixed = 0, FromEnd, TillEnd }; enum WillMessageType { OwnMessage = 0, Statistics, LastMessage }; enum WillUpdateType { TimePeriod = 0, OnClick }; enum WillStatistics { Minimum = 0, Maximum, ArithmeticMean, GeometricMean, HarmonicMean, ContraharmonicMean, Median, Variance, StandardDeviation, MeanDeviation, MeanDeviationAroundMedian, MedianDeviation, Skewness, Kurtosis, Entropy }; explicit MQTTClient(const QString& name); virtual ~MQTTClient() override; void ready(); UpdateType updateType() const; void setUpdateType(UpdateType); ReadingType readingType() const; void setReadingType(ReadingType); int sampleRate() const; void setSampleRate(int); bool isPaused() const; void setUpdateInterval(int); int updateInterval() const; void setKeepNvalues(int); int keepNvalues() const; void setKeepLastValues(bool); bool keepLastValues() const; void setMqttClientHostPort(const QString&, const quint16&); void setMqttClientAuthentication(const QString&, const QString&); void setMqttClientId(const QString&); QMqttClient mqttClient() const; void addMqttSubscriptions(const QMqttTopicFilter&, const quint8&); QVector mqttSubscribtions() const; bool checkTopicContains(const QString& superior, const QString& inferior); QString checkCommonLevel(const QString& first, const QString& second); QString clientHostName() const; quint16 clientPort() const; QString clientPassword() const; QString clientUserName() const; QString clientID () const; void updateNow(); void pauseReading(); void continueReading(); void setFilter(AbstractFileFilter*); AbstractFileFilter* filter() const; QIcon icon() const override; /*QMenu* createContextMenu() override; QWidget* view() const override;*/ void save(QXmlStreamWriter*) const override; bool load(XmlStreamReader*, bool preview) override; int topicNumber(); int topicIndex(const QString&); QVector topicNames() const; bool checkAllArrived(); void setMqttWillUse(bool); bool mqttWillUse() const; void setWillTopic(const QString&); QString willTopic() const; void setWillRetain(bool); bool willRetain() const; void setWillQoS(quint8); quint8 willQoS() const; void setWillMessageType(WillMessageType); WillMessageType willMessageType() const; void setWillOwnMessage(const QString&); QString willOwnMessage() const; WillUpdateType willUpdateType() const; void setWillUpdateType(WillUpdateType); int willTimeInterval() const; void setWillTimeInterval(int); void startWillTimer() const; void stopWillTimer() const; void setWillForMqtt() ; void setMqttRetain(bool); bool mqttRetain() const; void setMQTTUseID(bool); bool mqttUseID() const; void setMQTTUseAuthentication(bool); bool mqttUseAuthentication() const; void clearLastMessage(); void addWillStatistics(WillStatistics); void removeWillStatistics(WillStatistics); QVector willStatistics() const; void newMQTTSubscription(const QString&, quint8); void removeMQTTSubscription(const QString&); void addBeforeRemoveSubscription(const QString&, quint8); void reparentTopic(const QString& topic, const QString& parent); private: //void initActions(); UpdateType m_updateType; ReadingType m_readingType; bool m_paused; bool m_prepared; bool m_keepLastValues; int m_sampleRate; int m_keepNvalues; int m_updateInterval; AbstractFileFilter* m_filter; QTimer* m_updateTimer; /* QAction* m_reloadAction; QAction* m_toggleLinkAction; QAction* m_showEditorAction; QAction* m_showSpreadsheetAction; QAction* m_plotDataAction;*/ QMqttClient* m_client; QMap m_subscribedTopicNameQoS; QVector m_subscriptions; QVector m_topicNames; bool m_mqttTest; bool m_mqttUseWill; QString m_willMessage; QString m_willTopic; bool m_willRetain; quint8 m_willQoS; WillMessageType m_willMessageType; QString m_willOwnMessage; QString m_willLastMessage; QTimer* m_willTimer; int m_willTimeInterval; WillUpdateType m_willUpdateType; QVector m_willStatistics; bool m_mqttFirstConnectEstablished; bool m_mqttRetain; bool m_mqttUseID; bool m_mqttUseAuthentication; QVector m_mqttSubscriptions; bool m_disconnectForWill; bool m_loaded; int m_subscriptionsLoaded; - int m_subscriptionCount; + int m_subscriptionCountToLoad; public slots: void read(); private slots: void onMqttConnect(); void mqttSubscribtionMessageReceived(const QMqttMessage&); void mqttErrorChanged(QMqttClient::ClientError); void subscriptionLoaded(const QString&); signals: void mqttSubscribed(); void mqttNewTopicArrived(); void readFromTopics(); void clientAboutToBeDeleted(const QString&); }; #endif #endif // MQTTCLIENT_H