RiderMessageServiceImpl.java 3.95 KB
package com.diligrp.rider.service.impl;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.diligrp.rider.entity.RiderMessage;
import com.diligrp.rider.mapper.RiderMessageMapper;
import com.diligrp.rider.service.RiderMessageService;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import java.util.HashMap;
import java.util.List;
import java.util.Map;

@Slf4j
@Service
@RequiredArgsConstructor
public class RiderMessageServiceImpl implements RiderMessageService {

    private final RiderMessageMapper messageMapper;

    @Override
    public Map<String, Object> listByRider(Long riderId, Integer type, Integer page, Integer pageSize) {
        Page<RiderMessage> pageObj = new Page<>(page, pageSize);
        LambdaQueryWrapper<RiderMessage> wrapper = new LambdaQueryWrapper<>();
        wrapper.eq(RiderMessage::getRiderId, riderId)
                .eq(type != null, RiderMessage::getType, type)
                .orderByDesc(RiderMessage::getCreateTime);

        Page<RiderMessage> result = messageMapper.selectPage(pageObj, wrapper);

        Map<String, Object> data = new HashMap<>();
        data.put("list", result.getRecords());
        data.put("total", result.getTotal());
        data.put("page", page);
        data.put("pageSize", pageSize);
        return data;
    }

    @Override
    public Map<String, Integer> getUnreadCount(Long riderId) {
        LambdaQueryWrapper<RiderMessage> wrapper = new LambdaQueryWrapper<>();
        wrapper.eq(RiderMessage::getRiderId, riderId)
                .eq(RiderMessage::getIsRead, 0);

        Long total = messageMapper.selectCount(wrapper);

        wrapper.eq(RiderMessage::getType, 1);
        Long orderCount = messageMapper.selectCount(wrapper);

        wrapper.clear();
        wrapper.eq(RiderMessage::getRiderId, riderId)
                .eq(RiderMessage::getIsRead, 0)
                .eq(RiderMessage::getType, 2);
        Long systemCount = messageMapper.selectCount(wrapper);

        Map<String, Integer> result = new HashMap<>();
        result.put("total", total.intValue());
        result.put("order", orderCount.intValue());
        result.put("system", systemCount.intValue());
        return result;
    }

    @Override
    @Transactional
    public void markRead(Long riderId, Long messageId) {
        RiderMessage message = messageMapper.selectById(messageId);
        if (message != null && message.getRiderId().equals(riderId) && message.getIsRead() == 0) {
            message.setIsRead(1);
            message.setReadTime(System.currentTimeMillis());
            messageMapper.updateById(message);
        }
    }

    @Override
    @Transactional
    public void markReadBatch(Long riderId, List<Long> messageIds) {
        if (messageIds == null || messageIds.isEmpty()) {
            return;
        }

        LambdaUpdateWrapper<RiderMessage> wrapper = new LambdaUpdateWrapper<>();
        wrapper.eq(RiderMessage::getRiderId, riderId)
                .in(RiderMessage::getId, messageIds)
                .eq(RiderMessage::getIsRead, 0)
                .set(RiderMessage::getIsRead, 1)
                .set(RiderMessage::getReadTime, System.currentTimeMillis());

        messageMapper.update(null, wrapper);
    }

    @Override
    @Transactional
    public void markAllRead(Long riderId, Integer type) {
        LambdaUpdateWrapper<RiderMessage> wrapper = new LambdaUpdateWrapper<>();
        wrapper.eq(RiderMessage::getRiderId, riderId)
                .eq(type != null, RiderMessage::getType, type)
                .eq(RiderMessage::getIsRead, 0)
                .set(RiderMessage::getIsRead, 1)
                .set(RiderMessage::getReadTime, System.currentTimeMillis());

        messageMapper.update(null, wrapper);
    }
}