optimize: branch register resource only at RM server end

integration-tx-api/src/main/java/io/seata/integration/tx/api/interceptor/parser/DefaultInterfaceParser.java

@@ -64,4 +64,15 @@ public ProxyInvocationHandler parserInterfaceToProxy(Object target, String objec
         return null;
     }
 
+    @Override
+    public IfNeedEnhanceBean parseIfNeedEnhancement(Class<?> beanClass) {
+        for (InterfaceParser interfaceParser : ALL_INTERFACE_PARSERS) {
+            IfNeedEnhanceBean ifNeedEnhanceBean = interfaceParser.parseIfNeedEnhancement(beanClass);
+            if (ifNeedEnhanceBean.isIfNeed()) {
+                return ifNeedEnhanceBean;
+            }
+        }
+        return new IfNeedEnhanceBean();
+    }
+
 }

integration-tx-api/src/main/java/io/seata/integration/tx/api/interceptor/parser/GlobalTransactionalInterceptorParser.java

@@ -22,6 +22,7 @@
 
 import io.seata.common.ConfigurationKeys;
 import io.seata.common.util.CollectionUtils;
+import io.seata.common.util.ReflectionUtil;
 import io.seata.config.ConfigurationCache;
 import io.seata.config.ConfigurationChangeListener;
 import io.seata.integration.tx.api.interceptor.handler.GlobalTransactionalInterceptorHandler;
@@ -60,6 +61,19 @@ public ProxyInvocationHandler parserInterfaceToProxy(Object target, String objec
         return null;
     }
 
