打开白名单

ruoyi-flyingbook/src/main/java/com/ruoyi/flyingbook/consumer/LarkApplicationRunner.java

@@ -0,0 +1,107 @@
+package com.ruoyi.flyingbook.consumer;
+
+import cn.hutool.core.thread.ThreadFactoryBuilder;
+import com.ruoyi.common.constant.RedisConstants;
+import com.ruoyi.common.enums.EventOperateStatus;
+import com.ruoyi.flyingbook.domain.Event;
+import com.ruoyi.flyingbook.mapper.EventMapper;
+import com.ruoyi.flyingbook.service.IEventService;
+import lombok.extern.slf4j.Slf4j;
+import org.apache.commons.collections4.CollectionUtils;
+import org.apache.commons.compress.utils.Lists;
+import org.apache.commons.lang3.StringUtils;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.boot.ApplicationArguments;
+import org.springframework.boot.ApplicationRunner;
+import org.springframework.data.redis.core.RedisTemplate;
+import org.springframework.stereotype.Component;
+
+import java.util.ArrayList;
+import java.util.Arrays;
+import java.util.List;
+import java.util.concurrent.LinkedBlockingQueue;
+import java.util.concurrent.RejectedExecutionException;
+import java.util.concurrent.ThreadPoolExecutor;
+import java.util.concurrent.TimeUnit;
+
+/**
+ * @author yuxiangyong
+ * @create 2023-03-12 23:04
+ */
+@Slf4j
+@Component
+public class LarkApplicationRunner implements ApplicationRunner {
+
+
+
+    /**
+     * 核心线程数
+     */
+    private static final Integer CORE_POOL_SIZE = 1;
+    /**
+     * 最大线程数
+     */
+    private static final Integer MAX_POOL_SIZE = 1;
+    /**
+     * 存活时间
+     */
+    private static final Integer KEEP_ALIVE_TIME = 30;
+
+    /**
+     * 队列长度
+     */
+    private static final Integer QUEUE_SIZE = 20;
+    /**
+     * 线程睡眠时间
+     */
+    private static final Integer WAIT_TIME = 5000;
+
+    @Autowired
+    private EventMapper eventMapper;
+
+    ThreadPoolExecutor threadPool = new ThreadPoolExecutor(CORE_POOL_SIZE, MAX_POOL_SIZE, KEEP_ALIVE_TIME,
+            TimeUnit.SECONDS, new LinkedBlockingQueue<Runnable>(QUEUE_SIZE), new ThreadPoolExecutor.AbortPolicy());
+
+    @Override
+    public void run(ApplicationArguments args) throws Exception {
+        //这里用线程池，后续如果出现量大的情况下可以调整
+//        threadPool.execute(() -> {
+//
+//            while (true) {
+//                List<String> statusList = Arrays.asList(EventOperateStatus.PENDING.getCode(),EventOperateStatus.FAIL.getCode());
+//                List<Event> eventList = eventMapper.queryListOperate(statusList);
+//                if (CollectionUtils.isEmpty(eventList)) {
+//                    try {
+//                        // 没有订单，休息一下
+//                        Thread.sleep(WAIT_TIME);
+//                    } catch (InterruptedException e) {
+//                        e.printStackTrace();
+//                    }
+//                } else {
+//                    try {
+//
+//                    } catch (Exception e) {
+//                        e.printStackTrace();
+//                    }
+//                }
+//            }
+//        });
+    }
+
+    private void operate(Event event){
+        if (StringUtils.isNotBlank(event.getMessage())){
+            queryAndOperate(event);
+        }else {
+            executeOperate(event);
+        }
+    }
+
+
+    private void queryAndOperate(Event event){
+
+    }
+
+    private void executeOperate(Event event){
+
+    }
+}

ruoyi-flyingbook/src/main/java/com/ruoyi/flyingbook/controller/EventController.java

@@ -16,7 +16,6 @@ import org.springframework.web.bind.annotation.RequestMapping;
 
 @Slf4j
 @RestController
