feature: First support native-image for seata-client

all/pom.xml

@@ -212,6 +212,26 @@
                 </exclusion>
             </exclusions>
         </dependency>
+        <dependency>
+            <groupId>io.seata</groupId>
+            <artifactId>seata-spring-aot-core</artifactId>
+            <version>${project.version}</version>
+        </dependency>
+        <dependency>
+            <groupId>io.seata</groupId>
+            <artifactId>seata-spring-aot-client</artifactId>
+            <version>${project.version}</version>
+        </dependency>
+        <dependency>
+            <groupId>io.seata</groupId>
+            <artifactId>seata-spring-aot-nacos</artifactId>
+            <version>${project.version}</version>
+        </dependency>
+        <dependency>
+            <groupId>io.seata</groupId>
+            <artifactId>seata-spring-aot-zk</artifactId>
+            <version>${project.version}</version>
+        </dependency>
         <dependency>
             <groupId>io.seata</groupId>
             <artifactId>seata-tcc</artifactId>
@@ -623,6 +643,13 @@
             <plugin>
                 <groupId>org.apache.maven.plugins</groupId>
                 <artifactId>maven-shade-plugin</artifactId>
+                <dependencies>
+                    <dependency>
+                        <groupId>org.springframework.boot</groupId>
+                        <artifactId>spring-boot-maven-plugin</artifactId>
+                        <version>${spring-boot.version}</version>
+                    </dependency>
+                </dependencies>
                 <executions>
                     <execution>
                         <phase>package</phase>
@@ -641,17 +668,17 @@
                             </artifactSet>
                             <transformers>
                                 <!-- META-INF/services -->
-                                <transformer
-                                        implementation="org.apache.maven.plugins.shade.resource.ServicesResourceTransformer"/>
+                                <transformer implementation="org.apache.maven.plugins.shade.resource.ServicesResourceTransformer"/>
                                 <!-- spring相关 -->
-                                <transformer
-                                        implementation="org.apache.maven.plugins.shade.resource.AppendingTransformer">
+                                <transformer implementation="org.apache.maven.plugins.shade.resource.AppendingTransformer">
                                     <resource>META-INF/spring.handlers</resource>
                                 </transformer>
-                                <transformer
-                                        implementation="org.apache.maven.plugins.shade.resource.AppendingTransformer">
+                                <transformer implementation="org.apache.maven.plugins.shade.resource.AppendingTransformer">
                                     <resource>META-INF/spring.schemas</resource>
                                 </transformer>
+                                <transformer implementation="org.springframework.boot.maven.PropertiesMergingResourceTransformer">
+                                    <resource>META-INF/spring/aot.factories</resource>
+                                </transformer>
                             </transformers>
                             <filters>
                                 <filter>

build/pom.xml

@@ -103,7 +103,7 @@
         <maven-source-plugin.version>2.2.1</maven-source-plugin.version>
         <maven-resources-plugin.version>3.2.0</maven-resources-plugin.version>
         <maven-jar-plugin.version>3.2.2</maven-jar-plugin.version>
-        <maven-shade-plugin.version>2.4.3</maven-shade-plugin.version>
+        <maven-shade-plugin.version>3.2.4</maven-shade-plugin.version>
         <maven-dependency-plugin.version>3.0.2</maven-dependency-plugin.version>
         <maven-assembly-plugin.version>3.0.0</maven-assembly-plugin.version>
         <jib-maven-plugin.version>3.2.0</jib-maven-plugin.version>

common/src/main/java/io/seata/common/aot/NativeUtils.java

