用自定义类级注解校验两字段不能同时为空

背景:

有下面这么一个类,要校验两字段query、image不能同时为空,应该怎么实现?已知的@NotBlank等标签都只能检验单个字段。

import jakarta.validation.constraints.NotBlank;
import lombok.Data;
import org.springframework.http.codec.multipart.FilePart;

@Data
public class MultiModalQARequest {
。。。。

    /**
     * 用户提问
     */
    @NotBlank(message = "query不能为空")
    private String query;

    private FilePart image;

}


解决:

参考https://docs.jboss.org/hibernate/stable/validator/reference/en-US/html_single/?v=8.0#section-class-level-constraints中的在这里插入图片描述
我有以下实现:

import jakarta.validation.Constraint;
import jakarta.validation.Payload;

import java.lang.annotation.Documented;
import java.lang.annotation.Retention;
import java.lang.annotation.Target;

import static java.lang.annotation.ElementType.ANNOTATION_TYPE;
import static java.lang.annotation.ElementType.TYPE;
import static java.lang.annotation.RetentionPolicy.RUNTIME;

@Target({TYPE, ANNOTATION_TYPE})
@Retention(RUNTIME)
@Constraint(validatedBy = {NotBothBlankValidator.class})
@Documented
public @interface NotBothBlank {

    String message() default "两个字段不能同时为空";

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

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

    String firstFieldName();

    String secondFieldName();
}

import com.zju.design.aiplatformdemo2.utils.StringUtil;
import jakarta.validation.ConstraintValidator;
import jakarta.validation.ConstraintValidatorContext;
import lombok.extern.slf4j.Slf4j;

import java.lang.reflect.Field;
import java.util.function.BiFunction;

@Slf4j
public class NotBothBlankValidator
        implements ConstraintValidator<NotBothBlank, Object> {

    private NotBothBlank constraintAnnotation;

    @Override
    public void initialize(NotBothBlank constraintAnnotation) {
        this.constraintAnnotation = constraintAnnotation;
    }

    private static BiFunction<Object, Field, Boolean> isBlank = (object, field) ->
    {
        log.info("NotBothBlankValidator field type {}", field.getType());
        try {
            field.setAccessible(true);
            return (String.class.isAssignableFrom(field.getType()) ? StringUtil.isBlank((String) field.get(object)) : field.get(object) == null);
        } catch (IllegalAccessException e) {
            throw new RuntimeException(e);
        }
    };

    @Override
    public boolean isValid(Object validatedObject, ConstraintValidatorContext context) {
        if (validatedObject == null) {
            return true;
        }
        //通过反射拿到validatedObject对象的firstFieldName和secondFieldName字段的值,判断它们不能同时为空:
        try {
            Field firstField = validatedObject.getClass().getDeclaredField(constraintAnnotation.firstFieldName());
            Field secondField = validatedObject.getClass().getDeclaredField(constraintAnnotation.secondFieldName());
            return !(isBlank.apply(validatedObject, firstField) && isBlank.apply(validatedObject, secondField));
        } catch (Exception e) {
            log.error("NotBothBlankValidator error", e);
            throw new RuntimeException(e);
        }
    }
}

业务代码的变更:
在这里插入图片描述

测试可用。

其它相关文档:

https://stackoverflow.com/questions/54638255/spring-boot-validation-one-from-two-not-null

  • 3
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

qq_23204557

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值