/*
* Copyright 1999-2017 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 java.lang.reflect.Method;
import java.sql.Connection;
import javax.transaction.xa.XAException;
import javax.transaction.xa.XAResource;
import javax.transaction.xa.Xid;
import net.sourceforge.jtds.jdbc.XASupport;
import com.alibaba.druid.support.logging.Log;
import com.alibaba.druid.support.logging.LogFactory;
public class JtdsXAResource implements XAResource {
private final static Log LOG = LogFactory.getLog(JtdsXAResource.class);
private final Connection connection;
private final JtdsXAConnection xaConnection;
private String rmHost;
private static Method method;
public JtdsXAResource(JtdsXAConnection xaConnection, Connection connection){
this.xaConnection = xaConnection;
this.connection = connection;
if (method == null) {
try {
method = connection.getClass().getMethod("getRmHost");
} catch (Exception e) {
LOG.error("getRmHost method error", e);
}
}
if (method != null) {
try {
rmHost = (String) method.invoke(connection);
} catch (Exception e) {
LOG.error("getRmHost error", e);
}
}
}
protected JtdsXAConnection getResourceManager() {
return xaConnection;
}
protected String getRmHost() {
return this.rmHost;
}
@Override
public void commit(Xid xid, boolean commit) throws XAException {
XASupport.xa_commit(connection, xaConnection.getXAConnectionID(), xid, commit);
}
@Override
public void end(Xid xid, int flags) throws XAException {
XASupport.xa_end(connection, xaConnection.getXAConnectionID(), xid, flags);
}
@Override
public void forget(Xid xid) throws XAException {
XASupport.xa_forget(connection, xaConnection.getXAConnectionID(), xid);
}
@Override
public int getTransactionTimeout() throws XAException {
return 0;
}
@Override
public boolean isSameRM(XAResource xares) throws XAException {
if (xares instanceof JtdsXAResource) {
if (((JtdsXAResource) xares).getRmHost().equals(this.rmHost)) {
return true;
}
}
return false;
}
@Override
public int prepare(Xid xid) throws XAException {
return XASupport.xa_prepare(connection, xaConnection.getXAConnectionID(), xid);
}
@Override
public Xid[] recover(int flags) throws XAException {
return XASupport.xa_recover(connection, xaConnection.getXAConnectionID(), flags);
}
@Override
public void rollback(Xid xid) throws XAException {
XASupport.xa_rollback(connection, xaConnection.getXAConnectionID(), xid);
}
@Override
public boolean setTransactionTimeout(int seconds) throws XAException {
return false;
}
@Override
public void start(Xid xid, int flags) throws XAException {
XASupport.xa_start(connection, xaConnection.getXAConnectionID(), xid, flags);
}
}