ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/jsr166/jsr166/src/test/tck/StampedLockTest.java
(Generate patch)

Comparing jsr166/src/test/tck/StampedLockTest.java (file contents):
Revision 1.39 by jsr166, Sun Sep 24 15:24:34 2017 UTC vs.
Revision 1.45 by jsr166, Wed Aug 12 16:12:23 2020 UTC

# Line 8 | Line 8
8   import static java.util.concurrent.TimeUnit.DAYS;
9   import static java.util.concurrent.TimeUnit.MILLISECONDS;
10  
11 + import static java.util.concurrent.locks.StampedLock.isLockStamp;
12 + import static java.util.concurrent.locks.StampedLock.isOptimisticReadStamp;
13 + import static java.util.concurrent.locks.StampedLock.isReadLockStamp;
14 + import static java.util.concurrent.locks.StampedLock.isWriteLockStamp;
15 +
16   import java.util.ArrayList;
17   import java.util.List;
18 + import java.util.concurrent.Callable;
19 + import java.util.concurrent.CompletableFuture;
20   import java.util.concurrent.CountDownLatch;
21   import java.util.concurrent.Future;
22 + import java.util.concurrent.ThreadLocalRandom;
23   import java.util.concurrent.TimeUnit;
24 + import java.util.concurrent.atomic.AtomicBoolean;
25   import java.util.concurrent.locks.Lock;
26   import java.util.concurrent.locks.StampedLock;
27   import java.util.function.BiConsumer;
28 + import java.util.function.Consumer;
29   import java.util.function.Function;
30  
31   import junit.framework.Test;
# Line 70 | Line 80 | public class StampedLockTest extends JSR
80      }
81  
82      List<Action> lockLockers(Lock lock) {
83 <        List<Action> lockers = new ArrayList<>();
84 <        lockers.add(() -> lock.lock());
85 <        lockers.add(() -> lock.lockInterruptibly());
86 <        lockers.add(() -> lock.tryLock());
87 <        lockers.add(() -> lock.tryLock(Long.MIN_VALUE, DAYS));
88 <        lockers.add(() -> lock.tryLock(0L, DAYS));
89 <        lockers.add(() -> lock.tryLock(Long.MAX_VALUE, DAYS));
80 <        return lockers;
83 >        return List.of(
84 >            () -> lock.lock(),
85 >            () -> lock.lockInterruptibly(),
86 >            () -> lock.tryLock(),
87 >            () -> lock.tryLock(Long.MIN_VALUE, DAYS),
88 >            () -> lock.tryLock(0L, DAYS),
89 >            () -> lock.tryLock(Long.MAX_VALUE, DAYS));
90      }
91  
92      List<Function<StampedLock, Long>> readLockers() {
93 <        List<Function<StampedLock, Long>> readLockers = new ArrayList<>();
94 <        readLockers.add(sl -> sl.readLock());
95 <        readLockers.add(sl -> sl.tryReadLock());
96 <        readLockers.add(sl -> readLockInterruptiblyUninterrupted(sl));
97 <        readLockers.add(sl -> tryReadLockUninterrupted(sl, Long.MIN_VALUE, DAYS));
98 <        readLockers.add(sl -> tryReadLockUninterrupted(sl, 0L, DAYS));
99 <        readLockers.add(sl -> sl.tryConvertToReadLock(sl.tryOptimisticRead()));
91 <        return readLockers;
93 >        return List.of(
94 >            sl -> sl.readLock(),
95 >            sl -> sl.tryReadLock(),
96 >            sl -> readLockInterruptiblyUninterrupted(sl),
97 >            sl -> tryReadLockUninterrupted(sl, Long.MIN_VALUE, DAYS),
98 >            sl -> tryReadLockUninterrupted(sl, 0L, DAYS),
99 >            sl -> sl.tryConvertToReadLock(sl.tryOptimisticRead()));
100      }
101  
102      List<BiConsumer<StampedLock, Long>> readUnlockers() {
103 <        List<BiConsumer<StampedLock, Long>> readUnlockers = new ArrayList<>();
104 <        readUnlockers.add((sl, stamp) -> sl.unlockRead(stamp));
105 <        readUnlockers.add((sl, stamp) -> assertTrue(sl.tryUnlockRead()));
106 <        readUnlockers.add((sl, stamp) -> sl.asReadLock().unlock());
107 <        readUnlockers.add((sl, stamp) -> sl.unlock(stamp));
108 <        readUnlockers.add((sl, stamp) -> assertValid(sl, sl.tryConvertToOptimisticRead(stamp)));
101 <        return readUnlockers;
103 >        return List.of(
104 >            (sl, stamp) -> sl.unlockRead(stamp),
105 >            (sl, stamp) -> assertTrue(sl.tryUnlockRead()),
106 >            (sl, stamp) -> sl.asReadLock().unlock(),
107 >            (sl, stamp) -> sl.unlock(stamp),
108 >            (sl, stamp) -> assertValid(sl, sl.tryConvertToOptimisticRead(stamp)));
109      }
110  
111      List<Function<StampedLock, Long>> writeLockers() {
112 <        List<Function<StampedLock, Long>> writeLockers = new ArrayList<>();
113 <        writeLockers.add(sl -> sl.writeLock());
114 <        writeLockers.add(sl -> sl.tryWriteLock());
115 <        writeLockers.add(sl -> writeLockInterruptiblyUninterrupted(sl));
116 <        writeLockers.add(sl -> tryWriteLockUninterrupted(sl, Long.MIN_VALUE, DAYS));
117 <        writeLockers.add(sl -> tryWriteLockUninterrupted(sl, 0L, DAYS));
118 <        writeLockers.add(sl -> sl.tryConvertToWriteLock(sl.tryOptimisticRead()));
112 <        return writeLockers;
112 >        return List.of(
113 >            sl -> sl.writeLock(),
114 >            sl -> sl.tryWriteLock(),
115 >            sl -> writeLockInterruptiblyUninterrupted(sl),
116 >            sl -> tryWriteLockUninterrupted(sl, Long.MIN_VALUE, DAYS),
117 >            sl -> tryWriteLockUninterrupted(sl, 0L, DAYS),
118 >            sl -> sl.tryConvertToWriteLock(sl.tryOptimisticRead()));
119      }
120  
121      List<BiConsumer<StampedLock, Long>> writeUnlockers() {
122 <        List<BiConsumer<StampedLock, Long>> writeUnlockers = new ArrayList<>();
123 <        writeUnlockers.add((sl, stamp) -> sl.unlockWrite(stamp));
124 <        writeUnlockers.add((sl, stamp) -> assertTrue(sl.tryUnlockWrite()));
125 <        writeUnlockers.add((sl, stamp) -> sl.asWriteLock().unlock());
126 <        writeUnlockers.add((sl, stamp) -> sl.unlock(stamp));
127 <        writeUnlockers.add((sl, stamp) -> assertValid(sl, sl.tryConvertToOptimisticRead(stamp)));
122 <        return writeUnlockers;
122 >        return List.of(
123 >            (sl, stamp) -> sl.unlockWrite(stamp),
124 >            (sl, stamp) -> assertTrue(sl.tryUnlockWrite()),
125 >            (sl, stamp) -> sl.asWriteLock().unlock(),
126 >            (sl, stamp) -> sl.unlock(stamp),
127 >            (sl, stamp) -> assertValid(sl, sl.tryConvertToOptimisticRead(stamp)));
128      }
129  
130      /**
# Line 338 | Line 343 | public class StampedLockTest extends JSR
343       */
344      public void testInterruptibleOperationsThrowInterruptedExceptionWriteLockedInterrupted() {
345          final StampedLock lock = new StampedLock();
346 <        long s = lock.writeLock();
346 >        long stamp = lock.writeLock();
347  
348          Action[] interruptibleLockBlockingActions = {
349              () -> lock.writeLockInterruptibly(),
# Line 353 | Line 358 | public class StampedLockTest extends JSR
358          shuffle(interruptibleLockBlockingActions);
359  
360          assertThrowInterruptedExceptionWhenInterrupted(interruptibleLockBlockingActions);
361 +
362 +        releaseWriteLock(lock, stamp);
363      }
364  
365      /**
# Line 360 | Line 367 | public class StampedLockTest extends JSR
367       */
368      public void testInterruptibleOperationsThrowInterruptedExceptionReadLockedInterrupted() {
369          final StampedLock lock = new StampedLock();
370 <        long s = lock.readLock();
370 >        long stamp = lock.readLock();
371  
372          Action[] interruptibleLockBlockingActions = {
373              () -> lock.writeLockInterruptibly(),
# Line 371 | Line 378 | public class StampedLockTest extends JSR
378          shuffle(interruptibleLockBlockingActions);
379  
380          assertThrowInterruptedExceptionWhenInterrupted(interruptibleLockBlockingActions);
381 +
382 +        releaseReadLock(lock, stamp);
383      }
384  
385      /**
# Line 582 | Line 591 | public class StampedLockTest extends JSR
591          long s = lock.readLock();
592          Thread t = newStartedThread(new CheckedRunnable() {
593              public void realRun() {
594 <                threadAssertEquals(0L, lock.tryWriteLock());
594 >                assertEquals(0L, lock.tryWriteLock());
595              }});
596  
597          awaitTermination(t);
# Line 977 | Line 986 | public class StampedLockTest extends JSR
986       * IllegalMonitorStateException
987       */
988      public void testCannotUnlockOptimisticReadStamps() {
989 <        Runnable[] actions = {
990 <            () -> {
991 <                StampedLock sl = new StampedLock();
992 <                long stamp = assertValid(sl, sl.tryOptimisticRead());
993 <                sl.unlockRead(stamp);
994 <            },
995 <            () -> {
996 <                StampedLock sl = new StampedLock();
997 <                long stamp = sl.tryOptimisticRead();
998 <                sl.unlock(stamp);
999 <            },
1000 <
992 <            () -> {
993 <                StampedLock sl = new StampedLock();
994 <                long stamp = sl.tryOptimisticRead();
995 <                sl.writeLock();
996 <                sl.unlock(stamp);
997 <            },
998 <            () -> {
999 <                StampedLock sl = new StampedLock();
1000 <                sl.readLock();
1001 <                long stamp = assertValid(sl, sl.tryOptimisticRead());
1002 <                sl.unlockRead(stamp);
1003 <            },
1004 <            () -> {
1005 <                StampedLock sl = new StampedLock();
1006 <                sl.readLock();
1007 <                long stamp = assertValid(sl, sl.tryOptimisticRead());
1008 <                sl.unlock(stamp);
1009 <            },
989 >        {
990 >            StampedLock sl = new StampedLock();
991 >            long stamp = assertValid(sl, sl.tryOptimisticRead());
992 >            assertThrows(IllegalMonitorStateException.class,
993 >                () -> sl.unlockRead(stamp));
994 >        }
995 >        {
996 >            StampedLock sl = new StampedLock();
997 >            long stamp = sl.tryOptimisticRead();
998 >            assertThrows(IllegalMonitorStateException.class,
999 >                () -> sl.unlock(stamp));
1000 >        }
1001  
1002 <            () -> {
1003 <                StampedLock sl = new StampedLock();
1004 <                long stamp = sl.tryConvertToOptimisticRead(sl.writeLock());
1005 <                assertValid(sl, stamp);
1006 <                sl.writeLock();
1007 <                sl.unlockWrite(stamp);
1008 <            },
1009 <            () -> {
1010 <                StampedLock sl = new StampedLock();
1011 <                long stamp = sl.tryConvertToOptimisticRead(sl.writeLock());
1012 <                sl.writeLock();
1013 <                sl.unlock(stamp);
1014 <            },
1015 <            () -> {
1016 <                StampedLock sl = new StampedLock();
1017 <                long stamp = sl.tryConvertToOptimisticRead(sl.writeLock());
1018 <                sl.readLock();
1019 <                sl.unlockRead(stamp);
1020 <            },
1021 <            () -> {
1022 <                StampedLock sl = new StampedLock();
1032 <                long stamp = sl.tryConvertToOptimisticRead(sl.writeLock());
1033 <                sl.readLock();
1034 <                sl.unlock(stamp);
1035 <            },
1002 >        {
1003 >            StampedLock sl = new StampedLock();
1004 >            long stamp = sl.tryOptimisticRead();
1005 >            sl.writeLock();
1006 >            assertThrows(IllegalMonitorStateException.class,
1007 >                () -> sl.unlock(stamp));
1008 >        }
1009 >        {
1010 >            StampedLock sl = new StampedLock();
1011 >            sl.readLock();
1012 >            long stamp = assertValid(sl, sl.tryOptimisticRead());
1013 >            assertThrows(IllegalMonitorStateException.class,
1014 >                () -> sl.unlockRead(stamp));
1015 >        }
1016 >        {
1017 >            StampedLock sl = new StampedLock();
1018 >            sl.readLock();
1019 >            long stamp = assertValid(sl, sl.tryOptimisticRead());
1020 >            assertThrows(IllegalMonitorStateException.class,
1021 >                () -> sl.unlock(stamp));
1022 >        }
1023  
1024 <            () -> {
1025 <                StampedLock sl = new StampedLock();
1026 <                long stamp = sl.tryConvertToOptimisticRead(sl.readLock());
1027 <                assertValid(sl, stamp);
1028 <                sl.writeLock();
1029 <                sl.unlockWrite(stamp);
1030 <            },
1031 <            () -> {
1032 <                StampedLock sl = new StampedLock();
1033 <                long stamp = sl.tryConvertToOptimisticRead(sl.readLock());
1034 <                sl.writeLock();
1035 <                sl.unlock(stamp);
1036 <            },
1037 <            () -> {
1038 <                StampedLock sl = new StampedLock();
1039 <                long stamp = sl.tryConvertToOptimisticRead(sl.readLock());
1040 <                sl.readLock();
1041 <                sl.unlockRead(stamp);
1042 <            },
1043 <            () -> {
1044 <                StampedLock sl = new StampedLock();
1045 <                sl.readLock();
1046 <                long stamp = sl.tryConvertToOptimisticRead(sl.readLock());
1047 <                assertValid(sl, stamp);
1048 <                sl.readLock();
1049 <                sl.unlockRead(stamp);
1050 <            },
1051 <            () -> {
1052 <                StampedLock sl = new StampedLock();
1066 <                long stamp = sl.tryConvertToOptimisticRead(sl.readLock());
1067 <                sl.readLock();
1068 <                sl.unlock(stamp);
1069 <            },
1070 <            () -> {
1071 <                StampedLock sl = new StampedLock();
1072 <                sl.readLock();
1073 <                long stamp = sl.tryConvertToOptimisticRead(sl.readLock());
1074 <                sl.readLock();
1075 <                sl.unlock(stamp);
1076 <            },
1077 <        };
1024 >        {
1025 >            StampedLock sl = new StampedLock();
1026 >            long stamp = sl.tryConvertToOptimisticRead(sl.writeLock());
1027 >            assertValid(sl, stamp);
1028 >            sl.writeLock();
1029 >            assertThrows(IllegalMonitorStateException.class,
1030 >                () -> sl.unlockWrite(stamp));
1031 >        }
1032 >        {
1033 >            StampedLock sl = new StampedLock();
1034 >            long stamp = sl.tryConvertToOptimisticRead(sl.writeLock());
1035 >            sl.writeLock();
1036 >            assertThrows(IllegalMonitorStateException.class,
1037 >                () -> sl.unlock(stamp));
1038 >        }
1039 >        {
1040 >            StampedLock sl = new StampedLock();
1041 >            long stamp = sl.tryConvertToOptimisticRead(sl.writeLock());
1042 >            sl.readLock();
1043 >            assertThrows(IllegalMonitorStateException.class,
1044 >                () -> sl.unlockRead(stamp));
1045 >        }
1046 >        {
1047 >            StampedLock sl = new StampedLock();
1048 >            long stamp = sl.tryConvertToOptimisticRead(sl.writeLock());
1049 >            sl.readLock();
1050 >            assertThrows(IllegalMonitorStateException.class,
1051 >                () -> sl.unlock(stamp));
1052 >        }
1053  
1054 <        assertThrows(IllegalMonitorStateException.class, actions);
1054 >        {
1055 >            StampedLock sl = new StampedLock();
1056 >            long stamp = sl.tryConvertToOptimisticRead(sl.readLock());
1057 >            assertValid(sl, stamp);
1058 >            sl.writeLock();
1059 >            assertThrows(IllegalMonitorStateException.class,
1060 >                () -> sl.unlockWrite(stamp));
1061 >            }
1062 >        {
1063 >            StampedLock sl = new StampedLock();
1064 >            long stamp = sl.tryConvertToOptimisticRead(sl.readLock());
1065 >            sl.writeLock();
1066 >            assertThrows(IllegalMonitorStateException.class,
1067 >                () -> sl.unlock(stamp));
1068 >        }
1069 >        {
1070 >            StampedLock sl = new StampedLock();
1071 >            long stamp = sl.tryConvertToOptimisticRead(sl.readLock());
1072 >            sl.readLock();
1073 >            assertThrows(IllegalMonitorStateException.class,
1074 >                () -> sl.unlockRead(stamp));
1075 >        }
1076 >        {
1077 >            StampedLock sl = new StampedLock();
1078 >            sl.readLock();
1079 >            long stamp = sl.tryConvertToOptimisticRead(sl.readLock());
1080 >            assertValid(sl, stamp);
1081 >            sl.readLock();
1082 >            assertThrows(IllegalMonitorStateException.class,
1083 >                () -> sl.unlockRead(stamp));
1084 >        }
1085 >        {
1086 >            StampedLock sl = new StampedLock();
1087 >            long stamp = sl.tryConvertToOptimisticRead(sl.readLock());
1088 >            sl.readLock();
1089 >            assertThrows(IllegalMonitorStateException.class,
1090 >                () -> sl.unlock(stamp));
1091 >        }
1092 >        {
1093 >            StampedLock sl = new StampedLock();
1094 >            sl.readLock();
1095 >            long stamp = sl.tryConvertToOptimisticRead(sl.readLock());
1096 >            sl.readLock();
1097 >            assertThrows(IllegalMonitorStateException.class,
1098 >                () -> sl.unlock(stamp));
1099 >        }
1100      }
1101  
1102      static long writeLockInterruptiblyUninterrupted(StampedLock sl) {
# Line 1259 | Line 1279 | public class StampedLockTest extends JSR
1279                  } while (stamp == 0);
1280                  return Math.hypot(currentX, currentY);
1281              }
1282 +
1283 +            double distanceFromOrigin2() {
1284 +                long stamp = sl.tryOptimisticRead();
1285 +                try {
1286 +                    retryHoldingLock:
1287 +                    for (;; stamp = sl.readLock()) {
1288 +                        if (stamp == 0L)
1289 +                            continue retryHoldingLock;
1290 +                        // possibly racy reads
1291 +                        double currentX = x;
1292 +                        double currentY = y;
1293 +                        if (!sl.validate(stamp))
1294 +                            continue retryHoldingLock;
1295 +                        return Math.hypot(currentX, currentY);
1296 +                    }
1297 +                } finally {
1298 +                    if (StampedLock.isReadLockStamp(stamp))
1299 +                        sl.unlockRead(stamp);
1300 +                }
1301 +            }
1302 +
1303 +            void moveIfAtOrigin(double newX, double newY) {
1304 +                long stamp = sl.readLock();
1305 +                try {
1306 +                    while (x == 0.0 && y == 0.0) {
1307 +                        long ws = sl.tryConvertToWriteLock(stamp);
1308 +                        if (ws != 0L) {
1309 +                            stamp = ws;
1310 +                            x = newX;
1311 +                            y = newY;
1312 +                            return;
1313 +                        }
1314 +                        else {
1315 +                            sl.unlockRead(stamp);
1316 +                            stamp = sl.writeLock();
1317 +                        }
1318 +                    }
1319 +                } finally {
1320 +                    sl.unlock(stamp);
1321 +                }
1322 +            }
1323          }
1324  
1325          Point p = new Point();
1326          p.move(3.0, 4.0);
1327          assertEquals(5.0, p.distanceFromOrigin());
1328 +        p.moveIfAtOrigin(5.0, 12.0);
1329 +        assertEquals(5.0, p.distanceFromOrigin2());
1330      }
1331 <    
1331 >
1332 >    /**
1333 >     * Stamp inspection methods work as expected, and do not inspect
1334 >     * the state of the lock itself.
1335 >     */
1336 >    public void testStampStateInspectionMethods() {
1337 >        StampedLock lock = new StampedLock();
1338 >
1339 >        assertFalse(isWriteLockStamp(0L));
1340 >        assertFalse(isReadLockStamp(0L));
1341 >        assertFalse(isLockStamp(0L));
1342 >        assertFalse(isOptimisticReadStamp(0L));
1343 >
1344 >        {
1345 >            long stamp = lock.writeLock();
1346 >            for (int i = 0; i < 2; i++) {
1347 >                assertTrue(isWriteLockStamp(stamp));
1348 >                assertFalse(isReadLockStamp(stamp));
1349 >                assertTrue(isLockStamp(stamp));
1350 >                assertFalse(isOptimisticReadStamp(stamp));
1351 >                if (i == 0)
1352 >                    lock.unlockWrite(stamp);
1353 >            }
1354 >        }
1355 >
1356 >        {
1357 >            long stamp = lock.readLock();
1358 >            for (int i = 0; i < 2; i++) {
1359 >                assertFalse(isWriteLockStamp(stamp));
1360 >                assertTrue(isReadLockStamp(stamp));
1361 >                assertTrue(isLockStamp(stamp));
1362 >                assertFalse(isOptimisticReadStamp(stamp));
1363 >                if (i == 0)
1364 >                    lock.unlockRead(stamp);
1365 >            }
1366 >        }
1367 >
1368 >        {
1369 >            long optimisticStamp = lock.tryOptimisticRead();
1370 >            long readStamp = lock.tryConvertToReadLock(optimisticStamp);
1371 >            long writeStamp = lock.tryConvertToWriteLock(readStamp);
1372 >            for (int i = 0; i < 2; i++) {
1373 >                assertFalse(isWriteLockStamp(optimisticStamp));
1374 >                assertFalse(isReadLockStamp(optimisticStamp));
1375 >                assertFalse(isLockStamp(optimisticStamp));
1376 >                assertTrue(isOptimisticReadStamp(optimisticStamp));
1377 >
1378 >                assertFalse(isWriteLockStamp(readStamp));
1379 >                assertTrue(isReadLockStamp(readStamp));
1380 >                assertTrue(isLockStamp(readStamp));
1381 >                assertFalse(isOptimisticReadStamp(readStamp));
1382 >
1383 >                assertTrue(isWriteLockStamp(writeStamp));
1384 >                assertFalse(isReadLockStamp(writeStamp));
1385 >                assertTrue(isLockStamp(writeStamp));
1386 >                assertFalse(isOptimisticReadStamp(writeStamp));
1387 >                if (i == 0)
1388 >                    lock.unlockWrite(writeStamp);
1389 >            }
1390 >        }
1391 >    }
1392 >
1393 >    /**
1394 >     * Multiple threads repeatedly contend for the same lock.
1395 >     */
1396 >    public void testConcurrentAccess() throws Exception {
1397 >        final StampedLock sl = new StampedLock();
1398 >        final Lock wl = sl.asWriteLock();
1399 >        final Lock rl = sl.asReadLock();
1400 >        final long testDurationMillis = expensiveTests ? 1000 : 2;
1401 >        final int nTasks = ThreadLocalRandom.current().nextInt(1, 10);
1402 >        final AtomicBoolean done = new AtomicBoolean(false);
1403 >        final List<CompletableFuture> futures = new ArrayList<>();
1404 >        final List<Callable<Long>> stampedWriteLockers = List.of(
1405 >            () -> sl.writeLock(),
1406 >            () -> writeLockInterruptiblyUninterrupted(sl),
1407 >            () -> tryWriteLockUninterrupted(sl, LONG_DELAY_MS, MILLISECONDS),
1408 >            () -> {
1409 >                long stamp;
1410 >                do { stamp = sl.tryConvertToWriteLock(sl.tryOptimisticRead()); }
1411 >                while (stamp == 0L);
1412 >                return stamp;
1413 >            },
1414 >            () -> {
1415 >              long stamp;
1416 >              do { stamp = sl.tryWriteLock(); } while (stamp == 0L);
1417 >              return stamp;
1418 >            },
1419 >            () -> {
1420 >              long stamp;
1421 >              do { stamp = sl.tryWriteLock(0L, DAYS); } while (stamp == 0L);
1422 >              return stamp;
1423 >            });
1424 >        final List<Callable<Long>> stampedReadLockers = List.of(
1425 >            () -> sl.readLock(),
1426 >            () -> readLockInterruptiblyUninterrupted(sl),
1427 >            () -> tryReadLockUninterrupted(sl, LONG_DELAY_MS, MILLISECONDS),
1428 >            () -> {
1429 >                long stamp;
1430 >                do { stamp = sl.tryConvertToReadLock(sl.tryOptimisticRead()); }
1431 >                while (stamp == 0L);
1432 >                return stamp;
1433 >            },
1434 >            () -> {
1435 >              long stamp;
1436 >              do { stamp = sl.tryReadLock(); } while (stamp == 0L);
1437 >              return stamp;
1438 >            },
1439 >            () -> {
1440 >              long stamp;
1441 >              do { stamp = sl.tryReadLock(0L, DAYS); } while (stamp == 0L);
1442 >              return stamp;
1443 >            });
1444 >        final List<Consumer<Long>> stampedWriteUnlockers = List.of(
1445 >            stamp -> sl.unlockWrite(stamp),
1446 >            stamp -> sl.unlock(stamp),
1447 >            stamp -> assertTrue(sl.tryUnlockWrite()),
1448 >            stamp -> wl.unlock(),
1449 >            stamp -> sl.tryConvertToOptimisticRead(stamp));
1450 >        final List<Consumer<Long>> stampedReadUnlockers = List.of(
1451 >            stamp -> sl.unlockRead(stamp),
1452 >            stamp -> sl.unlock(stamp),
1453 >            stamp -> assertTrue(sl.tryUnlockRead()),
1454 >            stamp -> rl.unlock(),
1455 >            stamp -> sl.tryConvertToOptimisticRead(stamp));
1456 >        final Action writer = () -> {
1457 >            // repeatedly acquires write lock
1458 >            var locker = chooseRandomly(stampedWriteLockers);
1459 >            var unlocker = chooseRandomly(stampedWriteUnlockers);
1460 >            while (!done.getAcquire()) {
1461 >                long stamp = locker.call();
1462 >                try {
1463 >                    assertTrue(isWriteLockStamp(stamp));
1464 >                    assertTrue(sl.isWriteLocked());
1465 >                    assertFalse(isReadLockStamp(stamp));
1466 >                    assertFalse(sl.isReadLocked());
1467 >                    assertEquals(0, sl.getReadLockCount());
1468 >                    assertTrue(sl.validate(stamp));
1469 >                } finally {
1470 >                    unlocker.accept(stamp);
1471 >                }
1472 >            }
1473 >        };
1474 >        final Action reader = () -> {
1475 >            // repeatedly acquires read lock
1476 >            var locker = chooseRandomly(stampedReadLockers);
1477 >            var unlocker = chooseRandomly(stampedReadUnlockers);
1478 >            while (!done.getAcquire()) {
1479 >                long stamp = locker.call();
1480 >                try {
1481 >                    assertFalse(isWriteLockStamp(stamp));
1482 >                    assertFalse(sl.isWriteLocked());
1483 >                    assertTrue(isReadLockStamp(stamp));
1484 >                    assertTrue(sl.isReadLocked());
1485 >                    assertTrue(sl.getReadLockCount() > 0);
1486 >                    assertTrue(sl.validate(stamp));
1487 >                } finally {
1488 >                    unlocker.accept(stamp);
1489 >                }
1490 >            }
1491 >        };
1492 >        for (int i = nTasks; i--> 0; ) {
1493 >            Action task = chooseRandomly(writer, reader);
1494 >            futures.add(CompletableFuture.runAsync(checkedRunnable(task)));
1495 >        }
1496 >        Thread.sleep(testDurationMillis);
1497 >        done.setRelease(true);
1498 >        for (var future : futures)
1499 >            checkTimedGet(future, null);
1500 >    }
1501 >
1502   }

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines