[6주차 미션] 구현 완료했습니다. 리뷰 부탁드립니다.

build.gradle

@@ -34,4 +34,5 @@ dependencies {
     runtime('net.rakugakibox.spring.boot:logback-access-spring-boot-starter:2.7.1')
     testCompile('org.springframework.boot:spring-boot-starter-test')
     testCompile('org.assertj:assertj-core:3.10.0')
+    implementation 'org.projectlombok:lombok'
 }

src/main/java/codesquad/answer/Answer.java

@@ -2,10 +2,14 @@
 
 import codesquad.qua.Question;
 import codesquad.user.User;
+import lombok.Getter;
+import lombok.NoArgsConstructor;
 
 import javax.persistence.*;
 
 @Entity
+@Getter
+@NoArgsConstructor
 public class Answer {
 
     @Id
@@ -16,54 +20,35 @@ public class Answer {
     @JoinColumn(foreignKey = @ForeignKey(name = "fk_answer_to_question"))
     private Question question;
 
-    private String writer;
+    @ManyToOne(fetch = FetchType.LAZY)
+    @JoinColumn(foreignKey = @ForeignKey(name = "fk_answer_to_user"))
+    private User writer;
 
     private String comment;
 
-    private boolean deletedFlag = false;
+    private boolean deletedFlag;
 
-    public boolean isDeletedFlag() {
-        return deletedFlag;
+    public Answer(Question question, User writer, String comment) {
+        this.question = question;
+        this.writer = writer;
+        this.comment = comment;
+        deletedFlag = false;
     }
 
     public void changeDeletedFlag() {
         deletedFlag = true;
     }
 
-    public String getWriter() {
-        return writer;
-    }
-
-    public void setWriter(String writer) {
-        this.writer = writer;
-    }
-
-    public Long getId() {
-        return id;
-    }
-
-    public Question getQuestion() {
-        return question;
-    }
-
-    public void setQuestion(Question question) {
-        this.question = question;
-    }
-
-    public String getComment() {
-        return comment;
-    }
-
-    public void setComment(String comment) {
-        this.comment = comment;
-    }
-
     public void addQuestion(Question question) {
         question.getAnswers().add(this);
         this.question = question;
     }
 
     public boolean equalsWriter(User user) {
-        return writer.equals(user.getName());
+        return writer.equals(user);
+    }
+
+    public ResponseAnswerDto toResponseAnswerDto() {
+        return new ResponseAnswerDto(id, comment, question.getId(), writer.getName());
     }
 }

src/main/java/codesquad/answer/AnswerRepository.java

@@ -1,6 +1,12 @@
 package codesquad.answer;
 
+import org.springframework.data.jpa.repository.EntityGraph;
 import org.springframework.data.jpa.repository.JpaRepository;
 
+import java.util.Optional;
+
 public interface AnswerRepository extends JpaRepository<Answer, Long> {
+
+    @EntityGraph(attributePaths = {"question"})
+    Optional<Answer> findQuestionFetchJoinById(Long id);
 }

src/main/java/codesquad/answer/AnswerService.java

@@ -0,0 +1,62 @@
+package codesquad.answer;
+
+import codesquad.qua.Question;
+import codesquad.qua.QuestionRepository;
+import codesquad.response.Result;
+import codesquad.user.User;
+import codesquad.utils.SessionUtil;
+import lombok.RequiredArgsConstructor;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.stereotype.Service;
+import org.springframework.transaction.annotation.Transactional;
+
+import javax.servlet.http.HttpSession;
+import java.util.NoSuchElementException;
+
+@Service
+@Slf4j
+@RequiredArgsConstructor
+public class AnswerService {
+
+    private final AnswerRepository answerRepository;
+    private final QuestionRepository questionRepository;
+
+    public Result<ResponseAnswerDto> create(long questionId, RequestAnswerDto requestAnswerDto, HttpSession session) {
+        log.info("comment = {}", requestAnswerDto.getComment());
+
+        User user = SessionUtil.getUserBySession(session);
+
+        if (user == null) {
+            return Result.fail("로그인 하세요");
+        }
+
+        Question question = questionRepository.findById(questionId)
+                .orElseThrow(NoSuchElementException::new);
+
+        Answer answer = new Answer(question, user, requestAnswerDto.getComment());
+        answer.addQuestion(question);
+        answerRepository.save(answer);
+
+        return Result.ok(answer.toResponseAnswerDto());
+    }
+
+    @Transactional
+    public Result<ResponseAnswerDto> remove(long questionId, long answerId, HttpSession session) {
+        User user = SessionUtil.getUserBySession(session);
+
+        if (user == null) {
+            return Result.fail("로그인 하세요");
+        }
+
+        Answer answer = answerRepository.findQuestionFetchJoinById(answerId)
+                .orElseThrow(NoSuchElementException::new);
+
+        if (!answer.equalsWriter(user)) {
+            return Result.fail("다른 사람 답글은 삭제 못해요");
+        }
+
+        answer.changeDeletedFlag();
+
+        return Result.ok(answer.toResponseAnswerDto());
+    }
+}

src/main/java/codesquad/answer/ApiAnswerController.java

@@ -0,0 +1,27 @@
+package codesquad.answer;
+
+import codesquad.response.Result;
+import lombok.RequiredArgsConstructor;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.web.bind.annotation.*;
+
+import javax.servlet.http.HttpSession;
+
+@RestController
+@RequiredArgsConstructor
+@Slf4j
+public class ApiAnswerController {
+
+    private final AnswerService answerService;
+
+    @PostMapping("/questions/{question-id}/answers")
+    public Result<ResponseAnswerDto> create(@PathVariable("question-id") Long questionId, @RequestBody RequestAnswerDto requestAnswerDto, HttpSession session) {
+        return answerService.create(questionId, requestAnswerDto, session);
+    }
+
+    @DeleteMapping("/questions/{question-id}/answers/{answer-id}")
+    public Result<ResponseAnswerDto> remove(@PathVariable("question-id") Long questionId,
+                                            @PathVariable("answer-id") Long answerId, HttpSession session) {
+        return answerService.remove(questionId, answerId, session);
+    }
+}

src/main/java/codesquad/answer/RequestAnswerDto.java

@@ -0,0 +1,10 @@
+package codesquad.answer;
+
+import lombok.Getter;
+import lombok.Setter;
+
+@Getter
+@Setter
+public class RequestAnswerDto {
+    private String comment;
+}

src/main/java/codesquad/answer/ResponseAnswerDto.java

@@ -0,0 +1,16 @@
+package codesquad.answer;
+
+import lombok.AllArgsConstructor;
+import lombok.Getter;
+import lombok.Setter;
+
+@Getter
+@Setter
+@AllArgsConstructor
+public class ResponseAnswerDto {
+
+    private long id;
+    private String comment;
+    private long questionId;
+    private String writer;
+}

src/main/java/codesquad/exception/QuestionDeleteException.java

@@ -0,0 +1,11 @@
+package codesquad.exception;
+
+public class QuestionDeleteException extends RuntimeException {
+
+    public QuestionDeleteException() {
+    }
+
+    public QuestionDeleteException(String message) {
+        super(message);
+    }
+}

src/main/java/codesquad/exception/QuestionEditException.java

@@ -0,0 +1,11 @@
+package codesquad.exception;
+
+public class QuestionEditException extends RuntimeException {
+
+    public QuestionEditException() {
+    }
+
+    public QuestionEditException(String message) {
+        super(message);
+    }
+}