java - 使用 JndiObjectFactoryBean 在运行时设置动态模式

Question

我的客户希望在运行时有一个动态模式。我现在所做的是这样的:

对于我的 web.xml(仅重要部分):

<servlet>
  <servlet-name>servlet1</servlet-name>
  <servlet-class>org.springframework.web.servlet.DispatcherServlet</servlet-class>
  <load-on-startup>1</load-on-startup>
</servlet>

<servlet-mapping>
  <servlet-name>servlet1</servlet-name>
  <url-pattern>/myservlet/sample1/</url-pattern>
</servlet-mapping>

<servlet>
  <servlet-name>servlet2</servlet-name>
  <servlet-class>org.springframework.web.servlet.DispatcherServlet</servlet-class>
  <load-on-startup>2</load-on-startup>
</servlet>

<servlet-mapping>
  <servlet-name>servlet2</servlet-name>
  <url-pattern>/myservlet/sample2/</url-pattern>
</servlet-mapping>

<context-param>
  <param-name>contextConfigLocation</param-name>
  <param-value>
    /WEB-INF/servlet1-servlet.xml
    /WEB-INF/servlet2-servlet.xml
  </param-value>
</context-param>

... some properties ...

<resource-ref>
        <res-ref-name>DATASOURCE_1</res-ref-name>
        <res-type>javax.sql.DataSource</res-type>
        <res-auth>Container</res-auth>
        <res-sharing-scope>Shareable</res-sharing-scope>
</resource-ref>

<resource-ref>
        <res-ref-name>DATASOURCE_2</res-ref-name>
        <res-type>javax.sql.DataSource</res-type>
        <res-auth>Container</res-auth>
        <res-sharing-scope>Shareable</res-sharing-scope>
</resource-ref>

对于我的 servlet1-servlet.xml(servlet2-servlet.xml 的内容是相同的，除了 JNDI 名称和它的默认模式)

<bean id="dataSource" class="org.springframework.jndi.JndiObjectFactoryBean">
        <property name="jndiName" value="DATASOURCE_1" />
        <property name="lookupOnStartup" value="false" />
        <property name="cache" value="true" />
        <property name="proxyInterface" value="javax.sql.DataSource" />
</bean>

<bean id="entityManagerFactory"
    class="org.springframework.orm.jpa.LocalContainerEntityManagerFactoryBean">
    <property name="dataSource" ref="dataSource" />
    <property name="jpaVendorAdapter" ref="jpaVendorAdapter" />
    <property name="jpaDialect" ref="jpaDialect" />
    <property name="packagesToScan">
        <array>
            <value>ph.prj.domain.models</value>
        </array>
    </property>
    <property name="jpaProperties">
       <props>
            <prop key="hibernate.default_schema">SCHEMA1</prop>
       </props>
    </property>
</bean>

<bean id="jpaVendorAdapter"
    class="org.springframework.orm.jpa.vendor.HibernateJpaVendorAdapter">
    <property name="database" value="${jdbc.databaseName}" />
    <property name="databasePlatform" value="${hibernate.dialect}" />
    <property name="generateDdl" value="${init-db}" />
    <property name="showSql" value="${hibernate.show_sql}" />
</bean>

<bean id="jpaDialect" class="org.springframework.orm.jpa.vendor.HibernateJpaDialect" />

<bean id="transactionManager" class="org.springframework.orm.jpa.JpaTransactionManager">
    <property name="entityManagerFactory" ref="entityManagerFactory" />
    <property name="dataSource" ref="dataSource" />
    <property name="jpaDialect" ref="jpaDialect" />
</bean>

最后是我在运行时处理动态 jndi 名称的方法:

protected void setCurrentSchema(String identifier) throws Exception {
    if (StringUtils.equalsIgnoreCase(IdentifierEnum.SCHEMA_1.getValue(), identifier) || 
            StringUtils.equalsIgnoreCase(IdentifierEnum.SCHEMA_2.getValue(), identifier)) {
        // WEBSPHERE
        JndiObjectFactoryBean dataSource = context.getBean(JndiObjectFactoryBean.class);
        dataSource.setJndiName(ApplicationConstants.JNDI_ROOT_NAME + identifier);

    } else {
        logger.debug(String.format("INVALID identifier SPECIFIED - %s", identifier));
        throw new RequestParamException(String.format("INVALID identifier SPECIFIED - %s", identifier));
    }
}

为了访问我的 api，我需要使用 ff 调用。上下文路径:

/myservlet/sample1/

/myservlet/sample2/

不幸的是，我的客户不希望 sample1/sample2 部分出现在上下文路径中。

客户想访问为

/myservlet/

仅，与架构名称无关。

我计划利用此 link 中的 AbstractRoutingDataSource但是，示例使用的是 DriverManagerDataSource 而不是 JndiObjectFactoryBean，而且我找不到关于如何动态设置架构的方法。

我觉得这些方法可以帮助我实现我想要的输出 setJndiEnvironment/setJndiTemplate 但是，他们缺少一些关于如何使用它的示例，我找不到一些使用这些方法来更改其架构的来源。

希望你们能帮我解决这个问题。

Answer 1

您可以在运行时在 JpaTransactionManager 和 LocalContainerEntityManagerFactoryBean bean 上动态设置 jpaProperties。

protected void setCurrentSchema(String identifier) throws Exception {
    if (StringUtils.equalsIgnoreCase(IdentifierEnum.SCHEMA_1.getValue(), identifier) || 
        StringUtils.equalsIgnoreCase(IdentifierEnum.SCHEMA_2.getValue(), identifier)) {
        // WEBSPHERE
        Properties jpaProperties = new Properties();
        jpaProperties.put("hibernate.default_schema", identifier);

        LocalContainerEntityManagerFactoryBean entityManagerFactory = context.getBean(LocalContainerEntityManagerFactoryBean.class);
        entityManagerFactory.setJpaProperties(jpaProperties);
        entityManagerFactory.afterPropertiesSet();

        JpaTransactionManager transactionManager = context.getBean(JpaTransactionManager.class);
        transactionManager.setJpaProperties(jpaProperties);
        transactionManager.afterPropertiesSet();
    } else {
        logger.debug(String.format("INVALID identifier SPECIFIED - %s", identifier));
        throw new RequestParamException(String.format("INVALID identifier SPECIFIED - %s", identifier));
    }
}

有了这个，您可以只定义一个数据源，而不需要有一个默认模式，因为它将在运行时设置。您还可以摆脱映射到每个数据库模式的多个 *servlet.xml 文件，而且大多数情况下，您可以摆脱应用程序上下文路径上的额外部分