Refractor Operations to handle Policy Operations and remove joins

9 years ago · c3805364c9
parent 917c0cf965
commit c3805364c9
2 changed files with 274 additions and 0 deletions
--- a/components/device-mgt/org.wso2.carbon.device.mgt.core/src/main/java/org/wso2/carbon/device/mgt/core/operation/mgt/dao/impl/PolicyOperationDAOImpl.java
+++ b/components/device-mgt/org.wso2.carbon.device.mgt.core/src/main/java/org/wso2/carbon/device/mgt/core/operation/mgt/dao/impl/PolicyOperationDAOImpl.java
@ -0,0 +1,240 @@
+/*
+ *   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.device.mgt.core.operation.mgt.dao.impl;
+
+import org.apache.commons.logging.Log;
+import org.apache.commons.logging.LogFactory;
+import org.wso2.carbon.device.mgt.core.dto.operation.mgt.Operation;
+import org.wso2.carbon.device.mgt.core.dto.operation.mgt.PolicyOperation;
+import org.wso2.carbon.device.mgt.core.dto.operation.mgt.ProfileOperation;
+import org.wso2.carbon.device.mgt.core.operation.mgt.dao.OperationManagementDAOException;
+import org.wso2.carbon.device.mgt.core.operation.mgt.dao.OperationManagementDAOFactory;
+import org.wso2.carbon.device.mgt.core.operation.mgt.dao.OperationManagementDAOUtil;
+
+import java.io.*;
+import java.sql.*;
+import java.util.ArrayList;
+import java.util.List;
+
+public class PolicyOperationDAOImpl extends OperationDAOImpl {
+
+    private static final Log log = LogFactory.getLog(PolicyOperationDAOImpl.class);
+
+    @Override
+    public int addOperation(Operation operation) throws OperationManagementDAOException {
+
+        int operationId = super.addOperation(operation);
+        operation.setCreatedTimeStamp(new Timestamp(new java.util.Date().getTime()).toString());
+        operation.setId(operationId);
+        operation.setEnabled(true);
+        PolicyOperation policyOperation = (PolicyOperation) operation;
+        Connection conn = OperationManagementDAOFactory.getConnection();
+
+        PreparedStatement stmt = null;
+
+        try {
+            stmt = conn.prepareStatement("INSERT INTO DM_POLICY_OPERATION(OPERATION_ID, OPERATION_DETAILS) " +
+                    "VALUES(?, ?)");
+            stmt.setInt(1, operationId);
+            stmt.setObject(2, policyOperation);
+            stmt.executeUpdate();
+        } catch (SQLException e) {
+            throw new OperationManagementDAOException("Error occurred while adding policy operation", e);
+        } finally {
+            OperationManagementDAOUtil.cleanupResources(stmt);
+        }
+        return operationId;
+
+    }
+
+    @Override
+    public void updateOperation(Operation operation) throws OperationManagementDAOException {
+
+        PreparedStatement stmt = null;
+        ByteArrayOutputStream bao = null;
+        ObjectOutputStream oos = null;
+        super.updateOperation(operation);
+
+        try {
+            Connection connection = OperationManagementDAOFactory.getConnection();
+            stmt = connection.prepareStatement("UPDATE DM_POLICY_OPERATION O SET O.OPERATION_DETAILS=? " +
+                    "WHERE O.OPERATION_ID=?");
+
+            bao = new ByteArrayOutputStream();
+            oos = new ObjectOutputStream(bao);
+            oos.writeObject(operation);
+
+            stmt.setBytes(1, bao.toByteArray());
+            stmt.setInt(2, operation.getId());
+            stmt.executeUpdate();
+
+        } catch (SQLException e) {
+            throw new OperationManagementDAOException("Error occurred while update policy operation metadata", e);
+        } catch (IOException e) {
+            throw new OperationManagementDAOException("Error occurred while serializing policy operation object", e);
+        } finally {
+            if (bao != null) {
+                try {
+                    bao.close();
+                } catch (IOException e) {
+                    log.warn("Error occurred while closing ByteArrayOutputStream", e);
+                }
+            }
+            if (oos != null) {
+                try {
+                    oos.close();
+                } catch (IOException e) {
+                    log.warn("Error occurred while closing ObjectOutputStream", e);
+                }
+            }
+            OperationManagementDAOUtil.cleanupResources(stmt);
+        }
+    }
+
+    @Override
+    public void deleteOperation(int operationId) throws OperationManagementDAOException {
+
+        super.deleteOperation(operationId);
+        PreparedStatement stmt = null;
+        try {
+            Connection connection = OperationManagementDAOFactory.getConnection();
+            stmt = connection.prepareStatement("DELETE DM_POLICY_OPERATION WHERE OPERATION_ID=?");
+            stmt.setInt(1, operationId);
+            stmt.executeUpdate();
+        } catch (SQLException e) {
+            throw new OperationManagementDAOException("Error occurred while deleting operation metadata", e);
+        } finally {
+            OperationManagementDAOUtil.cleanupResources(stmt);
+        }
+    }
+
+    @Override
+    public Operation getOperation(int operationId) throws OperationManagementDAOException {
+
+        PreparedStatement stmt = null;
+        ResultSet rs = null;
+        PolicyOperation policyOperation = null;
+
+        ByteArrayInputStream bais;
+        ObjectInputStream ois;
+
+        try {
+            Connection conn = OperationManagementDAOFactory.getConnection();
+            String sql = "SELECT OPERATION_ID, ENABLED, OPERATION_DETAILS FROM DM_POLICY_OPERATION WHERE OPERATION_ID=?";
+
+            stmt = conn.prepareStatement(sql);
+            stmt.setInt(1, operationId);
+            rs = stmt.executeQuery();
+
+            if (rs.next()) {
+                byte[] operationDetails = rs.getBytes("OPERATION_DETAILS");
+                bais = new ByteArrayInputStream(operationDetails);
+                ois = new ObjectInputStream(bais);
+                policyOperation = (PolicyOperation) ois.readObject();
+            }
+
+        } catch (IOException e) {
+            String errorMsg = "IO Error occurred while de serialize the policy operation object";
+            log.error(errorMsg, e);
+            throw new OperationManagementDAOException(errorMsg, e);
+        } catch (ClassNotFoundException e) {
+            String errorMsg = "Class not found error occurred while de serialize the policy operation object";
+            log.error(errorMsg, e);
+            throw new OperationManagementDAOException(errorMsg, e);
+        } catch (SQLException e) {
+            String errorMsg = "SQL Error occurred while retrieving the policy operation object " + "available for " +
+                    "the id '"
+                    + operationId;
+            log.error(errorMsg, e);
+            throw new OperationManagementDAOException(errorMsg, e);
+        } finally {
+            OperationManagementDAOUtil.cleanupResources(stmt, rs);
+            OperationManagementDAOFactory.closeConnection();
+        }
+        return policyOperation;
+    }
+
+    @Override
+    public List<? extends Operation> getOperationsByDeviceAndStatus(int deviceId,
+            Operation.Status status) throws OperationManagementDAOException {
+
+        PreparedStatement stmt = null;
+        ResultSet rs = null;
+        PolicyOperation policyOperation;
+
+        List<Operation> operationList = new ArrayList<Operation>();
+
+        ByteArrayInputStream bais = null;
+        ObjectInputStream ois = null;
+
+        try {
+            Connection conn = OperationManagementDAOFactory.getConnection();
+            String sql = "Select OPERATION_ID, ENABLED, OPERATION_DETAILS from DM_POLICY_OPERATION po " +
+                    "INNER JOIN  " +
+                    "(Select * From DM_DEVICE_OPERATION_MAPPING WHERE DEVICE_ID=? " +
+                    "AND STATUS=?) dm ON dm.OPERATION_ID = po.OPERATION_ID";
+
+            stmt = conn.prepareStatement(sql);
+            stmt.setInt(1, deviceId);
+            stmt.setString(2, status.toString());
+
+            rs = stmt.executeQuery();
+
+            while (rs.next()) {
+                byte[] operationDetails = rs.getBytes("OPERATION_DETAILS");
+                bais = new ByteArrayInputStream(operationDetails);
+                ois = new ObjectInputStream(bais);
+                policyOperation = (PolicyOperation) ois.readObject();
+                operationList.add(policyOperation);
+            }
+
+        } catch (IOException e) {
+            String errorMsg = "IO Error occurred while de serialize the profile operation object";
+            log.error(errorMsg, e);
+            throw new OperationManagementDAOException(errorMsg, e);
+        } catch (ClassNotFoundException e) {
+            String errorMsg = "Class not found error occurred while de serialize the profile operation object";
+            log.error(errorMsg, e);
+            throw new OperationManagementDAOException(errorMsg, e);
+        } catch (SQLException e) {
+            String errorMsg = "SQL error occurred while retrieving the operation available for the device'" + deviceId +
+                    "' with status '" + status.toString();
+            log.error(errorMsg);
+            throw new OperationManagementDAOException(errorMsg, e);
+        } finally {
+            if (bais != null) {
+                try {
+                    bais.close();
+                } catch (IOException e) {
+                    log.warn("Error occurred while closing ByteArrayOutputStream", e);
+                }
+            }
+            if (ois != null) {
+                try {
+                    ois.close();
+                } catch (IOException e) {
+                    log.warn("Error occurred while closing ObjectOutputStream", e);
+                }
+            }
+            OperationManagementDAOUtil.cleanupResources(stmt, rs);
+            OperationManagementDAOFactory.closeConnection();
+        }
+        return operationList;
+    }
+}
--- a/components/device-mgt/org.wso2.carbon.device.mgt.core/src/main/java/org/wso2/carbon/device/mgt/core/operation/mgt/util/OperationCreateTimeComparator.java
+++ b/components/device-mgt/org.wso2.carbon.device.mgt.core/src/main/java/org/wso2/carbon/device/mgt/core/operation/mgt/util/OperationCreateTimeComparator.java
@ -0,0 +1,34 @@
+/*
+ *   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.device.mgt.core.operation.mgt.util;
+
+import org.wso2.carbon.device.mgt.common.operation.mgt.Operation;
+import java.util.Comparator;
+
+public class OperationCreateTimeComparator  implements Comparator<Operation>{
+
+    @Override
+    public int compare(Operation o1, Operation o2) {
+        long createdTime1 = java.sql.Timestamp.valueOf(o1.getCreatedTimeStamp()).getTime();
+        long createdTime2 = java.sql.Timestamp.valueOf(o1.getCreatedTimeStamp()).getTime();
+        return (int) (createdTime1 - createdTime2);
+    }
+
+
+}