增加主体

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

@@ -4,6 +4,7 @@ import com.alibaba.fastjson.JSONObject;
 import com.tzld.piaoquan.wecom.component.HttpPoolClient;
 import com.tzld.piaoquan.wecom.dao.mapper.StaffMapper;
 import com.tzld.piaoquan.wecom.dao.mapper.SubjectMapper;
+import com.tzld.piaoquan.wecom.model.bo.XxlJobParam;
 import com.tzld.piaoquan.wecom.model.po.Staff;
 import com.tzld.piaoquan.wecom.model.po.StaffExample;
 import com.tzld.piaoquan.wecom.model.po.Subject;
@@ -14,6 +15,7 @@ import com.tzld.piaoquan.wecom.utils.LarkRobotUtil;
 import com.xxl.job.core.biz.model.ReturnT;
 import com.xxl.job.core.handler.annotation.XxlJob;
 import lombok.extern.slf4j.Slf4j;
+import org.apache.commons.lang3.StringUtils;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Component;
 import org.springframework.util.CollectionUtils;
@@ -43,7 +45,14 @@ public class WeComStaffDataJob {
     @XxlJob("insertStaffJob")
     public ReturnT<String> insertStaff(String param) {
         try {
+            XxlJobParam xxlJobParam = new XxlJobParam();
+            if (StringUtils.isNotEmpty(param)) {
+                xxlJobParam = JSONObject.parseObject(param, XxlJobParam.class);
+            }
             SubjectExample subjectExample = new SubjectExample();
+            if (xxlJobParam.getSubjectId() != null) {
+                subjectExample.createCriteria().andIdEqualTo(xxlJobParam.getSubjectId());
+            }
             List<Subject> subjects = subjectMapper.selectByExample(subjectExample);
             if (CollectionUtils.isEmpty(subjects)) {
                 return ReturnT.SUCCESS;

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

@@ -65,6 +65,9 @@ public class WeComUserDataJob {
             subjectCriteria.andIdEqualTo(xxlJobParam.getSubjectId());
         }
         List<Subject> subjects = subjectMapper.selectByExample(subjectExample);
+        if (CollectionUtils.isEmpty(subjects)) {
+            return ReturnT.SUCCESS;
+        }
         for (Subject subject : subjects) {
             StaffExample staffExample = new StaffExample();
             StaffExample.Criteria staffCriteria = staffExample.createCriteria();