[Refactor] todo에서의 category save 기능

jiyunio · May 29, 2024 · e5a23bc · e5a23bc
1 parent 8220c05
commit e5a23bc
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 6 deletions.
diff --git 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
@@ -24,5 +24,5 @@ public class CreateTodoDTO {
     private LocalDate setDate;
 
     @NotNull
-    private ResponseCategoryDTO category;
+    private Category category;
 }