rentaiming
2024-05-27 b69edf91df77182944ccd376ec0ec1c033bf5b82
ruoyi-modules/ruoyi-article/src/main/java/com/ruoyi/article/service/impl/MemberLikeServiceImpl.java
@@ -1,10 +1,22 @@
package com.ruoyi.article.service.impl;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.ruoyi.article.domain.Article;
import com.ruoyi.article.domain.ArticleComments;
import com.ruoyi.article.domain.MemberLike;
import com.ruoyi.article.dto.MemberLikeDTO;
import com.ruoyi.article.mapper.MemberLikeMapper;
import com.ruoyi.article.service.IArticleCommentsService;
import com.ruoyi.article.service.IArticleService;
import com.ruoyi.article.service.IMemberLikeService;
import com.ruoyi.common.core.exception.ServiceException;
import org.springframework.stereotype.Service;
import javax.annotation.Resource;
import java.util.List;
/**
 * <p>
@@ -18,4 +30,69 @@
public class MemberLikeServiceImpl extends ServiceImpl<MemberLikeMapper, MemberLike> implements
        IMemberLikeService {
    @Resource
    private  IMemberLikeService memberLikeService;
    @Resource
    private IArticleService articleService;
    @Resource
    private IArticleCommentsService articleCommentsService;
    @Override
    public void saveMemberLike(MemberLikeDTO memberLikeDTO) {
        if (memberLikeDTO.getMemberId()!=null) {
            throw new ServiceException("用户ID不能为空");
        }
        if (memberLikeDTO.getState()!=null) {
            throw new ServiceException("类型不能为空");
        }
        if (memberLikeDTO.getState()==1){
            MemberLike memberLike=new MemberLike();
            memberLike.setMemberId(memberLikeDTO.getMemberId());
            memberLike.setArticleId(memberLikeDTO.getArticleId());
            memberLike.setType(memberLikeDTO.getType());
            if (memberLikeDTO.getType()==1){
                Article byId = articleService.getById(memberLikeDTO.getArticleId());
                byId.setLikeCount(byId.getLikeCount()+1);
                articleService.saveOrUpdate(byId);
            }
            if (memberLikeDTO.getType()==2){
                ArticleComments byId = articleCommentsService.getById(memberLikeDTO.getArticleId());
                byId.setLikeCount(byId.getLikeCount()+1);
                articleCommentsService.saveOrUpdate(byId);
            }
            if (memberLikeDTO.getType()==3){
                ArticleComments byId = articleCommentsService.getById(memberLikeDTO.getArticleId());
                byId.setLikeCount(byId.getLikeCount()+1);
                articleCommentsService.saveOrUpdate(byId);
            }
            memberLikeService.save(memberLike);
        }else{
            LambdaQueryWrapper< MemberLike> wrapper= Wrappers.lambdaQuery();
            wrapper.eq(MemberLike::getMemberId,memberLikeDTO.getMemberId());
            wrapper.eq(MemberLike::getArticleId,memberLikeDTO.getArticleId());
            wrapper.eq(MemberLike::getType,memberLikeDTO.getType());
            wrapper.eq( MemberLike::getDelFlag,0);
            List< MemberLike> memberLikes = memberLikeService.list(wrapper);
            for (MemberLike memberLike:memberLikes){
                memberLikeService.removeById(memberLike);
                if (memberLikeDTO.getType()==1){
                    Article byId = articleService.getById(memberLikeDTO.getArticleId());
                    byId.setLikeCount(byId.getLikeCount()-1);
                    articleService.saveOrUpdate(byId);
                }
                if (memberLikeDTO.getType()==2){
                    ArticleComments byId = articleCommentsService.getById(memberLikeDTO.getArticleId());
                    byId.setLikeCount(byId.getLikeCount()-1);
                    articleCommentsService.saveOrUpdate(byId);
                }
                if (memberLikeDTO.getType()==3){
                    ArticleComments byId = articleCommentsService.getById(memberLikeDTO.getArticleId());
                    byId.setLikeCount(byId.getLikeCount()-1);
                    articleCommentsService.saveOrUpdate(byId);
                }
            }
        }
    }
}