1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
package com.iailab.module.shasteel.job.service.impl;
 
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.iailab.framework.common.pojo.PageResult;
import com.iailab.framework.common.service.impl.BaseServiceImpl;
import com.iailab.module.shasteel.job.dao.MessageInfoDao;
import com.iailab.module.shasteel.job.entity.MessageInfoEntity;
import com.iailab.module.shasteel.job.service.MessageInfoService;
import org.springframework.stereotype.Service;
 
import java.util.List;
 
@Service
public class MessageInfoServiceImpl extends BaseServiceImpl<MessageInfoDao, MessageInfoEntity> implements MessageInfoService {
 
    @Override
    public void create(MessageInfoEntity messageInfoEntity) {
        baseDao.insert(messageInfoEntity);
    }
 
    @Override
    public PageResult<MessageInfoEntity> getMessagePage(Integer pageNo, Integer pageSize) {
        Page<MessageInfoEntity> params = new Page<>(pageNo, pageSize);
        IPage<MessageInfoEntity> page = baseDao.selectPage(params, null);
        return new PageResult<>(page.getRecords(), page.getTotal());
    }
 
    @Override
    public void readMessage(List<String> ids) {
        List<MessageInfoEntity> list = baseDao.selectBatchIds(ids);
        list.forEach(
                item->item.setReadStatus("1")
        );
        baseDao.updateById(list);
    }
 
    @Override
    public List<MessageInfoEntity> getMessageList() {
        return baseDao.selectList(new QueryWrapper<MessageInfoEntity>());
    }
 
    @Override
    public long getUnreadMessageCount() {
        return baseDao.selectCount(new QueryWrapper<MessageInfoEntity>().eq(
                "read_status", 0
        ));
    }
}