当前位置：首页 > news >正文

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

news 2025/6/27 6:58:17

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

创建自定义注解：首先，创建一个自定义注解来验证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;
}

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

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;@Overridepublic void initialize(ListSize constraintAnnotation) {this.min = constraintAnnotation.min();this.max = constraintAnnotation.max();}@Overridepublic 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;}
}

在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
}

在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 herereturn "Success";}
}