请帮我配置Custom LocalSessionFactoryBean

时间:2019-02-03 18:19:17

标签: java spring hibernate

作为休眠升级的一部分,我无法配置会话工厂。 在运行时,根据系统变量,我们使用schema或mock_schema。 我们在postProcessConfiguration中进行自定义。

在生成会话工厂的过程中,我不清楚在哪里挂接代码,也不清楚弹簧流。

从休眠状态我有一个解决方案。https://vladmihalcea.com/how-to-get-the-entity-mapping-to-database-table-binding-metadata-from-hibernate/#comments

  1. 如何让多个集成商说MetadataExtractorIntegratorEventListenerIntegrator
  2. 如何配置CustomLocalSessionFactoryBean(扩展LocalSessionFactoryBean)。
  3. 我可以在不覆盖LocalSessionFactoryBean的情况下做到吗?
  4. 如何从配置中获取/修改getCollectionMappings

    公共类CustomHibernateSessionFactoryBean扩展了org.springframework.orm.hibernate3.annotation.AnnotationSessionFactoryBean {

    @Inject
    private DatabasePlatform dbPlatform;
    
    @Inject
    private ApplicationContext applicationContext;
    
    @Override
    protected SessionFactory newSessionFactory(Configuration config) {
        config.setProperty(DatabasePlatform.HIBERNATE_DIALECT_KEY, this.dbPlatform.getDialectClass().getName());
    
        //Many such event listeners
        config.getEventListeners().setUpdateEventListeners(sortAndConcat(config.getEventListeners().getUpdateEventListeners(),saveOrUpdateEventListeners));
    
        return super.newSessionFactory(config);
    }
    
    @Override
    protected void postProcessConfiguration(Configuration config) {
        super.postProcessConfiguration(config);
    
        if (!DatabasePlatform.INVENTORY_SCHEMA.equals(DatabasePlatform.DEFAULT_INVENTORY_SCHEMA)) {
            logger.info("Overriding inventory schema name to %s", DatabasePlatform.INVENTORY_SCHEMA);
            Iterator<PersistentClass> classIterator = config.getClassMappings();
    
            while (classIterator.hasNext()) {
                PersistentClass persistentClass = classIterator.next();
                replaceSchemaName(persistentClass.getTable());
                Iterator<Join> joinIterator = persistentClass.getJoinIterator();
                while (joinIterator.hasNext()) {
                    replaceSchemaName(joinIterator.next().getTable());
                }
    
                Iterator<org.hibernate.mapping.Collection> collectionIterator = config.getCollectionMappings();
                while (collectionIterator.hasNext()) {
                    replaceSchemaName(collectionIterator.next().getCollectionTable());
                }
            }
        }
    }
    
    private void replaceSchemaName(Table table) {
        if (DatabasePlatform.DEFAULT_INVENTORY_SCHEMA.equals(table.getSchema())) {
            table.setSchema(DatabasePlatform.INVENTORY_SCHEMA);
        }
        if (table.getName().startsWith(DatabasePlatform.DEFAULT_INVENTORY_SCHEMA + ".")) {
            table.setName(table.getName().replaceFirst(DatabasePlatform.DEFAULT_INVENTORY_SCHEMA, DatabasePlatform.INVENTORY_SCHEMA));
        }
    }
    

    }

我正在为休眠5尝试的新代码

@SuppressWarnings("unused")
public class CustomHibernateSessionFactoryBean extends org.springframework.orm.hibernate5.LocalSessionFactoryBean {

    @Inject
    private ApplicationContext applicationContext;

    private static final SystemLogger logger = SystemLogger.getLogger(CustomHibernateSessionFactoryBean.class);

    @Inject
    private DatabasePlatform dbPlatform;

    private AsyncTaskExecutor bootstrapExecutor;

    private EventListenerIntegrator eventListenerIntegrator;

    private final RegionFactory regionFactory;

    private boolean metadataSourcesAccessed;

    private MetadataSources metadataSources;

    public CustomHibernateSessionFactoryBean() {
        // TODO: Check what is the effect of regionFactory not being set
        regionFactory = null;
    }

    CustomHibernateSessionFactoryBean(RegionFactory regionFactory, EventListenerIntegrator eventListenerIntegrator) {
        this.regionFactory = regionFactory;
        this.eventListenerIntegrator = eventListenerIntegrator;
    }

