in dubbo-filter-extensions/dubbo-filter-seata/src/main/java/org/apache/dubbo/seata/SeataTransactionPropagationProviderFilter.java [45:100]
public Result invoke(Invoker<?> invoker, Invocation invocation) throws RpcException {
String rpcXid = invocation.getAttachment(RootContext.KEY_XID);
if (rpcXid == null) {
rpcXid = invocation.getAttachment(LOWER_KEY_XID);
}
String rpcBranchType = invocation.getAttachment(RootContext.KEY_BRANCH_TYPE);
if (rpcBranchType == null) {
rpcBranchType = invocation.getAttachment(LOWER_KEY_BRANCH_TYPE);
}
if (LOGGER.isDebugEnabled()) {
LOGGER.debug("Server side xid in RpcContext[" + rpcXid + "]");
}
boolean bind = false;
if (rpcXid != null) {
RootContext.bind(rpcXid);
if (StringUtils.equals(BranchType.TCC.name(), rpcBranchType)) {
RootContext.bindBranchType(BranchType.TCC);
}
bind = true;
if (LOGGER.isDebugEnabled()) {
LOGGER.debug(String.format("bind xid [%s] branchType [%s] to RootContext", rpcXid, rpcBranchType));
}
}
try {
return invoker.invoke(invocation);
} finally {
if (bind) {
if (Constants.LOCAL_PROTOCOL.equals(invoker.getUrl().getProtocol())) {
LOGGER.info("invoke happens in same jvm, skip unbind xid");
} else {
BranchType previousBranchType = RootContext.getBranchType();
String unbindXid = RootContext.unbind();
if (BranchType.TCC == previousBranchType) {
RootContext.unbindBranchType();
}
if (LOGGER.isDebugEnabled()) {
LOGGER.debug(String.format("unbind xid [%s] branchType [%s] from RootContext", unbindXid, previousBranchType));
}
if (!rpcXid.equalsIgnoreCase(unbindXid)) {
LOGGER.warn(String
.format("xid in change during RPC from %s to %s,branchType from %s to %s", rpcXid, unbindXid,
rpcBranchType != null ? rpcBranchType : "AT", previousBranchType));
if (unbindXid != null) {
RootContext.bind(unbindXid);
LOGGER.warn(String.format("bind xid [%s] back to RootContext", unbindXid));
if (BranchType.TCC == previousBranchType) {
RootContext.bindBranchType(BranchType.TCC);
LOGGER.warn(String.format("bind branchType [%s] back to RootContext", previousBranchType));
}
}
}
}
}
}
}