市场上有些这样的应用，会展示某年某月一个月的数据，比如女神们经常用的“大姨妈”APP，一些游戏的用户签到信息，等等

那我们在写后台接口的时候，就需要返回这一个月的数据，今天我就分享一下笔者经常在工作中使用的方法。

数据库DB中存了许多用户的应用数据，每条数据有一个日期字段，可以是Integer类型(yyyyMMdd)，可以是String类型(yyyy-MM-dd)，还可以是Date类型.

首先定义了一个这样的类来保存某天的数据：

public class CalendarDate{publicInteger day;publicInteger weekDay;publicBoolean isToday;publicT info;

}

四个字段的意思分别是：day-这个月的第几天，weekDay-星期几，isToday-是否是今天，info-该天的用户数据。

请求接口如下，请求需要年和月两个参数：

@GetMapping("/calendar")public List>>calendarDates(@RequestParam Integer year, @RequestParam Integer month) {

Function>> function = day ->{

List datas = new ArrayList<>();

datas.add(day);//DB: datas = tableMapper.findByDay(day);

returnOptional.of(datas);

};returnDateUtils.calendar(year, month, function);

}

重点就是下面封装好的工具类：

public classDateUtils {public static List> calendar(int year, int month, Function>function) {

List> cdList = new ArrayList<>();int monthDays =monthDays(year, month);

CalendarDatecdR;for (int day = 1; day <= monthDays; day++) {

cdR= new CalendarDate<>();

cdR.day=day;

LocalDate date=LocalDate.of(year, month, day);

cdR.weekDay=dayOfWeek(date);

cdR.isToday=isToday(date);if (function != null) {

Optional optional =function.apply(date.toString());if(optional.isPresent()) {

cdR.info=optional.get();

}

}

cdList.add(cdR);

}returncdList;

}private static booleanisToday(LocalDate date) {

LocalDate today=LocalDate.now();return date.getYear() == today.getYear() &&date.getMonth()== today.getMonth() &&date.getDayOfMonth()==today.getDayOfMonth();

}private static intdayOfWeek(LocalDate date) {return null == date ? 0: date.getDayOfWeek().getValue();

}private static int monthDays(int year, intmonth) {

Calendar calendar=Calendar.getInstance();

calendar.set(Calendar.YEAR, year);

calendar.set(Calendar.MONTH, month- 1);

calendar.set(Calendar.DATE,1);

calendar.roll(Calendar.DATE,-1);returncalendar.get(Calendar.DATE);

}

}

View Code

针对不同的业务要求，编写不一样的function函数，就可以返回日历形式的用户数据了。