完善保存

epmet-module/gov-org/gov-org-server/src/main/java/com/epmet/service/impl/IcWorkLogServiceImpl.java

@@ -5,17 +5,17 @@ import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import com.baomidou.mybatisplus.core.metadata.IPage;
 import com.epmet.commons.mybatis.service.impl.BaseServiceImpl;
 import com.epmet.commons.tools.constant.FieldConstant;
-import com.epmet.commons.tools.constant.ServiceConstant;
 import com.epmet.commons.tools.exception.EpmetException;
 import com.epmet.commons.tools.page.PageData;
 import com.epmet.commons.tools.security.user.LoginUserUtil;
 import com.epmet.commons.tools.utils.ConvertUtils;
 import com.epmet.commons.tools.utils.Result;
+import com.epmet.dao.CustomerGridDao;
 import com.epmet.dao.IcWorkLogDao;
 import com.epmet.dto.CustomerStaffDTO;
 import com.epmet.dto.IcWorkLogDTO;
-import com.epmet.dto.form.UserResiInfoFormDTO;
-import com.epmet.dto.result.UserResiInfoResultDTO;
+import com.epmet.dto.form.CustomerGridFormDTO;
+import com.epmet.dto.result.GridDetailResultDTO;
 import com.epmet.entity.IcWorkLogEntity;
 import com.epmet.feign.EpmetUserFeignClient;
 import com.epmet.redis.IcWorkLogRedis;
@@ -25,6 +25,7 @@ import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
 
+import javax.annotation.Resource;
 import java.util.Arrays;
 import java.util.List;
 import java.util.Map;
@@ -47,6 +48,9 @@ public class IcWorkLogServiceImpl extends BaseServiceImpl<IcWorkLogDao, IcWorkLo
     @Autowired
     private EpmetUserFeignClient epmetUserFeignClient;
 
+    @Resource
+    private CustomerGridDao customerGridDao;
+
 
     @Override
     public PageData<IcWorkLogDTO> page(Map<String, Object> params) {
@@ -63,8 +67,8 @@ public class IcWorkLogServiceImpl extends BaseServiceImpl<IcWorkLogDao, IcWorkLo
         return ConvertUtils.sourceToTarget(entityList, IcWorkLogDTO.class);
     }
 
-    private QueryWrapper<IcWorkLogEntity> getWrapper(Map<String, Object> params){
-        String id = (String)params.get(FieldConstant.ID_HUMP);
+    private QueryWrapper<IcWorkLogEntity> getWrapper(Map<String, Object> params) {
+        String id = (String) params.get(FieldConstant.ID_HUMP);
 
         QueryWrapper<IcWorkLogEntity> wrapper = new QueryWrapper<>();
         wrapper.eq(StringUtils.isNotBlank(id), FieldConstant.ID, id);
@@ -86,9 +90,17 @@ public class IcWorkLogServiceImpl extends BaseServiceImpl<IcWorkLogDao, IcWorkLo
         customerStaffParam.setUserId(loginUserUtil.getLoginUserId());
         Result<CustomerStaffDTO> staffInfo = epmetUserFeignClient.getCustomerStaffInfoByUserId(customerStaffParam);
         if (!staffInfo.success() || null == staffInfo.getData()) {
-            throw new EpmetException("查询用户信息失败"+ JSON.toJSONString(customerStaffParam));
+            throw new EpmetException("查询用户信息失败" + JSON.toJSONString(customerStaffParam));
         } else {
             dto.setCreatedUser(staffInfo.getData().getRealName());
+            dto.setMobile(staffInfo.getData().getMobile());
+        }
+        CustomerGridFormDTO customerGridFormDTO = new CustomerGridFormDTO();
+        customerGridFormDTO.setGridId(dto.getGridId());
+        GridDetailResultDTO gridDetailResultDTO = customerGridDao.griddetail(customerGridFormDTO);
+        if (gridDetailResultDTO != null) {
+            dto.setAgencyId(gridDetailResultDTO.getAgencyId());
+            dto.setPids(gridDetailResultDTO.getPIds());
         }
         IcWorkLogEntity entity = ConvertUtils.sourceToTarget(dto, IcWorkLogEntity.class);
         insert(entity);