Wiredly enforcing uniqe-users with an exception

[demos/spring/data-jdbc] / src / main / java / de / juplo / boot / data / jdbc / UserController.java

diff --git a/src/main/java/de/juplo/boot/data/jdbc/UserController.java b/src/main/java/de/juplo/boot/data/jdbc/UserController.java
index 4eb1094..ac5ddbc 100644 (file)
--- a/src/main/java/de/juplo/boot/data/jdbc/UserController.java
+++ b/src/main/java/de/juplo/boot/data/jdbc/UserController.java
@@ -2,13 +2,20 @@ package de.juplo.boot.data.jdbc;
 
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
+import org.springframework.context.ApplicationEventPublisher;
+import org.springframework.dao.IncorrectResultSizeDataAccessException;
 import org.springframework.http.ResponseEntity;
 import org.springframework.transaction.annotation.Transactional;
 import org.springframework.web.bind.annotation.*;
 import org.springframework.web.servlet.support.ServletUriComponentsBuilder;
 import org.springframework.web.util.UriComponents;
 
+import java.time.Clock;
 import java.time.LocalDateTime;
+import java.time.ZonedDateTime;
+
+import static de.juplo.boot.data.jdbc.UserStatus.CREATED;
+import static de.juplo.boot.data.jdbc.UserStatus.DELETED;
 
 @RestController
 @Transactional
@@ -19,10 +26,18 @@ public class UserController {
 
 
     private final UserRepository repository;
+    private final Clock clock;
+    private final ApplicationEventPublisher publisher;
 
 
-    public UserController(UserRepository repository) {
+    public UserController(
+            UserRepository repository,
+            Clock clock,
+            ApplicationEventPublisher publisher)
+    {
         this.repository = repository;
+        this.clock = clock;
+        this.publisher = publisher;
     }
 
 
@@ -32,7 +47,20 @@ public class UserController {
             @RequestBody String username) {
         String sanitizedUsername = UserController.sanitize(username);
         User user = new User(sanitizedUsername, LocalDateTime.now(), false);
+
+        LOG.info("Request to create user: {}", user);
+
         repository.save(user);
+        publisher.publishEvent(
+            new UserEvent(
+                this,
+                sanitizedUsername,
+                CREATED,
+                ZonedDateTime.now(clock)));
+
+        // Triggers an IncorrectResultSizeDataAccessException, if the user already existed!
+        user = repository.findByUsername(sanitizedUsername);
+
         UriComponents uri =
             builder
                 .fromCurrentRequest()
@@ -59,6 +87,12 @@ public class UserController {
             return ResponseEntity.notFound().build();
 
         repository.delete(user);
+        publisher.publishEvent(
+            new UserEvent(
+                this,
+                user.getUsername(),
+                DELETED,
+                ZonedDateTime.now(clock)));
 
         return ResponseEntity.ok(user);
     }
@@ -75,4 +109,13 @@ public class UserController {
 
         return string.trim().toLowerCase();
     }
+
+    @ExceptionHandler
+    public ResponseEntity<?> incorrectResultSizeDataAccessException(
+        IncorrectResultSizeDataAccessException e
+        )
+    {
+      LOG.info("User already exists!");
+      return ResponseEntity.badRequest().build();
+    }
 }