GenericPrivateMessageDAO.java :  » Forum » JForum-2.1.8 » net » jforum » dao » generic » Java Open Source

Java Open Source » Forum » JForum 2.1.8 
JForum 2.1.8 » net » jforum » dao » generic » GenericPrivateMessageDAO.java
/*
 * Copyright (c) JForum Team
 * All rights reserved.
 * 
 * Redistribution and use in source and binary forms, 
 * with or without modification, are permitted provided 
 * that the following conditions are met:
 * 
 * 1) Redistributions of source code must retain the above 
 * copyright notice, this list of conditions and the 
 * following  disclaimer.
 * 2)  Redistributions in binary form must reproduce the 
 * above copyright notice, this list of conditions and 
 * the following disclaimer in the documentation and/or 
 * other materials provided with the distribution.
 * 3) Neither the name of "Rafael Steil" nor 
 * the names of its contributors may be used to endorse 
 * or promote products derived from this software without 
 * specific prior written permission.
 * 
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT 
 * HOLDERS AND CONTRIBUTORS "AS IS" AND ANY 
 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, 
 * BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF 
 * MERCHANTABILITY AND FITNESS FOR A PARTICULAR 
 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL 
 * THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE 
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, 
 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES 
 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF 
 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, 
 * OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER 
 * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER 
 * IN CONTRACT, STRICT LIABILITY, OR TORT 
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN 
 * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF 
 * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE
 * 
 * This file creation date: 20/05/2004 - 15:51:10
 * The JForum Project
 * http://www.jforum.net
 */
package net.jforum.dao.generic;

import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Timestamp;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;

import net.jforum.JForumExecutionContext;
import net.jforum.dao.DataAccessDriver;
import net.jforum.dao.UserDAO;
import net.jforum.entities.Post;
import net.jforum.entities.PrivateMessage;
import net.jforum.entities.PrivateMessageType;
import net.jforum.entities.User;
import net.jforum.exceptions.DatabaseException;
import net.jforum.util.DbUtils;
import net.jforum.util.preferences.ConfigKeys;
import net.jforum.util.preferences.SystemGlobals;

/**
 * @author Rafael Steil
 * @version $Id: GenericPrivateMessageDAO.java,v 1.12 2007/08/31 22:56:40 rafaelsteil Exp $
 */
public class GenericPrivateMessageDAO extends AutoKeys implements net.jforum.dao.PrivateMessageDAO
{
  /**
   * @see net.jforum.dao.PrivateMessageDAO#send(net.jforum.entities.PrivateMessage)
   */
  public void send(PrivateMessage pm)
  {
    // We should store 2 copies: one for the sendee's sent box
    // and another for the target user's inbox.
    PreparedStatement p = null;
    
    try {
      p = this.getStatementForAutoKeys("PrivateMessageModel.add");

      // Sendee's sent box
      this.addPm(pm, p);
      this.addPmText(pm);

      // Target user's inbox
      p.setInt(1, PrivateMessageType.NEW);
      pm.setId(this.executeAutoKeysQuery(p));

      this.addPmText(pm);
    }
    catch (Exception e) {
      throw new DatabaseException(e);
    }
    finally {
      DbUtils.close(p);
    }
  }

  protected void addPmText(PrivateMessage pm) throws Exception
  {
    PreparedStatement text = JForumExecutionContext.getConnection().prepareStatement(
      SystemGlobals.getSql("PrivateMessagesModel.addText"));

    text.setInt(1, pm.getId());
    text.setString(2, pm.getPost().getText());
    text.executeUpdate();

    text.close();
  }

  protected void addPm(PrivateMessage pm, PreparedStatement p) throws SQLException
  {
    p.setInt(1, PrivateMessageType.SENT);
    p.setString(2, pm.getPost().getSubject());
    p.setInt(3, pm.getFromUser().getId());
    p.setInt(4, pm.getToUser().getId());
    p.setTimestamp(5, new Timestamp(pm.getPost().getTime().getTime()));
    p.setInt(6, pm.getPost().isBbCodeEnabled() ? 1 : 0);
    p.setInt(7, pm.getPost().isHtmlEnabled() ? 1 : 0);
    p.setInt(8, pm.getPost().isSmiliesEnabled() ? 1 : 0);
    p.setInt(9, pm.getPost().isSignatureEnabled() ? 1 : 0);

    this.setAutoGeneratedKeysQuery(SystemGlobals.getSql("PrivateMessagesModel.lastGeneratedPmId"));
    pm.setId(this.executeAutoKeysQuery(p));
  }

  /**
   * @see net.jforum.dao.PrivateMessageDAO#delete(net.jforum.entities.PrivateMessage[], int)
   */
  public void delete(PrivateMessage[] pm, int userId)
  {
    PreparedStatement deleteMessage = null;
    PreparedStatement deleteText = null;
    PreparedStatement isDeleteAllowed = null;
    
    try {
      Connection connection = JForumExecutionContext.getConnection();
      
      deleteMessage = connection.prepareStatement(SystemGlobals.getSql("PrivateMessageModel.delete"));
      deleteText = connection.prepareStatement(SystemGlobals.getSql("PrivateMessagesModel.deleteText"));
      
      isDeleteAllowed = connection.prepareStatement(SystemGlobals.getSql("PrivateMessagesModel.isDeleteAllowed"));
      isDeleteAllowed.setInt(2, userId);
      isDeleteAllowed.setInt(3, userId);

      for (int i = 0; i < pm.length; i++) {
        PrivateMessage currentMessage = pm[i];
        
        isDeleteAllowed.setInt(1, currentMessage.getId());

        ResultSet rs = null;
        
        try {
          rs = isDeleteAllowed.executeQuery();
          
          if (rs.next()) {
            deleteText.setInt(1, currentMessage.getId());
            deleteText.executeUpdate();
    
            deleteMessage.setInt(1, currentMessage.getId());
            deleteMessage.executeUpdate();
          }
        }
        finally {
          DbUtils.close(rs);
        }
      }
    }
    catch (SQLException e) {
      throw new DatabaseException(e);
    }
    finally {
      DbUtils.close(deleteMessage);
      DbUtils.close(deleteText);
      DbUtils.close(isDeleteAllowed);
    }
  }

