bluetooth: Add characteristics to the device's attribute instance map

third_party/WebKit/Source/modules/bluetooth/BluetoothAttributeInstanceMap.cpp

 // Copyright 2016 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "modules/bluetooth/BluetoothAttributeInstanceMap.h"
 
 #include "modules/bluetooth/BluetoothDevice.h"
 #include "modules/bluetooth/BluetoothRemoteGATTService.h"
+#include "public/platform/modules/bluetooth/WebBluetoothRemoteGATTCharacteristicInit.h"
 #include "public/platform/modules/bluetooth/WebBluetoothRemoteGATTService.h"
 #include <memory>
 #include <utility>
 
 namespace blink {
 
 BluetoothAttributeInstanceMap::BluetoothAttributeInstanceMap(
     BluetoothDevice* device)
     : m_device(device) {}
 
@@ skipping 12 matching lines @@
   }
 
   return service;
 }
 
 bool BluetoothAttributeInstanceMap::containsService(
     const String& serviceInstanceId) {
   return m_serviceIdToObject.contains(serviceInstanceId);
 }
 
+BluetoothRemoteGATTCharacteristic*
+BluetoothAttributeInstanceMap::getOrCreateBluetoothRemoteGATTCharacteristic(
+    ExecutionContext* context,
+    std::unique_ptr<WebBluetoothRemoteGATTCharacteristicInit> webCharacteristic,
+    BluetoothRemoteGATTService* service) {
+  String characteristicInstanceId = webCharacteristic->characteristicInstanceID;
+
+  BluetoothRemoteGATTCharacteristic* characteristic =
+      m_characteristicIdToObject.get(characteristicInstanceId);
+
+  if (!characteristic) {
+    characteristic = BluetoothRemoteGATTCharacteristic::create(
+        context, std::move(webCharacteristic), service);
+    m_characteristicIdToObject.add(characteristicInstanceId, characteristic);
+  }
+
+  return characteristic;
+}
+
 void BluetoothAttributeInstanceMap::Clear() {
   m_serviceIdToObject.clear();
 }
 
 DEFINE_TRACE(BluetoothAttributeInstanceMap) {
   visitor->trace(m_device);
   visitor->trace(m_serviceIdToObject);
+  visitor->trace(m_characteristicIdToObject);
 }
 
 }  // namespace blink