|
@@ -3,7 +3,10 @@ package com.hz.employmentsite.controller.taskAndLog;
|
|
|
import com.github.pagehelper.PageInfo;
|
|
|
import com.hz.employmentsite.filter.exception.BaseResponse;
|
|
|
import com.hz.employmentsite.filter.exception.RespGenerstor;
|
|
|
+import com.hz.employmentsite.mapper.PcSiteMapper;
|
|
|
import com.hz.employmentsite.mapper.PcSiteUserMapper;
|
|
|
+import com.hz.employmentsite.model.PcSite;
|
|
|
+import com.hz.employmentsite.model.PcSiteExample;
|
|
|
import com.hz.employmentsite.model.PcSiteUserExample;
|
|
|
import com.hz.employmentsite.services.service.AccountService;
|
|
|
import com.hz.employmentsite.services.service.taskAndLog.DoWorkService;
|
|
@@ -28,6 +31,9 @@ public class DoWorkController {
|
|
|
@Autowired
|
|
|
private PcSiteUserMapper pcSiteUserMapper;
|
|
|
|
|
|
+ @Autowired
|
|
|
+ private PcSiteMapper pcSiteMapper;
|
|
|
+
|
|
|
@Autowired
|
|
|
private ExcelHelper excelHelper;
|
|
|
|
|
@@ -61,9 +67,18 @@ public class DoWorkController {
|
|
|
PcSiteUserExample.Criteria siteUserCro = siteUserExp.createCriteria();
|
|
|
siteUserCro.andUserIDEqualTo(accountService.getLoginUserID());
|
|
|
var curSiteUserInfo = pcSiteUserMapper.selectByExample(siteUserExp).stream().findFirst().orElse(null);
|
|
|
- if(curSiteUserInfo != null){
|
|
|
+ if (curSiteUserInfo != null) {
|
|
|
data.siteUserID = curSiteUserInfo.getSiteUserID();
|
|
|
data.userId = curSiteUserInfo.getUserID();
|
|
|
+ // 查询提交用户的所属驿站,获取区县与街道编码
|
|
|
+ PcSiteExample siteExample = new PcSiteExample();
|
|
|
+ PcSiteExample.Criteria siteExampleCriteria = siteExample.createCriteria();
|
|
|
+ siteExampleCriteria.andSiteIDEqualTo(curSiteUserInfo.getSiteID());
|
|
|
+ PcSite pcSites = pcSiteMapper.selectByExample(siteExample).stream().findFirst().orElse(null);
|
|
|
+ if (pcSites != null) {
|
|
|
+ data.setRegionCode(pcSites.getRegionCode());
|
|
|
+ data.setStreetCode(pcSites.getStreetCode());
|
|
|
+ }
|
|
|
}
|
|
|
}
|
|
|
return RespGenerstor.success(data);
|