ke.co.tawi.babblesms.server.persistence.maskcode.MaskDAO.java Source code

Introduction

Here is the source code for ke.co.tawi.babblesms.server.persistence.maskcode.MaskDAO.java

Source

/**
 * Copyright 2015 Tawi Commercial Services Ltd
 * 
 * Licensed under the Open Software License, Version 3.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://opensource.org/licenses/OSL-3.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 ke.co.tawi.babblesms.server.persistence.maskcode;

import ke.co.tawi.babblesms.server.beans.account.Account;
import ke.co.tawi.babblesms.server.beans.maskcode.Mask;
import ke.co.tawi.babblesms.server.persistence.GenericDAO;

import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.List;

import org.apache.commons.dbutils.BeanProcessor;
import org.apache.commons.lang3.exception.ExceptionUtils;
import org.apache.log4j.Logger;

/**
 * Persistence implementation for {@link Mask}
 * <p>
 *  
 * @author <a href="mailto:Wambua@tawi.mobi">Godfrey Wambua</a>
 * @author <a href="mailto:michael@tawi.mobi">Michael Wakahe</a>
 */

public class MaskDAO extends GenericDAO implements BabbleMaskDAO {

    private static MaskDAO maskDAO;

    private BeanProcessor beanProcessor = new BeanProcessor();

    private final Logger logger;

    /**
     * @return the singleton instance of {@link MaskDAO}
     */
    public static MaskDAO getInstance() {
        if (maskDAO == null) {
            maskDAO = new MaskDAO();
        }

        return maskDAO;
    }

    /**
     *
     */
    protected MaskDAO() {
        super();
        logger = Logger.getLogger(this.getClass());
    }

    /**
     * Used for testing purposes only.
     *
     * @param dbName
     * @param dbHost
     * @param dbUsername
     * @param dbPassword
     * @param dbPort
     */
    public MaskDAO(String dbName, String dbHost, String dbUsername, String dbPassword, int dbPort) {
        super(dbName, dbHost, dbUsername, dbPassword, dbPort);
        logger = Logger.getLogger(this.getClass());
    }

    /**
     * @see ke.co.tawi.babblesms.server.persistence.maskcode.BabbleMaskDAO#get(java.lang.String)
     */
    @Override
    public Mask get(String uuid) {
        Mask mask = null;

        try (Connection conn = dbCredentials.getConnection();
                PreparedStatement pstmt = conn.prepareStatement("SELECT * FROM Mask WHERE Uuid = ?;");) {
            pstmt.setString(1, uuid);

            try (ResultSet rset = pstmt.executeQuery();) {
                if (rset.next()) {
                    mask = beanProcessor.toBean(rset, Mask.class);
                }
            }

        } catch (SQLException e) {
            logger.error("SQL Exception when getting mask with uuid: " + uuid);
            logger.error(ExceptionUtils.getStackTrace(e));
        }

        return mask;
    }

    /**
     * @see ke.co.tawi.babblesms.server.persistence.maskcode.BabbleMaskDAO#put(ke.co.tawi.babblesms.server.beans.maskcode.Mask)
     */
    @Override
    public boolean put(Mask mask) {
        boolean success = true;

        try (Connection conn = dbCredentials.getConnection();
                PreparedStatement pstmt = conn.prepareStatement(
                        "INSERT INTO Mask (Uuid, maskname," + "accountuuid, networkuuid) VALUES (?,?,?,?);");) {

            pstmt.setString(1, mask.getUuid());
            pstmt.setString(2, mask.getMaskname());
            pstmt.setString(3, mask.getAccountuuid());
            pstmt.setString(4, mask.getNetworkuuid());

            pstmt.execute();

        } catch (SQLException e) {
            logger.error("SQL Exception when trying to put: " + mask);
            logger.error(ExceptionUtils.getStackTrace(e));
            success = false;
        }

        return success;
    }

    /**
     * @see ke.co.tawi.babblesms.server.persistence.maskcode.BabbleMaskDAO#getMasks(ke.co.tawi.babblesms.server.beans.account.Account)
     */
    @Override
    public List<Mask> getMasks(Account account) {
        List<Mask> masklist = null;

        try (Connection conn = dbCredentials.getConnection();
                PreparedStatement pstmt = conn.prepareStatement("SELECT * FROM Mask WHERE accountuuid=?;");) {

            pstmt.setString(1, account.getUuid());

            try (ResultSet rset = pstmt.executeQuery();) {
                masklist = beanProcessor.toBeanList(rset, Mask.class);
            }

        } catch (SQLException e) {
            logger.error("SQL Exception when getting all masks");
            logger.error(ExceptionUtils.getStackTrace(e));
        }

        return masklist;
    }

    /**
     * @see ke.co.tawi.babblesms.server.persistence.maskcode.BabbleMaskDAO#getAllMasks()
     */
    @Override
    public List<Mask> getAllMasks() {
        List<Mask> masklist = null;

        try (Connection conn = dbCredentials.getConnection();
                PreparedStatement pstmt = conn.prepareStatement("SELECT * FROM Mask;");
                ResultSet rset = pstmt.executeQuery();) {

            masklist = beanProcessor.toBeanList(rset, Mask.class);

        } catch (SQLException e) {
            logger.error("SQL Exception when getting all masks");
            logger.error(ExceptionUtils.getStackTrace(e));
        }

        return masklist;
    }

    /**
     * @see ke.co.tawi.babblesms.server.persistence.maskcode.BabbleMaskDAO#updateMask(ke.co.tawi.babblesms.server.beans.maskcode.Mask, java.lang.String)
     */
    @Override
    public boolean updateMask(Mask mask, String uuid) {
        boolean success = true;

        try (Connection conn = dbCredentials.getConnection();
                PreparedStatement pstmt = conn.prepareStatement(
                        "UPDATE Mask SET MASKName=?, accountuuid=?," + "networkuuid=? WHERE Uuid = ?;");) {

            pstmt.setString(1, mask.getMaskname());
            pstmt.setString(2, mask.getAccountuuid());
            pstmt.setString(3, mask.getNetworkuuid());
            pstmt.setString(4, mask.getUuid());

            pstmt.executeUpdate();

        } catch (SQLException e) {
            logger.error("SQL Exception when deleting " + mask + " with uuid " + uuid);
            logger.error(ExceptionUtils.getStackTrace(e));
            success = false;
        }

        return success;
    }

}