56 lines
2.2 KiB
Java
56 lines
2.2 KiB
Java
package com.dota.nexus.service.impl;
|
|
|
|
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
|
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
|
import com.dota.nexus.entity.Anniversary;
|
|
import com.dota.nexus.entity.AnniversaryEnum;
|
|
import com.dota.nexus.entity.vo.AnniversaryVO;
|
|
import com.dota.nexus.mapper.AnniversaryMapper;
|
|
import com.dota.nexus.service.AnniversaryService;
|
|
import org.springframework.stereotype.Service;
|
|
|
|
import java.time.LocalDate;
|
|
import java.time.temporal.ChronoUnit;
|
|
import java.util.ArrayList;
|
|
import java.util.Comparator;
|
|
import java.util.List;
|
|
import java.util.stream.Collectors;
|
|
|
|
@Service
|
|
public class AnniversaryServiceImpl extends ServiceImpl<AnniversaryMapper, Anniversary> implements AnniversaryService {
|
|
|
|
@Override
|
|
public List<AnniversaryVO> getFuture() {
|
|
var anniversaries = this.baseMapper.selectList(null);
|
|
var list = new ArrayList<AnniversaryVO>();
|
|
var now = LocalDate.now();
|
|
for (Anniversary anniversary : anniversaries) {
|
|
// 周年，变成今年或者明年
|
|
if (anniversary.getType() == AnniversaryEnum.ANNIVERSARY) {
|
|
LocalDate date = anniversary.getDate();
|
|
date = date.withYear(now.getYear());
|
|
if (date.isBefore(now)) {
|
|
date = date.withYear(now.getYear() + 1);
|
|
}
|
|
|
|
list.add(new AnniversaryVO(anniversary.getName(), date.toString(), (int) ChronoUnit.DAYS.between(now, date)));
|
|
} else if (anniversary.getType() == AnniversaryEnum.COMMEMORATION && !anniversary.getDate().isBefore(now)) {
|
|
list.add(new AnniversaryVO(anniversary.getName(), anniversary.getDate().toString(), (int) ChronoUnit.DAYS.between(now, anniversary.getDate())));
|
|
}
|
|
}
|
|
list.sort(Comparator.comparingInt(AnniversaryVO::distance));
|
|
|
|
return list;
|
|
}
|
|
|
|
|
|
@Override
|
|
public List<AnniversaryVO> getPast() {
|
|
var q = new LambdaQueryWrapper<Anniversary>();
|
|
q.eq(Anniversary::getType, AnniversaryEnum.COMMEMORATION);
|
|
var anniversaries = this.baseMapper.selectList(q);
|
|
var now = LocalDate.now();
|
|
return anniversaries.stream().map(a -> new AnniversaryVO(a.getName(), a.getDate().toString(), (int) ChronoUnit.DAYS.between(now, a.getDate()))).sorted((a,b)->b.distance()-a.distance()).collect(Collectors.toList());
|
|
}
|
|
}
|