💻 Today I Learned

drf에서 게시글 조회수, 추천 기능 구현하기

조회수와 추천하기 기능을 구현하기 위해 우선 모델에 likes와 views 필드(ManyToManyField)를 추가했다.

	likes = models.ManyToManyField(User, blank=True, related_name='article_likes')
    views = models.ManyToManyField(User, related_name='article_views')

그 다음 게시글 상세 serializer에서 views를 SerializerMethodField()로 obj.views.count()로 커스텀 한 후 count + 1을 해주었는데 save가 되지 않았다.

class ArticleDetailSerializer(serializers.ModelSerializer):
    views = serializers.SerializerMethodField()
    
    def views(self, obj):
        return obj.views.count() +1

    class Meta:
        model = Article
        exclude = ('created_at',)

다시 views에서 serializer를 is_valid()로 검증 후 save를 했더니 에러가 발생하면서 views의 count를 늘릴 수 없었다.

또 다른 방식으로 like 로직처럼 아래와 같이 시도하여 views 테이블에 request user가 없을 경우 add로 추가하는 방식으로 성공했으나 로그인 된 사용자가 조회했을 때만 조회 수가 올라가도록 할 수 있었고, 팀원들과 상의하에 비로그인 유저도 조회했을 때 조회수가 증가하는 것으로 결정되어 다시 다른 방법으로 수정했다.

class ArticleDetailView(APIView):
	permission_classes = [permissions.IsAuthenticatedOrReadOnly]
    def get(self, request, article_id):
    	article = get_object_or_404(Article, id=article_id)

		if request.user.is_authenticated:
        	article.views.add(request.user)

		serializer = ArticleDetailSerializer(article)

		return Response(serializer.data, status=status.HTTP_200_OK)

먼저 모델부터 ManyToManyField 대신 BigIntegerField를 사용하여 serializer에서 views필드를 .count로 커스텀할 필요가 없이 views를 바로 +1로 올릴 수 있도록 수정했다.

    like = models.ManyToManyField(User, blank=True, related_name='article_like')
    views = models.BigIntegerField(default=0)

그리고 이번엔 serializer로 save하는 방법과 views 테이블에 add로 추가하는 방법 대신 views.py의 ArticleDetailView에서 article을 특정한 후 article의 views필드에 +1로 views가 늘어나게 하고, save로 저장해보았다.

class ArticleDetailView(APIView):
    permission_classes = [permissions.IsAuthenticatedOrReadOnly]
    def get(self, request, article_id):
        article = get_object_or_404(Article, id=article_id)
        article.views += 1
        article.save()
        
        serializer = ArticleDetailSerializer(article)
        return Response(serializer.data, status=status.HTTP_200_OK)

성공적으로 GET요청이 올 때 마다 해당 게시글의 조회수가 늘어나는 것을 확인할 수 있었고 추천 기능은 위에서 시도한 방법으로 article의 Like 테이블에 request.user가 없는 경우 add하는 방식으로 기능을 구현할 수 있었다.