Spring: уникальный идентификатор валидатора без использования Hibernate - PullRequest
Новая
       37

Spring: уникальный идентификатор валидатора без использования Hibernate

1 голос
/ 25 апреля 2020

Каков наилучший способ проверки с помощью валидатора, чтобы он не имел дубликатов при выполнении POST?

Я пытался создать собственный валидатор, но всегда получаю Internal Server Error

{"отметка времени": "2020-04-25T14: 37: 19.158 + 0000", "состояние": 500, "ошибка": "Внутренняя ошибка сервера", "сообщение": "HV000030: проверка не может быть найден для ограничения 'com.omega.mtest.validator.IdConstraint', проверяющего тип 'java .lang.Integer'. Проверьте конфигурацию для 'id' "," path ":" / user "}

Это мой класс модели 

    public class User {

        @IdConstraint
        @Min(value = 1, message = "ID can't be zero or null")
        @Max(value = 1000000, message = "We collect a billion records")
        private int id;



        @Pattern(regexp = "^[a-zA-Z ]*$", message = "Input doesn't match for a full name")
        private String name;

        @Min(value = 10)
        @Max(value = 120, message = "We didn't expect that age")
        private int age;

        @Pattern(regexp = "^[a-zA-Z ]*$", message = "Input doesn't match for a city name")
        private String city;

        public User(int id, String name, int age, String city) {
            this.id = id;
            this.name = name;
            this.age = age;
            this.city = city;
        }

       //getters
    }

Пользовательский интерфейс валидатора: 

@Constraint(validatedBy = UserValidator.class)
@Target({ElementType.METHOD, ElementType.FIELD})
@Retention(RUNTIME)
public @interface IdConstraint {

    String message() default "The input list cannot contain two ugual IDs";
    Class<?>[] groups() default {};
    Class<? extends Payload>[] payload() default {};
}

Реализация класса интерфейса 

public class UserValidator implements ConstraintValidator<IdConstraint, List<User>> {

    @Override
    public void initialize(IdConstraint constraintAnnotation) {

    }

    @Override
    public boolean isValid(List<User> users, ConstraintValidatorContext context) {
        if (users.size() == 1) {
            return true;
        } else {
            for (int i = 0; i < users.size(); i++) {
                for (int j = 0; j < users.size(); j++) {
                    if (i != j) {
                        if (users.get(i).getId() == users.get(j).getId()) {
                            return true;
                        }
                    }
                }
            }
            return false;
        }
    }

Класс контроллера: 

@Validated
@RestController
public class UsersController {

    @Autowired
    public UserService userService;

    @RequestMapping(value = "/user", method = RequestMethod.POST)
    public ResponseEntity<User> createUser(@Valid @RequestBody User user) {
        return new ResponseEntity<>(userService.createUser(user), HttpStatus.CREATED);
    }

    @ExceptionHandler(ConstraintViolationException.class)
    @ResponseStatus(HttpStatus.BAD_REQUEST)
    ResponseEntity<String> handleConstraintViolationException(ConstraintViolationException e) {
        return new ResponseEntity<>("Not valid due to validation error: " + e.getMessage(), HttpStatus.BAD_REQUEST);
    }

    @RequestMapping(value = "/users", method = RequestMethod.GET)
    public List<User> getUsers() {
        return userService.getAllUsers();
    }

    @RequestMapping(value = "/user/{id}", method = RequestMethod.GET)
    public User getUserById(@PathVariable int id) {
        return userService.findUserById(id);
    }
}

UserService 

public interface UserService {

    List<User> getAllUsers();

    User createUser(User user);

    User findUserById(int id);

    List<User> list();
}

Репо 

@Component
public class UserRepository implements UserService {

    private List<User> users;

    public UserRepository() {
        users = new ArrayList<>();
    }

    @Override
    public List<User> getAllUsers() {
        return users.stream().collect(Collectors.toList());
    }

    @Override
    public User createUser(User user) {
        users.add(user);
        return user;
    }


    @Override
    public User findUserById(int id) {
        return users.stream()
                .filter(t -> t.getId() == id)
                .findAny()
                .orElse(null);
    }

    @Override
    public List<User> list() {
        return users;
    }
}

Ответы [ 3 ]

1 голос
/ 25 апреля 2020

Прежде всего вы используете @IdConstraint в неправильном месте, значит private int id. Вы должны поставить это на List<User>. И ваш валидатор работает, когда у вас есть список пользователей и любой из двух пользователей не имеют одинаковый идентификатор в теле запроса. Это можно решить двумя способами.

Проверка - это услуга:

Здесь, в конечной точке сообщения, вы используете одного пользователя в качестве тела запроса. Таким образом, вы можете проверить, существует ли идентификатор в базе данных в службе, используя getById() хранилища. Если существует, то создайте исключение, в противном случае вставьте. 

User user= userRepository.getById(user.getId());
if (ObjectUtils.isEmpty(user)) {
  throw new UserNotFoundException();
}

Использование валидатора:

Не лучший подход к вызову репозитория в функции валидатора. Но все же, вы хотите, чтобы вы могли делать так. 

public class UserIdValidator implements ConstraintValidator<IdConstraint, Integer> {

    @Autowired
    Private UserService userService;

    @Override
    public void initialize(IdConstraint constraintAnnotation) {

    }

    @Override
    public boolean isValid(Integer id, ConstraintValidatorContext context) {
      User user= userService.findUserById(id);
      if (ObjectUtils.isEmpty(user)) {
        return false;
      }
      return true;
    }
}
1 голос
/ 25 апреля 2020

Ваша подпись валидатора неверна.

Вы не валидируете список пользователей здесь, но целое число.

Как это 

public class UserIdValidator implements ConstraintValidator<IdConstraint, Integer> {

Тогда вам нужно обратитесь в службу поддержки, если идентификатор уже используется. 

public class UserIdValidator implements ConstraintValidator<IdConstraint, Integer> {

    @Autowired
    Private UserService userService;

    @Override
    public void initialize(IdConstraint constraintAnnotation) {

    }

    @Override
    public boolean isValid(Integer id, ConstraintValidatorContext context) {
         return !userService.existsById(id);
        }
    }
0 голосов
/ 25 апреля 2020

Я так и сделал, и это работает: 

  @RequestMapping(value = "/user", method = RequestMethod.POST)
public ResponseEntity<User> createUser(@Valid @RequestBody User user) {
    Optional<User> check = userService.findUserById(user.getId());
    if (check.isPresent()) {
        return new ResponseEntity<>(userService.createUser(user), HttpStatus.CREATED);
    }
    return new ResponseEntity<>(null, HttpStatus.BAD_REQUEST);
}



@RequestMapping(value = "/user/{id}", method = RequestMethod.GET)
    public ResponseEntity<User> getUserById(@PathVariable int id) {
        if (userService.findUserById(id).isPresent()) {
            return new ResponseEntity<>(userService.findUserById(id).get(), HttpStatus.OK);
        }
        return new ResponseEntity<>(null, HttpStatus.BAD_REQUEST);
    }
...