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 sendMessage(@RequestBody @NotNull SendMessageDTO sendMessageDTO) { + public Result sendMessage(@RequestBody @Validated SendMessageDTO sendMessageDTO) { Message message = SendMessageUtil.sendMessage(sendMessageDTO); messageService.save(message); return new Result<>(200, "发送成功", "");