+    @Override
+    public IfNeedEnhanceBean parseIfNeedEnhancement(Class<?> beanClass) {
+        Set<Class<?>> interfaceClasses = ReflectionUtil.getInterfaces(beanClass);
+        Class<?>[] interfaceClasseArray = interfaceClasses.toArray(new Class<?>[0]);
+
+        IfNeedEnhanceBean ifNeedEnhanceBean = new IfNeedEnhanceBean();
+        if (existsAnnotation(beanClass) || existsAnnotation(interfaceClasseArray)) {
+            ifNeedEnhanceBean.setIfNeed(true);
+            ifNeedEnhanceBean.setNeedEnhanceEnum(NeedEnhanceEnum.GLOBAL_TRANSACTIONAL_BEAN);
+        }
+        return ifNeedEnhanceBean;
+    }
+
     private boolean existsAnnotation(Class<?>... classes) {
         boolean result = false;
         if (CollectionUtils.isNotEmpty(classes)) {

integration-tx-api/src/main/java/io/seata/integration/tx/api/interceptor/parser/IfNeedEnhanceBean.java

@@ -0,0 +1,41 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements.  See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You 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 io.seata.integration.tx.api.interceptor.parser;
+
+public class IfNeedEnhanceBean {
+
+    private boolean ifNeed;
+
+    private NeedEnhanceEnum needEnhanceEnum;
+
+    public boolean isIfNeed() {
+        return ifNeed;
+    }
+
+    public void setIfNeed(boolean ifNeed) {
+        this.ifNeed = ifNeed;
+    }
+
+    public NeedEnhanceEnum getNeedEnhanceEnum() {
+        return needEnhanceEnum;
+    }
+
+    public void setNeedEnhanceEnum(NeedEnhanceEnum needEnhanceEnum) {
+        this.needEnhanceEnum = needEnhanceEnum;
+    }
+
+}

integration-tx-api/src/main/java/io/seata/integration/tx/api/interceptor/parser/InterfaceParser.java

@@ -25,5 +25,7 @@ public interface InterfaceParser {
 
     ProxyInvocationHandler parserInterfaceToProxy(Object target, String objectName) throws Exception;
 
+    IfNeedEnhanceBean parseIfNeedEnhancement(Class<?> beanClass);
+
 
 }

integration-tx-api/src/main/java/io/seata/integration/tx/api/interceptor/parser/NeedEnhanceEnum.java

@@ -0,0 +1,33 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements.  See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You 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 io.seata.integration.tx.api.interceptor.parser;
+
+/**
+ * @author ruishansun
+ */
+
+public enum NeedEnhanceEnum {
+
+    /**
+     * global transactional bean type
+     */
+    GLOBAL_TRANSACTIONAL_BEAN,
+    /**
+     * service bean type
+     */
+    SERVICE_BEAN;
+}

integration-tx-api/src/main/java/io/seata/integration/tx/api/remoting/RemotingParser.java

@@ -55,6 +55,15 @@ public interface RemotingParser {
      */
     boolean isService(Object bean, String beanName) throws FrameworkException;
 
+    /**
+     * if it is service bean ?
+     *
+     * @param beanClass the bean class
+     * @return boolean boolean
+     * @throws FrameworkException the framework exception
+     */
+    boolean isService(Class<?> beanClass) throws FrameworkException;
+
     /**
      * get the remoting bean info
      *

integration-tx-api/src/main/java/io/seata/integration/tx/api/remoting/parser/DefaultRemotingParser.java

@@ -134,6 +134,21 @@ public boolean isService(Object bean, String beanName) {
         return false;
     }
 
+    /**
+     * is service bean ?
+     *
+     * @param beanClass the bean class
+     * @return boolean boolean
+     */
+    public boolean isService(Class<?> beanClass) {
+        for (RemotingParser remotingParser : allRemotingParsers) {
+            if (remotingParser.isService(beanClass)) {
+                return true;
+            }
+        }
+        return false;
+    }
+
     /**
      * get the remoting Service desc
      *

integration-tx-api/src/main/java/io/seata/integration/tx/api/remoting/parser/DubboRemotingParser.java

@@ -42,6 +42,12 @@ public boolean isService(Object bean, String beanName) throws FrameworkException
                 || "org.apache.dubbo.config.spring.ServiceBean".equals(c.getName());
     }
 
+    @Override
+    public boolean isService(Class<?> beanClass) throws FrameworkException {
+        return "com.alibaba.dubbo.config.spring.ServiceBean".equals(beanClass.getName())
+                || "org.apache.dubbo.config.spring.ServiceBean".equals(beanClass.getName());
+    }
+
     @Override
     public RemotingDesc getServiceDesc(Object bean, String beanName) throws FrameworkException {
         if (!this.isRemoting(bean, beanName)) {

integration-tx-api/src/main/java/io/seata/integration/tx/api/remoting/parser/HSFRemotingParser.java

@@ -65,6 +65,12 @@ public boolean isService(Object bean, String beanName) {
         return isHsf && "com.taobao.hsf.app.spring.util.HSFSpringProviderBean".equals(beanClassName);
     }
 
+    @Override
+    public boolean isService(Class<?> beanClass) throws FrameworkException {
+        String beanClassName = beanClass.getName();
+        return isHsf && "com.taobao.hsf.app.spring.util.HSFSpringProviderBean".equals(beanClassName);
+    }
+
     @Override
     public RemotingDesc getServiceDesc(Object bean, String beanName) throws FrameworkException {
         if (!this.isRemoting(bean, beanName)) {

integration-tx-api/src/main/java/io/seata/integration/tx/api/remoting/parser/SofaRpcRemotingParser.java

@@ -43,6 +43,12 @@ public boolean isService(Object bean, String beanName) throws FrameworkException
         return "com.alipay.sofa.runtime.spring.factory.ServiceFactoryBean".equals(beanClassName);
     }
 
+    @Override
+    public boolean isService(Class<?> beanClass) throws FrameworkException {
+        String beanClassName = beanClass.getName();
+        return "com.alipay.sofa.runtime.spring.factory.ServiceFactoryBean".equals(beanClassName);
+    }
+
     @Override
     public RemotingDesc getServiceDesc(Object bean, String beanName) throws FrameworkException {
         if (!this.isRemoting(bean, beanName)) {

spring/src/main/java/io/seata/rm/fence/SpringFenceHandler.java

@@ -164,7 +164,9 @@ public boolean commitFence(Method commitMethod, Object targetTCCBean,
                     }
                     return false;
                 }
-                return updateStatusAndInvokeTargetMethod(conn, commitMethod, targetTCCBean, xid, branchId, CommonFenceConstant.STATUS_COMMITTED, status, args);
+                boolean result = updateStatusAndInvokeTargetMethod(conn, commitMethod, targetTCCBean, xid, branchId, CommonFenceConstant.STATUS_COMMITTED, status, args);
+                LOGGER.info("Common fence commit result: {}. xid: {}, branchId: {}", result, xid, branchId);
+                return result;
             } catch (Throwable t) {
                 status.setRollbackOnly();
                 throw new SkipCallbackWrapperException(t);
@@ -211,7 +213,9 @@ public boolean rollbackFence(Method rollbackMethod, Object targetTCCBean,
                         return false;
                     }
                 }
-                return updateStatusAndInvokeTargetMethod(conn, rollbackMethod, targetTCCBean, xid, branchId, CommonFenceConstant.STATUS_ROLLBACKED, status, args);
+                boolean result = updateStatusAndInvokeTargetMethod(conn, rollbackMethod, targetTCCBean, xid, branchId, CommonFenceConstant.STATUS_ROLLBACKED, status, args);
+                LOGGER.info("Common fence rollback result: {}. xid: {}, branchId: {}", result, xid, branchId);
+                return result;
             } catch (Throwable t) {
                 status.setRollbackOnly();
                 throw new SkipCallbackWrapperException(t);
@@ -238,13 +242,13 @@ private static boolean insertCommonFenceLog(Connection conn, String xid, Long br
     }
 
     /**
-     * Update TCC Fence status and invoke target method
+     * Update Common Fence status and invoke target method
      *
      * @param method                target method
      * @param targetTCCBean         target bean
      * @param xid                   the global transaction id
      * @param branchId              the branch transaction id
-     * @param status                the tcc fence status
+     * @param status                the common fence status
      * @return the boolean
      */
     private static boolean updateStatusAndInvokeTargetMethod(Connection conn, Method method, Object targetTCCBean,

spring/src/main/java/io/seata/spring/annotation/GlobalTransactionScanner.java

@@ -25,6 +25,7 @@
 import java.util.concurrent.atomic.AtomicBoolean;
 import javax.annotation.Nullable;
 
+import com.google.common.collect.ImmutableSet;
 import io.seata.common.util.CollectionUtils;
 import io.seata.common.util.StringUtils;
 import io.seata.config.ConfigurationCache;
@@ -42,6 +43,8 @@
 import io.seata.integration.tx.api.interceptor.handler.GlobalTransactionalInterceptorHandler;
 import io.seata.integration.tx.api.interceptor.handler.ProxyInvocationHandler;
 import io.seata.integration.tx.api.interceptor.parser.DefaultInterfaceParser;
+import io.seata.integration.tx.api.interceptor.parser.IfNeedEnhanceBean;
+import io.seata.integration.tx.api.interceptor.parser.NeedEnhanceEnum;
 import io.seata.integration.tx.api.remoting.parser.DefaultRemotingParser;
 import io.seata.rm.RMClient;
 import io.seata.spring.annotation.scannercheckers.PackageScannerChecker;
@@ -62,10 +65,13 @@
 import org.springframework.aop.framework.autoproxy.AbstractAutoProxyCreator;
 import org.springframework.aop.support.AopUtils;
 import org.springframework.beans.BeansException;
+import org.springframework.beans.PropertyValue;
 import org.springframework.beans.factory.DisposableBean;
 import org.springframework.beans.factory.FactoryBean;
 import org.springframework.beans.factory.InitializingBean;
+import org.springframework.beans.factory.config.BeanDefinition;
 import org.springframework.beans.factory.config.ConfigurableListableBeanFactory;
+import org.springframework.beans.factory.config.RuntimeBeanReference;
 import org.springframework.context.ApplicationContext;
 import org.springframework.context.ApplicationContextAware;
 import org.springframework.context.ConfigurableApplicationContext;
@@ -115,6 +121,8 @@ public class GlobalTransactionScanner extends AbstractAutoProxyCreator
 
     private ApplicationContext applicationContext;
 
+    private static final Set<String> NEED_ENHANCE_BEAN_NAME_SET = new HashSet<>();
+
 
     /**
      * Instantiates a new Global transaction scanner.
@@ -281,6 +289,9 @@ protected Object wrapIfNecessary(Object bean, String beanName, Object cacheKey)
                 if (PROXYED_SET.contains(beanName)) {
                     return bean;
                 }
+                if (!NEED_ENHANCE_BEAN_NAME_SET.contains(beanName)) {
+                    return bean;
+                }
                 interceptor = null;
                 ProxyInvocationHandler proxyInvocationHandler = DefaultInterfaceParser.get().parserInterfaceToProxy(bean, beanName);
                 if (proxyInvocationHandler == null) {
@@ -470,8 +481,68 @@ public void afterPropertiesSet() {
         if (initialized.compareAndSet(false, true)) {
             initClient();
         }
+
+        this.findBusinessBeanNamesNeededEnhancement();
+    }
+
+    private void findBusinessBeanNamesNeededEnhancement() {
+        if (applicationContext instanceof ConfigurableApplicationContext) {
+            ConfigurableApplicationContext configurableApplicationContext = (ConfigurableApplicationContext) applicationContext;
+            ConfigurableListableBeanFactory configurableListableBeanFactory = configurableApplicationContext.getBeanFactory();
+
+            String[] beanNames = applicationContext.getBeanDefinitionNames();
+            for (String contextBeanName : beanNames) {
+                BeanDefinition beanDefinition = configurableListableBeanFactory.getBeanDefinition(contextBeanName);
+                if (IGNORE_ENHANCE_CHECK_SET.contains(beanDefinition.getBeanClassName())) {
+                    continue;
+                }
+                try {
+                    // get the class by bean definition class name
+                    Class<?> beanClass = Class.forName(beanDefinition.getBeanClassName());
+                    // check if it needs enhancement by the class
+                    IfNeedEnhanceBean ifNeedEnhanceBean = DefaultInterfaceParser.get().parseIfNeedEnhancement(beanClass);
+                    if (!ifNeedEnhanceBean.isIfNeed()) {
+                        continue;
+                    }
+                    if (ifNeedEnhanceBean.getNeedEnhanceEnum().equals(NeedEnhanceEnum.SERVICE_BEAN)) {
+                        // the native bean which dubbo, sofa bean service bean referenced
+                        PropertyValue propertyValue = beanDefinition.getPropertyValues().getPropertyValue("ref");
+                        if (propertyValue == null) {
+                            // the native bean which HSF service bean referenced
+                            propertyValue = beanDefinition.getPropertyValues().getPropertyValue("target");
+                        }
+                        if (propertyValue != null) {
+                            RuntimeBeanReference r = (RuntimeBeanReference) propertyValue.getValue();
+                            if (r != null && StringUtils.isNotBlank(r.getBeanName())) {
+                                NEED_ENHANCE_BEAN_NAME_SET.add(r.getBeanName());
+                                continue;
+                            }
+                        }
+                        // the native bean which local tcc service bean referenced
+                        NEED_ENHANCE_BEAN_NAME_SET.add(contextBeanName);
+                    } else if (ifNeedEnhanceBean.getNeedEnhanceEnum().equals(NeedEnhanceEnum.GLOBAL_TRANSACTIONAL_BEAN)) {
+                        // global transactional bean
+                        NEED_ENHANCE_BEAN_NAME_SET.add(contextBeanName);
+                    }
+                } catch (ClassNotFoundException e) {
+                    LOGGER.warn("check if need enhance bean error, it can be ignore", e);
+                }
+            }
+            LOGGER.info("The needed enhancement business beans are : {}", NEED_ENHANCE_BEAN_NAME_SET);
+        }
     }
 
+    private static final Set<String> IGNORE_ENHANCE_CHECK_SET = ImmutableSet.of(
+            "io.seata.spring.annotation.GlobalTransactionScanner"
+            , "io.seata.rm.fence.SpringFenceConfig"
+            , "org.springframework.context.annotation.internalConfigurationAnnotationProcessor"
+            , "org.springframework.context.annotation.internalAutowiredAnnotationProcessor"
+            , "org.springframework.context.annotation.internalCommonAnnotationProcessor"
+            , "org.springframework.context.event.internalEventListenerProcessor"
+            , "org.springframework.context.event.internalEventListenerFactory"
+    );
+
+
     @Override
     public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
         this.applicationContext = applicationContext;

spring/src/main/java/io/seata/spring/remoting/parser/RemotingFactoryBeanParser.java

@@ -74,6 +74,11 @@ public boolean isService(Object bean, String beanName) {
         return DefaultRemotingParser.get().isService(factoryBean, getFactoryBeanName(beanName));
     }
 
+    @Override
+    public boolean isService(Class<?> beanClass) throws FrameworkException {
+        return false;
+    }
+
     @Override
     public RemotingDesc getServiceDesc(Object bean, String beanName) throws FrameworkException {
         Object factoryBean = getRemotingFactoryBean(bean, beanName);