package cn.exrick.xboot.core.serviceimpl;
|
|
import cn.exrick.xboot.core.common.vo.SearchVo;
|
import cn.exrick.xboot.core.dao.StopWordDao;
|
import cn.exrick.xboot.core.entity.StopWord;
|
import cn.exrick.xboot.core.service.StopWordService;
|
import cn.hutool.core.date.DateUtil;
|
import cn.hutool.core.util.StrUtil;
|
import cn.hutool.dfa.WordTree;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.data.domain.Page;
|
import org.springframework.data.domain.Pageable;
|
import org.springframework.data.jpa.domain.Specification;
|
import org.springframework.lang.Nullable;
|
import org.springframework.stereotype.Service;
|
import org.springframework.transaction.annotation.Transactional;
|
|
import javax.persistence.criteria.*;
|
import java.util.ArrayList;
|
import java.util.Date;
|
import java.util.List;
|
|
/**
|
* 禁用词管理接口实现
|
* @author Exrick
|
*/
|
@Slf4j
|
@Service
|
@Transactional
|
public class StopWordServiceImpl implements StopWordService {
|
|
@Autowired
|
private StopWordDao stopWordDao;
|
|
@Override
|
public StopWordDao getRepository() {
|
return stopWordDao;
|
}
|
|
private static WordTree wordTree;
|
|
public WordTree getInstance() {
|
|
if (wordTree == null) {
|
// 初始加载数据
|
wordTree = new WordTree();
|
stopWordDao.findAll().forEach(e -> wordTree.addWord(e.getTitle()));
|
}
|
return wordTree;
|
}
|
|
@Override
|
public Page<StopWord> findByCondition(StopWord stopWord, SearchVo searchVo, Pageable pageable) {
|
|
return stopWordDao.findAll(new Specification<StopWord>() {
|
@Nullable
|
@Override
|
public Predicate toPredicate(Root<StopWord> root, CriteriaQuery<?> cq, CriteriaBuilder cb) {
|
|
Path<String> titleField = root.get("title");
|
Path<Date> createTimeField = root.get("createTime");
|
|
List<Predicate> list = new ArrayList<>();
|
|
if (StrUtil.isNotBlank(stopWord.getTitle())) {
|
list.add(cb.like(titleField, "%" + stopWord.getTitle() + "%"));
|
}
|
|
// 创建时间
|
if (StrUtil.isNotBlank(searchVo.getStartDate()) && StrUtil.isNotBlank(searchVo.getEndDate())) {
|
Date start = DateUtil.parse(searchVo.getStartDate());
|
Date end = DateUtil.parse(searchVo.getEndDate());
|
list.add(cb.between(createTimeField, start, DateUtil.endOfDay(end)));
|
}
|
|
Predicate[] arr = new Predicate[list.size()];
|
cq.where(list.toArray(arr));
|
return null;
|
}
|
}, pageable);
|
}
|
|
}
|
