From b9bceadec88d582711eedc037359155de2799915 Mon Sep 17 00:00:00 2001
From: Saad Sahibjan <saadsahibjan@gmail.com>
Date: Thu, 8 Aug 2019 16:51:21 +0530
Subject: [PATCH] Add the implementation of DeviceTypePluginExtensionService

---
 .../DeviceTypePluginExtensionServiceImpl.java | 43 +++++++++++++++++++
 1 file changed, 43 insertions(+)
 create mode 100644 components/device-mgt/org.wso2.carbon.device.mgt.extensions/src/main/java/org/wso2/carbon/device/mgt/extensions/device/type/template/DeviceTypePluginExtensionServiceImpl.java

diff --git a/components/device-mgt/org.wso2.carbon.device.mgt.extensions/src/main/java/org/wso2/carbon/device/mgt/extensions/device/type/template/DeviceTypePluginExtensionServiceImpl.java b/components/device-mgt/org.wso2.carbon.device.mgt.extensions/src/main/java/org/wso2/carbon/device/mgt/extensions/device/type/template/DeviceTypePluginExtensionServiceImpl.java
new file mode 100644
index 00000000000..4ccaeadb604
--- /dev/null
+++ b/components/device-mgt/org.wso2.carbon.device.mgt.extensions/src/main/java/org/wso2/carbon/device/mgt/extensions/device/type/template/DeviceTypePluginExtensionServiceImpl.java
@@ -0,0 +1,43 @@
+/*
+ *   Copyright (c) 2019, Entgra (Pvt) Ltd. (https://entgra.io) All Rights Reserved.
+ *
+ *   Entgra (Pvt) Ltd. licenses this file to you under the Apache License,
+ *   Version 2.0 (the "License"); you may not use this file except
+ *   in compliance with the License.
+ *   You may obtain a copy of the License at
+ *
+ *   http://www.apache.org/licenses/LICENSE-2.0
+ *
+ *   Unless required by applicable law or agreed to in writing,
+ *   software distributed under the License is distributed on an
+ *   "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ *   KIND, either express or implied. See the License for the
+ *   specific language governing permissions and limitations
+ *   under the License.
+ */
+package org.wso2.carbon.device.mgt.extensions.device.type.template;
+
+import org.wso2.carbon.device.mgt.extensions.device.type.template.dao.DeviceTypePluginDAOManager;
+import org.wso2.carbon.device.mgt.extensions.spi.DeviceTypePluginExtensionService;
+
+import java.util.HashMap;
+import java.util.Map;
+
+public class DeviceTypePluginExtensionServiceImpl implements DeviceTypePluginExtensionService {
+
+    private static volatile Map<String, DeviceTypePluginDAOManager> pluginDAOManagers = new HashMap<>();
+
+    @Override
+    public void addPluginDAOManager(String deviceType, DeviceTypePluginDAOManager pluginDAOManager) {
+        if (pluginDAOManager != null) {
+            if (!pluginDAOManagers.containsKey(deviceType)) {
+                pluginDAOManagers.put(deviceType, pluginDAOManager);
+            }
+        }
+    }
+
+    @Override
+    public DeviceTypePluginDAOManager getPluginDAOManager(String deviceType) {
+        return pluginDAOManagers.get(deviceType);
+    }
+}