枚举常用方法
方法名 | 描述 |
---|---|
toString | 返回当前对象名,子类可以重写 |
name | 返回当前对象名,子类中不能重写 |
ordinal | 返回当前对象的位置号,按照枚举类中的定义顺序从0开始 |
values | 以数组形式返回当前枚举类中所有的常量,是一个静态方法 |
valueOf | 将字符串常量转换为枚举常量,要求字符串常量对应的枚举常量必须存在,否则抛异常,这是一个静态方法 |
compareTo | 比较两个枚举常量的位置号,返回位置号之差 |
应用场景
前端页面查询条件为一个起止日期
前端传到后台直接为 今日 本周 本月 本年等汉字,后端需要查询必须把这些汉字转化为日期范围。
这种第一想到的是if(本周)
{}else if(本年)
{}
等,这样写很难维护代码,下面用枚举类来实现
下面用代码来实现。
枚举类
import cn.hutool.core.date.DateTime;
import cn.hutool.core.date.DateUtil;
import java.time.ZoneId;
import java.util.Date;
public enum DateEnum implements DateRange {
JR("今日"){
public DateRangeQuery getTimeRange(String dataStr) {
DateRangeQuery dateRangeQuery = new DateRangeQuery();
Date date = DateUtil.date();
Date beginOfDay = DateUtil.beginOfDay(date);
dateRangeQuery.setTimeGe(beginOfDay.toInstant()
.atZone(ZoneId.systemDefault())
.toLocalDateTime());
//一天的结束,
Date endOfDay = DateUtil.endOfDay(date);
dateRangeQuery.setTimeLe(endOfDay.toInstant()
.atZone(ZoneId.systemDefault())
.toLocalDateTime());
return dateRangeQuery;
}
},
BZ("本周"){
public DateRangeQuery getTimeRange(String dataStr) {
DateRangeQuery dateRangeQuery = new DateRangeQuery();
Date date = DateUtil.date();
Date beginOfDay = DateUtil.beginOfWeek(date);
dateRangeQuery.setTimeGe(beginOfDay.toInstant()
.atZone(ZoneId.systemDefault())
.toLocalDateTime());
Date endOfDay = DateUtil.endOfWeek(date);
dateRangeQuery.setTimeLe(endOfDay.toInstant()
.atZone(ZoneId.systemDefault())
.toLocalDateTime());
return dateRangeQuery;
}
},
BN("本年"){
public DateRangeQuery getTimeRange(String dataStr) {
DateRangeQuery dateRangeQuery = new DateRangeQuery();
Date date = DateUtil.date();
Date beginOfDay = DateUtil.beginOfYear(date);
dateRangeQuery.setTimeGe(beginOfDay.toInstant()
.atZone(ZoneId.systemDefault())
.toLocalDateTime());
//
Date endOfDay = DateUtil.endOfYear(date);
dateRangeQuery.setTimeLe(endOfDay.toInstant()
.atZone(ZoneId.systemDefault())
.toLocalDateTime());
return dateRangeQuery;
}
},
BY("本月"){
public DateRangeQuery getTimeRange(String dataStr) {
DateRangeQuery dateRangeQuery = new DateRangeQuery();
Date date = DateUtil.date();
Date beginOfDay = DateUtil.beginOfMonth(date);
dateRangeQuery.setTimeGe(beginOfDay.toInstant()
.atZone(ZoneId.systemDefault())
.toLocalDateTime());
//
Date endOfDay = DateUtil.endOfMonth(date);
dateRangeQuery.setTimeLe(endOfDay.toInstant()
.atZone(ZoneId.systemDefault())
.toLocalDateTime());
return dateRangeQuery;
}
};
public String msg;
DateEnum(String msg){
this.msg = msg;
}
}
PO
@Data
@Builder
@ToString
@NoArgsConstructor
@AllArgsConstructor
public class DateRangeQuery {
@Schema(description = "待办生成时间-开始")
private LocalDateTime timeGe;
@Schema(description = "待办生成时间-结束")
private LocalDateTime timeLe;
}
接口类
public interface DateRange {
DateRangeQuery getTimeRange(String dataStr);
}
调用类方法
if(StrUtil.isNotEmpty(taskQuery.getGenTime())){
DateRangeQuery d = DateEnum.valueOf(TaskConst.constTime.get(taskQuery.getGenTimeEq())).getTimeRange(taskQuery.getGenTimeEq());
taskQuery.setGenTimeGe(d.getTimeGe());
taskQuery.setGenTimeLe(d.getTimeLe());
}
常量类
public class TaskConst {
public static String SYSTEM_TYPE = "SYSTEM_TYPE";
public static String JR = "今日";
public static String BY = "本月";
public static String BN = "本年";
public static String BZ = "本周";
public static final Map<String, String> constTime = new HashMap<String, String>() {
{
put(JR, "JR");
put(BY , "BY ");
put(BN , "BN ");
put(BZ , "BZ ");
}
};
}