/*
* Copyright 1999-2011 Alibaba Group Holding Ltd.
*
* 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 com.alibaba.druid.pool.xa;
import com.alibaba.druid.pool.DruidDataSource;
import com.alibaba.druid.pool.DruidPooledConnection;
import com.alibaba.druid.support.logging.Log;
import com.alibaba.druid.support.logging.LogFactory;
import com.alibaba.druid.util.H2Utils;
import com.alibaba.druid.util.JdbcUtils;
import com.alibaba.druid.util.MySqlUtils;
import com.alibaba.druid.util.OracleUtils;
import com.alibaba.druid.util.PGUtils;
import javax.sql.XAConnection;
import javax.sql.XADataSource;
import javax.transaction.xa.XAException;
import java.sql.Connection;
import java.sql.SQLException;
public class DruidXADataSource extends DruidDataSource implements XADataSource {
private final static Log LOG = LogFactory.getLog(DruidXADataSource.class);
private static final long serialVersionUID = 1L;
private Object h2Factory = null;
@Override
public XAConnection getXAConnection() throws SQLException {
DruidPooledConnection conn = this.getConnection();
Connection physicalConn = conn.unwrap(Connection.class);
XAConnection rawXAConnection = createPhysicalXAConnection(physicalConn);
return new DruidPooledXAConnection(conn, rawXAConnection);
}
protected void initCheck() throws SQLException {
super.initCheck();
if (JdbcUtils.H2.equals(this.dbType)) {
h2Factory = H2Utils.createJdbcDataSourceFactory();
}
}
private XAConnection createPhysicalXAConnection(Connection physicalConn) throws SQLException {
if (JdbcUtils.ORACLE.equals(dbType)) {
try {
return OracleUtils.OracleXAConnection(physicalConn);
} catch (XAException xae) {
LOG.error("create xaConnection error", xae);
return null;
}
}
if (JdbcUtils.MYSQL.equals(dbType)) {
return MySqlUtils.createXAConnection(physicalConn);
}
if (JdbcUtils.POSTGRESQL.equals(dbType)) {
return PGUtils.createXAConnection(physicalConn);
}
if (JdbcUtils.H2.equals(dbType)) {
return H2Utils.createXAConnection(h2Factory, physicalConn);
}
if (JdbcUtils.JTDS.equals(dbType)) {
return new JtdsXAConnection(physicalConn);
}
throw new SQLException("xa not support dbType : " + this.dbType);
}
@Override
public XAConnection getXAConnection(String user, String password) throws SQLException {
throw new UnsupportedOperationException("Not supported by DruidDataSource");
}
}