init

ruoyi-common/ruoyi-common-job/src/main/java/org/dromara/common/job/config/SnailJobConfig.java

@@ -0,0 +1,78 @@
+package org.dromara.common.job.config;
+
+import ch.qos.logback.classic.Logger;
+import ch.qos.logback.classic.LoggerContext;
+import ch.qos.logback.classic.spi.ILoggingEvent;
+import cn.hutool.core.collection.CollUtil;
+import com.aizuda.snailjob.client.common.appender.SnailLogbackAppender;
+import com.aizuda.snailjob.client.common.event.SnailChannelReconnectEvent;
+import com.aizuda.snailjob.client.common.event.SnailClientStartingEvent;
+import com.aizuda.snailjob.client.starter.EnableSnailJob;
+import org.dromara.common.core.utils.StringUtils;
+import org.dromara.common.job.config.properties.SnailJobServerProperties;
+import org.slf4j.LoggerFactory;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.boot.autoconfigure.AutoConfiguration;
+import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
+import org.springframework.boot.context.properties.EnableConfigurationProperties;
+import org.springframework.cloud.client.ServiceInstance;
+import org.springframework.cloud.client.discovery.DiscoveryClient;
+import org.springframework.context.event.EventListener;
+import org.springframework.scheduling.annotation.EnableScheduling;
+
+import java.util.List;
+
+/**
+ * 启动定时任务
+ *
+ * @author dhb52
+ * @since 2024/3/12
+ */
+@AutoConfiguration
+@EnableConfigurationProperties(SnailJobServerProperties.class)
+@ConditionalOnProperty(prefix = "snail-job", name = "enabled", havingValue = "true")
+@EnableScheduling
+@EnableSnailJob
+public class SnailJobConfig {
+
+    @Autowired
+    private SnailJobServerProperties properties;
+    @Autowired
+    private DiscoveryClient discoveryClient;
+
+    @EventListener(SnailClientStartingEvent.class)
+    public void onStarting(SnailClientStartingEvent event) {
+        // 从 nacos 获取 server 服务连接
+        registerServer();
+        // 注册 日志监控配置
+        registerLogging();
+    }
+
+    @EventListener(SnailChannelReconnectEvent.class)
+    public void onReconnect(SnailChannelReconnectEvent event) {
+        // 连接中断 重新从 nacos 获取存活的服务连接(高可用配置)
+        registerServer();
+    }
+
+    private void registerServer() {
+        String serverName = properties.getServerName();
+        if (StringUtils.isNotBlank(serverName)) {
+            List<ServiceInstance> instances = discoveryClient.getInstances(serverName);
+            if (CollUtil.isNotEmpty(instances)) {
+                ServiceInstance instance = instances.get(0);
+                System.setProperty("snail-job.server.host", instance.getHost());
+                System.setProperty("snail-job.server.port", properties.getPort());
+            }
+        }
+    }
+
+    private void registerLogging() {
+        LoggerContext lc = (LoggerContext) LoggerFactory.getILoggerFactory();
+        SnailLogbackAppender<ILoggingEvent> ca = new SnailLogbackAppender<>();
+        ca.setName("snail_log_appender");
+        ca.start();
+        Logger rootLogger = lc.getLogger(Logger.ROOT_LOGGER_NAME);
+        rootLogger.addAppender(ca);
+    }
+
+}

ruoyi-common/ruoyi-common-job/src/main/java/org/dromara/common/job/config/properties/SnailJobServerProperties.java

@@ -0,0 +1,14 @@
+package org.dromara.common.job.config.properties;
+
+import lombok.Data;
+import org.springframework.boot.context.properties.ConfigurationProperties;
+
+@Data
+@ConfigurationProperties(prefix = "snail-job.server")
+public class SnailJobServerProperties {
+
+    private String serverName;
+
+    private String port;
+
+}

ruoyi-common/ruoyi-common-job/src/main/resources/META-INF/spring/org.springframework.boot.autoconfigure.AutoConfiguration.imports

@@ -0,0 +1 @@
+org.dromara.common.job.config.SnailJobConfig