oasys-service/oasys-chat/src/main/java/cn/linter/oasys/chat/service/impl/MessageServiceImpl.java

微信扫码关注公众号，代码整理好后第一时间通知你。

package cn.linter.oasys.chat.service.impl;

import cn.linter.oasys.chat.entity.Message;
import cn.linter.oasys.chat.repository.MessageRepository;
import cn.linter.oasys.chat.service.MessageService;
import org.springframework.data.domain.Page;
import org.springframework.data.domain.PageRequest;
import org.springframework.data.domain.Pageable;
import org.springframework.stereotype.Service;

import java.time.LocalDateTime;

/**
 * 聊天消息服务实现类
 *
 * @author wangxiaoyang
 * @since 2021/1/5
 */
@Service
public class MessageServiceImpl implements MessageService {

    private final MessageRepository repository;

    public MessageServiceImpl(MessageRepository repository) {
        this.repository = repository;
    }

    @Override
    public Page<Message> listMessage(LocalDateTime start, LocalDateTime end, int pageNumber, int pageSize) {
        Pageable pageable = PageRequest.of(pageNumber, pageSize);
        if (start == null && end == null) {
            return repository.findAll(pageable);
        }
        if (start == null) {
            return repository.findAllByCreateTimeBefore(end, pageable);
        }
        if (end == null) {
            return repository.findAllByCreateTimeAfter(start, pageable);
        }
        return repository.findAllByCreateTimeBetween(start, end, pageable);
    }

}