userdao.java

java web开发者突击程序 Struts+Hibernate实例源码

package com.demo.hibernate.dao;

import org.hibernate.HibernateException;
import org.hibernate.Query;
import org.hibernate.Session;
import org.hibernate.Transaction;

import com.demo.hibernate.beans.User;
import com.demo.hibernate.util.HibernateSessionFactory;

public class UserDAO {

	public User valid(String username, String password)
			throws HibernateException {
		Session session = null;
		Transaction tx = null;
		User user = null;
		try {
			session = HibernateSessionFactory.currentSession();
			tx = session.beginTransaction();

			Query query = session
					.createQuery("from User where username=? and password=?");
			query.setString(0, username.trim());
			query.setString(1, password.trim());

			user = (User) query.uniqueResult();

			query = null;

			tx.commit();
		} catch (HibernateException e) {
			throw e;
		} finally {
			if (tx != null) {
				tx.rollback();
			}
			HibernateSessionFactory.closeSession();
		}
		return user;
	}

	public boolean isExist(String username) throws HibernateException {
		Session session = null;
		Transaction tx = null;
		boolean b = false;
		try {
			session = HibernateSessionFactory.currentSession();
			tx = session.beginTransaction();

			Query query = session.createQuery("from User where username=?");
			query.setString(0, username.trim());

			User user = (User) query.uniqueResult();
			if (user != null)
				b = true;
			query = null;

			tx.commit();
		} catch (HibernateException e) {
			throw e;
		} finally {
			if (tx != null) {
				tx.rollback();
			}
			HibernateSessionFactory.closeSession();
		}
		return b;
	}

	public boolean add(String username, String password, String email) throws HibernateException {
		Session session = null;
		Transaction tx = null;
		boolean b = true;
		try {
			session = HibernateSessionFactory.currentSession();
			tx = session.beginTransaction();
			
			User user = new User();
			user.setUsername(username);
			user.setPassword(password);
			user.setEmail(email);
			
			session.save(user);
			tx.commit();
		} catch (HibernateException e) {
			b = false;
			throw e;
		} finally {
			if (tx != null) {
				tx.rollback();
			}
			HibernateSessionFactory.closeSession();
		}
		return b;
	}

	public static void main(String[] args) {
		UserDAO service = new UserDAO();
		User user = service.valid("admin", "admin");
		System.out.println(user.getEmail());
		System.out.println(service.isExist("admin"));
		System.out.println(service.isExist("ddd"));
		System.out.println(service.add("test2","123","[e-mail]"));

	}

}