prevent multiple locks by same user

baloise-incubator · Oct 20, 2023 · 22f96ef · 22f96ef
1 parent cabe4ca
commit 22f96ef
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/spring-backend/src/main/java/com/baloise/collab/springbackend/items/ItemsRepository.java b/spring-backend/src/main/java/com/baloise/collab/springbackend/items/ItemsRepository.java
@@ -3,7 +3,9 @@
 import org.springframework.data.repository.CrudRepository;
 import org.springframework.stereotype.Repository;
 
+import java.util.List;
+
 @Repository
 public interface ItemsRepository extends CrudRepository<ItemEntity, Long> {
-
+    List<ItemEntity> findByUserLock(String userLock);
 }
diff --git a/...g-backend/src/main/java/com/baloise/collab/springbackend/items/ItemsWebSocketHandler.java b/...g-backend/src/main/java/com/baloise/collab/springbackend/items/ItemsWebSocketHandler.java
@@ -49,6 +49,7 @@ public void handleItemLock(Message<String> message) throws JsonProcessingExcepti
         String userName = optionalToken.map(AbstractAuthenticationToken::getName).orElse("dummyUser");
         if (dto.lock()) {
             if (item.getUserLock() == null) {
+                resetUserLocks(userName);
                 item.setUserLock(userName);
                 itemsRepository.save(item);
                 notifyInventoryChange();
@@ -62,6 +63,14 @@ public void handleItemLock(Message<String> message) throws JsonProcessingExcepti
         }
     }
 
+    private void resetUserLocks(String userName) {
+        var entities = itemsRepository.findByUserLock(userName);
+        for (var entity : entities) {
+            entity.setUserLock(null);
+        }
+        itemsRepository.saveAll(entities);
+    }
+
     @MessageMapping("/itemtransfer")
     public void handleItemTransfer(Message<String> message) throws JsonProcessingException {
         var dto = objectMapper.readValue(message.getPayload(), ItemTransferDto.class);