TinyMessageSubscriptionToken AddSubscriptionInternal <TMessage>(Action <TMessage> deliveryAction, Func <TMessage, bool> messageFilter, bool strongReference, ITinyMessageProxy proxy) where TMessage : class { if (deliveryAction == null) { throw new ArgumentNullException(nameof(deliveryAction)); } if (messageFilter == null) { throw new ArgumentNullException(nameof(messageFilter)); } if (proxy == null) { throw new ArgumentNullException(nameof(proxy)); } lock (_subscriptionsPadlock) { if (!_subscriptions.TryGetValue(typeof(TMessage), out var currentSubscriptions)) { currentSubscriptions = new List <SubscriptionItem>(); _subscriptions[typeof(TMessage)] = currentSubscriptions; } var subscriptionToken = new TinyMessageSubscriptionToken(this, typeof(TMessage)); ITinyMessageSubscription subscription; if (strongReference) { subscription = new StrongTinyMessageSubscription <TMessage>(subscriptionToken, deliveryAction, messageFilter); } else { subscription = new WeakTinyMessageSubscription <TMessage>(subscriptionToken, deliveryAction, messageFilter); } currentSubscriptions.Add(new SubscriptionItem(proxy, subscription)); return(subscriptionToken); } }
void RemoveSubscriptionInternal <TMessage>(TinyMessageSubscriptionToken subscriptionToken) where TMessage : class { if (subscriptionToken == null) { throw new ArgumentNullException(nameof(subscriptionToken)); } lock (_subscriptionsPadlock) { if (!_subscriptions.TryGetValue(typeof(TMessage), out var currentSubscriptions)) { return; } var currentlySubscribed = (from sub in currentSubscriptions where ReferenceEquals(sub.Subscription.SubscriptionToken, subscriptionToken) select sub).ToList(); currentlySubscribed.ForEach(sub => currentSubscriptions.Remove(sub)); } }
/// <summary> /// Initializes a new instance of the TinyMessageSubscription class. /// </summary> /// <param name="subscriptionToken">Destination object</param> /// <param name="deliveryAction">Delivery action</param> /// <param name="messageFilter">Filter function</param> public StrongTinyMessageSubscription(TinyMessageSubscriptionToken subscriptionToken, Action <TMessage> deliveryAction, Func <TMessage, bool> messageFilter) { SubscriptionToken = subscriptionToken ?? throw new ArgumentNullException(nameof(subscriptionToken)); _deliveryAction = deliveryAction ?? throw new ArgumentNullException(nameof(deliveryAction)); _messageFilter = messageFilter ?? throw new ArgumentNullException(nameof(messageFilter)); }
/// <summary> /// Unsubscribe from a particular message type. /// Does not throw an exception if the subscription is not found. /// </summary> /// <typeparam name="TMessage">Type of message</typeparam> /// <param name="subscriptionToken">Subscription token received from Subscribe</param> public void Unsubscribe <TMessage>(TinyMessageSubscriptionToken subscriptionToken) where TMessage : class { RemoveSubscriptionInternal <TMessage>(subscriptionToken); }