public void ProcessSessionPublishQueue() { var publish = (MqttMsgPublish)sessionsPublishQueue.Take(); if (publish != null) { // get all sessions var sessions = sessionManager.GetSessions(); if (sessions != null && sessions.Count > 0) { foreach (var session in sessions) { var query = from s in session.Subscriptions where new Regex(s.Topic).IsMatch(publish.Topic) select s; var comparer = new MqttSubscriptionComparer( MqttSubscriptionComparer.MqttSubscriptionComparerType.OnClientId); // consider only session active for client disconnected (not online) if (session.Connection == null) { foreach (var subscription in query.Distinct(comparer)) { // TODO There has to be something wrong here session.OutgoingMessages.Enqueue(publish); } } } } } }
private void SessionPublishThread() { while (this.isRunning) { try { var publish = (MqttMsgPublish)this.sessionPublishQueue.Take(); if (publish != null) { // get all sessions List <MqttBrokerSession> sessions = MqttSessionManager.GetSessions(); if ((sessions != null) && (sessions.Count > 0)) { foreach (MqttBrokerSession session in sessions) { var query = from s in session.Subscriptions where (new Regex(s.Topic)).IsMatch(publish.Topic) select s; MqttSubscriptionComparer comparer = new MqttSubscriptionComparer(MqttSubscriptionComparer.MqttSubscriptionComparerType.OnClientId); // consider only session active for client disconnected (not online) if (session.ClientConnection == null) { foreach (MqttSubscription subscription in query.Distinct(comparer)) { // save PUBLISH message for client disconnected (not online) session.OutgoingMessages.Enqueue(publish); } } } } } } catch (Exception exception) { logger.LogException(this, exception); } } }
static MqttSubscriptionManager() { NonWildcardSubscriptions = new ConcurrentDictionary <string, List <MqttSubscription> >(); WildcardSubscriptions = new ConcurrentDictionary <string, List <MqttSubscription> >(); Comparer = new MqttSubscriptionComparer(MqttSubscriptionComparer.MqttSubscriptionComparerType.OnClientId); }
/// <summary> /// Process the message queue to publish /// </summary> public void PublishThread() { int count; byte qosLevel; MqttMsgPublish publish; // create event to signal that current thread is ended this.publishEventEnd = new AutoResetEvent(false); while (this.isRunning) { // wait on message queueud to publish this.publishQueueWaitHandle.WaitOne(); // first check new subscribers to send retained messages ... lock (this.subscribersForRetained) { count = this.subscribersForRetained.Count; // publish retained messages to subscribers (new or reconnected) while (count > 0) { count--; MqttSubscription subscription = this.subscribersForRetained.Dequeue(); var query = from p in this.retainedMessages where (new Regex(subscription.Topic)).IsMatch(p.Key) // check for topics based also on wildcard with regex select p.Value; if (query.Count() > 0) { foreach (MqttMsgPublish retained in query) { qosLevel = (subscription.QosLevel < retained.QosLevel) ? subscription.QosLevel : retained.QosLevel; // send PUBLISH message to the current subscriber subscription.Client.Publish(retained.Topic, retained.Message, qosLevel, retained.Retain); } } } } // ... then check clients to send outgoing session messages lock (this.clientsForSession) { count = this.clientsForSession.Count; // publish outgoing session messages to clients (reconnected) while (count > 0) { count--; string clientId = this.clientsForSession.Dequeue(); MqttBrokerSession session = this.sessionManager.GetSession(clientId); while (session.OutgoingMessages.Count > 0) { MqttMsgPublish outgoingMsg = session.OutgoingMessages.Dequeue(); var query = from s in session.Subscriptions where (new Regex(s.Topic)).IsMatch(outgoingMsg.Topic) // check for topics based also on wildcard with regex select s; MqttSubscription subscription = query.First(); if (subscription != null) { qosLevel = (subscription.QosLevel < outgoingMsg.QosLevel) ? subscription.QosLevel : outgoingMsg.QosLevel; session.Client.Publish(outgoingMsg.Topic, outgoingMsg.Message, qosLevel, outgoingMsg.Retain); } } } } // ... then pass to process publish queue lock (this.publishQueue) { publish = null; count = this.publishQueue.Count; // publish all queued messages while (count > 0) { count--; publish = (MqttMsgPublish)this.publishQueue.Dequeue(); if (publish != null) { // get all subscriptions for a topic List<MqttSubscription> subscriptions = this.subscriberManager.GetSubscriptionsByTopic(publish.Topic); if ((subscriptions != null) && (subscriptions.Count > 0)) { foreach (MqttSubscription subscription in subscriptions) { qosLevel = (subscription.QosLevel < publish.QosLevel) ? subscription.QosLevel : publish.QosLevel; // send PUBLISH message to the current subscriber subscription.Client.Publish(publish.Topic, publish.Message, qosLevel, publish.Retain); } } // get all sessions List<MqttBrokerSession> sessions = this.sessionManager.GetSessions(); if ((sessions != null) && (sessions.Count > 0)) { foreach (MqttBrokerSession session in sessions) { var query = from s in session.Subscriptions where (new Regex(s.Topic)).IsMatch(publish.Topic) select s; MqttSubscriptionComparer comparer = new MqttSubscriptionComparer(MqttSubscriptionComparer.MqttSubscriptionComparerType.OnClientId); // consider only session active for client disconnected (not online) if (session.Client == null) { foreach (MqttSubscription subscription in query.Distinct(comparer)) { qosLevel = (subscription.QosLevel < publish.QosLevel) ? subscription.QosLevel : publish.QosLevel; // save PUBLISH message for client disconnected (not online) session.OutgoingMessages.Enqueue(publish); } } } } } } } } // signal thread end this.publishEventEnd.Set(); }