ProfileManagerImpl.java

/*
*  Copyright (c) 2015 WSO2 Inc. (http://www.wso2.org) All Rights Reserved.
*
*  WSO2 Inc. 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.policy.mgt.core.mgt.impl;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.wso2.carbon.device.mgt.core.dao.DeviceDAO;
import org.wso2.carbon.device.mgt.core.dao.DeviceManagementDAOException;
import org.wso2.carbon.device.mgt.core.dao.DeviceManagementDAOFactory;
import org.wso2.carbon.device.mgt.core.dao.DeviceTypeDAO;
import org.wso2.carbon.device.mgt.core.dto.DeviceType;
import org.wso2.carbon.policy.mgt.common.Profile;
import org.wso2.carbon.policy.mgt.common.ProfileFeature;
import org.wso2.carbon.policy.mgt.common.ProfileManagementException;
import org.wso2.carbon.policy.mgt.core.dao.*;
import org.wso2.carbon.policy.mgt.core.mgt.ProfileManager;

import java.sql.Timestamp;
import java.util.ArrayList;
import java.util.Calendar;
import java.util.List;

public class ProfileManagerImpl implements ProfileManager {

    private static Log log = LogFactory.getLog(ProfileManagerImpl.class);
    private ProfileDAO profileDAO;
    private FeatureDAO featureDAO;
    private DeviceDAO deviceDAO;
    private DeviceTypeDAO deviceTypeDAO;

    public ProfileManagerImpl() {
        profileDAO = PolicyManagementDAOFactory.getProfileDAO();
        featureDAO = PolicyManagementDAOFactory.getFeatureDAO();
        deviceDAO = DeviceManagementDAOFactory.getDeviceDAO();
        deviceTypeDAO = DeviceManagementDAOFactory.getDeviceTypeDAO();
    }

    @Override
    public Profile addProfile(Profile profile) throws ProfileManagementException {

        Timestamp currentTimestamp = new Timestamp(Calendar.getInstance().getTime().getTime());
        profile.setCreatedDate(currentTimestamp);
        profile.setUpdatedDate(currentTimestamp);

        try {
            PolicyManagementDAOFactory.beginTransaction();
            profile = profileDAO.addProfile(profile);
            featureDAO.addProfileFeatures(profile.getProfileFeaturesList(), profile.getProfileId());
            PolicyManagementDAOFactory.commitTransaction();
        } catch (ProfileManagerDAOException e) {
            try {
                PolicyManagementDAOFactory.rollbackTransaction();
            } catch (PolicyManagerDAOException e1) {
                log.warn("Error occurred while roll backing the transaction.");
            }
            String msg = "Error occurred while adding the profile (" + profile.getProfileName() + ")";
            log.error(msg, e);
            throw new ProfileManagementException(msg, e);
        } catch (FeatureManagerDAOException e) {
            try {
                PolicyManagementDAOFactory.rollbackTransaction();
            } catch (PolicyManagerDAOException e1) {
                log.warn("Error occurred while roll backing the transaction.");
            }
            String msg = "Error occurred while adding the profile features (" + profile.getProfileName() + ")";
            log.error(msg, e);
            throw new ProfileManagementException(msg, e);
        } catch (PolicyManagerDAOException e) {
            try {
                PolicyManagementDAOFactory.rollbackTransaction();
            } catch (PolicyManagerDAOException e1) {
                log.warn("Error occurred while roll backing the transaction.");
            }
            String msg = "Error occurred while adding the profile (" + profile.getProfileName() + ") to the database";
            log.error(msg, e);
            throw new ProfileManagementException(msg, e);
        }

        return profile;
    }

    @Override
    public Profile updateProfile(Profile profile) throws ProfileManagementException {

        Timestamp currentTimestamp = new Timestamp(Calendar.getInstance().getTime().getTime());
        profile.setUpdatedDate(currentTimestamp);

        try {
            PolicyManagementDAOFactory.beginTransaction();
            profile = profileDAO.updateProfile(profile);
            featureDAO.updateProfileFeatures(profile.getProfileFeaturesList(), profile.getProfileId());
            PolicyManagementDAOFactory.commitTransaction();
        } catch (ProfileManagerDAOException e) {
            try {
                PolicyManagementDAOFactory.rollbackTransaction();
            } catch (PolicyManagerDAOException e1) {
                log.warn("Error occurred while roll backing the transaction.");
            }
            String msg = "Error occurred while updating the profile (" + profile.getProfileName() + ")";
            log.error(msg, e);
            throw new ProfileManagementException(msg, e);
        } catch (FeatureManagerDAOException e) {
            try {
                PolicyManagementDAOFactory.rollbackTransaction();
            } catch (PolicyManagerDAOException e1) {
                log.warn("Error occurred while roll backing the transaction.");
            }
            String msg = "Error occurred while updating the profile features (" + profile.getProfileName() + ")";
            log.error(msg, e);
            throw new ProfileManagementException(msg, e);
        } catch (PolicyManagerDAOException e) {
            try {
                PolicyManagementDAOFactory.rollbackTransaction();
            } catch (PolicyManagerDAOException e1) {
                log.warn("Error occurred while roll backing the transaction.");
            }
            String msg = "Error occurred while updating the profile (" + profile.getProfileName() + ") to the database";
            log.error(msg, e);
            throw new ProfileManagementException(msg, e);
        }

        return profile;
    }

    @Override
    public boolean deleteProfile(Profile profile) throws ProfileManagementException {
        boolean bool = true;
        try {
            PolicyManagementDAOFactory.beginTransaction();
            featureDAO.deleteFeaturesOfProfile(profile);
            bool = profileDAO.deleteProfile(profile);
            PolicyManagementDAOFactory.commitTransaction();
        } catch (ProfileManagerDAOException e) {
            try {
                PolicyManagementDAOFactory.rollbackTransaction();
            } catch (PolicyManagerDAOException e1) {
                log.warn("Error occurred while roll backing the transaction.");
            }
            String msg = "Error occurred while deleting the profile (" + profile.getProfileName() + ")";
            log.error(msg, e);
            throw new ProfileManagementException(msg, e);
        } catch (FeatureManagerDAOException e) {
            try {
                PolicyManagementDAOFactory.rollbackTransaction();
            } catch (PolicyManagerDAOException e1) {
                log.warn("Error occurred while roll backing the transaction.");
            }
            String msg = "Error occurred while deleting the features from profile (" + profile.getProfileName() + ")";
            log.error(msg, e);
            throw new ProfileManagementException(msg, e);
        } catch (PolicyManagerDAOException e) {
            try {
                PolicyManagementDAOFactory.rollbackTransaction();
            } catch (PolicyManagerDAOException e1) {
                log.warn("Error occurred while roll backing the transaction.");
            }
            String msg = "Error occurred while deleting the profile (" + profile.getProfileName() + ") from database";
            log.error(msg, e);
            throw new ProfileManagementException(msg, e);
        }
        return bool;
    }

    @Override
    public Profile getProfile(int profileId) throws ProfileManagementException {
        Profile profile;
        List<ProfileFeature> featureList;
        DeviceType deviceType;

        try {
            profile = profileDAO.getProfiles(profileId);
            featureList = featureDAO.getFeaturesForProfile(profileId);
            deviceType = deviceTypeDAO.getDeviceType(profile.getDeviceType().getId());

            profile.setProfileFeaturesList(featureList);
            profile.setDeviceType(deviceType);

        } catch (ProfileManagerDAOException e) {
            String msg = "Error occurred while getting profile id (" + profileId + ")";
            log.error(msg, e);
            throw new ProfileManagementException(msg, e);
        } catch (FeatureManagerDAOException e) {
            String msg = "Error occurred while getting features related profile id (" + profileId + ")";
            log.error(msg, e);
            throw new ProfileManagementException(msg, e);
        } catch (DeviceManagementDAOException e) {
            String msg = "Error occurred while getting device type related profile id (" + profileId + ")";
            log.error(msg, e);
            throw new ProfileManagementException(msg, e);
        }
        return profile;
    }

    @Override
    public List<Profile> getAllProfiles() throws ProfileManagementException {
        List<Profile> profileList;
        try {
            profileList = profileDAO.getAllProfiles();
            List<ProfileFeature> featureList = featureDAO.getAllProfileFeatures();
            List<DeviceType> deviceTypes = deviceTypeDAO.getDeviceTypes();
            for (Profile profile : profileList) {

                List<ProfileFeature> list = new ArrayList<ProfileFeature>();
                for (ProfileFeature profileFeature : featureList) {
                    if (profile.getProfileId() == profileFeature.getProfileId()) {
                        list.add(profileFeature);
                    }
                }
                profile.setProfileFeaturesList(list);

                for (DeviceType deviceType : deviceTypes) {
                    if (profile.getDeviceType().getId() == deviceType.getId()) {
                        profile.setDeviceType(deviceType);
                    }
                }
            }
        } catch (ProfileManagerDAOException e) {
            String msg = "Error occurred while getting profiles";
            log.error(msg, e);
            throw new ProfileManagementException(msg, e);
        } catch (FeatureManagerDAOException e) {
            String msg = "Error occurred while getting features related to profiles";
            log.error(msg, e);
            throw new ProfileManagementException(msg, e);
        } catch (DeviceManagementDAOException e) {
            String msg = "Error occurred while getting device types related to profiles";
            log.error(msg, e);
            throw new ProfileManagementException(msg, e);
        }
        return profileList;
    }

    @Override
    public List<Profile> getProfilesOfDeviceType(String deviceTypeName) throws ProfileManagementException {
        List<Profile> profileList;
        List<ProfileFeature> featureList;
        try {
            DeviceType deviceType = deviceTypeDAO.getDeviceType(deviceTypeName);
            profileList = profileDAO.getProfilesOfDeviceType(deviceType);
            featureList = featureDAO.getAllProfileFeatures();

            for (Profile profile : profileList) {
                List<ProfileFeature> profileFeatureList = new ArrayList<ProfileFeature>();
                for (ProfileFeature profileFeature : featureList) {
                    if (profile.getProfileId() == profileFeature.getProfileId()) {
                        profileFeatureList.add(profileFeature);
                    }
                }
                profile.setProfileFeaturesList(profileFeatureList);

            }
        } catch (ProfileManagerDAOException e) {
            String msg = "Error occurred while getting profiles";
            log.error(msg, e);
            throw new ProfileManagementException(msg, e);
        } catch (DeviceManagementDAOException e) {
            String msg = "Error occurred while getting device types";
            log.error(msg, e);
            throw new ProfileManagementException(msg, e);
        } catch (FeatureManagerDAOException e) {
            String msg = "Error occurred while getting profile features types";
            log.error(msg, e);
            throw new ProfileManagementException(msg, e);
        }
        return profileList;
    }
}