From e5a23bc3ae7d51d3e1620b402eb16cf6a2fa0576 Mon Sep 17 00:00:00 2001 From: jiyunio Date: Wed, 29 May 2024 16:28:19 +0900 Subject: [PATCH] =?UTF-8?q?[Refactor]=20todo=EC=97=90=EC=84=9C=EC=9D=98=20?= =?UTF-8?q?category=20save=20=EA=B8=B0=EB=8A=A5?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/main/java/com/jiyunio/todolist/todo/TodoService.java | 9 ++++----- .../com/jiyunio/todolist/todo/dto/CreateTodoDTO.java | 2 +- 2 files changed, 5 insertions(+), 6 deletions(-) diff --git a/src/main/java/com/jiyunio/todolist/todo/TodoService.java b/src/main/java/com/jiyunio/todolist/todo/TodoService.java index 53d8bf8..85aaf99 100644 --- a/src/main/java/com/jiyunio/todolist/todo/TodoService.java +++ b/src/main/java/com/jiyunio/todolist/todo/TodoService.java @@ -1,6 +1,7 @@ package com.jiyunio.todolist.todo; import com.jiyunio.todolist.category.Category; +import com.jiyunio.todolist.category.CategoryRepository; import com.jiyunio.todolist.customError.CustomException; import com.jiyunio.todolist.customError.ErrorCode; import com.jiyunio.todolist.member.Member; @@ -21,17 +22,15 @@ public class TodoService { private final MemberRepository memberRepository; private final TodoRepository todoRepository; + private final CategoryRepository categoryRepository; public ResponseTodoDTO createTodo(Long memberId, CreateTodoDTO createTodo) { Member member = memberRepository.findById(memberId).orElseThrow( // 회원 존재 안함 () -> new CustomException(HttpStatus.NOT_FOUND, ErrorCode.NOT_EXIST_MEMBER) ); - Category category = Category.builder() - .member(member) - .content(createTodo.getCategory().getContent()) - .color(createTodo.getCategory().getColor()) - .build(); + + Category category = categoryRepository.findById(createTodo.getCategory().getId()).get(); Todo todo = Todo.builder() .member(member) diff --git a/src/main/java/com/jiyunio/todolist/todo/dto/CreateTodoDTO.java b/src/main/java/com/jiyunio/todolist/todo/dto/CreateTodoDTO.java index 2e64558..fa91cfd 100644 --- a/src/main/java/com/jiyunio/todolist/todo/dto/CreateTodoDTO.java +++ b/src/main/java/com/jiyunio/todolist/todo/dto/CreateTodoDTO.java @@ -24,5 +24,5 @@ public class CreateTodoDTO { private LocalDate setDate; @NotNull - private ResponseCategoryDTO category; + private Category category; }