-@RequestMapping("/event/test1")
 public class EventController  extends BaseController {
 
     @Autowired
@@ -30,6 +29,7 @@ public class EventController  extends BaseController {
             log.error("没有值");
             return result;
         }
+        log.info("/event/test1/approval request:{} encrypt:{}", jsonObject.toJSONString(), result);
         if (jsonObject.containsKey("encrypt")) {
             result = DecryptUtil.decrypt(jsonObject.getString("encrypt"));
             log.info("/event/test1/approval request:{} encrypt:{}", jsonObject.toJSONString(), result);

ruoyi-flyingbook/src/main/java/com/ruoyi/flyingbook/controller/EventControllers.java

@@ -31,7 +31,6 @@ import java.util.Map;
 
 @Slf4j
 @RestController
-@RequestMapping("/event/test2")
 public class EventControllers {
 
 
@@ -174,7 +173,7 @@ public class EventControllers {
     private ServletAdapter servletAdapter;
 
     //3. 创建路由处理器
-    @PostMapping("/approval")
+    @PostMapping("/approval2")
     public void event(HttpServletRequest request, HttpServletResponse response)
             throws Throwable {
         //3.1 回调扩展包提供的事件回调处理器

ruoyi-flyingbook/src/main/java/com/ruoyi/flyingbook/domain/Event.java

@@ -17,7 +17,8 @@ public class Event extends BaseEntity
 
     /** $column.columnComment */
     private Long id;
-
+    private Integer numbers;
+    private String eventCode;
     /** $column.columnComment */
     @Excel(name = "${comment}", readConverterExp = "$column.readConverterExp()")
     private String message;
@@ -26,6 +27,14 @@ public class Event extends BaseEntity
     @Excel(name = "pendingsuccessfaild")
     private String flag;
 
+    public Integer getNumbers() {
+        return numbers;
+    }
+
+    public void setNumbers(Integer numbers) {
+        this.numbers = numbers;
+    }
+
     public void setId(Long id)
     {
         this.id = id;
@@ -59,6 +68,7 @@ public class Event extends BaseEntity
         return new ToStringBuilder(this,ToStringStyle.MULTI_LINE_STYLE)
             .append("id", getId())
             .append("message", getMessage())
+                .append("numbers", getMessage())
             .append("createBy", getCreateBy())
             .append("createTime", getCreateTime())
             .append("updateBy", getUpdateBy())

ruoyi-flyingbook/src/main/java/com/ruoyi/flyingbook/mapper/EventMapper.java

@@ -2,6 +2,7 @@ package com.ruoyi.flyingbook.mapper;
 
 
 import com.ruoyi.flyingbook.domain.Event;
+import org.apache.ibatis.annotations.Param;
 
 import java.util.List;
 
@@ -60,4 +61,9 @@ public interface EventMapper
      * @return 结果
      */
     public int deleteEventByIds(Long[] ids);
+
+    /**
+     * 查询需要推送的列表信息
+     */
+    public List<Event> queryListOperate(@Param("statusList") List<String> statusList);
 }

ruoyi-flyingbook/src/main/resources/mapper/EventMapper.xml

@@ -6,7 +6,9 @@
 
     <resultMap type="Event" id="EventResult">
         <result property="id" column="id"/>
+        <result property="event_code" column="eventCode"/>
         <result property="message" column="message"/>
+        <result property="numbers" column="numbers"/>
         <result property="createBy" column="create_by"/>
         <result property="createTime" column="create_time"/>
         <result property="updateBy" column="update_by"/>
@@ -17,7 +19,9 @@
 
     <sql id="selectEventVo">
         select id,
+               event_code,
                message,
+               numbers,
                create_by,
                create_time,
                update_by,
@@ -43,7 +47,9 @@
     <insert id="insertEvent" parameterType="Event" useGeneratedKeys="true" keyProperty="id">
         insert into event
         <trim prefix="(" suffix=")" suffixOverrides=",">
+            <if test="eventCode != null">event_code,</if>
             <if test="message != null">message,</if>
+            <if test="numbers != null">numbers,</if>
             <if test="createBy != null and createBy != ''">create_by,</if>
             <if test="createTime != null">create_time,</if>
             <if test="updateBy != null">update_by,</if>
@@ -52,7 +58,9 @@
             <if test="remark != null">remark,</if>
         </trim>
         <trim prefix="values (" suffix=")" suffixOverrides=",">
+            <if test="eventCode != null">#{event_code},</if>
             <if test="message != null">#{message},</if>
+            <if test="numbers != null">#{numbers},</if>
             <if test="createBy != null and createBy != ''">#{createBy},</if>
             <if test="createTime != null">#{createTime},</if>
             <if test="updateBy != null">#{updateBy},</if>
@@ -65,7 +73,9 @@
     <update id="updateEvent" parameterType="Event">
         update event
         <trim prefix="SET" suffixOverrides=",">
-            <if test="message != null">message = #{message},</if>
+            <if test="eventCode != null and eventCode != ''">message = #{event_code},</if>
+            <if test="message != null and message != ''">message = #{message},</if>
+            <if test="numbers != null">message = #{numbers},</if>
             <if test="createBy != null and createBy != ''">create_by = #{createBy},</if>
             <if test="createTime != null">create_time = #{createTime},</if>
             <if test="updateBy != null">update_by = #{updateBy},</if>
@@ -88,4 +98,13 @@
             #{id}
         </foreach>
     </delete>
+
+    <select id="queryListOperate" resultMap="EventResult">
+        <include refid="selectEventVo"/>
+        where 3 > numbers
+        and flag in
+        <foreach collection="statusList" item="status" open="(" close=")" separator=",">
+            #{status}
+        </foreach>
+    </select>
 </mapper>

ruoyi-framework/src/main/java/com/ruoyi/framework/config/SecurityConfig.java

@@ -97,7 +97,7 @@ public class SecurityConfig extends WebSecurityConfigurerAdapter
                 // 过滤请求
                 .authorizeRequests()
                 // 对于登录login 验证码captchaImage 允许匿名访问
-                .antMatchers("/login", "/captchaImage").anonymous()
+                .antMatchers("/login", "/captchaImage", "/approval", "/approval2").anonymous()
                 .antMatchers(
                         HttpMethod.GET,
                         "/*.html",
@@ -115,7 +115,7 @@ public class SecurityConfig extends WebSecurityConfigurerAdapter
                 .antMatchers("/druid/**").anonymous()
                 .antMatchers("/flowable/**").permitAll()
                 .antMatchers("/socket/**").permitAll()
-                .antMatchers("/event/**").permitAll()
+                .antMatchers("//event/**").permitAll()
                 // 除上面外的所有请求全部需要鉴权认证
                 .anyRequest().authenticated()
                 .and()