基于javax.validation自定义日期范围校验注解

实际项目中经常需要对传入的日期时间进行判断,如是否为一年内,几个月之内,几天前,几天之内等等的需求。如要求前端传入的日期是要为当前日期一年内的某个日期,基于jdk8的LocalDateTime or LocalDate等常用的做法如下:

  // 前端传字符串如‘2020-07-13 09:09:09’ springmvc接收并转换为LocalDateTime类型
   @JsonFormat(shape = JsonFormat.Shape.STRING, pattern = "yyyy-MM-dd HH:mm:ss", timezone = "GMT+8")
   private LocalDateTime endTime;
  

  LocalDateTime now = LocalDateTime.now(); 
  // jdk8校验传入日期是否为一年内
  boolean flag = endTime.isBefore(now.plusYears(1))

基于上述的做法通常是比较通用的模式,如果每个日期时间都重复如此判断,略微繁琐,于是可以通过javax.validation的自定义校验注解来作用于实体属性上,借住hibernate-validate与springmvc结合来解决此类日期时间的范围校验。

  • DateTimeRange.java 用于LocalDateTime or String类型日期时间范围校验
/**
 * 日期时间范围校验注解,可作用于LocalDateTime or 字符串型年月日时分秒(格式可通过pattern属性指定)
 * 
 * @author meilin.huang
 * @date 2020-07-09 1:51 下午
 */
@Target({ElementType.METHOD, ElementType.FIELD, ElementType.ANNOTATION_TYPE, ElementType.CONSTRUCTOR, ElementType.PARAMETER, ElementType.TYPE_USE})
@Retention(RetentionPolicy.RUNTIME)
@Documented
@Constraint(validatedBy = DateTimeRange.DateTimeRangeValidator.class)
public @interface DateTimeRange {

    /**
     * 最小时间范围(为负数即为前n unit)
     */
    int min() default 0;

    int max() default Integer.MAX_VALUE;

    /**
     * 时间单位 (年月日)
     */
    RangeUnit unit() default RangeUnit.DAYS;

    /**
     * 作用于字符串时,指定的格式,包含年月日时分秒
     */
    String pattern() default "yyyy-MM-dd HH:mm:ss";

    /**
     * 是否忽略更小的单位,即比当前指定的unit更小的单位(如unit为Days,则忽略hours,minutes, second)
     */
    boolean ignoreLowerUnit() default false;

    /**
     * 错误提示
     */
    String message() default "日期时间错误";

    /**
     * 用于分组校验
     */
    Class<?>[] groups() default {};

    Class<? extends Payload>[] payload() default {};


    class DateTimeRangeValidator implements ConstraintValidator<DateTimeRange, Object> {

        private DateTimeRange dateTimeRange;

        @Override
        public void initialize(DateTimeRange dateRange) {
            this.dateTimeRange = dateRange;
        }

        @Override
        public boolean isValid(Object value, ConstraintValidatorContext context) {
            if (value == null) {
                return true;
            }
            LocalDateTime ta = getByValue(value);
            if (ta == null) {
                return false;
            }

            RangeUnit unit = dateTimeRange.unit();
            int min = dateTimeRange.min();
            int max = dateTimeRange.max();

            // 忽略更小单位时,计算两个时间的单位差值比较即可
            if (dateTimeRange.ignoreLowerUnit()) {
                long between = RangeUnit.getBetween(dateTimeRange.unit(), LocalDateTime.now(), ta);
                return between >= min && between <= max;
            }

            LocalDateTime now = LocalDateTime.now();
            return ta.isAfter((ChronoLocalDateTime<?>) RangeUnit.plus(now, unit, min))
                    && ta.isBefore((ChronoLocalDateTime<?>) RangeUnit.plus(now, unit, max));
        }

        private LocalDateTime getByValue(Object value) {
            if (value instanceof LocalDateTime) {
                return (LocalDateTime) value;
            }
            if (value instanceof String) {
                try {
                    return LocalDateTime.parse((String) value, DateTimeFormatter.ofPattern(dateTimeRange.pattern()));
                } catch (Exception e) {
                    return null;
                }
            }
            return null;
        }
    }
}
  • DateRange.java 用于LocalDate or String类型日期范围校验
/**
 * 日期范围校验,作用于 {@link LocalDate} or 字符串(年月日,格式可通过pattern属性指定,默认格式为: yyyy-MM-dd)
 *
 * @author meilin.huang
 * @date 2020-07-08 5:15 下午
 */
@Target({ElementType.METHOD, ElementType.FIELD, ElementType.ANNOTATION_TYPE, ElementType.CONSTRUCTOR, ElementType.PARAMETER, ElementType.TYPE_USE})
@Retention(RetentionPolicy.RUNTIME)
@Documented
@Constraint(validatedBy = DateRange.DateRangeValidator.class)
public @interface DateRange {

    /**
     * 最小时间范围(为负数即为前n unit)
     */
    int min() default 0;

    int max() default Integer.MAX_VALUE;

    /**
     * 时间单位 (年月日)
     */
    RangeUnit unit() default RangeUnit.DAYS;

    /**
     * 作用于字符串时,指定的格式,只能包含年月日不包含时间
     */
    String pattern() default "yyyy-MM-dd";

    /**
     * 是否忽略更小的单位,即比当前指定的unit更小的单位(如unit为Month,则忽略Days)
     */
    boolean ignoreLowerUnit() default false;

