java注解-入参校验(标注于单个参数上)

新增基类:

复制代码
package com.yhsp.pay.common.enums.basenum;

public interface BaseEnum {

    String getKey();

    String getValue();

}
复制代码

枚举类实现基类:

复制代码
package com.yhsp.pay.common.enums;

import com.yhsp.pay.common.enums.basenum.BaseEnum;

/**
 * @author guocz
 * @date 20211122
 * 业务平台枚举
 */
public enum BizModeEnum implements BaseEnum {

    /**
     *
     */
    BAR_CODE("1", "默认"),
    APPLET_MALL("3", "小程序商城"),
    PAYMENT_CARD("4", "支付台牌");

    BizModeEnum(String key, String name) {
        this.key = key;
        this.name = name;
    }

    private String key;

    private String name;


    @Override
    public String getKey() {
        return key;
    }

    public String getName() {
        return name;
    }

    public String getKeyByName(String name) {
        for (BizModeEnum s : BizModeEnum.values()) {
            if (s.getName().equals(name)) {
                return s.getKey();
            }
        }
        return "";
    }

    public static String getNameByKey(String key) {
        for (BizModeEnum s : BizModeEnum.values()) {
            if (s.getKey().equals(key)) {
                return s.getName();
            }
        }
        return "";
    }

    @Override
    public String getValue() {
        return name;
    }

}
复制代码

1、创建注解,添加属性,采用@Constraint(validatedBy = {IfInEnumValidator.class})绑定注解实现类。

例:枚举校验

复制代码
package com.yhsp.pay.model.aspect;

import com.yhsp.pay.common.enums.basenum.BaseEnum;
import com.yhsp.pay.model.aspect.impl.IfInEnumValidator;

import javax.validation.Constraint;
import javax.validation.Payload;
import java.lang.annotation.*;

/**
 * @author guocz
 * @date 20211123
 * 不为空判断是否在枚举中
 */
@Target({ElementType.METHOD, ElementType.FIELD, ElementType.ANNOTATION_TYPE, ElementType.CONSTRUCTOR})
@Retention(RetentionPolicy.RUNTIME)
@Constraint(validatedBy = {IfInEnumValidator.class})
public @interface IfInEnum {

    String message() default "枚举值不正确";

    Class<? extends BaseEnum> enumClass();

    Class<?>[] groups() default { };

    Class<? extends Payload>[] payload() default { };
}
复制代码

2、创建注解实现类,实现ConstraintValidator<IfInEnum, String>接口,用于注解绑定该类。

复制代码
package com.yhsp.pay.model.aspect.impl;

import com.yhsp.pay.common.enums.basenum.BaseEnum;
import com.yhsp.pay.model.aspect.IfInEnum;
import org.apache.commons.lang3.StringUtils;
import org.hibernate.validator.internal.engine.constraintvalidation.ConstraintValidatorContextImpl;

import javax.validation.ConstraintValidator;
import javax.validation.ConstraintValidatorContext;

/**
 * @author guocz
 * @date 20211123
 * 不为空判断是否在枚举中
 */
public class IfInEnumValidator implements ConstraintValidator<IfInEnum, String> {

    @Override
    public void initialize(IfInEnum constraintAnnotation) {
    }

    @Override
    public boolean isValid(String s, ConstraintValidatorContext constraintValidatorContext) {
        if (StringUtils.isBlank(s)) {
            return true;
        }

        Class c = (Class) ((ConstraintValidatorContextImpl) constraintValidatorContext).getConstraintDescriptor().getAttributes().get("enumClass");
        BaseEnum[] baseEnum = (BaseEnum[]) c.getEnumConstants();
        for (BaseEnum e : baseEnum) {
            if (e.getKey().equals(s)) {
                return true;
            }
        }

        return false;
    }

}
复制代码

 

posted @   幻影黑子  阅读(1482)  评论(0编辑  收藏  举报
编辑推荐:
· AI与.NET技术实操系列(二):开始使用ML.NET
· 记一次.NET内存居高不下排查解决与启示
· 探究高空视频全景AR技术的实现原理
· 理解Rust引用及其生命周期标识(上)
· 浏览器原生「磁吸」效果!Anchor Positioning 锚点定位神器解析
阅读排行:
· 全程不用写代码,我用AI程序员写了一个飞机大战
· DeepSeek 开源周回顾「GitHub 热点速览」
· 记一次.NET内存居高不下排查解决与启示
· MongoDB 8.0这个新功能碉堡了,比商业数据库还牛
· .NET10 - 预览版1新功能体验(一)
点击右上角即可分享
微信分享提示