@@ -0,0 +1,63 @@
+/*
+ *  Copyright 1999-2019 Seata.io Group.
+ *
+ *  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 io.seata.common.aot;
+
+/**
+ * The native utils
+ *
+ * @author wang.liang
+ */
+public class NativeUtils {
+
+    /**
+     * The native-image code
+     *
+     * @see <a href="https://github.com/oracle/graal/blob/master/sdk/src/org.graalvm.nativeimage/src/org/graalvm/nativeimage/ImageInfo.java">ImageInfo.java</a>
+     */
+    private static final String NATIVE_IMAGE_CODE = System.getProperty("org.graalvm.nativeimage.imagecode");
+
+
+    public static final String SPRING_AOT_PROCESSING = "spring.aot.processing";
+
+
+    /**
+     * Whether Spring-AOT processing
+     *
+     * @return the boolean
+     */
+    public static boolean isSpringAotProcessing() {
+        return "true".equalsIgnoreCase(System.getProperty(SPRING_AOT_PROCESSING));
+    }
+
+    /**
+     * Gets the native-image code.
+     *
+     * @return the native-image code
+     */
+    public static String getNativeImageCode() {
+        return NATIVE_IMAGE_CODE;
+    }
+
+    /**
+     * Whether run in native-image
+     *
+     * @return the boolean
+     * @see org.springframework.core.NativeDetector#inNativeImage()
+     */
+    public static boolean inNativeImage() {
+        return NATIVE_IMAGE_CODE != null;
+    }
+}

common/src/main/java/io/seata/common/util/ReflectionUtil.java

@@ -130,6 +130,25 @@ public static Class<?> getWrappedClass(Class<?> clazz) {
         return clazz;
     }
 
+    public static boolean isJavaClass(Class<?> clazz) {
+        return clazz != null && clazz.getClassLoader() == null;
+    }
+
+    /**
+     * Whether the class exists
+     *
+     * @param className the class name
+     * @return the boolean
+     */
+    public static boolean existsClass(String className) {
+        try {
+            getClassByName(className);
+            return true;
+        } catch (ClassNotFoundException e) {
+            return false;
+        }
+    }
+
     //endregion
 
 

config/seata-config-core/pom.xml

@@ -40,10 +40,6 @@
             <groupId>org.yaml</groupId>
             <artifactId>snakeyaml</artifactId>
         </dependency>
-        <dependency>
-            <groupId>net.bytebuddy</groupId>
-            <artifactId>byte-buddy</artifactId>
-        </dependency>
     </dependencies>
 
 </project>

config/seata-config-core/src/main/java/io/seata/config/ConfigurationCache.java

@@ -15,17 +15,16 @@
  */
 package io.seata.config;
 
+import java.lang.reflect.Proxy;
 import java.util.HashMap;
 import java.util.HashSet;
 import java.util.Map;
 import java.util.Objects;
 import java.util.concurrent.ConcurrentHashMap;
+
 import io.seata.common.util.CollectionUtils;
 import io.seata.common.util.DurationUtil;
 import io.seata.common.util.StringUtils;
