重写ElasticJobAutoConfiguration获取注解方法

brave-spring-boot-starter/pom.xml

@@ -52,6 +52,12 @@
             <version>${elastic.job.version}</version>
             <optional>true</optional>
         </dependency>
+        <dependency>
+            <groupId>com.github.kuhn-he</groupId>
+            <artifactId>elastic-job-lite-spring-boot-starter</artifactId>
+            <version>${elastic.job.version}</version>
+            <optional>true</optional>
+        </dependency>
         <dependency>
             <groupId>org.apache.httpcomponents</groupId>
             <artifactId>httpclient</artifactId>

brave-spring-boot-starter/src/main/java/cn/quantgroup/tech/brave/configuration/BraveAutoConfiguration.java

@@ -16,22 +16,32 @@ import cn.quantgroup.tech.brave.service.*;
 import cn.quantgroup.tech.brave.service.impl.*;
 import cn.quantgroup.tech.brave.slf4j.MDCCurrentTraceContext;
 import com.dangdang.ddframe.job.api.ElasticJob;
+import com.dangdang.ddframe.job.api.simple.SimpleJob;
+import com.dangdang.ddframe.job.config.JobCoreConfiguration;
+import com.dangdang.ddframe.job.config.simple.SimpleJobConfiguration;
+import com.dangdang.ddframe.job.event.rdb.JobEventRdbConfiguration;
+import com.dangdang.ddframe.job.lite.config.LiteJobConfiguration;
+import com.dangdang.ddframe.job.lite.spring.api.SpringJobScheduler;
+import com.dangdang.ddframe.job.reg.zookeeper.ZookeeperConfiguration;
+import com.dangdang.ddframe.job.reg.zookeeper.ZookeeperRegistryCenter;
+import com.dangdang.elasticjob.lite.annotation.ElasticSimpleJob;
 import lombok.extern.slf4j.Slf4j;
 import okhttp3.Dispatcher;
 import okhttp3.OkHttpClient;
+import org.apache.commons.lang3.StringUtils;
 import org.apache.http.client.HttpClient;
 import org.apache.http.impl.client.HttpClientBuilder;
 import org.aspectj.lang.ProceedingJoinPoint;
 import org.aspectj.lang.annotation.Around;
 import org.aspectj.lang.annotation.Aspect;
 import org.springframework.amqp.rabbit.core.RabbitTemplate;
+import org.springframework.aop.support.AopUtils;
 import org.springframework.beans.factory.annotation.Autowired;
-import org.springframework.boot.autoconfigure.condition.ConditionalOnClass;
-import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;
-import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
-import org.springframework.boot.autoconfigure.condition.ConditionalOnWebApplication;
+import org.springframework.beans.factory.annotation.Value;
+import org.springframework.boot.autoconfigure.condition.*;
 import org.springframework.boot.context.properties.EnableConfigurationProperties;
 import org.springframework.boot.web.servlet.FilterRegistrationBean;
+import org.springframework.context.ApplicationContext;
 import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.Configuration;
 import org.springframework.context.annotation.Import;
@@ -45,9 +55,12 @@ import zipkin2.reporter.AsyncReporter;
 import zipkin2.reporter.Sender;
 import zipkin2.reporter.kafka11.KafkaSender;
 
+import javax.annotation.PostConstruct;
 import javax.servlet.Filter;
+import javax.sql.DataSource;
 import java.util.ArrayList;
 import java.util.List;
+import java.util.Map;
 
 /**
  * 注册brave
@@ -445,6 +458,54 @@ public class BraveAutoConfiguration {
                 }
             }
         }
+
+        @Configuration
+        @ConditionalOnClass(ElasticSimpleJob.class)
+        @ConditionalOnExpression("'${elaticjob.zookeeper.server-lists}'.length() > 0")
+        @ConditionalOnProperty(prefix = "tech.brave", name = "enabled", havingValue = "true", matchIfMissing = true)
+        public class ElasticJobAutoConfiguration {
+
+            @Value("${elaticjob.zookeeper.server-lists}")
+            private String serverList;
+
+            @Value("${elaticjob.zookeeper.namespace}")
+            private String namespace;
+
+            @Autowired
+            private ApplicationContext applicationContext;
+
+            @PostConstruct
+            public void initElasticJob() {
+                ZookeeperRegistryCenter regCenter = new ZookeeperRegistryCenter(new ZookeeperConfiguration(serverList, namespace));
+                regCenter.init();
+                Map<String, SimpleJob> map = applicationContext.getBeansOfType(SimpleJob.class);
+
+                for (Map.Entry<String, SimpleJob> entry : map.entrySet()) {
+                    SimpleJob simpleJob = entry.getValue();
+                    ElasticSimpleJob elasticSimpleJobAnnotation = AopUtils.getTargetClass(simpleJob).getAnnotation(ElasticSimpleJob.class);
+
+                    String cron = StringUtils.defaultIfBlank(elasticSimpleJobAnnotation.cron(), elasticSimpleJobAnnotation.value());
+                    SimpleJobConfiguration simpleJobConfiguration = new SimpleJobConfiguration(JobCoreConfiguration.newBuilder(simpleJob.getClass().getName(), cron, elasticSimpleJobAnnotation.shardingTotalCount()).shardingItemParameters(elasticSimpleJobAnnotation.shardingItemParameters()).build(), simpleJob.getClass().getCanonicalName());
+                    LiteJobConfiguration liteJobConfiguration = LiteJobConfiguration.newBuilder(simpleJobConfiguration).overwrite(true).build();
+
+                    String dataSourceRef = elasticSimpleJobAnnotation.dataSource();
+                    if (StringUtils.isNotBlank(dataSourceRef)) {
+
+                        if (!applicationContext.containsBean(dataSourceRef)) {
+                            throw new RuntimeException("not exist datasource [" + dataSourceRef + "] !");
+                        }
+
+                        DataSource dataSource = (DataSource) applicationContext.getBean(dataSourceRef);
+                        JobEventRdbConfiguration jobEventRdbConfiguration = new JobEventRdbConfiguration(dataSource);
+                        SpringJobScheduler jobScheduler = new SpringJobScheduler(simpleJob, regCenter, liteJobConfiguration, jobEventRdbConfiguration);
+                        jobScheduler.init();
+                    } else {
+                        SpringJobScheduler jobScheduler = new SpringJobScheduler(simpleJob, regCenter, liteJobConfiguration);
+                        jobScheduler.init();
+                    }
+                }
+            }
+        }
     }
 }