完成收藏模块

Reviewed-on: #3

src/main/java/com/bao/dating/controller/PostController.java

@@ -20,7 +20,7 @@ public class PostController {
     private PostService postService;
 
     /**
-     * 上传媒体文件接口
+     * 上传媒体文件接口 like
      * @param files 媒体文件数组
      * @return 上传后的文件URL列表
      */

src/main/java/com/bao/dating/controller/PostFavoriteController.java

@@ -0,0 +1,34 @@
+package com.bao.dating.controller;
+
+import com.bao.dating.common.Result;
+import com.bao.dating.common.ResultCode;
+import com.bao.dating.pojo.entity.User;
+import com.bao.dating.service.PostFavoriteService;
+import com.bao.dating.service.PostService;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.web.bind.annotation.*;
+
+import java.util.Map;
+
+@RestController
+@RequestMapping("/posts")
+public class PostFavoriteController {
+    @Autowired
+    private PostFavoriteService postFavoriteService;
+    @PostMapping("/{post_id}/favorites")
+    public Result<Map<String,Long>> addPostFavorite(@PathVariable("post_id")Long postId, User user){
+        if (user == null){
+            return Result.error(ResultCode.PARAM_ERROR);
+        }
+        Long userId = user.getUserId();
+        return postFavoriteService.postFavorite(userId,postId);
+    }
+    @DeleteMapping("/{post_id}/favorites")
+    public Result<?> deletePostFavorite(@PathVariable("post_id")Long postId, User user){
+        if (user == null){
+            return Result.error(ResultCode.PARAM_ERROR);
+        }
+        Long userId = user.getUserId();
+        return postFavoriteService.deletePostFavorite(userId, postId);
+    }
+}

src/main/java/com/bao/dating/controller/text.java

@@ -0,0 +1,4 @@
+package com.bao.dating.controller;
+
+public class text {
+}

src/main/java/com/bao/dating/mapper/PostFavoriteMapper.java

@@ -0,0 +1,15 @@
+package com.bao.dating.mapper;
+
+import com.bao.dating.pojo.entity.PostFavorite;
+import org.apache.ibatis.annotations.Mapper;
+import org.apache.ibatis.annotations.Param;
+
+import java.util.List;
+
+@Mapper
+public interface PostFavoriteMapper {
+    //查询当前已收藏所有用户
+    List<Long> selectUserIDByPostID(@Param("postId") Long postId);
+    int addPostFavorite(PostFavorite postFavorite);
+    int deletePostFavorite(@Param("postId") Long postId);
+}

src/main/java/com/bao/dating/mapper/PostLikeMapper.java

@@ -6,11 +6,29 @@ import org.apache.ibatis.annotations.Param;
 
 @Mapper
 public interface PostLikeMapper {
-    //根据 postId 和 userId 查询点赞记录 用于判断是否已经点赞
+    /**
+     * 根据 postId 和 userId 查询点赞记录 用于判断是否已经点赞
+     * @param postId
+     * @param userId
+     * @return
+     */
     PostLike findByPostAndUser(@Param("postId") Long postId, @Param("userId") Long userId);
 
+    /**
+     * 插入点赞记录
+     *
+     * @param postLike
+     * @return
+     */
     int insert(PostLike postLike);
 
-    // 根据 postId + userId 删除点赞
+    /**
+     * 删除点赞记录
+     * 根据 postId + userId 删除点赞
+     *
+     * @param postId
+     * @param userId
+     * @return
+     */
     int deleteByPostIdAndUserId(@Param("postId") Long postId, @Param("userId") Long userId);
 }

src/main/java/com/bao/dating/mapper/PostMapper.java

@@ -2,16 +2,77 @@ package com.bao.dating.mapper;
 
 import com.bao.dating.pojo.entity.Post;
 import org.apache.ibatis.annotations.Mapper;
+import org.apache.ibatis.annotations.Param;
 
 @Mapper
 public interface PostMapper {
+    /**
+     * 插入动态
+     *
+     * @param post 动态对象
+     * @return 插入的行数
+     */
     int insert(Post post);
 
-    int increaseLikeCount(Long postId);
+    /**
+     * 根据ID删除动态
+     *
+     * @param postId 动态ID
+     * @return 删除的行数
+     */
+    int deletePostById(Integer postId);
 
+    /**
+     * 查询点赞数
+     *
+     * @param postId
+     * @return
+     */
     int selectLikeCount(Long postId);
 
+    /**
+     * 点赞数+1
+     *
+     * @param postId 动态ID
+     * @return 影响行数
+     */
+    int increaseLikeCount(Long postId);
+
+    /**
+     * 点赞数-1
+     *
+     * @param postId 动态ID
+     * @return 影响行数
+     */
     int decreaseLikeCount(Long postId);
 
-    int deletePostById(Integer postId);
+    /**
+     * 查询当前动态属于哪个用户id
+     * @param postId 动态id
+     * @return 用户id
+     */
+    Long selectUserIdByPostId(@Param("post_id") Long postId);
+
+    /**
+     * 查询点赞数
+     *
+     * @param postId
+     * @return
+     */
+    int selectFavoriteCount(Long postId);
+
+    /**
+     * 收藏数+1
+     *
+     * @param postId 动态ID
+     * @return 影响行数
+     */
+    int increaseFavoriteCount(Long postId);
+    /**
+     * 收藏数-1
+     *
+     * @param postId 动态ID
+     * @return 影响行数
+     */
+    int decreaseFavoriteCount(Long postId);
 }

src/main/java/com/bao/dating/pojo/dto/PostRequestDTO.java

@@ -7,7 +7,7 @@ import java.util.List;
 
 @Data
 public class PostRequestDTO {
-    private Integer userId;
+    private Long userId;
     private String content;
     private List<String> tags;
     private List<String> mediaUrls;

src/main/java/com/bao/dating/service/PostFavoriteService.java

@@ -0,0 +1,10 @@
+package com.bao.dating.service;
+
+import com.bao.dating.common.Result;
+
+import java.util.Map;
+
+public interface PostFavoriteService {
+    Result<Map<String,Long>> postFavorite(Long userid,Long postId);
+    Result<?> deletePostFavorite(Long userid,Long postId);
+}

src/main/java/com/bao/dating/service/PostLikeService.java

@@ -11,6 +11,12 @@ public interface PostLikeService {
      * @return 统一返回结果
      */
     Result<?> likePost(Long postId, Long userId);
-    //取消点赞
+
+    /**
+     * 取消点赞指定的 post
+     *
+     * @param postId 帖子ID
+     * @param userId 用户ID
+     */
     void unlike(Long postId, Long userId);
 }

src/main/java/com/bao/dating/service/PostService.java

@@ -27,4 +27,11 @@ public interface PostService {
      * @return 删除的动态对象
      */
     void deletePostById(Integer postId);
+
+    /**
+     * 查询
+     * @param postId 动态id
+     * @return 用户id
+     */
+    Long selectUserIdByPostId(Long postId);
 }

src/main/java/com/bao/dating/service/impl/PostFavoriteServiceImpl.java

@@ -0,0 +1,61 @@
+package com.bao.dating.service.impl;
+
+import com.bao.dating.common.Result;
+import com.bao.dating.common.ResultCode;
+import com.bao.dating.mapper.PostFavoriteMapper;
+import com.bao.dating.mapper.PostMapper;
+import com.bao.dating.pojo.entity.PostFavorite;
+import com.bao.dating.service.PostFavoriteService;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Service;
+import org.springframework.transaction.annotation.Transactional;
+
+import java.time.LocalDateTime;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+
+@Service
+public class PostFavoriteServiceImpl implements PostFavoriteService {
+    @Autowired
+    private PostFavoriteMapper postFavoriteMapper;
+    @Autowired
+    private PostMapper postMapper;
+    @Override
+    @Transactional
+    public Result<Map<String, Long>> postFavorite(Long userid, Long postId) {
+        Long userId = postMapper.selectUserIdByPostId(postId);
+        if (userid.equals(userId)){
+            return Result.error(ResultCode.FORBIDDEN,"无法收藏自己发布动态");
+        }
+        List<Long> allUserId = postFavoriteMapper.selectUserIDByPostID(postId);
+        if (allUserId.contains(userid)){
+            return Result.error(ResultCode.FORBIDDEN,"已收藏");
+        }
+        PostFavorite postFavorite = new PostFavorite();
+        postFavorite.setPostId(postId);
+        postFavorite.setUserId(userid);
+        postFavorite.setCreatedAt(LocalDateTime.now());
+        postFavoriteMapper.addPostFavorite(postFavorite);
+        postMapper.increaseFavoriteCount(postId);
+        Long count = (long) postMapper.selectFavoriteCount(postId);
+        Map<String, Long> data = new HashMap<>();
+        data.put("favorite_id",postFavorite.getFavoriteId());
+        data.put("post_id",postId);
+        data.put("user_id",userid);
+        data.put("current_favorites_count",count);
+        return Result.success(ResultCode.SUCCESS_REVIEW,"收藏成功",data);
+    }
+
+    @Override
+    @Transactional
+    public Result<?> deletePostFavorite(Long userid, Long postId) {
+        List<Long> allUserId = postFavoriteMapper.selectUserIDByPostID(postId);
+        if (! allUserId.contains(userid)){
+            return Result.error(ResultCode.FORBIDDEN,"请先收藏");
+        }
+        postFavoriteMapper.deletePostFavorite(postId);
+        postMapper.decreaseFavoriteCount(postId);
+        return null;
+    }
+}

src/main/java/com/bao/dating/service/impl/PostLikeServiceImpl.java

@@ -24,6 +24,13 @@ public class PostLikeServiceImpl implements PostLikeService {
 
 
 
+    /**
+     * 点赞指定的 post
+     *
+     * @param postId 帖子ID
+     * @param userId 用户ID
+     * @return 统一返回结果
+     */
     @Override
     @Transactional // 保证点赞 + 点赞数更新是一个事务
     public Result<?> likePost(Long postId, Long userId) {
@@ -51,6 +58,12 @@ public class PostLikeServiceImpl implements PostLikeService {
         return Result.success(ResultCode.SUCCESS_REVIEW,"点赞成功",data);
     }
 
+    /**
+     * 取消点赞指定的 post
+     *
+     * @param postId 帖子ID
+     * @param userId 用户ID
+     */
     @Override
     @Transactional
     public void unlike(Long postId, Long userId) {

src/main/java/com/bao/dating/service/impl/PostServiceImpl.java

@@ -147,4 +147,15 @@ public class PostServiceImpl implements PostService {
     public void deletePostById(Integer postId) {
         postMapper.deletePostById(postId);
     }
+
+    /**
+     * 查询用户id
+     * @param postId 动态id
+     * @return
+     */
+
+    @Override
+    public Long selectUserIdByPostId(Long postId) {
+        return postMapper.selectUserIdByPostId(postId);
+    }
 }

src/main/resources/com/bao/dating/mapper/PostFavoriteMapper.xml

@@ -0,0 +1,17 @@
+<?xml version="1.0" encoding="UTF-8" ?>
+<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN"
+        "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
+<mapper namespace="com.bao.dating.mapper.PostFavoriteMapper">
+    <!--添加动态收藏-->
+    <insert id="addPostFavorite">
+        insert into post_favorite values (null,#{userId},#{postId},#{createdAt})
+    </insert>
+    <!--删除动态收藏-->
+    <delete id="deletePostFavorite">
+        delete from post_favorite where post_id = #{postId}
+    </delete>
+    <!--查询当前已收藏所有用户-->
+    <select id="selectUserIDByPostID" resultType="java.lang.Long">
+        SELECT user_id FROM post_favorite WHERE post_id = #{postId}
+    </select>
+</mapper>

src/main/resources/com/bao/dating/mapper/PostMapper.xml

@@ -3,7 +3,8 @@
         "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
 
 <mapper namespace="com.bao.dating.mapper.PostMapper">
-    <!--  动态添加  -->
+
+    <!--动态添加-->
     <insert id="insert" useGeneratedKeys="true" keyProperty="postId">
         INSERT INTO post
         (user_id, content,
@@ -26,21 +27,41 @@
         </if>
          #{isPublic}, 0, 0, #{createdAt}, #{updatedAt})
     </insert>
+
+    <!--动态删除-->
+    <delete id="deletePostById" parameterType="int">
+        DELETE FROM post WHERE post_id = #{postId}
+    </delete>
+
     <!--增加点赞数量-->
     <update id="increaseLikeCount">
         update dating.post set like_count = like_count + 1 where post.post_id = #{postId}
     </update>
+
+    <!--减少点赞数量-->
     <update id="decreaseLikeCount">
         update dating.post set like_count= like_count - 1 where post.post_id = #{postId}
     </update>
+    <!--增加收藏数量-->
+    <update id="increaseFavoriteCount">
+        update post set favorite_count = favorite_count + 1 where post_id = #{post_id}
+    </update>
+    <!--减少收藏数量-->
+    <update id="decreaseFavoriteCount">
+        update post set favorite_count = favorite_count - 1 where post_id = #{post_id}
+    </update>
 
     <!--查询点赞当前数量-->
     <select id="selectLikeCount" resultType="java.lang.Integer">
         select dating.post.like_count from dating.post where post.post_id = #{postId}
     </select>
-
+    <!--查询当前动态的用户id-->
-    <delete id="deletePostById" parameterType="int">
+    <select id="selectUserIdByPostId" resultType="java.lang.Long">
-        DELETE FROM post WHERE post_id = #{postId}
+        SELECT user_id FROM post WHERE post_id = #{post_id}
-    </delete>
+    </select>
+    <!--查询当前收藏数量-->
+    <select id="selectFavoriteCount" resultType="java.lang.Integer">
+        select dating.post.favorite_count from dating.post where post.post_id = #{postId}
+    </select>
 
 </mapper>