    /**
     * 错误提示
     */
    String message() default "日期错误";

    /**
     * 用于分组校验
     */
    Class<?>[] groups() default {};

    Class<? extends Payload>[] payload() default {};


    class DateRangeValidator implements ConstraintValidator<DateRange, Object> {

        private DateRange dateRange;

        @Override
        public void initialize(DateRange dateRange) {
            this.dateRange = dateRange;
        }

        @Override
        public boolean isValid(Object value, ConstraintValidatorContext context) {
            if (value == null) {
                return true;
            }
            LocalDate ta = getByValue(value);
            if (ta == null) {
                return false;
            }
            // 忽略更小单位时,计算两个时间的单位差值比较即可
            if (dateRange.ignoreLowerUnit()) {
                long between = RangeUnit.getBetween(dateRange.unit(), LocalDate.now(), ta);
                return between >= dateRange.min() && between <= dateRange.max();
            }

            LocalDate now = LocalDate.now();
            RangeUnit unit = dateRange.unit();
            ChronoLocalDate min = (ChronoLocalDate) RangeUnit.plus(now, unit, dateRange.min());
            ChronoLocalDate max = (ChronoLocalDate) RangeUnit.plus(now, unit, dateRange.max());
            return (ta.isAfter(min) || ta.isEqual(min))
                    && (ta.isBefore(max) || ta.isEqual(max));
        }

        private LocalDate getByValue(Object value) {
            if (value instanceof LocalDate) {
                return (LocalDate) value;
            }
            if (value instanceof String) {
                try {
                    return LocalDate.parse((String) value, DateTimeFormatter.ofPattern(dateRange.pattern()));
                } catch (Exception e) {
                    return null;
                }
            }
            if (value instanceof LocalDateTime) {
                return ((LocalDateTime) value).toLocalDate();
            }
            return null;
        }
    }
}
  • RangeUnit.java 范围单位枚举,用于以上两注解的unit属性
/**
 * @author meilin.huang
 * @date 2020-07-09 2:06 下午
 */
public enum RangeUnit {

    /**
     * 年
     */
    YEAR,

    MONTH,

    DAYS,

    WEEKS,

    HOURS,

    MINUTES,

    SECOND;

    public static Temporal plus(Temporal date, RangeUnit unit, long value) {
        switch (unit) {
            case DAYS:
                return date.plus(value, ChronoUnit.DAYS);
            case MONTH:
                return date.plus(value, ChronoUnit.MONTHS);
            case YEAR:
                return date.plus(value, ChronoUnit.YEARS);
            case WEEKS:
                return date.plus(value, ChronoUnit.WEEKS);
            case HOURS:
                return date.plus(value, ChronoUnit.HOURS);
            case MINUTES:
                return date.plus(value, ChronoUnit.MINUTES);
            case SECOND:
                return date.plus(value, ChronoUnit.SECONDS);
            default:
                return date;
        }
    }

    public static long getBetween(RangeUnit unit, Temporal date, Temporal date2) {
        switch (unit) {
            case DAYS:
                return ChronoUnit.DAYS.between(date, date2);
            case MONTH:
                return ChronoUnit.MONTHS.between(date, date2);
            case YEAR:
                return ChronoUnit.YEARS.between(date, date2);
            case WEEKS:
                return ChronoUnit.WEEKS.between(date, date2);
            case HOURS:
                return ChronoUnit.HOURS.between(date, date2);
            case MINUTES:
                return ChronoUnit.MINUTES.between(date, date2);
            case SECOND:
                return ChronoUnit.SECONDS.between(date, date2);
            default:
                return 0;
        }
    }

}
  • 测试类
public class ValidatorUtilsTest extends TestCase {

    public static class User {

        /**
         * createTime必须在当前日期的前一个月以及后一个月访问内,忽略day的比较
         * 即计算当前时间与传入createTime的月份差是否在最小与最大值范围内
         */
        @DateRange(min = -1, max = 1, unit = RangeUnit.MONTH, ignoreLowerUnit = true, message = "createTime范围错误")
        String createTime = "2020-06-09";

        /**
         * date必须为当前日期时间的后一个月至后两月内
         */
        @DateTimeRange(min = 1, max = 2, unit = RangeUnit.MONTH, message = "date范围错误")
        String date = "2020-08-15 09:18:00";

        @DateTimeRange(min = 10, max = 30, message = "endTime日期范围错误")
        LocalDateTime endTime = LocalDateTime.now().plusDays(30);
    }

    @Test
    public void testValidate() {
        ValidationResult validate = ValidatorUtils.validate(new User());
        System.out.println(validate);
    }
}

  通过注解的方式对参数的日期时间属性进行范围校验,可以简化代码,去除冗余重复繁琐的代码,舒服不是一点点。当然以上是基于jdk8的localdatetime和localdate实现,如需要对Date类型范围校验,稍作修改即可。更多方便有趣的代码可以前往个人业余总结以及练手项目中获取哈https://gitee.com/objs/mayfly

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 212,542评论 6 493
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 90,596评论 3 385
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 158,021评论 0 348
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 56,682评论 1 284
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 65,792评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 49,985评论 1 291
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,107评论 3 410
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 37,845评论 0 268
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,299评论 1 303
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,612评论 2 327
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,747评论 1 341
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,441评论 4 333
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,072评论 3 317
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,828评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,069评论 1 267
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 46,545评论 2 362
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 43,658评论 2 350