From 1714328db04a15522c4cd5b0a8fdcb01bb527bff Mon Sep 17 00:00:00 2001
From: Vishanth <vishanth911@gmail.com>
Date: Thu, 15 Dec 2016 10:32:26 +0530
Subject: [PATCH] datasource access for multi tenancy

---
 .../deployer/template/DeviceTypeManager.java  | 45 ++++++++++++-------
 1 file changed, 28 insertions(+), 17 deletions(-)

diff --git a/components/device-mgt-extensions/org.wso2.carbon.device.mgt.extensions.device.type.deployer/src/main/java/org/wso2/carbon/device/mgt/extensions/device/type/deployer/template/DeviceTypeManager.java b/components/device-mgt-extensions/org.wso2.carbon.device.mgt.extensions.device.type.deployer/src/main/java/org/wso2/carbon/device/mgt/extensions/device/type/deployer/template/DeviceTypeManager.java
index 8a9bcd7ed5..eba2f18721 100644
--- a/components/device-mgt-extensions/org.wso2.carbon.device.mgt.extensions.device.type.deployer/src/main/java/org/wso2/carbon/device/mgt/extensions/device/type/deployer/template/DeviceTypeManager.java
+++ b/components/device-mgt-extensions/org.wso2.carbon.device.mgt.extensions.device.type.deployer/src/main/java/org/wso2/carbon/device/mgt/extensions/device/type/deployer/template/DeviceTypeManager.java
@@ -20,6 +20,7 @@ package org.wso2.carbon.device.mgt.extensions.device.type.deployer.template;
 
 import org.apache.commons.logging.Log;
 import org.apache.commons.logging.LogFactory;
+import org.wso2.carbon.context.PrivilegedCarbonContext;
 import org.wso2.carbon.device.mgt.common.Device;
 import org.wso2.carbon.device.mgt.common.DeviceIdentifier;
 import org.wso2.carbon.device.mgt.common.DeviceManagementException;
@@ -41,6 +42,7 @@ import org.wso2.carbon.device.mgt.extensions.device.type.deployer.template.util.
 import org.wso2.carbon.device.mgt.extensions.license.mgt.registry.RegistryBasedLicenseManager;
 import org.wso2.carbon.registry.api.RegistryException;
 import org.wso2.carbon.registry.api.Resource;
+import org.wso2.carbon.utils.multitenancy.MultitenantConstants;
 
 import javax.xml.bind.JAXBContext;
 import javax.xml.bind.JAXBException;
@@ -118,25 +120,34 @@ public class DeviceTypeManager implements DeviceManager {
                     throw new DeviceTypeDeployerFileException("Could not find definition for table: " + tableName);
                 }
                 propertiesExist = true;
-                DeviceDAODefinition deviceDAODefinition = new DeviceDAODefinition(deviceDefinitionTable);
-                String datasourceName = deviceTypeConfiguration.getDataSource().getJndiConfig().getName();
-                if (datasourceName != null && !datasourceName.isEmpty()) {
-                    String setupOption = System.getProperty("setup");
-                    if (setupOption != null) {
-                        if (log.isDebugEnabled()) {
-                            log.debug("-Dsetup is enabled. Device management repository schema initialization is about " +
-                                    "to begin");
-                        }
-                        try {
-                            DeviceTypeUtils.setupDeviceManagementSchema(datasourceName, deviceType,
-                                    deviceDAODefinition.getDeviceTableName());
-                        } catch (DeviceTypeMgtPluginException e) {
-                            log.error("Exception occurred while initializing device management database schema", e);
+
+                try {
+                    PrivilegedCarbonContext.startTenantFlow();
+                    PrivilegedCarbonContext.getThreadLocalCarbonContext()
+                            .setTenantDomain(MultitenantConstants.SUPER_TENANT_DOMAIN_NAME, true);
+
+                    DeviceDAODefinition deviceDAODefinition = new DeviceDAODefinition(deviceDefinitionTable);
+                    String datasourceName = deviceTypeConfiguration.getDataSource().getJndiConfig().getName();
+                    if (datasourceName != null && !datasourceName.isEmpty()) {
+                        String setupOption = System.getProperty("setup");
+                        if (setupOption != null) {
+                            if (log.isDebugEnabled()) {
+                                log.debug("-Dsetup is enabled. Device management repository schema initialization is about " +
+                                        "to begin");
+                            }
+                            try {
+                                DeviceTypeUtils.setupDeviceManagementSchema(datasourceName, deviceType,
+                                        deviceDAODefinition.getDeviceTableName());
+                            } catch (DeviceTypeMgtPluginException e) {
+                                log.error("Exception occurred while initializing device management database schema", e);
+                            }
                         }
+                        deviceTypePluginDAOManager = new DeviceTypePluginDAOManager(datasourceName, deviceDAODefinition);
+                    } else {
+                        throw new DeviceTypeDeployerFileException("Invalid datasource name.");
                     }
-                    deviceTypePluginDAOManager = new DeviceTypePluginDAOManager(datasourceName, deviceDAODefinition);
-                } else {
-                    throw new DeviceTypeDeployerFileException("Invalid datasource name.");
+                } finally {
+                    PrivilegedCarbonContext.endTenantFlow();
                 }
             }
         }