-import net.bytebuddy.ByteBuddy;
-import net.bytebuddy.implementation.InvocationHandlerAdapter;
-import net.bytebuddy.matcher.ElementMatchers;
 
 /**
  * @author funkye
@@ -101,8 +100,8 @@ public void onChangeEvent(ConfigurationChangeEvent event) {
     }
 
     public Configuration proxy(Configuration originalConfiguration) throws Exception {
-        return new ByteBuddy().subclass(Configuration.class).method(ElementMatchers.any())
-            .intercept(InvocationHandlerAdapter.of((proxy, method, args) -> {
+        return (Configuration)Proxy.newProxyInstance(this.getClass().getClassLoader(), new Class[]{Configuration.class}
+            , (proxy, method, args) -> {
                 String methodName = method.getName();
                 if (methodName.startsWith(METHOD_PREFIX) && !methodName.equalsIgnoreCase(METHOD_LATEST_CONFIG)) {
                     String rawDataId = (String)args[0];
@@ -126,8 +125,8 @@ public Configuration proxy(Configuration originalConfiguration) throws Exception
                     return wrapper == null ? null : wrapper.convertData(type);
                 }
                 return method.invoke(originalConfiguration, args);
-            })).make().load(originalConfiguration.getClass().getClassLoader()).getLoaded().getDeclaredConstructor()
-            .newInstance();
+            }
+        );
     }
 
     private static class ConfigurationCacheInstance {

config/seata-config-zk/src/main/java/io/seata/config/zk/ZookeeperConfiguration.java

@@ -68,13 +68,13 @@ public class ZookeeperConfiguration extends AbstractConfiguration {
     private static final String CONNECT_TIMEOUT_KEY = "connectTimeout";
     private static final String AUTH_USERNAME = "username";
     private static final String AUTH_PASSWORD = "password";
-    private static final String SERIALIZER_KEY = "serializer";
+    public static final String SERIALIZER_KEY = "serializer";
     private static final String CONFIG_PATH_KEY = "nodePath";
     private static final int THREAD_POOL_NUM = 1;
     private static final int DEFAULT_SESSION_TIMEOUT = 6000;
     private static final int DEFAULT_CONNECT_TIMEOUT = 2000;
     private static final String DEFAULT_CONFIG_PATH = ROOT_PATH + "/seata.properties";
-    private static final String FILE_CONFIG_KEY_PREFIX = FILE_ROOT_CONFIG + FILE_CONFIG_SPLIT_CHAR + CONFIG_TYPE
+    public static final String FILE_CONFIG_KEY_PREFIX = FILE_ROOT_CONFIG + FILE_CONFIG_SPLIT_CHAR + CONFIG_TYPE
             + FILE_CONFIG_SPLIT_CHAR;
     private static final ExecutorService CONFIG_EXECUTOR = new ThreadPoolExecutor(THREAD_POOL_NUM, THREAD_POOL_NUM,
             Integer.MAX_VALUE, TimeUnit.MILLISECONDS, new LinkedBlockingQueue<>(),

core/src/main/java/io/seata/core/serializer/SerializerServiceLoader.java

@@ -30,7 +30,7 @@ private SerializerServiceLoader() {
     }
 
 
-    private static final String PROTOBUF_SERIALIZER_CLASS_NAME = "io.seata.serializer.protobuf.ProtobufSerializer";
+    public static final String PROTOBUF_SERIALIZER_CLASS_NAME = "io.seata.serializer.protobuf.ProtobufSerializer";
 
     /**
      * Load the service of {@link Serializer}

discovery/seata-discovery-eureka/src/main/resources/META-INF/native-image/io.seata/seata-discovery-eureka/generated/handlers/reflect-config.json

@@ -0,0 +1,9 @@
+[
+  {
+    "condition": {
+      "typeReachable": "com.netflix.discovery.DiscoveryClient"
+    },
+    "name": "com.thoughtworks.xstream.XStream",
+    "allPublicMethods": true
+  }
+]

discovery/seata-discovery-eureka/src/main/resources/META-INF/native-image/io.seata/seata-discovery-eureka/native-image.properties

@@ -0,0 +1,2 @@
+Args = --initialize-at-build-time=org.apache.commons.logging.LogFactory \
+       --initialize-at-build-time=org.apache.commons.logging.LogFactoryService

pom.xml

@@ -53,6 +53,8 @@
         <module>rm</module>
         <module>rm-datasource</module>
         <module>spring</module>
+        <module>spring-aot</module>
+        <module>spring-framework-fake-for-java8</module>
         <module>tcc</module>
         <module>test</module>
         <module>tm</module>

rm-datasource/src/main/resources/META-INF/native-image/io.seata/seata-rm-datasource/native-image.properties

@@ -0,0 +1 @@
+Args = -H:+AddAllCharsets

seata-spring-autoconfigure/seata-spring-autoconfigure-core/pom.xml

@@ -36,5 +36,19 @@
             <version>${project.version}</version>
             <scope>test</scope>
         </dependency>
+
+        <dependency>
+            <groupId>${project.groupId}</groupId>
+            <artifactId>seata-spring-framework-fake-for-java8</artifactId>
+            <version>${project.version}</version>
+            <scope>provided</scope>
+        </dependency>
+        <!-- During development, replace `seata-spring-framework-fake-for-java8` with the following dependency. -->
+        <!--<dependency>
+            <groupId>org.springframework</groupId>
+            <artifactId>spring-beans</artifactId>
+            <version>6.0.4</version>
+            <scope>provided</scope>
+        </dependency>-->
     </dependencies>
 </project>

seata-spring-autoconfigure/seata-spring-autoconfigure-core/src/main/java/io/seata/spring/aot/SeataPropertiesBeanRegistrationAotProcessor.java

@@ -0,0 +1,71 @@
+/*
+ *  Copyright 1999-2019 Seata.io Group.
+ *
+ *  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 io.seata.spring.aot;
+
+import io.seata.common.util.ReflectionUtil;
+import io.seata.spring.boot.autoconfigure.StarterConstants;
+import org.springframework.aot.generate.GenerationContext;
+import org.springframework.aot.hint.MemberCategory;
+import org.springframework.aot.hint.ReflectionHints;
+import org.springframework.beans.factory.aot.BeanRegistrationAotContribution;
+import org.springframework.beans.factory.aot.BeanRegistrationAotProcessor;
+import org.springframework.beans.factory.aot.BeanRegistrationCode;
+import org.springframework.beans.factory.support.RegisteredBean;
+
+/**
+ * The seata properties bean registration AOT processor
+ *
+ * @author wang.liang
+ */
+class SeataPropertiesBeanRegistrationAotProcessor implements BeanRegistrationAotProcessor {
+
+    @Override
+    public BeanRegistrationAotContribution processAheadOfTime(RegisteredBean registeredBean) {
+        if (ReflectionUtil.existsClass("io.seata.spring.boot.autoconfigure.StarterConstants")) {
+            Class<?> beanClass = registeredBean.getBeanClass();
+            if (StarterConstants.PROPERTY_BEAN_MAP.containsValue(beanClass)) {
+                return new SeataPropertiesBeanRegistrationAotContribution(beanClass);
+            }
+        }
+        return null;
+    }
+
+
+    /**
+     * The seata properties bean registration AOT contribution
+     */
+    private static class SeataPropertiesBeanRegistrationAotContribution implements BeanRegistrationAotContribution {
+
+        private final Class<?> propertiesBeanClass;
+
+
+        public SeataPropertiesBeanRegistrationAotContribution(Class<?> propertiesBeanClass) {
+            this.propertiesBeanClass = propertiesBeanClass;
+        }
+
+
+        @Override
+        public void applyTo(GenerationContext generationContext, BeanRegistrationCode beanRegistrationCode) {
+            ReflectionHints reflectionHints = generationContext.getRuntimeHints().reflection();
+
+            // register the properties bean class
+            // See SpringBootConfigurationProvider#getDefaultValueFromPropertyObject(...)
+            AotUtils.registerType(reflectionHints, propertiesBeanClass, MemberCategory.DECLARED_FIELDS);
+        }
+
+    }
+
+}

