messagedao.java

ajax博客系统,该文件内容部分综合网上的资源,自己进行了改进,转载请注明 汪建伟

package cn.hxex.blog.dao.hibernate;

import java.util.List;
import java.util.Iterator;

import cn.hxex.blog.dao.IMessageDAO;
import cn.hxex.blog.exception.BlogDAOException;
import cn.hxex.blog.hibernate.HibernateDAO;
import cn.hxex.blog.model.Message;
import cn.hxex.blog.model.User;
import cn.hxex.blog.model.ReplyMessage;

public class MessageDAO extends HibernateDAO implements IMessageDAO {

	public void deleteMessage(String id, String userId) {
		User user = (User) super.getObject(User.class, userId);
		if (user == null) {
			throw new BlogDAOException("Couldn't user!");
		}
		Message message = null;
		Iterator it = user.getMessages().iterator();
		while (it.hasNext()) {
			Message m = (Message) it.next();
			if(m.getId().equals(id))
			{
				message = m;
			}
		}
		if (message != null) {
			super.deleteObject(message);
			user.getMessages().remove(message);
		}
	}

	public List getMessages() {
		List list;
		String hsql = "from Message";
		list = super.getObjects(hsql);
		return list;
	}

	public Message getMessage(String id) {
		Message message = (Message) super.getObject(Message.class, id);
		return message;
	}

	public void saveMessage(Message message) {
		if(message == null)
		{
			return;
		}
		super.saveObject(message);
	}

	public void updateMessage(Message message) {
		if(message != null)
		{
			super.updateObject(message);
		}
	}

	public void deleteReplyMessage(String id, String userId) {
		ReplyMessage replyMessage = (ReplyMessage) super.getObject(ReplyMessage.class, id);
		if (replyMessage == null) {
			throw new BlogDAOException("找不到你所要删除得日志评论！");
		}
		Message message = replyMessage.getMessage();
		Iterator it = message.getReplies().iterator();
		ReplyMessage r = null;
		while (it.hasNext()) {
			ReplyMessage rm = (ReplyMessage) it.next();
			if (rm.getId().equals(id)) {
				r = rm;
			}
		}
		if (r != null) {
			message.getReplies().remove(r);
		}
	}

}