    CustomHibernateSessionFactoryBean(EventListenerIntegrator eventListenerIntegrator) {
        this.regionFactory = null;
        this.eventListenerIntegrator = eventListenerIntegrator;
    }

    @Override
    public void setMetadataSources(MetadataSources metadataSources) {
        Assert.notNull(metadataSources, "MetadataSources must not be null");
        this.metadataSourcesAccessed = true;
        this.metadataSources = metadataSources;
    }

    @Override
    public MetadataSources getMetadataSources() {
        this.metadataSourcesAccessed = true;
        if (this.metadataSources == null) {
            BootstrapServiceRegistryBuilder builder = new BootstrapServiceRegistryBuilder();
            ResourcePatternResolver resourcePatternResolver = (ResourcePatternResolver) getResourceLoader();
            if (resourcePatternResolver != null) {
                builder = builder.applyClassLoader(resourcePatternResolver.getClassLoader()).applyIntegrator(eventListenerIntegrator);
            }
            this.metadataSources = new MetadataSources(builder.build());
        }
        return this.metadataSources;
    }

    @Override
    protected SessionFactory buildSessionFactory(LocalSessionFactoryBuilder sfb) {

        getConfiguration().setProperty(DatabasePlatform.HIBERNATE_DIALECT_KEY, this.dbPlatform.getDialectClass().getName());
        if (!DatabasePlatform.INVENTORY_SCHEMA.equals(DatabasePlatform.DEFAULT_INVENTORY_SCHEMA)) {
            logger.info("Overriding inventory schema name to %s", DatabasePlatform.INVENTORY_SCHEMA);
            Metadata metadata = getMetadataSources().getMetadataBuilder().build();
            Collection<PersistentClass> entityBindings = metadata.getEntityBindings();

            for (Iterator<PersistentClass> iterator = entityBindings.iterator(); iterator.hasNext();) {
                PersistentClass persistentClass = iterator.next();
                replaceSchemaName(persistentClass.getTable());
                @SuppressWarnings("unchecked")
                Iterator<Join> joinIterator = persistentClass.getJoinIterator();
                while (joinIterator.hasNext()) {
                    replaceSchemaName(joinIterator.next().getTable());
                }
            }
        }
        return (this.bootstrapExecutor != null ? sfb.buildSessionFactory(this.bootstrapExecutor) : sfb.buildSessionFactory());

    }

    private void replaceSchemaName(Table table) {
        if (DatabasePlatform.DEFAULT_INVENTORY_SCHEMA.equals(table.getSchema())) {
            table.setSchema(DatabasePlatform.INVENTORY_SCHEMA);
        }
        if (table.getName().startsWith(DatabasePlatform.DEFAULT_INVENTORY_SCHEMA + ".")) {
            table.setName(table.getName().replaceFirst(DatabasePlatform.DEFAULT_INVENTORY_SCHEMA, DatabasePlatform.INVENTORY_SCHEMA));
        }
    }
}

为什么需要自定义localsessionFactoryBean?

I have 2 need for using SessionFactoryBean
    1. Normal application
    2. Dbtest

In normal application, schema1 is populated by other process
In Dbtest, i need to use mock_schema1 instead of schema1 so that we will not rewrite data populated by other application.

@ Table(name = “cluster”, schema = “schema1”)
@ SecondaryTable(name = “Cluster”, schema = “schema2”, pkJoinColumns = @ PrimaryKeyJoinColumn(name = “clusterId”, referencedColumnName = “objid”))
@ org.hibernate.annotations.Table(appliesTo = “Cluster”, optional = false)
@ Entity
@ OptimisticLocking(type = OptimisticLockType.DIRTY)
@ DynamicUpdate
public class Cluster implements Serializable {

}         但是,如果您看到我的实体定义在运行时使用“ schema1”,那么如果我正在运行Dbtest,我希望实体使用“ mock_schema1”而不是“ schema1”

================================================ ======================== 这是对 M的答复。 Deinum 的问题:

在我的测试中,设置 hibernate.default_schema 不会达到我的目的。

例如作为dbtest的一部分,如果我设置并保存集群实体,它将尝试填充嘲笑_schema2和schema1。 取而代之的是我想要schema2和mock_schema1。

0 个答案:

没有答案