git.basschouten.com Git - openhab-addons.git/blob

   1 /**
   2  * Copyright (c) 2010-2021 Contributors to the openHAB project
   3  *
   4  * See the NOTICE file(s) distributed with this work for additional
   5  * information.
   6  *
   7  * This program and the accompanying materials are made available under the
   8  * terms of the Eclipse Public License 2.0 which is available at
   9  * http://www.eclipse.org/legal/epl-2.0
  10  *
  11  * SPDX-License-Identifier: EPL-2.0
  12  */
  13 package org.openhab.binding.bluetooth.discovery.internal;
  14
  15 import java.time.Duration;
  16 import java.util.HashMap;
  17 import java.util.HashSet;
  18 import java.util.Map;
  19 import java.util.Optional;
  20 import java.util.Set;
  21 import java.util.concurrent.CompletableFuture;
  22 import java.util.concurrent.ConcurrentHashMap;
  23 import java.util.concurrent.CopyOnWriteArraySet;
  24 import java.util.function.BiConsumer;
  25
  26 import org.eclipse.jdt.annotation.NonNullByDefault;
  27 import org.eclipse.jdt.annotation.Nullable;
  28 import org.openhab.binding.bluetooth.BluetoothAdapter;
  29 import org.openhab.binding.bluetooth.BluetoothAddress;
  30 import org.openhab.binding.bluetooth.BluetoothBindingConstants;
  31 import org.openhab.binding.bluetooth.BluetoothDevice;
  32 import org.openhab.binding.bluetooth.BluetoothDiscoveryListener;
  33 import org.openhab.binding.bluetooth.discovery.BluetoothDiscoveryParticipant;
  34 import org.openhab.core.cache.ExpiringCache;
  35 import org.openhab.core.config.discovery.AbstractDiscoveryService;
  36 import org.openhab.core.config.discovery.DiscoveryResult;
  37 import org.openhab.core.config.discovery.DiscoveryResultBuilder;
  38 import org.openhab.core.config.discovery.DiscoveryService;
  39 import org.openhab.core.thing.ThingTypeUID;
  40 import org.openhab.core.thing.ThingUID;
  41 import org.osgi.service.component.annotations.Activate;
  42 import org.osgi.service.component.annotations.Component;
  43 import org.osgi.service.component.annotations.Deactivate;
  44 import org.osgi.service.component.annotations.Modified;
  45 import org.osgi.service.component.annotations.Reference;
  46 import org.osgi.service.component.annotations.ReferenceCardinality;
  47 import org.osgi.service.component.annotations.ReferencePolicy;
  48 import org.slf4j.Logger;
  49 import org.slf4j.LoggerFactory;
  50
  51 /**
  52  * The {@link BluetoothDiscoveryService} handles searching for BLE devices.
  53  *
  54  * @author Chris Jackson - Initial Contribution
  55  * @author Kai Kreuzer - Introduced BluetoothAdapters and BluetoothDiscoveryParticipants
  56  * @author Connor Petty - Introduced connection based discovery and added roaming support
  57  */
  58 @NonNullByDefault
  59 @Component(service = DiscoveryService.class, configurationPid = "discovery.bluetooth")
  60 public class BluetoothDiscoveryService extends AbstractDiscoveryService implements BluetoothDiscoveryListener {
  61
  62     private final Logger logger = LoggerFactory.getLogger(BluetoothDiscoveryService.class);
  63
  64     private static final int SEARCH_TIME = 15;
  65
  66     private final Set<BluetoothAdapter> adapters = new CopyOnWriteArraySet<>();
  67     private final Set<BluetoothDiscoveryParticipant> participants = new CopyOnWriteArraySet<>();
  68     @NonNullByDefault({})
  69     private final Map<BluetoothAddress, DiscoveryCache> discoveryCaches = new ConcurrentHashMap<>();
  70
  71     private final Set<ThingTypeUID> supportedThingTypes = new CopyOnWriteArraySet<>();
  72
  73     public BluetoothDiscoveryService() {
  74         super(SEARCH_TIME);
  75         supportedThingTypes.add(BluetoothBindingConstants.THING_TYPE_BEACON);
  76     }
  77
  78     @Override
  79     @Activate
  80     protected void activate(@Nullable Map<String, Object> configProperties) {
  81         logger.debug("Activating Bluetooth discovery service");
  82         super.activate(configProperties);
  83     }
  84
  85     @Override
  86     @Modified
  87     protected void modified(@Nullable Map<String, Object> configProperties) {
  88         super.modified(configProperties);
  89     }
  90
  91     @Override
  92     @Deactivate
  93     public void deactivate() {
  94         logger.debug("Deactivating Bluetooth discovery service");
  95     }
  96
  97     @Reference(cardinality = ReferenceCardinality.MULTIPLE, policy = ReferencePolicy.DYNAMIC)
  98     protected void addBluetoothAdapter(BluetoothAdapter adapter) {
  99         this.adapters.add(adapter);
 100         adapter.addDiscoveryListener(this);
 101     }
 102
 103     protected void removeBluetoothAdapter(BluetoothAdapter adapter) {
 104         this.adapters.remove(adapter);
 105         adapter.removeDiscoveryListener(this);
 106     }
 107
 108     @Reference(cardinality = ReferenceCardinality.MULTIPLE, policy = ReferencePolicy.DYNAMIC)
 109     protected void addBluetoothDiscoveryParticipant(BluetoothDiscoveryParticipant participant) {
 110         this.participants.add(participant);
 111         supportedThingTypes.addAll(participant.getSupportedThingTypeUIDs());
 112     }
 113
 114     protected void removeBluetoothDiscoveryParticipant(BluetoothDiscoveryParticipant participant) {
 115         supportedThingTypes.removeAll(participant.getSupportedThingTypeUIDs());
 116         this.participants.remove(participant);
 117     }
 118
 119     @Override
 120     public Set<ThingTypeUID> getSupportedThingTypes() {
 121         return supportedThingTypes;
 122     }
 123
 124     @Override
 125     public void startScan() {
 126         for (BluetoothAdapter adapter : adapters) {
 127             adapter.scanStart();
 128         }
 129     }
 130
 131     @Override
 132     public void stopScan() {
 133         for (BluetoothAdapter adapter : adapters) {
 134             adapter.scanStop();
 135         }
 136         removeOlderResults(getTimestampOfLastScan());
 137     }
 138
 139     @Override
 140     public void deviceRemoved(BluetoothDevice device) {
 141         discoveryCaches.computeIfPresent(device.getAddress(), (addr, cache) -> cache.removeDiscoveries(device));
 142     }
 143
 144     @Override
 145     public void deviceDiscovered(BluetoothDevice device) {
 146         logger.debug("Discovered bluetooth device '{}': {}", device.getName(), device);
 147
 148         DiscoveryCache cache = discoveryCaches.computeIfAbsent(device.getAddress(), addr -> new DiscoveryCache());
 149         cache.handleDiscovery(device);
 150     }
 151
 152     private static ThingUID createThingUIDWithBridge(DiscoveryResult result, BluetoothAdapter adapter) {
 153         return new ThingUID(result.getThingTypeUID(), adapter.getUID(), result.getThingUID().getId());
 154     }
 155
 156     private static DiscoveryResult copyWithNewBridge(DiscoveryResult result, BluetoothAdapter adapter) {
 157         String label = result.getLabel();
 158
 159         return DiscoveryResultBuilder.create(createThingUIDWithBridge(result, adapter))//
 160                 .withBridge(adapter.getUID())//
 161                 .withProperties(result.getProperties())//
 162                 .withRepresentationProperty(result.getRepresentationProperty())//
 163                 .withTTL(result.getTimeToLive())//
 164                 .withLabel(label)//
 165                 .build();
 166     }
 167
 168     private class DiscoveryCache {
 169
 170         private final Map<BluetoothAdapter, SnapshotFuture> discoveryFutures = new HashMap<>();
 171         private final Map<BluetoothAdapter, Set<DiscoveryResult>> discoveryResults = new ConcurrentHashMap<>();
 172
 173         private ExpiringCache<BluetoothDeviceSnapshot> latestSnapshot = new ExpiringCache<>(Duration.ofMinutes(1),
 174                 () -> null);
 175
 176         /**
 177          * This is meant to be used as part of a Map.compute function
 178          *
 179          * @param device the device to remove from this cache
 180          * @return this DiscoveryCache if there are still snapshots, null otherwise
 181          */
 182         public synchronized @Nullable DiscoveryCache removeDiscoveries(final BluetoothDevice device) {
 183             // we remove any discoveries that have been published for this device
 184             BluetoothAdapter adapter = device.getAdapter();
 185             if (discoveryFutures.containsKey(adapter)) {
 186                 discoveryFutures.remove(adapter).future.thenAccept(result -> retractDiscoveryResult(adapter, result));
 187             }
 188             if (discoveryFutures.isEmpty()) {
 189                 return null;
 190             }
 191             return this;
 192         }
 193
 194         public synchronized void handleDiscovery(BluetoothDevice device) {
 195             if (!discoveryFutures.isEmpty()) {
 196                 CompletableFuture
 197                         // we have an ongoing futures so lets create our discovery after they all finish
 198                         .allOf(discoveryFutures.values().stream().map(sf -> sf.future)
 199                                 .toArray(CompletableFuture[]::new))
 200                         .thenRun(() -> createDiscoveryFuture(device));
 201             } else {
 202                 createDiscoveryFuture(device);
 203             }
 204         }
 205
 206         private synchronized void createDiscoveryFuture(BluetoothDevice device) {
 207             BluetoothAdapter adapter = device.getAdapter();
 208             CompletableFuture<DiscoveryResult> future = null;
 209
 210             BluetoothDeviceSnapshot snapshot = new BluetoothDeviceSnapshot(device);
 211             BluetoothDeviceSnapshot latestSnapshot = this.latestSnapshot.getValue();
 212             if (latestSnapshot != null) {
 213                 snapshot.merge(latestSnapshot);
 214
 215                 if (snapshot.equals(latestSnapshot)) {
 216                     // this means that snapshot has no newer fields than the latest snapshot
 217                     if (discoveryFutures.containsKey(adapter)
 218                             && discoveryFutures.get(adapter).snapshot.equals(latestSnapshot)) {
 219                         // This adapter has already produced the most up-to-date result, so no further processing is
 220                         // necessary
 221                         return;
 222                     }
 223
 224                     /*
 225                      * This isn't a new snapshot, but an up-to-date result from this adapter has not been produced yet.
 226                      * Since a result must have been produced for this snapshot, we search the results of the other
 227                      * adapters to find the future for the latest snapshot, then we modify it to make it look like it
 228                      * came from this adapter. This way we don't need to recompute the DiscoveryResult.
 229                      */
 230                     Optional<CompletableFuture<DiscoveryResult>> otherFuture = discoveryFutures.values().stream()
 231                             // make sure that we only get futures for the current snapshot
 232                             .filter(sf -> sf.snapshot.equals(latestSnapshot)).findAny().map(sf -> sf.future);
 233                     if (otherFuture.isPresent()) {
 234                         future = otherFuture.get();
 235                     }
 236                 }
 237             }
 238             this.latestSnapshot.putValue(snapshot);
 239
 240             if (future == null) {
 241                 // we pass in the snapshot since it acts as a delegate for the device. It will also retain any new
 242                 // fields added to the device as part of the discovery process.
 243                 future = startDiscoveryProcess(snapshot);
 244             }
 245
 246             if (discoveryFutures.containsKey(adapter)) {
 247                 // now we need to make sure that we remove the old discovered result if it is different from the new
 248                 // one.
 249                 SnapshotFuture oldSF = discoveryFutures.get(adapter);
 250                 future = oldSF.future.thenCombine(future, (oldResult, newResult) -> {
 251                     logger.trace("\n old: {}\n new: {}", oldResult, newResult);
 252                     if (!oldResult.getThingUID().equals(newResult.getThingUID())) {
 253                         retractDiscoveryResult(adapter, oldResult);
 254                     }
 255                     return newResult;
 256                 });
 257             }
 258             /*
 259              * this appends a post-process to any ongoing or completed discoveries with this device's address.
 260              * If this discoveryFuture is ongoing then this post-process will run asynchronously upon the future's
 261              * completion.
 262              * If this discoveryFuture is already completed then this post-process will run in the current thread.
 263              * We need to make sure that this is part of the future chain so that the call to 'thingRemoved'
 264              * in the 'removeDiscoveries' method above can be sure that it is running after the 'thingDiscovered'
 265              */
 266             future = future.thenApply(result -> {
 267                 publishDiscoveryResult(adapter, result);
 268                 return result;
 269             }).whenComplete((r, t) -> {
 270                 if (t != null) {
 271                     logger.warn("Error occured during discovery of {}", device.getAddress(), t);
 272                 }
 273             });
 274
 275             // now save this snapshot for later
 276             discoveryFutures.put(adapter, new SnapshotFuture(snapshot, future));
 277         }
 278
 279         private void publishDiscoveryResult(BluetoothAdapter adapter, DiscoveryResult result) {
 280             Set<DiscoveryResult> results = new HashSet<>();
 281             BiConsumer<BluetoothAdapter, DiscoveryResult> publisher = (a, r) -> {
 282                 results.add(copyWithNewBridge(r, a));
 283             };
 284
 285             publisher.accept(adapter, result);
 286             for (BluetoothDiscoveryParticipant participant : participants) {
 287                 participant.publishAdditionalResults(result, publisher);
 288             }
 289             results.forEach(BluetoothDiscoveryService.this::thingDiscovered);
 290             discoveryResults.put(adapter, results);
 291         }
 292
 293         private void retractDiscoveryResult(BluetoothAdapter adapter, DiscoveryResult result) {
 294             Set<DiscoveryResult> results = discoveryResults.remove(adapter);
 295             if (results != null) {
 296                 for (DiscoveryResult r : results) {
 297                     thingRemoved(r.getThingUID());
 298                 }
 299             }
 300         }
 301
 302         private CompletableFuture<DiscoveryResult> startDiscoveryProcess(BluetoothDeviceSnapshot device) {
 303             return CompletableFuture.supplyAsync(new BluetoothDiscoveryProcess(device, participants, adapters),
 304                     scheduler);
 305         }
 306     }
 307
 308     private static class SnapshotFuture {
 309         public final BluetoothDeviceSnapshot snapshot;
 310         public final CompletableFuture<DiscoveryResult> future;
 311
 312         public SnapshotFuture(BluetoothDeviceSnapshot snapshot, CompletableFuture<DiscoveryResult> future) {
 313             this.snapshot = snapshot;
 314             this.future = future;
 315         }
 316     }
 317 }