  /**
   * @see net.jforum.dao.PrivateMessageDAO#selectFromInbox(net.jforum.entities.User)
   */
  public List selectFromInbox(User user)
  {
    String query = SystemGlobals.getSql("PrivateMessageModel.baseListing");
    query = query.replaceAll("#FILTER#", SystemGlobals.getSql("PrivateMessageModel.inbox"));

    PreparedStatement p = null;
    ResultSet rs = null;
    try {
      p = JForumExecutionContext.getConnection().prepareStatement(query);
      p.setInt(1, user.getId());

      List pmList = new ArrayList();

      rs = p.executeQuery();
      while (rs.next()) {
        PrivateMessage pm = this.getPm(rs, false);

        User fromUser = new User();
        fromUser.setId(rs.getInt("user_id"));
        fromUser.setUsername(rs.getString("username"));

        pm.setFromUser(fromUser);

        pmList.add(pm);
      }

      return pmList;
    }
    catch (SQLException e) {
      throw new DatabaseException(e);
    }
    finally {
      DbUtils.close(rs, p);
    }
  }

  /**
   * @see net.jforum.dao.PrivateMessageDAO#selectFromSent(net.jforum.entities.User)
   */
  public List selectFromSent(User user)
  {
    String query = SystemGlobals.getSql("PrivateMessageModel.baseListing");
    query = query.replaceAll("#FILTER#", SystemGlobals.getSql("PrivateMessageModel.sent"));

    PreparedStatement p = null;
    ResultSet rs = null;
    try {
      p = JForumExecutionContext.getConnection().prepareStatement(query);
      p.setInt(1, user.getId());

      List pmList = new ArrayList();

      rs = p.executeQuery();
      while (rs.next()) {
        PrivateMessage pm = this.getPm(rs, false);

        User toUser = new User();
        toUser.setId(rs.getInt("user_id"));
        toUser.setUsername(rs.getString("username"));

        pm.setToUser(toUser);

        pmList.add(pm);
      }
      return pmList;
    }
    catch (SQLException e) {
      throw new DatabaseException(e);
    }
    finally {
      DbUtils.close(rs, p);
    }
  }

  protected PrivateMessage getPm(ResultSet rs) throws SQLException
  {
    return this.getPm(rs, true);
  }

  protected PrivateMessage getPm(ResultSet rs, boolean full) throws SQLException
  {
    PrivateMessage pm = new PrivateMessage();
    Post p = new Post();

    pm.setId(rs.getInt("privmsgs_id"));
    pm.setType(rs.getInt("privmsgs_type"));
    p.setTime(new Date(rs.getTimestamp("privmsgs_date").getTime()));
    p.setSubject(rs.getString("privmsgs_subject"));

    SimpleDateFormat df = new SimpleDateFormat(SystemGlobals.getValue(ConfigKeys.DATE_TIME_FORMAT));
    pm.setFormatedDate(df.format(p.getTime()));

    if (full) {
      UserDAO um = DataAccessDriver.getInstance().newUserDAO();
      pm.setFromUser(um.selectById(rs.getInt("privmsgs_from_userid")));
      pm.setToUser(um.selectById(rs.getInt("privmsgs_to_userid")));

      p.setBbCodeEnabled(rs.getInt("privmsgs_enable_bbcode") == 1);
      p.setSignatureEnabled(rs.getInt("privmsgs_attach_sig") == 1);
      p.setHtmlEnabled(rs.getInt("privmsgs_enable_html") == 1);
      p.setSmiliesEnabled(rs.getInt("privmsgs_enable_smilies") == 1);
      p.setText(this.getPmText(rs));
    }

    pm.setPost(p);

    return pm;
  }

  protected String getPmText(ResultSet rs) throws SQLException
  {
    return rs.getString("privmsgs_text");
  }

  /**
   * @see net.jforum.dao.PrivateMessageDAO#selectById(net.jforum.entities.PrivateMessage)
   */
  public PrivateMessage selectById(PrivateMessage pm)
  {
    PreparedStatement p = null;
    ResultSet rs = null;
    try {
      p = JForumExecutionContext.getConnection().prepareStatement(
          SystemGlobals.getSql("PrivateMessageModel.selectById"));
      p.setInt(1, pm.getId());

      rs = p.executeQuery();
      if (rs.next()) {
        pm = this.getPm(rs);
      }

      return pm;
    }
    catch (SQLException e) {
      throw new DatabaseException(e);
    }
    finally {
      DbUtils.close(rs, p);
    }
  }

  /**
   * @see net.jforum.dao.PrivateMessageDAO#updateType(net.jforum.entities.PrivateMessage)
   */
  public void updateType(PrivateMessage pm)
  {
    PreparedStatement p = null;
    try {
      p = JForumExecutionContext.getConnection().prepareStatement(
          SystemGlobals.getSql("PrivateMessageModel.updateType"));
      p.setInt(1, pm.getType());
      p.setInt(2, pm.getId());
      p.executeUpdate();
    }
    catch (SQLException e) {
      throw new DatabaseException(e);
    }
    finally {
      DbUtils.close(p);
    }
  }
}
java2s.com  | Contact Us | Privacy Policy
Copyright 2009 - 12 Demo Source and Support. All rights reserved.
All other trademarks are property of their respective owners.