org.hibernate.solon.integration.HibernateConfiguration Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of hibernate-solon-plugin Show documentation

The newest version!

package org.hibernate.solon.integration;

import org.hibernate.HibernateException;
import org.hibernate.SessionFactory;
import org.hibernate.cfg.AvailableSettings;
import org.hibernate.cfg.Configuration;
import org.hibernate.context.internal.ThreadLocalSessionContext;
import org.noear.solon.Utils;
import org.noear.solon.core.util.ResourceUtil;

import javax.sql.DataSource;
import java.util.Collection;
import java.util.Properties;

/**
 * @author lingkang
 * @author bai
 * @since 2.5
 */
public class HibernateConfiguration extends Configuration {

    public HibernateConfiguration() {

    }

    /**
     * 添加实体映射，有hibernate的@Table、@Entity
     */
    public HibernateConfiguration addMapping(String basePackage) {
        if (Utils.isNotEmpty(basePackage)) {
            Collection> classes = ResourceUtil.scanClasses(basePackage);
            for (Class clazz : classes)
                addAnnotatedClass(clazz);
        }
        return this;
    }

    /**
     * 设置数据源
     */
    public HibernateConfiguration setDataSource(DataSource dataSource) {
        if (dataSource != null) {
            this.getProperties().put(AvailableSettings.DATASOURCE, dataSource);
        }
        return this;
    }

    /**
     * 设置属性
     */
    public HibernateConfiguration setProperties(Properties properties) {
        if (properties != null) {
            properties.entrySet().forEach(obj -> {
                getProperties().put(obj.getKey(), obj.getValue());
            });
        }
        return this;
    }

    /**
     * 构建会话工具
     */
    @Override
    public SessionFactory buildSessionFactory() throws HibernateException {
        getProperties().put(AvailableSettings.TRANSACTION_COORDINATOR_STRATEGY, "jdbc");
        getProperties().put(AvailableSettings.CURRENT_SESSION_CONTEXT_CLASS, ThreadLocalSessionContext.class.getName());

        SessionFactory sessionFactory = super.buildSessionFactory();
        return new JpaTranSessionFactory(sessionFactory);
    }
}