修改客户信息的提交功能

jeeplus-module/jeeplus-test/src/main/java/com/jeeplus/test/workClientInfo/controller/WorkClientController.java

@@ -1,7 +1,11 @@
 package com.jeeplus.test.workClientInfo.controller;
 
+import cn.hutool.core.util.ObjectUtil;
+import cn.hutool.core.util.StrUtil;
+import com.baomidou.mybatisplus.core.metadata.IPage;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.jeeplus.sys.utils.StringUtils;
+import com.jeeplus.test.projectRecords.domain.Project;
 import com.jeeplus.test.workClientInfo.domain.WorkClientInfo;
 import com.jeeplus.test.workClientInfo.domain.dto.PageInfoDto;
 import com.jeeplus.test.workClientInfo.domain.dto.WorkClientInfosDto;
@@ -9,6 +13,7 @@ import com.jeeplus.test.workClientInfo.service.WorkClientService;
 import io.swagger.annotations.Api;
 import io.swagger.annotations.ApiOperation;
 import lombok.extern.slf4j.Slf4j;
+import org.springframework.http.ResponseEntity;
 import org.springframework.web.bind.annotation.*;
 
 import javax.annotation.Resource;
@@ -37,13 +42,19 @@ public class WorkClientController {
      */
     @ApiOperation(value = "客户信息新增/修改")
     @PostMapping(value = "save")
-    public String save(@RequestBody WorkClientInfosDto workClientInfosDto) {
-        String id = workClientInfosDto.getWorkClientInfo().getId();
-        if (StringUtils.isNotEmpty(id)) {
-            return workClientService.update(workClientInfosDto);
-        }else {
-            return workClientService.save(workClientInfosDto);
+    public ResponseEntity save(@RequestBody WorkClientInfosDto workClientInfosDto) {
+        if(ObjectUtil.isNotEmpty(workClientInfosDto)){
+            if(ObjectUtil.isNotEmpty(workClientInfosDto.getWorkClientInfo())){
+                if(StrUtil.isNotEmpty(workClientInfosDto.getWorkClientInfo().getId())){
+                    return ResponseEntity.ok(workClientService.update(workClientInfosDto));
+                }else{
+                    return ResponseEntity.ok(workClientService.save(workClientInfosDto));
+                }
+            }else{
+                return ResponseEntity.badRequest().body("保存信息失败，没有获取到数据");
+            }
         }
+        return ResponseEntity.badRequest().body("保存信息失败，没有获取到数据");
     }
 
     /**