From 3b1fb1aa13f3f9daad7518625a0469c8df18644a Mon Sep 17 00:00:00 2001
From: yjxx <1055683039@qq.com>
Date: Fri, 8 Nov 2024 15:36:25 +0800
Subject: [PATCH] =?UTF-8?q?=E7=AE=A1=E7=90=86=E5=91=98=E5=8A=9F=E8=83=BD?=
 =?UTF-8?q?=E5=85=A8=E9=83=A8=E5=AE=9E=E7=8E=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../jty/expressdistributionsystem/DTO/SendMessageDTO.java     | 4 ++++
 .../expressdistributionsystem/controller/AdminController.java | 3 ++-
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/src/main/java/jty/expressdistributionsystem/DTO/SendMessageDTO.java b/src/main/java/jty/expressdistributionsystem/DTO/SendMessageDTO.java
index 690db4b..b828540 100644
--- a/src/main/java/jty/expressdistributionsystem/DTO/SendMessageDTO.java
+++ b/src/main/java/jty/expressdistributionsystem/DTO/SendMessageDTO.java
@@ -2,6 +2,8 @@ package jty.expressdistributionsystem.DTO;
 
 import com.fasterxml.jackson.databind.annotation.JsonSerialize;
 import com.fasterxml.jackson.databind.ser.std.ToStringSerializer;
+import jakarta.validation.constraints.NotBlank;
+import jakarta.validation.constraints.NotNull;
 import lombok.Data;
 
 import java.io.Serializable;
@@ -9,7 +11,9 @@ import java.io.Serializable;
 @Data
 public class SendMessageDTO implements Serializable {
     @JsonSerialize(using = ToStringSerializer.class)
+    @NotNull(message = "请选择你的聊天对象")
     private Long Id;
 
+    @NotBlank(message = "消息不能为空")
     private String content;
 }
diff --git a/src/main/java/jty/expressdistributionsystem/controller/AdminController.java b/src/main/java/jty/expressdistributionsystem/controller/AdminController.java
index 51b7f85..2d1b019 100644
--- a/src/main/java/jty/expressdistributionsystem/controller/AdminController.java
+++ b/src/main/java/jty/expressdistributionsystem/controller/AdminController.java
@@ -12,6 +12,7 @@ import jty.expressdistributionsystem.utils.SendMessageUtil;
 import jty.expressdistributionsystem.utils.ThreadLocalUtil;
 import lombok.extern.slf4j.Slf4j;
 import org.jetbrains.annotations.NotNull;
+import org.springframework.validation.annotation.Validated;
 import org.springframework.web.bind.annotation.*;
 
 import java.util.ArrayList;
@@ -78,7 +79,7 @@ public class AdminController {
 
     // 管理员发送消息
     @PostMapping("/sendMessage")
-    public Result<String> sendMessage(@RequestBody @NotNull SendMessageDTO sendMessageDTO) {
+    public Result<String> sendMessage(@RequestBody @Validated SendMessageDTO sendMessageDTO) {
         Message message = SendMessageUtil.sendMessage(sendMessageDTO);
         messageService.save(message);
         return new Result<>(200, "发送成功", "");