增加配置参数通过IP地址限制定时任务执行

肖超群
1 parent aeea98bd
Showing 5 changed files with 77 additions and 6 deletions
huaheng-wms-core/src/main/java/org/jeecg/JeecgSystemApplication.java
huaheng-wms-core/src/main/java/org/jeecg/modules/quartz/service/impl/QuartzJobServiceImpl.java
huaheng-wms-core/src/main/java/org/jeecg/modules/wms/monitor/job/MyTriggerListener.java
huaheng-wms-core/src/main/java/org/jeecg/utils/constant/QuantityConstant.java
huaheng-wms-core/src/main/resources/application.yml
@@ -6,6 +6,10 @@ import java.util.Enumeration;
 import javax.transaction.SystemException;
  
 import org.jeecg.common.util.oConvertUtils;
+import org.jeecg.modules.wms.monitor.job.MyTriggerListener;
+import org.jeecg.utils.SpringUtils;
+import org.quartz.Scheduler;
+import org.springframework.boot.CommandLineRunner;
 import org.springframework.boot.SpringApplication;
 import org.springframework.boot.autoconfigure.EnableAutoConfiguration;
 import org.springframework.boot.autoconfigure.SpringBootApplication;
@@ -14,6 +18,7 @@ import org.springframework.boot.builder.SpringApplicationBuilder;
 import org.springframework.boot.web.servlet.support.SpringBootServletInitializer;
 import org.springframework.cache.annotation.EnableCaching;
 import org.springframework.context.ConfigurableApplicationContext;
+import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.ComponentScan;
 import org.springframework.context.annotation.ComponentScan.Filter;
 import org.springframework.context.annotation.FilterType;
@@ -53,6 +58,7 @@ public class JeecgSystemApplication extends SpringBootServletInitializer {
             + "WEB External: \thttp://" + ip + ":" + port + path + "/index.html\n\t" + "API External: \thttp://" + ip + ":" + port + path + "/\n\n\t"
             + "Swagger文档: \thttp://" + ip + ":" + port + path + "/doc.html\n\n\t" + "The following profiles are active: [" + profiles + "]\n"
             + "----------------------------------------------------------");
+
     }
  
     public static InetAddress getLocalHostExactAddress() {
@@ -76,4 +82,14 @@ public class JeecgSystemApplication extends SpringBootServletInitializer {
         }
         return null;
     }
+
+    @Bean
+    public CommandLineRunner run() {
+        return args -> {
+            // 在这里编写启动时需要执行的代码
+            MyTriggerListener myTriggerListener = new MyTriggerListener();
+            SpringUtils.getBean(Scheduler.class).getListenerManager().addTriggerListener(myTriggerListener);
+            System.out.println("Spring Boot 已启动完成，可以在这里执行自定义的启动逻辑...");
+        };
+    }
 }
 package org.jeecg.modules.quartz.service.impl;
  
-import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
-import lombok.extern.slf4j.Slf4j;
+import java.util.Date;
+import java.util.List;
+
 import org.jeecg.common.constant.CommonConstant;
 import org.jeecg.common.exception.JeecgBootException;
 import org.jeecg.common.util.DateUtils;
@@ -13,8 +14,9 @@ import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
  
-import java.util.Date;
-import java.util.List;
+import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+
+import lombok.extern.slf4j.Slf4j;
  
 /**
  * @Description: 定时任务在线管理
+package org.jeecg.modules.wms.monitor.job;
+
+import org.jeecg.JeecgSystemApplication;
+import org.jeecg.modules.wms.config.parameterConfiguration.service.IParameterConfigurationService;
+import org.jeecg.utils.SpringUtils;
+import org.jeecg.utils.StringUtils;
+import org.jeecg.utils.constant.QuantityConstant;
+import org.quartz.JobExecutionContext;
+import org.quartz.Trigger;
+import org.quartz.TriggerListener;
+
+import lombok.extern.slf4j.Slf4j;
+
+/**
+ * @author 游杰
+ */
+@Slf4j
+public class MyTriggerListener implements TriggerListener {
+
+    @Override
+    public String getName() {
+        return "MyTriggerListener";
+    }
+
+    @Override
+    public void triggerFired(Trigger trigger, JobExecutionContext context) {
+
+    }
+
+    @Override
+    public boolean vetoJobExecution(Trigger trigger, JobExecutionContext context) {
+        String localIp = JeecgSystemApplication.getLocalHostExactAddress().getHostAddress();
+        String ip = SpringUtils.getBean(IParameterConfigurationService.class).getValueByCode(QuantityConstant.RULE_JOB_IP);
+        if (StringUtils.isNotEmpty(ip)) {
+            if (!ip.equals(localIp)) {
+                return true;
+            }
+        }
+        return false;
+    }
+
+    @Override
+    public void triggerMisfired(Trigger trigger) {
+
+    }
+
+    @Override
+    public void triggerComplete(Trigger trigger, JobExecutionContext context, Trigger.CompletedExecutionInstruction triggerInstructionCode) {
+
+    }
+
+}
@@ -535,6 +535,7 @@ public class QuantityConstant {
     public static final String RULE_SHIPMENT_ZONE = "shipmentZoneRule";
     public static final String SHIPMENT_WHOLE_TASK_CLEAR_INVENTORY = "shipmentWholeTaskClearInventory";
     public static final String RULE_SHIPMENT_CONTAINER_ADVICE = "shipmentContainerAdvice";
+    public static final String RULE_JOB_IP = "jobIp";
  
     public static final int STATUS_CLOSE = 0;
     public static final int STATUS_OPEN = 1;
@@ -2,11 +2,11 @@ spring:
   application:
     name: huaheng-wms-core
   profiles:
-    active: test
+    active: dev
  
  
 huaheng:
   system:
     version: '@project.version@'
     artifactId: '@project.artifactId@'
-    checkSystemActivationCode: true
+    checkSystemActivationCode: false