ServiceBrowser: add Services property which is synchronized with the …

…user events
tmds · Nov 25, 2013 · ab361ba · ab361ba
1 parent 0a6210e
commit ab361ba
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/Tmds/MDns/ServiceBrowser.cs b/Tmds/MDns/ServiceBrowser.cs
@@ -56,6 +56,7 @@ public void StartBrowse(string serviceType)
         public QueryParameters QueryParameters { get; private set; }
         public SynchronizationContext SynchronizationContext { get; set; }
         public bool IsBrowsing { get; private set; }
+        public IEnumerator<ServiceAnnouncement> Services { get { return _services.GetEnumerator(); } }
 
         public event EventHandler<ServiceAnnouncementEventArgs> ServiceAdded;
         public event EventHandler<ServiceAnnouncementEventArgs> ServiceRemoved;
@@ -78,6 +79,10 @@ internal void OnServiceAdded(ServiceInfo service)
             _serviceAnnouncements.Add(Tuple.Create(service.NetworkInterface.Id, service.Name), announcement);
             SynchronizationContextPost(o =>
             {
+                lock (_services)
+                {
+                    _services.Add(announcement);
+                }
                 if (ServiceAdded != null)
                 {
                     ServiceAdded(this, new ServiceAnnouncementEventArgs(announcement));
@@ -93,6 +98,10 @@ internal void OnServiceRemoved(ServiceInfo service)
             SynchronizationContextPost(o =>
             {
                 announcement.IsRemoved = true;
+                lock (_services)
+                {
+                    _services.Remove(announcement);
+                }
                 if (ServiceRemoved != null)
                 {
                     ServiceRemoved(this, new ServiceAnnouncementEventArgs(announcement));
@@ -229,6 +238,7 @@ private void SynchronizationContextPost(SendOrPostCallback cb)
             }
         }
 
+        private readonly HashSet<ServiceAnnouncement> _services = new HashSet<ServiceAnnouncement>();
         private readonly Dictionary<Tuple<string, Name>, ServiceAnnouncement> _serviceAnnouncements = new Dictionary<Tuple<string, Name>, ServiceAnnouncement>();
         private readonly List<string> _serviceTypes = new List<string>();
         private Dictionary<int, NetworkInterfaceHandler> _interfaceHandlers;