seata-spring-autoconfigure/seata-spring-autoconfigure-core/src/main/java/io/seata/spring/boot/autoconfigure/provider/SpringBootConfigurationProvider.java

@@ -16,6 +16,7 @@
 package io.seata.spring.boot.autoconfigure.provider;
 
 import java.lang.reflect.Field;
+import java.lang.reflect.Proxy;
 import java.time.Duration;
 import java.util.Map;
 import java.util.Objects;
@@ -32,8 +33,6 @@
 import org.apache.commons.lang.StringUtils;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
-import org.springframework.cglib.proxy.Enhancer;
-import org.springframework.cglib.proxy.MethodInterceptor;
 import org.springframework.core.env.ConfigurableEnvironment;
 import org.springframework.lang.Nullable;
 
@@ -60,8 +59,8 @@ public class SpringBootConfigurationProvider implements ExtConfigurationProvider
 
     @Override
     public Configuration provide(Configuration originalConfiguration) {
-        return (Configuration)Enhancer.create(originalConfiguration.getClass(),
-            (MethodInterceptor)(proxy, method, args, methodProxy) -> {
+        return (Configuration)Proxy.newProxyInstance(this.getClass().getClassLoader(), new Class[]{Configuration.class}
+            , (proxy, method, args) -> {
                 if (method.getName().startsWith(INTERCEPT_METHOD_PREFIX) && args.length > 0) {
                     Object result;
                     String rawDataId = (String)args[0];

seata-spring-autoconfigure/seata-spring-autoconfigure-core/src/main/resources/META-INF/spring/aot.factories

@@ -0,0 +1,2 @@
+org.springframework.beans.factory.aot.BeanRegistrationAotProcessor=\
+io.seata.spring.aot.SeataPropertiesBeanRegistrationAotProcessor