From 80ae2fb8d3509ec8e78a3a465c3ff4abd9b1ee24 Mon Sep 17 00:00:00 2001
From: heshunme <heshunme@outlook.com>
Date: Mon, 1 Jul 2024 17:21:13 +0800
Subject: [PATCH] =?UTF-8?q?=E9=80=9A=E8=BF=87DTO=E8=A7=84=E8=8C=83?=
 =?UTF-8?q?=E5=8C=96=E6=95=B0=E6=8D=AE=E4=BA=A4=E6=8D=A2?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../controller/AuthenticationController.java  | 22 ++++++++++---------
 1 file changed, 12 insertions(+), 10 deletions(-)

diff --git a/src/main/java/org/cmh/backend/authentication/controller/AuthenticationController.java b/src/main/java/org/cmh/backend/authentication/controller/AuthenticationController.java
index 1f2c243..35568cd 100644
--- a/src/main/java/org/cmh/backend/authentication/controller/AuthenticationController.java
+++ b/src/main/java/org/cmh/backend/authentication/controller/AuthenticationController.java
@@ -1,6 +1,7 @@
 package org.cmh.backend.authentication.controller;
 
 
+import lombok.AllArgsConstructor;
 import lombok.Getter;
 import lombok.Setter;
 import org.cmh.backend.authentication.service.UserService;
@@ -12,9 +13,6 @@ import org.springframework.web.bind.annotation.PostMapping;
 import org.springframework.web.bind.annotation.RequestBody;
 import org.springframework.web.bind.annotation.RestController;
 
-import java.util.HashMap;
-import java.util.Map;
-
 @RestController
 class AuthenticationController {
 
@@ -26,17 +24,14 @@ class AuthenticationController {
         return "Hello SpringBoot!";
     }
 
-    @PostMapping("/register")
-    public ResponseEntity<Map<String, Object>> register(@RequestBody RegisterRequest request) {
-        Map<String, Object> response = new HashMap<>();
 
+    @PostMapping("/register")
+    public ResponseEntity<RegisterResponse> register(@RequestBody RegisterRequest request) {
         boolean isRegistered = userService.registerUser(request.getUsername(), request.getPassword(), request.getContactInfo());
         if (isRegistered) {
-            response.put("message", "Registration successful!");
-            return new ResponseEntity<>(response, HttpStatus.OK);
+            return new ResponseEntity<>(new RegisterResponse("Registration successful!"), HttpStatus.OK);
         } else {
-            response.put("message", "Registration failed! User already exist");
-            return new ResponseEntity<>(response, HttpStatus.BAD_REQUEST);
+            return new ResponseEntity<>(new RegisterResponse("Registration failed! User already exist"), HttpStatus.BAD_REQUEST);
         }
     }
 
@@ -52,3 +47,10 @@ class RegisterRequest {
     private String contactInfo;
 }
 
+@Getter
+@Setter
+@AllArgsConstructor
+class RegisterResponse {
+    private String message;
+}
+