package sp.config;

import com.atomikos.icatch.jta.J2eeUserTransaction;
import com.atomikos.icatch.jta.UserTransactionImp;
import com.atomikos.icatch.jta.UserTransactionManager;
import com.atomikos.jdbc.AtomikosDataSourceBean;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import com.mysql.jdbc.jdbc2.optional.MysqlXADataSource;
import org.springframework.orm.jpa.JpaVendorAdapter;
import org.springframework.orm.jpa.LocalContainerEntityManagerFactoryBean;
import org.springframework.orm.jpa.vendor.Database;
import org.springframework.orm.jpa.vendor.HibernateJpaVendorAdapter;
import org.springframework.transaction.jta.JtaTransactionManager;

import javax.sql.DataSource;

@Configuration
public class NewDbConfig {

    @Bean(name="dataSourceA")
    public AtomikosDataSourceBean dataSourceA() {
        AtomikosDataSourceBean dataSource = new AtomikosDataSourceBean();
        dataSource.setUniqueResourceName("DataSourceA");

        MysqlXADataSource mysqlXADataSource = new MysqlXADataSource();
        mysqlXADataSource.setDatabaseName("atomikos_one");
        mysqlXADataSource.setServerName("localhost");
        mysqlXADataSource.setPort(3306);
        mysqlXADataSource.setUser("root");
        mysqlXADataSource.setPassword("mypass");
        mysqlXADataSource.setUrl("jdbc:mysql://localhost:3306/atomikos_one");

        dataSource.setXaDataSource(mysqlXADataSource);

        return dataSource;
    }

    @Bean(name="dataSourceB")
    public AtomikosDataSourceBean dataSourceB() {
        AtomikosDataSourceBean dataSource = new AtomikosDataSourceBean();
        dataSource.setUniqueResourceName("DataSourceB");

        MysqlXADataSource mysqlXADataSource = new MysqlXADataSource();
        mysqlXADataSource.setDatabaseName("atomikos_two");
        mysqlXADataSource.setServerName("localhost");
        mysqlXADataSource.setPort(3306);
        mysqlXADataSource.setUser("root");
        mysqlXADataSource.setPassword("mypass");
        mysqlXADataSource.setUrl("jdbc:mysql://localhost:3306/atomikos_two");

        dataSource.setXaDataSource(mysqlXADataSource);

        return dataSource;
    }

    @Bean(name="entityManagerFactoryA")
    public LocalContainerEntityManagerFactoryBean entityManagerFactoryA(
            @Qualifier("dataSourceA") DataSource dataSource,
            @Qualifier("vendorAdapterA") JpaVendorAdapter vendorAdapter
    ) {
        LocalContainerEntityManagerFactoryBean factoryBean = new LocalContainerEntityManagerFactoryBean();
        factoryBean.setPackagesToScan("sp.model.atomikosA");
        factoryBean.setDataSource(dataSource);
        factoryBean.setJpaVendorAdapter(vendorAdapter);

        return factoryBean;
    }

    @Bean(name="vendorAdapterA")
    public JpaVendorAdapter hibernateJpaVendorAdapterA() {
        HibernateJpaVendorAdapter vendorAdapter = new HibernateJpaVendorAdapter();
        vendorAdapter.setShowSql(true);
        vendorAdapter.setDatabase(Database.MYSQL);
        vendorAdapter.setDatabasePlatform("org.hibernate.dialect.MySQL5Dialect");
        return vendorAdapter;
    }

    @Bean(name="entityManagerFactoryB")
    public LocalContainerEntityManagerFactoryBean entityManagerFactoryB(
            @Qualifier("dataSourceB") DataSource dataSource,
            @Qualifier("vendorAdapterB") JpaVendorAdapter vendorAdapter
    ) {
        LocalContainerEntityManagerFactoryBean factoryBean = new LocalContainerEntityManagerFactoryBean();
        factoryBean.setPackagesToScan("sp.model.atomikosB");
        factoryBean.setDataSource(dataSource);
        factoryBean.setJpaVendorAdapter(vendorAdapter);

        return factoryBean;
    }

