diff --git a/src/main/java/be/gyu/urlShortener/controller/MainController.java b/src/main/java/be/gyu/urlShortener/controller/MainController.java index e557326..b05fa60 100644 --- a/src/main/java/be/gyu/urlShortener/controller/MainController.java +++ b/src/main/java/be/gyu/urlShortener/controller/MainController.java @@ -9,9 +9,11 @@ import org.springframework.stereotype.Controller; import org.springframework.web.bind.annotation.GetMapping; import org.springframework.web.bind.annotation.PathVariable; import org.springframework.web.bind.annotation.PostMapping; +import org.springframework.web.bind.annotation.RequestBody; import org.springframework.web.bind.annotation.ResponseBody; import org.springframework.web.util.UriComponentsBuilder; +import be.gyu.urlShortener.dto.GenerateShortUrlRequestDto; import be.gyu.urlShortener.service.MainService; import jakarta.servlet.http.HttpServletRequest; import jakarta.servlet.http.HttpServletResponse; @@ -37,9 +39,9 @@ public class MainController { */ @PostMapping("/short") @ResponseBody - public ResponseEntity> postGenerateShortUrl(HttpServletRequest request){ + public ResponseEntity> postGenerateShortUrl(@RequestBody GenerateShortUrlRequestDto data, HttpServletRequest request){ // 서비스 로직 메소드 호출 - Map resultMap=mainService.generateShortUrl(request.getParameter("url")); + Map resultMap=mainService.generateShortUrl(data.getOriginalUrl()); if(!resultMap.containsKey("status") || resultMap.get("status").equals("failed")){ resultMap.put("status","failed"); diff --git a/src/main/java/be/gyu/urlShortener/dto/GenerateShortUrlRequestDto.java b/src/main/java/be/gyu/urlShortener/dto/GenerateShortUrlRequestDto.java new file mode 100644 index 0000000..a0f950e --- /dev/null +++ b/src/main/java/be/gyu/urlShortener/dto/GenerateShortUrlRequestDto.java @@ -0,0 +1,13 @@ +package be.gyu.urlShortener.dto; + +import lombok.AllArgsConstructor; +import lombok.Getter; +import lombok.NoArgsConstructor; + +@Getter +// @Setter +@NoArgsConstructor +@AllArgsConstructor +public class GenerateShortUrlRequestDto { + private String originalUrl; +}