web-工作任务[驿站站长]数据范围调整

src/main/java/com/hz/employmentsite/services/impl/taskAndLog/DoTaskImpl.java

@@ -125,16 +125,9 @@ public class DoTaskImpl implements DoTaskService {
     public int save(DoTaskVo data, String userId) {
         int result = 0;
         PcDotask dbData = doTaskMapper.selectByPrimaryKey(data.getDoTaskID());
-        PcDotaskUserExample exp = new PcDotaskUserExample();
-        PcDotaskUserExample.Criteria cro = exp.createCriteria();
-        cro.andDotaskIDEqualTo(data.getDoTaskID());
-        PcDotaskUser taskUser = doTaskUserMapper.selectByExample(exp).stream().findFirst().orElse(null);
-        PcSiteUserExample siteUserExp = new PcSiteUserExample();
-        siteUserExp.or().andUserIDEqualTo(userId);
-        var curSiteInfo = pcSiteUserMapper.selectByExample(siteUserExp).stream().findFirst().orElse(null);
-        if( curSiteInfo != null){
-            data.setSiteID(curSiteInfo.getSiteID());
-        }
+        PcDotaskUserExample taskUserExp = new PcDotaskUserExample();
+        taskUserExp.or().andDotaskIDEqualTo(data.getDoTaskID());
+        PcDotaskUser taskUser = doTaskUserMapper.selectByExample(taskUserExp).stream().findFirst().orElse(null);
         if (dbData == null) {
             dbData = new PcDotask();
             dbData.setDotaskID(data.getDoTaskID());
@@ -143,7 +136,14 @@ public class DoTaskImpl implements DoTaskService {
             dbData.setDotaskName(data.getDoTaskName());
             dbData.setRegionCode(data.getRegionCode());
             dbData.setStreetCode(data.getStreetCode());
-            dbData.setSiteID(data.getSiteID());
+            PcSiteUserExample siteUserExp = new PcSiteUserExample();
+            siteUserExp.or().andUserIDEqualTo(userId);
+            var curSiteInfo = pcSiteUserMapper.selectByExample(siteUserExp).stream().findFirst().orElse(null);
+            if( curSiteInfo != null){
+                dbData.setSiteID(curSiteInfo.getSiteID());
+            }else{
+                dbData.setSiteID(data.getSiteID());
+            }
             dbData.setFinishTime(data.getFinishTime());
             dbData.setCreateTime(new Date());
             dbData.setCreateUserID(userId);
@@ -156,7 +156,7 @@ public class DoTaskImpl implements DoTaskService {
             dbData.setDotaskName(data.doTaskName);
             dbData.setRegionCode(data.regionCode);
             dbData.setStreetCode(data.streetCode);
-            dbData.setSiteID(data.siteID);
+            //dbData.setSiteID(data.siteID);
             dbData.setFinishTime(data.finishTime);
             dbData.setModifyTime(new Date());
             dbData.setModifyUserID(userId);

src/main/java/com/hz/employmentsite/util/datarange/DataRangeInterceptor.java

@@ -82,7 +82,7 @@ public class DataRangeInterceptor implements Interceptor {
                 whereClause = getJoinSql(entityClass, dataRange);
 
                 if (!whereClause.equals("")) {
-                    if(mainSql.contains("task.dotaskID,task.siteID")&&dataRange.getRange()==4){
+                    if(mainSql.contains("task.dotaskID,task.siteID")){
                         whereClause += " or x.siteID is null ";
                     }
                     mainSql = "select x.* from" +