net.sasasin.sreader.commons.dao.impl.GenericDaoHibernateImpl.java Source code

Introduction

Here is the source code for net.sasasin.sreader.commons.dao.impl.GenericDaoHibernateImpl.java

Source

/*
 * SReader is RSS/Atom feed reader with full text.
 *
 * Copyright (C) 2011, Shinnosuke Suzuki <sasasin@sasasin.net>
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as 
 * published by the Free Software Foundation, either version 3 of
 * the License, or any later version.
 *   
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this program.
 * If not, see <http://www.gnu.org/licenses/>.
 */
package net.sasasin.sreader.commons.dao.impl;

import java.io.Serializable;
import java.lang.reflect.ParameterizedType;
import java.util.List;

import net.sasasin.sreader.commons.dao.GenericDao;

import org.hibernate.Criteria;
import org.hibernate.HibernateException;
import org.hibernate.Session;
import org.hibernate.SessionFactory;
import org.hibernate.Transaction;
import org.hibernate.cfg.Configuration;
import org.hibernate.service.ServiceRegistry;
import org.hibernate.service.ServiceRegistryBuilder;

/**
 * {@link GenericDao}?Hibernate?
 * 
 * @author sasasin
 * 
 */
public class GenericDaoHibernateImpl<T, PK extends Serializable> implements GenericDao<T, PK> {

    @SuppressWarnings("unchecked")
    private final Class<T> type = (Class<T>) ((ParameterizedType) getClass().getGenericSuperclass())
            .getActualTypeArguments()[0];

    private SessionFactory sessionFactory;
    private Session session;

    protected SessionFactory getSessionFactory() {
        if (sessionFactory == null) {
            Configuration configuration = new Configuration().configure();
            ServiceRegistry serviceRegistry = new ServiceRegistryBuilder()
                    .applySettings(configuration.getProperties()).buildServiceRegistry();
            sessionFactory = new Configuration().configure().buildSessionFactory(serviceRegistry);
        }
        return sessionFactory;
    }

    public void setSessionFactory(SessionFactory sessionFactory) {
        this.sessionFactory = sessionFactory;
    }

    private Session getSession() {
        if (session == null || !session.isOpen()) {
            session = getSessionFactory().openSession();
        }
        return session;
    }

    protected Class<T> getType() {

        return this.type;
    }

    @SuppressWarnings("unchecked")
    @Override
    public T get(PK id) {

        T entity = (T) getSession().get(getType(), id);

        return entity;

    }

    @SuppressWarnings("unchecked")
    @Override
    public List<T> findAll() {
        Session s = getSession();

        List<T> result = s.createCriteria(getType()).setResultTransformer(Criteria.DISTINCT_ROOT_ENTITY).list();

        return result;
    }

    @SuppressWarnings("unchecked")
    @Override
    public PK save(T entity) {
        Session s = getSession();
        Transaction tx = s.beginTransaction();
        try {
            PK id = (PK) s.save(entity);
            tx.commit();
            s.close();
            return id;
        } catch (HibernateException e) {
            // insert?????
            e.printStackTrace();
            tx.rollback();
            s.close();
            throw e;
        }
    }

    @Override
    public void update(T entity) {
        Session s = getSession();
        Transaction tx = s.beginTransaction();
        try {
            s.update(entity);
            tx.commit();
            s.close();
        } catch (HibernateException e) {
            // update?????
            e.printStackTrace();
            tx.rollback();
            s.close();
            throw e;
        }
    }

    @Override
    public void delete(T entity) {
        Session s = getSession();
        Transaction tx = s.beginTransaction();
        try {
            s.delete(entity);
            tx.commit();
            s.close();
        } catch (HibernateException e) {
            // delete?????
            e.printStackTrace();
            tx.rollback();
            s.close();
            throw e;
        }
    }

}