diff --git a/src/main/java/com/nelioalves/workshopmongo/services/UserService.java b/src/main/java/com/nelioalves/workshopmongo/services/UserService.java index 599bf21..ee19e09 100644 --- a/src/main/java/com/nelioalves/workshopmongo/services/UserService.java +++ b/src/main/java/com/nelioalves/workshopmongo/services/UserService.java @@ -16,14 +16,18 @@ public class UserService { @Autowired private UserRepository repo; + + private static ObjectNotFoundException get() { + return new ObjectNotFoundException("Objeto não encontrado"); + } public List findAll() { return repo.findAll(); } public User findById(String id) { - Optional obj = repo.findById(id); - return obj.orElseThrow(() -> new ObjectNotFoundException("Objeto não encontrado")); + Optional obj = userRepository.findById(id); + return obj.orElseThrow(UserService::get); } public User insert(User obj) {