From 8c89eb107ab5d9ecb779326c8cafc353d8007af3 Mon Sep 17 00:00:00 2001
From: zss <zss@example.com>
Date: 星期二, 15 八月 2023 14:56:06 +0800
Subject: [PATCH] 校验2.0

---
 inventory-server/src/main/java/com/yuanchu/mom/controller/ImportRepertoryController.java |    3 ++-
 inventory-server/src/main/java/com/yuanchu/mom/controller/SaleController.java            |    3 ++-
 inventory-server/src/main/java/com/yuanchu/mom/pojo/dto/ImportRepertoryDto.java          |    8 ++++++++
 3 files changed, 12 insertions(+), 2 deletions(-)

diff --git a/inventory-server/src/main/java/com/yuanchu/mom/controller/ImportRepertoryController.java b/inventory-server/src/main/java/com/yuanchu/mom/controller/ImportRepertoryController.java
index 80634e3..648f37c 100644
--- a/inventory-server/src/main/java/com/yuanchu/mom/controller/ImportRepertoryController.java
+++ b/inventory-server/src/main/java/com/yuanchu/mom/controller/ImportRepertoryController.java
@@ -13,6 +13,7 @@
 import io.swagger.annotations.ApiImplicitParams;
 import io.swagger.annotations.ApiOperation;
 import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.validation.annotation.Validated;
 import org.springframework.web.bind.annotation.*;
 import com.yuanchu.mom.service.ImportRepertoryService;
 
@@ -41,7 +42,7 @@
 
     @ApiOperation(value = "鏂板鎴愬搧鍏ュ簱")
     @PostMapping("/addImpRep")
-    public Result addSale(@RequestHeader("token") String token, @RequestBody ImportRepertoryDto importRepertoryDto) throws Exception {
+    public Result addSale(@RequestHeader("token") String token,@Validated @RequestBody ImportRepertoryDto importRepertoryDto) throws Exception {
         Map<String, String> data = JackSonUtil.unmarshal(jwt.readJWT(token).get("data"), Map.class);
         importRepertoryService.addImpRep(data.get("name").replaceAll("\"", ""), importRepertoryDto);
         return Result.success("鏂板鎴愬姛!");
diff --git a/inventory-server/src/main/java/com/yuanchu/mom/controller/SaleController.java b/inventory-server/src/main/java/com/yuanchu/mom/controller/SaleController.java
index 7108114..29f37af 100644
--- a/inventory-server/src/main/java/com/yuanchu/mom/controller/SaleController.java
+++ b/inventory-server/src/main/java/com/yuanchu/mom/controller/SaleController.java
@@ -14,6 +14,7 @@
 import io.swagger.annotations.ApiImplicitParam;
 import io.swagger.annotations.ApiImplicitParams;
 import io.swagger.annotations.ApiOperation;
+import org.springframework.validation.annotation.Validated;
 import org.springframework.web.bind.annotation.*;
 
 import javax.annotation.Resource;
@@ -67,7 +68,7 @@
 
     @ApiOperation(value = "鏂板閿�鍞崟")
     @PostMapping("/addSale")
-    public Result addSale(@RequestHeader("token") String token, @RequestBody SaleDto saleDto) throws Exception {
+    public Result addSale(@RequestHeader("token") String token,@Validated @RequestBody SaleDto saleDto) throws Exception {
         Map<String, String> data = JackSonUtil.unmarshal(jwt.readJWT(token).get("data"), Map.class);
         saleService.addSale(data.get("name").replaceAll("\"", ""), saleDto);
         return Result.success("鏂板鎴愬姛!");
diff --git a/inventory-server/src/main/java/com/yuanchu/mom/pojo/dto/ImportRepertoryDto.java b/inventory-server/src/main/java/com/yuanchu/mom/pojo/dto/ImportRepertoryDto.java
index 81fde89..94f16a8 100644
--- a/inventory-server/src/main/java/com/yuanchu/mom/pojo/dto/ImportRepertoryDto.java
+++ b/inventory-server/src/main/java/com/yuanchu/mom/pojo/dto/ImportRepertoryDto.java
@@ -3,42 +3,50 @@
 import com.fasterxml.jackson.databind.annotation.JsonSerialize;
 import lombok.Data;
 
+import javax.validation.constraints.NotBlank;
+
 @Data
 //鏂板鎴愬搧鍏ュ簱
 public class ImportRepertoryDto {
     /**
      * 璁㈠崟缂栫爜
      **/
+    @NotBlank(message = "璁㈠崟缂栧彿涓嶈兘涓虹┖")
     @JsonSerialize
     private String orderCode;
 
     /**
      * 鏉愭枡缂栫爜
      **/
+    @NotBlank(message = "鏉愭枡缂栫爜涓嶈兘涓虹┖")
     @JsonSerialize
     private String code;
 
     /**
      * 鏉愭枡鍚嶇О
      **/
+    @NotBlank(message = "鏉愭枡鍚嶇О涓嶈兘涓虹┖")
     @JsonSerialize
     private String name;
 
     /**
      * 鍨嬪彿瑙勬牸
      **/
+    @NotBlank(message = "鍨嬪彿瑙勬牸涓嶈兘涓虹┖")
     @JsonSerialize
     private String specifications;
 
     /**
      * 鍗曚綅
      **/
+    @NotBlank(message = "鍗曚綅涓嶈兘涓虹┖")
     @JsonSerialize
     private String unit;
 
     /**
      * 鏁伴噺
      **/
+    @NotBlank(message = "鏁伴噺涓嶈兘涓虹┖")
     @JsonSerialize
     private Integer number;
 

--
Gitblit v1.9.3