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

Introduction

Here is the source code for ke.co.tawi.babblesms.server.persistence.maskcode.ShortcodeDAO.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.Shortcode;
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 Shortcode}
 * <p>
 *  
 * @author <a href="mailto:wambua@tawi.mobi">Godfrey Wambua</a>
 * @author <a href="mailto:michael@tawi.mobi">Michael Wakahe</a>
 */

public class ShortcodeDAO extends GenericDAO implements BabbleShortcodeDAO {

    private static ShortcodeDAO shortcodeDAO;

    private BeanProcessor beanProcessor = new BeanProcessor();

    private final Logger logger;

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

        return shortcodeDAO;
    }

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

    /**
     * Used for testing purposes only.
     *
     * @param dbName
     * @param dbHost
     * @param dbUsername
     * @param dbPassword
     * @param dbPort
     */
    public ShortcodeDAO(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.BabbleShortcodeDAO#get(java.lang.String)
     */
    @Override
    public Shortcode get(String uuid) {
        Shortcode shortcode = null;

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

            pstmt.setString(1, uuid);
            try (ResultSet rset = pstmt.executeQuery();) {

                if (rset.next()) {
                    shortcode = beanProcessor.toBean(rset, Shortcode.class);
                }
            }

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

        return shortcode;
    }

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

        try (Connection conn = dbCredentials.getConnection();
                PreparedStatement pstmt = conn.prepareStatement("INSERT INTO Shortcode (Uuid, codenumber,"
                        + "accountuuid, networkuuid) VALUES (?,?,?,?);");) {
            pstmt.setString(1, shortcode.getUuid());
            pstmt.setString(2, shortcode.getCodenumber());
            pstmt.setString(3, shortcode.getAccountuuid());
            pstmt.setString(4, shortcode.getNetworkuuid());

            pstmt.execute();

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

        return success;
    }

    /**
    *
    */
    @Override
    public List<Shortcode> getShortcodes(Account account) {
        List<Shortcode> list = null;

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

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

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

        } catch (SQLException e) {
            logger.error("SQL Exception when getting shortcodes of " + account);
            logger.error(ExceptionUtils.getStackTrace(e));
        }

        return list;
    }

    /**
    * @see ke.co.tawi.babblesms.server.persistence.maskcode.BabbleShortcodeDAO#update(Shortcode, String)
    */
    @Override
    public boolean update(Shortcode shortcode, String uuid) {
        boolean success = true;

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

            pstmt.setString(1, shortcode.getCodenumber());
            pstmt.setString(2, shortcode.getAccountuuid());
            pstmt.setString(3, shortcode.getNetworkuuid());
            pstmt.setString(4, shortcode.getUuid());

            pstmt.executeUpdate();

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

        return success;
    }

    /**
     * @see ke.co.tawi.babblesms.server.persistence.maskcode.BabbleShortcodeDAO#getAllShortcodes()
     */
    @Override
    public List<Shortcode> getAllShortcodes() {
        List<Shortcode> list = null;

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

            list = beanProcessor.toBeanList(rset, Shortcode.class);

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

        return list;
    }

}