客户修改

fhKeeper/formulahousekeeper/management-crm/src/main/java/com/management/platform/service/impl/StageServiceImpl.java

@@ -39,12 +39,18 @@ public class StageServiceImpl extends ServiceImpl<StageMapper, Stage> implements
     @Transactional(rollbackFor = Exception.class)
     public HttpRespMsg changeStage(String stages, HttpServletRequest request) {
         HttpRespMsg msg = new HttpRespMsg();
-        int i = 0;
         List<Stage> stages1 = JSONObject.parseArray(stages, Stage.class);
+        int i = 0;
         User user = userMapper.selectById(request.getHeader("Token"));
-        stageMapper.delete(new QueryWrapper<Stage>().eq("company_id", user.getCompanyId()));
         for (Stage stage : stages1) {
-            stageMapper.insert(stage);
+            stage.setSeq(i);
+            stage.setCompanyId(user.getCompanyId());
+            if(null != stage.getId()){
+                stageMapper.updateById(stage);
+            }else {
+                stageMapper.insert(stage);
+            }
+            i++;
         }
         msg.setMsg("操作成功");
         return msg;
@@ -53,7 +59,7 @@ public class StageServiceImpl extends ServiceImpl<StageMapper, Stage> implements
     @Override
     public HttpRespMsg getStage(HttpServletRequest request) {
         HttpRespMsg msg = new HttpRespMsg();
-        msg.setData(stageMapper.selectList(new QueryWrapper<Stage>().eq("companyId",userMapper.selectById(request.getHeader("Token")).getCompanyId()).orderBy(true, true, "seq")));
+        msg.setData(stageMapper.selectList(new QueryWrapper<Stage>().eq("company_id",userMapper.selectById(request.getHeader("Token")).getCompanyId()).orderBy(true, true, "seq")));
         return msg;
     }
 }