修改生成任务

we-com-server/src/main/java/com/tzld/piaoquan/wecom/job/WeComMessageDataJob.java

@@ -79,7 +79,7 @@ public class WeComMessageDataJob {
     Map<String, String> pageMap = new HashMap<>();
 
     //初始化操作
-    private void init() {
+    private void init(List<Long> staffIds) {
         //历史优质视频获取
         String sql = String.format("SELECT * FROM loghubods.history_good_video_can_push_user_list where dt = %s;",
                 DateUtil.getBeforeDayDateString());
@@ -128,6 +128,9 @@ public class WeComMessageDataJob {
                 LarkRobotUtil.sendMessage("保底视频获取异常，StaffId为空" + DateUtil.getThatDayDateString());
                 throw new RuntimeException();
             }
+            if (!CollectionUtils.isEmpty(staffIds) && !staffIds.contains(videoParam.getStaffId())) {
+                continue;
+            }
             //默认组视频不做查询
             if (videoParam.getStaffId() == 0L) {
                 if (videoParam.getVideoIds().size() < MAX_VIDEO_NUM) {
@@ -197,7 +200,6 @@ public class WeComMessageDataJob {
         if (StringUtils.isNotEmpty(param)) {
             xxlJobParam = JSONObject.parseObject(param, XxlJobParam.class);
         }
-        init();
         CorpExample corpExample = new CorpExample();
         if (xxlJobParam.getCorpId() != null) {
             corpExample.createCriteria().andIdEqualTo(xxlJobParam.getCorpId());
@@ -213,6 +215,7 @@ public class WeComMessageDataJob {
         if (!CollectionUtils.isEmpty(xxlJobParam.getStaffIds())) {
             staffIds.addAll(xxlJobParam.getStaffIds());
         }
+        init(staffIds);
         for (Corp corp : corps) {
             UserExample userExample = new UserExample();
             userExample.createCriteria().andExternalUserIdIsNotNull().andCorpIdEqualTo(corp.getId());

we-com-server/src/main/java/com/tzld/piaoquan/wecom/model/bo/XxlJobParam.java

@@ -1,10 +1,12 @@
 package com.tzld.piaoquan.wecom.model.bo;
 
 import lombok.Data;
+import lombok.ToString;
 
 import java.util.List;
 
 @Data
+@ToString
 public class XxlJobParam {
 
     private Long corpId;