Added API to get specific service + synchronized services list operat…

…ions
devgianlu · Jun 20, 2021 · ac30381 · ac30381
1 parent 88b9e80
commit ac30381
Showing 1 changed file with 15 additions and 1 deletion.
diff --git a/src/main/java/xyz/gianlu/zeroconf/Zeroconf.java b/src/main/java/xyz/gianlu/zeroconf/Zeroconf.java
@@ -1,6 +1,7 @@
 package xyz.gianlu.zeroconf;
 
 import org.jetbrains.annotations.NotNull;
+import org.jetbrains.annotations.Nullable;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
@@ -555,7 +556,9 @@ private void addRelated(@NotNull Record record) {
             if (!record.getName().endsWith(serviceName))
                 return;
 
-            services.stream().filter(s -> s.serviceName.equals(record.getName())).forEach(s -> s.addRelatedRecord(record));
+            synchronized (services) {
+                services.stream().filter(s -> s.serviceName.equals(record.getName())).forEach(s -> s.addRelatedRecord(record));
+            }
         }
 
         private void addService(@NotNull RecordSRV record) {
@@ -575,6 +578,17 @@ public Collection<DiscoveredService> getServices() {
             return Collections.unmodifiableCollection(services);
         }
 
+        @Nullable
+        public DiscoveredService getService(@NotNull String name) {
+            synchronized (services) {
+                for (DiscoveredService service : services)
+                    if (service.name.equals(name))
+                        return service;
+
+                return null;
+            }
+        }
+
         @Override
         public void run() {
             while (!shouldStop) {