mirror of
				https://github.com/IntellectualSites/PlotSquared.git
				synced 2025-11-03 18:53:43 +01:00 
			
		
		
		
	Make sure lock is free'd even on error (fixes #3116)
This commit is contained in:
		@@ -71,11 +71,9 @@ public final class LockRepository {
 | 
				
			|||||||
     * @param runnable Action to run when the lock is available
 | 
					     * @param runnable Action to run when the lock is available
 | 
				
			||||||
     */
 | 
					     */
 | 
				
			||||||
    public void useLock(final @NonNull LockKey key, final @NonNull Runnable runnable) {
 | 
					    public void useLock(final @NonNull LockKey key, final @NonNull Runnable runnable) {
 | 
				
			||||||
        this.useLock(key, lock -> {
 | 
					       try (LockAccess ignored = lock(key)) {
 | 
				
			||||||
            lock.lock();
 | 
					 | 
				
			||||||
           runnable.run();
 | 
					           runnable.run();
 | 
				
			||||||
            lock.unlock();
 | 
					       }
 | 
				
			||||||
        });
 | 
					 | 
				
			||||||
    }
 | 
					    }
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    /**
 | 
					    /**
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -0,0 +1,40 @@
 | 
				
			|||||||
 | 
					package com.plotsquared.core.synchronization;
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					import org.junit.jupiter.api.BeforeEach;
 | 
				
			||||||
 | 
					import org.junit.jupiter.api.DisplayName;
 | 
				
			||||||
 | 
					import org.junit.jupiter.api.Test;
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					import java.util.concurrent.locks.Lock;
 | 
				
			||||||
 | 
					import java.util.concurrent.locks.ReentrantLock;
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					import static org.junit.jupiter.api.Assertions.assertFalse;
 | 
				
			||||||
 | 
					import static org.junit.jupiter.api.Assertions.assertThrows;
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					class LockRepositoryTest {
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    private LockKey key;
 | 
				
			||||||
 | 
					    private LockRepository lockRepository;
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    @BeforeEach
 | 
				
			||||||
 | 
					    void setUp() {
 | 
				
			||||||
 | 
					        this.key = LockKey.of("test");
 | 
				
			||||||
 | 
					        this.lockRepository = new LockRepository();
 | 
				
			||||||
 | 
					    }
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    @Test
 | 
				
			||||||
 | 
					    @DisplayName("Unlock even if there is an error")
 | 
				
			||||||
 | 
					    void useLockUnlock() {
 | 
				
			||||||
 | 
					        Lock l = this.lockRepository.getLock(this.key);
 | 
				
			||||||
 | 
					        // Striped uses a ReentrantLock internally, and we need its isLocked method for this test
 | 
				
			||||||
 | 
					        if (!(l instanceof ReentrantLock lock)) {
 | 
				
			||||||
 | 
					            throw new IllegalStateException("Expected a ReentrantLock");
 | 
				
			||||||
 | 
					        }
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        assertThrows(IllegalStateException.class, () -> {
 | 
				
			||||||
 | 
					            this.lockRepository.useLock(this.key, () -> {
 | 
				
			||||||
 | 
					                throw new IllegalStateException();
 | 
				
			||||||
 | 
					            });
 | 
				
			||||||
 | 
					        });
 | 
				
			||||||
 | 
					        assertFalse(lock.isLocked());
 | 
				
			||||||
 | 
					    }
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
		Reference in New Issue
	
	Block a user