Merge branch 'feature/1.0.0' of github.com:ab-inbev-apac/abi-cloud-qr-platform into feature/1.0.0

abi-cloud-qr-platform-server/src/main/java/com/abi/qms/platform/runner/CheckTableStoreRunner.java

@@ -1,5 +1,6 @@
 package com.abi.qms.platform.runner;
 
+import com.abi.qms.platform.dao.tablestore.entity.QrCode;
 import com.abi.task.common.api.exception.BusinessException;
 import com.abi.task.common.tablestore.TableStoreUtils;
 import com.abi.task.common.tablestore.common.TableStore;
@@ -9,6 +10,9 @@ import org.apache.commons.lang3.ObjectUtils;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.boot.ApplicationArguments;
 import org.springframework.boot.ApplicationRunner;
+import org.springframework.boot.actuate.beans.BeansEndpoint;
+import org.springframework.context.ApplicationContext;
+import org.springframework.context.ApplicationContextAware;
 import org.springframework.core.io.Resource;
 import org.springframework.core.io.support.PathMatchingResourcePatternResolver;
 import org.springframework.core.type.classreading.CachingMetadataReaderFactory;
@@ -27,13 +31,16 @@ import java.util.concurrent.atomic.AtomicInteger;
  * @version: 1.0
  * @todo:
  */
-//@Component
+@Component
 @Slf4j
 public class CheckTableStoreRunner implements ApplicationRunner {
 
     @Autowired
     private TableStoreUtils tableStoreUtils;
 
+    @Autowired
+    private ApplicationContextAware applicationContextAware;
+
     private static AtomicInteger retryTimes = new AtomicInteger(0);
     private static String RESOURCE_PATH = "classpath*:com/abi/qms/platform/dao/tablestore/entity/*.class";
 
@@ -54,34 +61,35 @@ public class CheckTableStoreRunner implements ApplicationRunner {
 
     private static Set<TableStore> getAllEntity(){
         Set<TableStore> ret = Sets.newHashSet();
-        PathMatchingResourcePatternResolver pathMatchingResourcePatternResolver = new PathMatchingResourcePatternResolver();
-        CachingMetadataReaderFactory cachingMetadataReaderFactory = new CachingMetadataReaderFactory();
-        try {
-            Resource[] resources = pathMatchingResourcePatternResolver.getResources(RESOURCE_PATH);
-            ClassLoader loader = ClassLoader.getSystemClassLoader();
-            for (Resource resource : resources) {
-                MetadataReader reader = cachingMetadataReaderFactory.getMetadataReader(resource);
-                String className = reader.getClassMetadata().getClassName();
-                Class aClass = loader.loadClass(className);
-                TableStore annotation = (TableStore) aClass.getAnnotation(TableStore.class);
-                if (ObjectUtils.isNotEmpty(annotation)){
-                    ret.add(annotation);
-                }
-            }
-        } catch (Exception e) {
-            if (retryTimes.getAndIncrement()>2){
-                log.error("TableStore 初始化检查失败",e);
-                throw new BusinessException(500,"TableStore 初始化检查失败");
-            }
-            try {
-                Thread.sleep(retryTimes.get() * 1000);
-            } catch (InterruptedException interruptedException) {
-                // Do Nothing
-            }
-            //进行当前重试
-            log.info("当前开始重试,第{}次",retryTimes.get());
-            return getAllEntity();
-        }
+        ret.add(QrCode.class.getAnnotation(TableStore.class));
+//        PathMatchingResourcePatternResolver pathMatchingResourcePatternResolver = new PathMatchingResourcePatternResolver();
+//        CachingMetadataReaderFactory cachingMetadataReaderFactory = new CachingMetadataReaderFactory();
+//        try {
+//            Resource[] resources = pathMatchingResourcePatternResolver.getResources(RESOURCE_PATH);
+//            ClassLoader loader = ClassLoader.getSystemClassLoader();
+//            for (Resource resource : resources) {
+//                MetadataReader reader = cachingMetadataReaderFactory.getMetadataReader(resource);
+//                String className = reader.getClassMetadata().getClassName();
+//                Class aClass = loader.loadClass(className);
+//                TableStore annotation = (TableStore) aClass.getAnnotation(TableStore.class);
+//                if (ObjectUtils.isNotEmpty(annotation)){
+//                    ret.add(annotation);
+//                }
+//            }
+//        } catch (Exception e) {
+//            if (retryTimes.getAndIncrement()>2){
+//                log.error("TableStore 初始化检查失败",e);
+//                throw new BusinessException(500,"TableStore 初始化检查失败");
+//            }
+//            try {
+//                Thread.sleep(retryTimes.get() * 1000);
+//            } catch (InterruptedException interruptedException) {
+//                // Do Nothing
+//            }
+//            //进行当前重试
+//            log.info("当前开始重试,第{}次",retryTimes.get());
+//            return getAllEntity();
+//        }
         return ret;
     }