    @Bean(name="vendorAdapterB")
    public JpaVendorAdapter hibernateJpaVendorAdapterB() {
        HibernateJpaVendorAdapter vendorAdapter = new HibernateJpaVendorAdapter();
        vendorAdapter.setShowSql(true);
        vendorAdapter.setDatabase(Database.MYSQL);
        vendorAdapter.setDatabasePlatform("org.hibernate.dialect.MySQL5Dialect");
        return vendorAdapter;
    }

    @Bean(name="atomikosTransactionManager")
    public UserTransactionManager userTransactionManager() {
        UserTransactionManager transactionManager = new UserTransactionManager();
        transactionManager.setForceShutdown(false);
        return transactionManager;
    }

    @Bean(name="atomikosUserTransaction")
    public UserTransactionImp atomikosUserTransaction() {
        UserTransactionImp j2eeUserTransaction = new UserTransactionImp();
        try {
            j2eeUserTransaction.setTransactionTimeout(300);
        } catch (Exception e) {

        }
        return j2eeUserTransaction;
    }

    @Bean(name="transactionManager")
    public JtaTransactionManager jtaTransactionManager(
            @Qualifier("atomikosTransactionManager") UserTransactionManager userTransactionManager,
            @Qualifier("atomikosUserTransaction") UserTransactionImp j2eeUserTransaction
    ) {
        JtaTransactionManager jtaTransactionManager = new JtaTransactionManager();
        jtaTransactionManager.setTransactionManager(userTransactionManager);
        jtaTransactionManager.setUserTransaction(j2eeUserTransaction);
        jtaTransactionManager.setAllowCustomIsolationLevels(true);
        return jtaTransactionManager;
    }

}

package sp.dao.atomikosA;

import sp.model.atomikosA.Customer;
import org.springframework.stereotype.Service;

import javax.persistence.EntityManager;
import javax.persistence.PersistenceContext;

@Service
public class CustomerADaoImpl implements CustomerADao {

    @PersistenceContext(unitName="entityManagerFactoryA")
    protected EntityManager em;

    public void persistItem(Customer customer) {
        em.persist(customer);
    }
}

package sp.dao.atomikosB;

import sp.model.atomikosB.Order;
import org.springframework.stereotype.Service;

import javax.persistence.EntityManager;
import javax.persistence.PersistenceContext;

@Service
public class OrderBDaoImpl implements OrderBDao {

    @PersistenceContext(unitName="entityManagerFactoryB")
    protected EntityManager em;

    public void persistItem(Order order) {
        em.persist(order);
    }
}

package sp.services;

import sp.dao.atomikosA.CustomerADao;
import sp.dao.atomikosB.OrderBDao;
import sp.model.atomikosA.Customer;
import sp.model.atomikosB.Order;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

@Service("employeeService")
public class EmployeeServiceImpl implements EmployeeService {

    @Autowired
    private CustomerADao customerADao;

    @Autowired
    private OrderBDao orderBDao;

    @Transactional(rollbackFor=Exception.class)
    public void persistEmployees(Customer employeeA, Order employeeB) throws Exception {
        System.out.println("Persist A");
        customerADao.persistItem(employeeA);
        System.out.println("Persist A OK - persist B");
        orderBDao.persistItem(employeeB);
        System.out.println("Persist B okk");
    }

    @Transactional
    public void persistCustomer(Customer customer) throws Exception {
        System.out.println("Persist A");
        customerADao.persistItem(customer);
    }

}

@RequestMapping("/")
    @ResponseBody
    public String index() {

        sp.model.atomikosA.Customer customer = new sp.model.atomikosA.Customer();
        customer.setName("Maksat");
        customer.setAge(27);
        customer.setCustId(3L);

        Order order = new Order();
        order.setCustId(3L);
        order.setCode(1);
        order.setQuantity(1);

        try {
            employeeService.persistEmployees(customer, order);
            employeeService.persistCustomer(customer);
        } catch (Exception e) {
            int del = 2;
        }

        return "success";
    }