@RequestBody接收到的参数中如何限制List的长度？

在Spring MVC中，你可以使用@Valid注解和自定义的验证注解来限制List的长度，防止DOS攻击。具体步骤如下：

1. 创建自定义注解：首先，创建一个自定义注解来验证List的长度。

import javax.validation.Constraint;
import javax.validation.Payload;
import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;

@Constraint(validatedBy = ListSizeValidator.class)
@Target({ ElementType.METHOD, ElementType.FIELD, ElementType.PARAMETER })
@Retention(RetentionPolicy.RUNTIME)
public @interface ListSize {
    String message() default "List size is out of bounds";
    Class<?>[] groups() default {};
    Class<? extends Payload>[] payload() default {};
    int min() default 0;
    int max() default Integer.MAX_VALUE;
}

2. 创建验证器：然后，创建一个验证器来实现自定义注解的逻辑。

import javax.validation.ConstraintValidator;
import javax.validation.ConstraintValidatorContext;
import java.util.List;

public class ListSizeValidator implements ConstraintValidator<ListSize, List<?>> {
    private int min;
    private int max;

    @Override
    public void initialize(ListSize constraintAnnotation) {
        this.min = constraintAnnotation.min();
        this.max = constraintAnnotation.max();
    }

    @Override
    public boolean isValid(List<?> list, ConstraintValidatorContext context) {
        if (list == null) {
            return true; // Consider null as valid, use @NotNull for null check
        }
        return list.size() >= min && list.size() <= max;
    }
}

3. 在DTO中使用自定义注解：在你的DTO类中使用自定义注解来限制List的长度。

import javax.validation.constraints.NotNull;
import java.util.List;

public class MyRequestDTO {

    @NotNull
    @ListSize(min = 1, max = 10, message = "List size must be between 1 and 10")
    private List<String> myList;

    // getters and setters
}

4. 在Controller中使用@Valid注解：在你的Controller中使用@Valid注解来触发验证。

import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RestController;
import javax.validation.Valid;

@RestController
public class MyController {

    @PostMapping("/submit")
    public String submit(@Valid @RequestBody MyRequestDTO request) {
        // Your logic here
        return "Success";
    }
}

这样，当你发送一个包含List的请求时，Spring MVC会自动验证List的长度是否在指定范围内。如果不符合范围，将返回验证错误信息。

原文地址：https://blog.csdn.net/weixin_43349479/article/details/140531991

免责声明：本站文章内容转载自网络资源，如本站内容侵犯了原著者的合法权益，可联系本站删除。更多内容请关注自学内容网（zxcms.com）！