Browse Source

优化代码

wangfeng 5 years ago
parent
commit
a58dd4e96a

+ 7 - 5
src/main/java/com/diagbot/vo/ModelHospitalUpdataVO.java

@@ -3,6 +3,9 @@ package com.diagbot.vo;
 import lombok.Getter;
 import lombok.Setter;
 
+import javax.validation.constraints.NotBlank;
+import javax.validation.constraints.NotNull;
+
 /**
  * @author wangfeng
  * @Description:
@@ -12,15 +15,14 @@ import lombok.Setter;
 @Getter
 public class ModelHospitalUpdataVO {
 
-
     private Long id;
-
+    @NotNull(message = "请输入医院Id")
     private Long hospitalId;
-
+    @NotBlank(message = "请输入医院数据模块名")
     private String hospitalModelName;
-
+    @NotNull(message = "请输入数据模块Id")
     private Long standModelId;
-
+    @NotBlank(message = "请输入数据模块名")
     private String standModelName;
     /**
      * 备注

+ 4 - 3
src/main/java/com/diagbot/web/ModelHospitalController.java

@@ -17,6 +17,7 @@ import org.springframework.web.bind.annotation.RequestBody;
 import org.springframework.web.bind.annotation.RequestMapping;
 import org.springframework.web.bind.annotation.RestController;
 
+import javax.validation.Valid;
 import java.util.List;
 
 /**
@@ -49,7 +50,7 @@ public class ModelHospitalController {
             notes = "根据id获取医院数据")
     @PostMapping("/getModelHospitalById")
     @SysLogger("getModelHospitalById")
-    public RespDTO<ModelHospitalAllDTO> getModelHospitalById(@RequestBody ModelHospitalIdVO modelHospitalIdVO) {
+    public RespDTO<ModelHospitalAllDTO> getModelHospitalById(@RequestBody @Valid ModelHospitalIdVO modelHospitalIdVO) {
         ModelHospitalAllDTO data = modelHospitalFacade.getModelHospitalById(modelHospitalIdVO.getModeId());
         return RespDTO.onSuc(data);
     }
@@ -57,7 +58,7 @@ public class ModelHospitalController {
             notes = "添加和修改医院数据模块")
     @PostMapping("/saveAndupdataModel")
     @SysLogger("saveAndupdataModel")
-    public RespDTO<Boolean> updataModelHospital(@RequestBody ModelHospitalUpdataVO modelHospitalUpdataVO) {
+    public RespDTO<Boolean> updataModelHospital(@RequestBody @Valid ModelHospitalUpdataVO modelHospitalUpdataVO) {
         Boolean res = modelHospitalFacade.updataModelHospitals(modelHospitalUpdataVO);
         return RespDTO.onSuc(res);
     }
@@ -66,7 +67,7 @@ public class ModelHospitalController {
             notes = "删除医院数据模块")
     @PostMapping("/clearModelHospital")
     @SysLogger("clearModelHospital")
-    public RespDTO<Boolean> clearModelHospital(@RequestBody ModelHospitalIdVO modelHospitalIdVO) {
+    public RespDTO<Boolean> clearModelHospital(@RequestBody @Valid ModelHospitalIdVO modelHospitalIdVO) {
         Boolean res = modelHospitalFacade.clearModelHospital(modelHospitalIdVO.getModeId());
         return RespDTO.onSuc(res);
     }