[Spring Boot] 글쓴이 항목 추가

점프 투 스프링부트를 참고하여 학습,작성하였습니다.

---

 

1.글쓴이 속성추가

질문과 답변 모델에 author 속성을 추가한다. 
한 글쓴이는 여러개의 질문과 답변을 할수있으므로 @ManyToOne으로 만든다.

public class Question {
    @Id
    @GeneratedValue(strategy = GenerationType.IDENTITY)
    private Integer id;

    @Column(length = 200)
    private String subject;

    @Column(columnDefinition = "TEXT")
    private String content;

    private LocalDateTime createDate;
    
    @OneToMany(mappedBy = "question", cascade = CascadeType.REMOVE) 
    private List<Answer> answerList; 

    @ManyToOne
    private SiteUser author;
}

public class Answer {
    @Id
    @GeneratedValue(strategy = GenerationType.IDENTITY)
    private Integer id;

    @Column(columnDefinition = "TEXT")
    private String content;

    private LocalDateTime createDate; 

    @ManyToOne 
    private Question question;  

    @ManyToOne
    private SiteUser author;
}

 

 

 

 

 

2.답변 컨트롤러 수정

createAnswer 메소드의 매개변수에 스프링 시큐리티가 제공하는 Principal 객체를추가한다.
principal.getName()을 호출하면 현재 로그인한 사용자의 사용자명(사용자ID)을 알 수 있다.

public String createAnswer(Model model, @PathVariable("id") Integer id, @Valid AnswerForm answerForm, BindingResult bindingResult, Principal principal) {

 

 

 

3.유저 서비스 수정

UserService에 getUser메서드를 추가하였다.
findByUsername 메서드를 사용하여 있으면 사용자명을 리턴하고 없으면  DataNoFoundException이 작동하도록 만든 메서드이다.

public SiteUser getUser(String username) {
    Optional<SiteUser> siteUser = this.userRepository.findByusername(username);
    if (siteUser.isPresent()) {
        return siteUser.get();
    } else {
        throw new DataNotFoundException("siteuser not found");
    }
}

 

4.답변 서비스 수정

방금만든 getUser메서드를 이용하여 글쓴이를 얻어 답변을 저장하는곳에 추가해준다.

public void create(Question question, String content,SiteUser author) {
    Answer answer = new Answer();
    answer.setContent(content);
    answer.setCreateDate(LocalDateTime.now());
    answer.setQuestion(question);
    answer.setAuthor(author);
    this.answerRepository.save(answer);
}

 

 

 

 

5.컨트롤러 수정

서비스의 매개변수가 변화했으므로 컨트롤러또한 수정해준다.

public class AnswerController {

    private final AnswerService answerService;
    private final QuestionService questionService;
    private final UserService userService;

    @PostMapping("/create/{id}")
    public String createAnswer(Model model, @PathVariable("id") Integer id, @Valid AnswerForm answerForm, BindingResult bindingResult, Principal principal) {
        Question question = this.questionService.getQuestion(id);
        SiteUser siteUser = this.userService.getUser(principal.getName());
        if (bindingResult.hasErrors()) {
            model.addAttribute("question", question);
            return "question_detail";
        }
        this.answerService.create(question, answerForm.getContent(), siteUser);
        return String.format("redirect:/question/detail/%s", id);
    }
}

 

 

 

 

6.질문 수정

질문도 답변과 동일한 방법으로 수정해준다.

6-1.질문 서비스

public void create(String subject, String content, SiteUser user) {
    Question q = new Question();
    q.setSubject(subject);
    q.setContent(content);
    q.setCreateDate(LocalDateTime.now());
    q.setAuthor(user);
    this.questionRepository.save(q);
}

 

 

 

6-2.질문 컨트롤러

@PostMapping("/create")
public String questionCreate(@Valid QuestionForm questionForm, BindingResult bindingResult, Principal principal) {
    if (bindingResult.hasErrors()) {
        return "question_form";
    }
    SiteUser siteUser = this.userService.getUser(principal.getName());
    this.questionService.create(questionForm.getSubject(), questionForm.getContent(), siteUser);

    return "redirect:/question/list"; // 질문 저장후 질문목록으로 이동
}
}

 

 

 

 

 

 

7.로그아웃상태일때 처리

7-1.@PreAuthorize("isAuthenticated()")

@PreAuthorize("isAuthenticated()") 애너테이션이 붙은 메서드는 로그인한 경우에만 실행된다.

