yunsu/cloud-model/src/main/java/com/hssx/cloudmodel/service/impl/NewsNoticeServiceImpl.java

package com.hssx.cloudmodel.service.impl;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.github.pagehelper.PageHelper;
import com.github.pagehelper.PageInfo;
import com.hssx.cloudmodel.entity.NewsNotice;
import com.hssx.cloudmodel.entity.User;
import com.hssx.cloudmodel.entity.vo.NewsNoticeVO;
import com.hssx.cloudmodel.mapper.NewsNoticeMapper;
import com.hssx.cloudmodel.mapper.UserMapper;
import com.hssx.cloudmodel.service.NewsNoticeService;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.hssx.cloudmodel.util.HttpRespMsg;
import com.hssx.cloudmodel.util.PageUtil;
import org.springframework.stereotype.Service;

import javax.annotation.Resource;
import java.util.ArrayList;
import java.util.List;

/**
 * <p>
 *  服务实现类
 * </p>
 *
 * @author 吴涛涛
 * @since 2019-08-09
 */
@Service
public class NewsNoticeServiceImpl extends ServiceImpl<NewsNoticeMapper, NewsNotice> implements NewsNoticeService {
    @Resource
    NewsNoticeMapper newsNoticeMapper;
    @Resource
    UserMapper userMapper;

    @Override
    public HttpRespMsg pageList(NewsNotice newsNotice,PageUtil page,Integer flag,String token) {
        HttpRespMsg msg = new HttpRespMsg();
        List<NewsNoticeVO> list = new ArrayList<>();
        User user = userMapper.selectOne(new QueryWrapper<User>().eq("head_imgurl",token));
        if(user != null){
            if(flag == 0){
                //显示最近五条的未读消息
                list = newsNoticeMapper.selectNewestFiveUnreadNewsByUserId(user);
                msg.data = list;
            }else if(flag == 1){
                //加载全部
                PageHelper.startPage(page.getPageNum(),page.getPageSize());
                list = newsNoticeMapper.selectAllUnreadNewsByUserId(user,newsNotice);
                PageInfo<NewsNoticeVO> pageInfo = new PageInfo<>(list);
                msg.data = pageInfo;
            }
        }else{
            msg.setError("用户不存在或者未登录");
        }
        return msg;
    }
}