/* * Copyright 2020 Red Hat, Inc, and individual contributors. * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ package me.snowdrop.boot.narayana.autoconfigure; import java.io.File; import javax.jms.Message; import javax.transaction.Transaction; import javax.transaction.TransactionManager; import javax.transaction.UserTransaction; import com.arjuna.ats.arjuna.recovery.RecoveryManager; import com.arjuna.ats.internal.jta.recovery.arjunacore.XARecoveryModule; import com.arjuna.ats.jbossatx.jta.RecoveryManagerService; import me.snowdrop.boot.narayana.core.jdbc.GenericXADataSourceWrapper; import me.snowdrop.boot.narayana.core.jdbc.PooledXADataSourceWrapper; import me.snowdrop.boot.narayana.core.jms.NarayanaXAConnectionFactoryWrapper; import me.snowdrop.boot.narayana.core.properties.NarayanaProperties; import me.snowdrop.boot.narayana.core.properties.NarayanaPropertiesInitializer; import org.jboss.tm.XAResourceRecoveryRegistry; import org.springframework.beans.factory.ObjectProvider; import org.springframework.boot.autoconfigure.AutoConfigureBefore; import org.springframework.boot.autoconfigure.condition.ConditionalOnClass; import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean; import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty; import org.springframework.boot.autoconfigure.transaction.TransactionManagerCustomizers; import org.springframework.boot.autoconfigure.transaction.jta.JtaAutoConfiguration; import org.springframework.boot.autoconfigure.transaction.jta.JtaProperties; import org.springframework.boot.context.properties.EnableConfigurationProperties; import org.springframework.boot.jdbc.XADataSourceWrapper; import org.springframework.boot.jms.XAConnectionFactoryWrapper; import org.springframework.boot.system.ApplicationHome; import org.springframework.context.annotation.Bean; import org.springframework.context.annotation.Configuration; import org.springframework.context.annotation.DependsOn; import org.springframework.transaction.PlatformTransactionManager; import org.springframework.transaction.jta.JtaTransactionManager; import org.springframework.util.StringUtils; /** * JTA Configuration for <a href="http://narayana.io/">Narayana</a>. * * @author <a href="mailto:[email protected]">Gytis Trikleris</a> */ @Configuration @EnableConfigurationProperties({ JtaProperties.class, NarayanaProperties.class }) @ConditionalOnProperty(prefix = "spring.jta", value = "enabled", matchIfMissing = true) @ConditionalOnClass({ Transaction.class, JtaTransactionManager.class, XAResourceRecoveryRegistry.class, com.arjuna.ats.jta.UserTransaction.class }) @ConditionalOnMissingBean(PlatformTransactionManager.class) @AutoConfigureBefore(JtaAutoConfiguration.class) public class NarayanaConfiguration { private final JtaProperties jtaProperties; private final TransactionManagerCustomizers transactionManagerCustomizers; public NarayanaConfiguration(JtaProperties jtaProperties, ObjectProvider<TransactionManagerCustomizers> transactionManagerCustomizers) { this.jtaProperties = jtaProperties; this.transactionManagerCustomizers = transactionManagerCustomizers.getIfAvailable(); } @Bean @ConditionalOnMissingBean public static NarayanaBeanFactoryPostProcessor narayanaBeanFactoryPostProcessor() { return new NarayanaBeanFactoryPostProcessor(); } @Bean @ConditionalOnMissingBean public NarayanaPropertiesInitializer narayanaPropertiesInitializer(NarayanaProperties properties) { initLogDir(properties); initTransactionManagerId(properties); return new NarayanaPropertiesInitializer(properties); } @Bean @DependsOn("narayanaPropertiesInitializer") @ConditionalOnMissingBean public UserTransaction narayanaUserTransaction() { return com.arjuna.ats.jta.UserTransaction.userTransaction(); } @Bean @DependsOn("narayanaPropertiesInitializer") @ConditionalOnMissingBean public TransactionManager narayanaTransactionManager() { return com.arjuna.ats.jta.TransactionManager.transactionManager(); } @Bean @ConditionalOnMissingBean public JtaTransactionManager transactionManager(UserTransaction userTransaction, TransactionManager transactionManager) { JtaTransactionManager jtaTransactionManager = new JtaTransactionManager(userTransaction, transactionManager); if (this.transactionManagerCustomizers != null) { this.transactionManagerCustomizers.customize(jtaTransactionManager); } return jtaTransactionManager; } @Bean(destroyMethod = "stop") @DependsOn("narayanaPropertiesInitializer") @ConditionalOnMissingBean public RecoveryManagerService recoveryManagerService() { RecoveryManager.delayRecoveryManagerThread(); RecoveryManagerService recoveryManagerService = new RecoveryManagerService(); recoveryManagerService.create(); recoveryManagerService.start(); return recoveryManagerService; } @Bean @DependsOn("recoveryManagerService") @ConditionalOnMissingBean public XARecoveryModule xaRecoveryModule() { return XARecoveryModule.getRegisteredXARecoveryModule(); } private void initLogDir(NarayanaProperties properties) { if (!StringUtils.isEmpty(properties.getLogDir())) { return; } if (!StringUtils.isEmpty(this.jtaProperties.getLogDir())) { properties.setLogDir(this.jtaProperties.getLogDir()); } else { properties.setLogDir(getLogDir().getAbsolutePath()); } } private void initTransactionManagerId(NarayanaProperties properties) { if (!StringUtils.isEmpty(properties.getTransactionManagerId())) { return; } if (!StringUtils.isEmpty(this.jtaProperties.getTransactionManagerId())) { properties.setTransactionManagerId(this.jtaProperties.getTransactionManagerId()); } } private File getLogDir() { if (StringUtils.hasLength(this.jtaProperties.getLogDir())) { return new File(this.jtaProperties.getLogDir()); } File home = new ApplicationHome().getDir(); return new File(home, "transaction-logs"); } /** * Generic data source wrapper configuration. */ @ConditionalOnProperty(name = "narayana.dbcp.enabled", havingValue = "false", matchIfMissing = true) static class GenericJdbcConfiguration { @Bean @ConditionalOnMissingBean(XADataSourceWrapper.class) public XADataSourceWrapper xaDataSourceWrapper(NarayanaProperties narayanaProperties, XARecoveryModule xaRecoveryModule) { return new GenericXADataSourceWrapper(narayanaProperties, xaRecoveryModule); } } /** * Pooled data source wrapper configuration. */ @ConditionalOnProperty(name = "narayana.dbcp.enabled", havingValue = "true") static class PooledJdbcConfiguration { @Bean @ConditionalOnMissingBean(XADataSourceWrapper.class) public XADataSourceWrapper xaDataSourceWrapper(NarayanaProperties narayanaProperties, XARecoveryModule xaRecoveryModule, TransactionManager transactionManager) { return new PooledXADataSourceWrapper(narayanaProperties, xaRecoveryModule, transactionManager); } } /** * JMS connection factory wrapper configuration. */ @Configuration @ConditionalOnClass(Message.class) static class NarayanaJmsConfiguration { @Bean @ConditionalOnMissingBean(XAConnectionFactoryWrapper.class) public XAConnectionFactoryWrapper xaConnectionFactoryWrapper(TransactionManager transactionManager, XARecoveryModule xaRecoveryModule, NarayanaProperties narayanaProperties) { return new NarayanaXAConnectionFactoryWrapper(transactionManager, xaRecoveryModule, narayanaProperties); } } }