@PreAuthorize("isAuthenticated()")
    @GetMapping("/create")
    public String questionCreate(QuestionForm questionForm) {
        return "question_form";
    }

    @PreAuthorize("isAuthenticated()")
    @PostMapping("/create")
    public String questionCreate(@Valid QuestionForm questionForm, BindingResult bindingResult, Principal principal) {
        if (bindingResult.hasErrors()) {
            return "question_form";
        }
        SiteUser siteUser = this.userService.getUser(principal.getName());
        this.questionService.create(questionForm.getSubject(), questionForm.getContent(), siteUser);

        return "redirect:/question/list"; // 질문 저장후 질문목록으로 이동
    }
}

 

 @PreAuthorize("isAuthenticated()")
    @PostMapping("/create/{id}")
    public String createAnswer(Model model, @PathVariable("id") Integer id, @Valid AnswerForm answerForm, BindingResult bindingResult, Principal principal) {
        Question question = this.questionService.getQuestion(id);
        SiteUser siteUser = this.userService.getUser(principal.getName());
        if (bindingResult.hasErrors()) {
            model.addAttribute("question", question);
            return "question_detail";
        }
        this.answerService.create(question, answerForm.getContent(), siteUser);
        return String.format("redirect:/question/detail/%s", id);
    }
}

 

 

7-2.SecurityConfig.java

@EnableMethodSecurity 애너테이션의prePostEnabled = true는
QuestionController와 AnswerController에서 로그인 여부를 판별할 때 사용한
@PreAuthorize 애너테이션을 사용하기 위해 반드시 필요한 설정이다.

import org.springframework.security.config.annotation.method.configuration.EnableMethodSecurity;

@Configuration
@EnableWebSecurity
@EnableMethodSecurity(prePostEnabled = true)
public class SecurityConfig {

 

 

7-3.답변 비활성화

sec:authorize를 사용하여 로그아웃 상태일때는 답변창이 비활성화되게만든다.

<form th:action="@{|/answer/create/${question.id}|}" th:object="${answerForm}" method="post">
    <div th:replace="~{form_errors :: formErrorsFragment}"></div>
    <textarea sec:authorize="isAnonymous()" disabled th:field="*{content}" class="form-control" rows="10"></textarea>
        <textarea sec:authorize="isAuthenticated()" th:field="*{content}" class="form-control" rows="10"></textarea>
    <input type="submit" value="답변등록">
</form>
</div>

 

 

 

 

 

 

8.글쓴이 표시

글쓴이가 보이도록 템플릿들을 수정한다.

<table class="table">
    <thead class="table-dark">
        <tr>
            <th>번호</th>
            <th>제목</th>
            <th>글쓴이</th>
            <th>작성일시</th>
        </tr>
    </thead>
    <tbody>
        <tr th:each="question ,loop: ${paging}">

            <td th:text="${paging.getTotalElements - (paging.number * paging.size) - loop.index}"></td>
            <td>
                <a th:href="@{|/question/detail/${question.id}|}" th:text="${question.subject}"></a>
                <span class="text-danger small ms-2"
                    th:if="${#lists.size(question.answerList) > 0}" 
                    th:text="${#lists.size(question.answerList)}">
                </span>
            </td>
            <td><span th:if="${question.author != null}" th:text="${question.author.username}"></span></td>
            <td th:text="${#temporals.format(question.createDate, 'yyyy-MM-dd HH:mm')}"></td>
        </tr>
    </tbody>
</table>

 

 

 

 

 

<html layout:decorate="~{base}">

<div layout:fragment="content">
<h1 th:text="${question.subject}"></h1>
<div th:text="${question.content}"></div>



<h5 th:text="|${#lists.size(question.answerList)}개의 답변이 있습니다.|"></h5>
<div class="card my-3" th:each="answer : ${question.answerList}">
    <div class="card-body">
        <div class="card-text" style="white-space: pre-line;" th:text="${answer.content}"></div>
        <div class="d-flex justify-content-end">
            <div class="badge bg-light text-dark p-2 text-start">
                <div class="mb-2">
                    <span th:if="${answer.author != null}" th:text="${answer.author.username}"></span>
                </div>
                <div th:text="${#temporals.format(answer.createDate, 'yyyy-MM-dd HH:mm')}"></div>
            </div>
        </div>
    </div>
</div>

<form th:action="@{|/answer/create/${question.id}|}" th:object="${answerForm}" method="post">
    <div th:replace="~{form_errors :: formErrorsFragment}"></div>
    <textarea sec:authorize="isAnonymous()" disabled th:field="*{content}" class="form-control" rows="10"></textarea>
        <textarea sec:authorize="isAuthenticated()" th:field="*{content}" class="form-control" rows="10"></textarea>
    <input type="submit" value="답변등록">
</form>
</div>