mirror of
https://github.com/InsanusMokrassar/MicroUtils.git
synced 2025-12-18 04:05:47 +00:00
Compare commits
12 Commits
v0.26.4
...
cb56bf9793
| Author | SHA1 | Date | |
|---|---|---|---|
| cb56bf9793 | |||
| b152986b4e | |||
| fce47897d5 | |||
| 0b701a3e99 | |||
| 9dad353957 | |||
| 89e16b7bdb | |||
| c2965da341 | |||
| ffb072dc5f | |||
| a247dbcb02 | |||
| 1dd71175f4 | |||
| bbe62c0e7b | |||
| 9822ff321b |
17
CHANGELOG.md
17
CHANGELOG.md
@@ -1,5 +1,22 @@
|
|||||||
# Changelog
|
# Changelog
|
||||||
|
|
||||||
|
## 0.26.7
|
||||||
|
|
||||||
|
## 0.26.6
|
||||||
|
|
||||||
|
* `Versions`:
|
||||||
|
* `Ktor`: `3.3.0` -> `3.3.1`
|
||||||
|
* `Okio`: `3.16.0` -> `3.16.2`
|
||||||
|
|
||||||
|
## 0.26.5
|
||||||
|
|
||||||
|
* `Versions`:
|
||||||
|
* `Kotlin`: `2.2.10` -> `2.2.20`
|
||||||
|
* `KSLog`: `1.5.0` -> `1.5.1`
|
||||||
|
* `Ktor`: `3.2.3` -> `3.3.0`
|
||||||
|
* `KotlinX Browser`: `0.3` -> `0.5.0`
|
||||||
|
* `Koin`: `4.1.0` -> `4.1.1`
|
||||||
|
|
||||||
## 0.26.4
|
## 0.26.4
|
||||||
|
|
||||||
* `Common`:
|
* `Common`:
|
||||||
|
|||||||
@@ -72,7 +72,7 @@ sealed interface SmartSemaphore {
|
|||||||
acquiredPermits != checkedPermits
|
acquiredPermits != checkedPermits
|
||||||
}
|
}
|
||||||
if (shouldContinue) {
|
if (shouldContinue) {
|
||||||
waitRelease()
|
waitRelease(checkedPermits - acquiredPermits)
|
||||||
}
|
}
|
||||||
} while (shouldContinue && currentCoroutineContext().isActive)
|
} while (shouldContinue && currentCoroutineContext().isActive)
|
||||||
} catch (e: Throwable) {
|
} catch (e: Throwable) {
|
||||||
|
|||||||
@@ -4,6 +4,7 @@ import kotlinx.coroutines.flow.first
|
|||||||
import kotlinx.coroutines.sync.Mutex
|
import kotlinx.coroutines.sync.Mutex
|
||||||
import kotlinx.coroutines.sync.withLock
|
import kotlinx.coroutines.sync.withLock
|
||||||
import kotlinx.coroutines.test.runTest
|
import kotlinx.coroutines.test.runTest
|
||||||
|
import kotlin.test.BeforeTest
|
||||||
import kotlin.test.Test
|
import kotlin.test.Test
|
||||||
import kotlin.test.assertEquals
|
import kotlin.test.assertEquals
|
||||||
import kotlin.test.assertFails
|
import kotlin.test.assertFails
|
||||||
@@ -13,184 +14,517 @@ import kotlin.time.Duration.Companion.milliseconds
|
|||||||
import kotlin.time.Duration.Companion.seconds
|
import kotlin.time.Duration.Companion.seconds
|
||||||
|
|
||||||
class SmartKeyRWLockerTests {
|
class SmartKeyRWLockerTests {
|
||||||
|
private lateinit var locker: SmartKeyRWLocker<String>
|
||||||
|
|
||||||
|
@BeforeTest
|
||||||
|
fun setup() {
|
||||||
|
locker = SmartKeyRWLocker()
|
||||||
|
}
|
||||||
|
|
||||||
|
// ==================== Global Read Tests ====================
|
||||||
|
|
||||||
@Test
|
@Test
|
||||||
fun writeLockKeyFailedOnGlobalWriteLockTest() = runTest {
|
fun testGlobalReadAllowsMultipleConcurrentReads() = runTest {
|
||||||
val locker = SmartKeyRWLocker<String>()
|
val results = mutableListOf<Boolean>()
|
||||||
val testKey = "test"
|
|
||||||
|
locker.acquireRead()
|
||||||
|
|
||||||
|
val jobs = List(5) {
|
||||||
|
launch {
|
||||||
|
locker.acquireRead()
|
||||||
|
delay(100.milliseconds)
|
||||||
|
results.add(true)
|
||||||
|
locker.releaseRead()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
jobs.joinAll()
|
||||||
|
locker.releaseRead()
|
||||||
|
|
||||||
|
assertEquals(5, results.size)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun testGlobalReadBlocksGlobalWrite() = runTest {
|
||||||
|
locker.acquireRead()
|
||||||
|
|
||||||
|
var writeAcquired = false
|
||||||
|
val writeJob = launch {
|
||||||
|
locker.lockWrite()
|
||||||
|
writeAcquired = true
|
||||||
|
locker.unlockWrite()
|
||||||
|
}
|
||||||
|
|
||||||
|
delay(200.milliseconds)
|
||||||
|
assertFalse(writeAcquired, "Write should be blocked by global read")
|
||||||
|
|
||||||
|
locker.releaseRead()
|
||||||
|
writeJob.join()
|
||||||
|
|
||||||
|
assertTrue(writeAcquired, "Write should succeed after read released")
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun testGlobalReadBlocksAllKeyWrites() = runTest {
|
||||||
|
locker.acquireRead()
|
||||||
|
|
||||||
|
val writeFlags = mutableMapOf<String, Boolean>()
|
||||||
|
val keys = listOf("key1", "key2", "key3")
|
||||||
|
|
||||||
|
val jobs = keys.map { key ->
|
||||||
|
launch {
|
||||||
|
locker.lockWrite(key)
|
||||||
|
writeFlags[key] = true
|
||||||
|
locker.unlockWrite(key)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
delay(200.milliseconds)
|
||||||
|
assertTrue(writeFlags.isEmpty(), "No writes should succeed while global read active")
|
||||||
|
|
||||||
|
locker.releaseRead()
|
||||||
|
jobs.joinAll()
|
||||||
|
|
||||||
|
assertEquals(keys.size, writeFlags.size, "All writes should succeed after global read released")
|
||||||
|
}
|
||||||
|
|
||||||
|
// ==================== Global Write Tests ====================
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun testGlobalWriteBlocksAllOperations() = runTest {
|
||||||
locker.lockWrite()
|
locker.lockWrite()
|
||||||
|
|
||||||
assertTrue { locker.isWriteLocked() }
|
var globalReadAcquired = false
|
||||||
|
var keyReadAcquired = false
|
||||||
|
var keyWriteAcquired = false
|
||||||
|
|
||||||
assertFails {
|
val jobs = listOf(
|
||||||
realWithTimeout(1.seconds) {
|
launch {
|
||||||
locker.lockWrite(testKey)
|
locker.acquireRead()
|
||||||
|
globalReadAcquired = true
|
||||||
|
locker.releaseRead()
|
||||||
|
},
|
||||||
|
launch {
|
||||||
|
locker.acquireRead("key1")
|
||||||
|
keyReadAcquired = true
|
||||||
|
locker.releaseRead("key1")
|
||||||
|
},
|
||||||
|
launch {
|
||||||
|
locker.lockWrite("key2")
|
||||||
|
keyWriteAcquired = true
|
||||||
|
locker.unlockWrite("key2")
|
||||||
}
|
}
|
||||||
}
|
)
|
||||||
assertFalse { locker.isWriteLocked(testKey) }
|
|
||||||
|
delay(200.milliseconds)
|
||||||
|
assertFalse(globalReadAcquired, "Global read should be blocked")
|
||||||
|
assertFalse(keyReadAcquired, "Key read should be blocked")
|
||||||
|
assertFalse(keyWriteAcquired, "Key write should be blocked")
|
||||||
|
|
||||||
locker.unlockWrite()
|
locker.unlockWrite()
|
||||||
assertFalse { locker.isWriteLocked() }
|
jobs.joinAll()
|
||||||
|
|
||||||
realWithTimeout(1.seconds) {
|
assertTrue(globalReadAcquired)
|
||||||
locker.lockWrite(testKey)
|
assertTrue(keyReadAcquired)
|
||||||
}
|
assertTrue(keyWriteAcquired)
|
||||||
assertTrue { locker.isWriteLocked(testKey) }
|
|
||||||
assertTrue { locker.unlockWrite(testKey) }
|
|
||||||
assertFalse { locker.isWriteLocked(testKey) }
|
|
||||||
}
|
}
|
||||||
|
|
||||||
@Test
|
@Test
|
||||||
fun writeLockKeyFailedOnGlobalReadLockTest() = runTest {
|
fun testGlobalWriteIsExclusive() = runTest {
|
||||||
val locker = SmartKeyRWLocker<String>()
|
|
||||||
val testKey = "test"
|
|
||||||
locker.acquireRead()
|
|
||||||
|
|
||||||
assertEquals(Int.MAX_VALUE - 1, locker.readSemaphore().freePermits)
|
|
||||||
|
|
||||||
assertFails {
|
|
||||||
realWithTimeout(1.seconds) {
|
|
||||||
locker.lockWrite(testKey)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
assertFalse { locker.isWriteLocked(testKey) }
|
|
||||||
|
|
||||||
locker.releaseRead()
|
|
||||||
assertEquals(Int.MAX_VALUE, locker.readSemaphore().freePermits)
|
|
||||||
|
|
||||||
realWithTimeout(1.seconds) {
|
|
||||||
locker.lockWrite(testKey)
|
|
||||||
}
|
|
||||||
assertTrue { locker.isWriteLocked(testKey) }
|
|
||||||
assertTrue { locker.unlockWrite(testKey) }
|
|
||||||
assertFalse { locker.isWriteLocked(testKey) }
|
|
||||||
}
|
|
||||||
@Test
|
|
||||||
fun readLockFailedOnWriteLockKeyTest() = runTest {
|
|
||||||
val locker = SmartKeyRWLocker<String>()
|
|
||||||
val testKey = "test"
|
|
||||||
locker.lockWrite(testKey)
|
|
||||||
|
|
||||||
assertTrue { locker.isWriteLocked(testKey) }
|
|
||||||
|
|
||||||
assertFails {
|
|
||||||
realWithTimeout(1.seconds) {
|
|
||||||
locker.acquireRead()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
assertEquals(locker.readSemaphore().maxPermits - 1, locker.readSemaphore().freePermits)
|
|
||||||
|
|
||||||
locker.unlockWrite(testKey)
|
|
||||||
assertFalse { locker.isWriteLocked(testKey) }
|
|
||||||
|
|
||||||
realWithTimeout(1.seconds) {
|
|
||||||
locker.acquireRead()
|
|
||||||
}
|
|
||||||
assertEquals(locker.readSemaphore().maxPermits - 1, locker.readSemaphore().freePermits)
|
|
||||||
assertTrue { locker.releaseRead() }
|
|
||||||
assertEquals(locker.readSemaphore().maxPermits, locker.readSemaphore().freePermits)
|
|
||||||
}
|
|
||||||
@Test
|
|
||||||
fun writeLockFailedOnWriteLockKeyTest() = runTest {
|
|
||||||
val locker = SmartKeyRWLocker<String>()
|
|
||||||
val testKey = "test"
|
|
||||||
locker.lockWrite(testKey)
|
|
||||||
|
|
||||||
assertTrue { locker.isWriteLocked(testKey) }
|
|
||||||
|
|
||||||
assertFails {
|
|
||||||
realWithTimeout(1.seconds) {
|
|
||||||
locker.lockWrite()
|
locker.lockWrite()
|
||||||
}
|
|
||||||
}
|
|
||||||
assertFalse(locker.isWriteLocked())
|
|
||||||
|
|
||||||
locker.unlockWrite(testKey)
|
var secondWriteAcquired = false
|
||||||
assertFalse { locker.isWriteLocked(testKey) }
|
val job = launch {
|
||||||
|
|
||||||
realWithTimeout(1.seconds) {
|
|
||||||
locker.lockWrite()
|
locker.lockWrite()
|
||||||
|
secondWriteAcquired = true
|
||||||
|
locker.unlockWrite()
|
||||||
}
|
}
|
||||||
assertTrue(locker.isWriteLocked())
|
|
||||||
assertTrue { locker.unlockWrite() }
|
delay(200.milliseconds)
|
||||||
assertFalse(locker.isWriteLocked())
|
assertFalse(secondWriteAcquired, "Second global write should be blocked")
|
||||||
|
|
||||||
|
locker.unlockWrite()
|
||||||
|
job.join()
|
||||||
|
|
||||||
|
assertTrue(secondWriteAcquired)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// ==================== Key Read Tests ====================
|
||||||
|
|
||||||
@Test
|
@Test
|
||||||
fun readsBlockingGlobalWrite() = runTest {
|
fun testKeyReadAllowsMultipleConcurrentReadsForSameKey() = runTest {
|
||||||
val locker = SmartKeyRWLocker<String>()
|
val key = "testKey"
|
||||||
|
val results = mutableListOf<Boolean>()
|
||||||
|
|
||||||
val testKeys = (0 until 100).map { "test$it" }
|
locker.acquireRead(key)
|
||||||
|
|
||||||
for (i in testKeys.indices) {
|
val jobs = List(5) {
|
||||||
val it = testKeys[i]
|
launch {
|
||||||
locker.acquireRead(it)
|
locker.acquireRead(key)
|
||||||
val previous = testKeys.take(i)
|
delay(50.milliseconds)
|
||||||
val next = testKeys.drop(i + 1)
|
results.add(true)
|
||||||
|
locker.releaseRead(key)
|
||||||
previous.forEach {
|
|
||||||
assertTrue { locker.readSemaphoreOrNull(it) ?.freePermits == Int.MAX_VALUE - 1 }
|
|
||||||
}
|
|
||||||
next.forEach {
|
|
||||||
assertTrue { locker.readSemaphoreOrNull(it) ?.freePermits == null }
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
for (i in testKeys.indices) {
|
jobs.joinAll()
|
||||||
val it = testKeys[i]
|
locker.releaseRead(key)
|
||||||
assertFails {
|
|
||||||
realWithTimeout(13.milliseconds) { locker.lockWrite() }
|
assertEquals(5, results.size)
|
||||||
}
|
}
|
||||||
val readPermitsBeforeLock = locker.readSemaphore().freePermits
|
|
||||||
realWithTimeout(1.seconds) { locker.acquireRead() }
|
@Test
|
||||||
|
fun testKeyReadAllowsReadsForDifferentKeys() = runTest {
|
||||||
|
val results = mutableMapOf<String, Boolean>()
|
||||||
|
|
||||||
|
locker.acquireRead("key1")
|
||||||
|
|
||||||
|
val jobs = listOf("key2", "key3", "key4").map { key ->
|
||||||
|
launch {
|
||||||
|
locker.acquireRead(key)
|
||||||
|
delay(50.milliseconds)
|
||||||
|
results[key] = true
|
||||||
|
locker.releaseRead(key)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
jobs.joinAll()
|
||||||
|
locker.releaseRead("key1")
|
||||||
|
|
||||||
|
assertEquals(3, results.size)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun testKeyReadBlocksWriteForSameKey() = runTest {
|
||||||
|
val key = "testKey"
|
||||||
|
locker.acquireRead(key)
|
||||||
|
|
||||||
|
var writeAcquired = false
|
||||||
|
val job = launch {
|
||||||
|
locker.lockWrite(key)
|
||||||
|
writeAcquired = true
|
||||||
|
locker.unlockWrite(key)
|
||||||
|
}
|
||||||
|
|
||||||
|
delay(200.milliseconds)
|
||||||
|
assertFalse(writeAcquired, "Write for same key should be blocked")
|
||||||
|
|
||||||
|
locker.releaseRead(key)
|
||||||
|
job.join()
|
||||||
|
|
||||||
|
assertTrue(writeAcquired)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun testKeyReadBlocksGlobalWrite() = runTest {
|
||||||
|
locker.acquireRead("key1")
|
||||||
|
|
||||||
|
var globalWriteAcquired = false
|
||||||
|
val job = launch {
|
||||||
|
locker.lockWrite()
|
||||||
|
globalWriteAcquired = true
|
||||||
|
locker.unlockWrite()
|
||||||
|
}
|
||||||
|
|
||||||
|
delay(200.milliseconds)
|
||||||
|
assertFalse(globalWriteAcquired, "Global write should be blocked by key read")
|
||||||
|
|
||||||
|
locker.releaseRead("key1")
|
||||||
|
job.join()
|
||||||
|
|
||||||
|
assertTrue(globalWriteAcquired)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun testKeyReadAllowsWriteForDifferentKey() = runTest {
|
||||||
|
locker.acquireRead("key1")
|
||||||
|
|
||||||
|
var writeAcquired = false
|
||||||
|
val job = launch {
|
||||||
|
locker.lockWrite("key2")
|
||||||
|
writeAcquired = true
|
||||||
|
locker.unlockWrite("key2")
|
||||||
|
}
|
||||||
|
|
||||||
|
job.join()
|
||||||
|
assertTrue(writeAcquired, "Write for different key should succeed")
|
||||||
|
|
||||||
|
locker.releaseRead("key1")
|
||||||
|
}
|
||||||
|
|
||||||
|
// ==================== Key Write Tests ====================
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun testKeyWriteBlocksReadForSameKey() = runTest {
|
||||||
|
val key = "testKey"
|
||||||
|
locker.lockWrite(key)
|
||||||
|
|
||||||
|
var readAcquired = false
|
||||||
|
val job = launch {
|
||||||
|
locker.acquireRead(key)
|
||||||
|
readAcquired = true
|
||||||
|
locker.releaseRead(key)
|
||||||
|
}
|
||||||
|
|
||||||
|
delay(200.milliseconds)
|
||||||
|
assertFalse(readAcquired, "Read for same key should be blocked")
|
||||||
|
|
||||||
|
locker.unlockWrite(key)
|
||||||
|
job.join()
|
||||||
|
|
||||||
|
assertTrue(readAcquired)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun testKeyWriteBlocksGlobalRead() = runTest {
|
||||||
|
locker.lockWrite("key1")
|
||||||
|
|
||||||
|
var globalReadAcquired = false
|
||||||
|
val job = launch {
|
||||||
|
locker.acquireRead()
|
||||||
|
globalReadAcquired = true
|
||||||
locker.releaseRead()
|
locker.releaseRead()
|
||||||
assertEquals(readPermitsBeforeLock, locker.readSemaphore().freePermits)
|
|
||||||
|
|
||||||
locker.releaseRead(it)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
assertTrue { locker.readSemaphore().freePermits == Int.MAX_VALUE }
|
delay(200.milliseconds)
|
||||||
realWithTimeout(1.seconds) { locker.lockWrite() }
|
assertFalse(globalReadAcquired, "Global read should be blocked by key write")
|
||||||
assertFails {
|
|
||||||
realWithTimeout(13.milliseconds) { locker.acquireRead() }
|
locker.unlockWrite("key1")
|
||||||
}
|
job.join()
|
||||||
assertTrue { locker.unlockWrite() }
|
|
||||||
assertTrue { locker.readSemaphore().freePermits == Int.MAX_VALUE }
|
assertTrue(globalReadAcquired)
|
||||||
}
|
}
|
||||||
|
|
||||||
@Test
|
@Test
|
||||||
fun writesBlockingGlobalWrite() = runTest {
|
fun testKeyWriteIsExclusiveForSameKey() = runTest {
|
||||||
val locker = SmartKeyRWLocker<String>()
|
val key = "testKey"
|
||||||
|
locker.lockWrite(key)
|
||||||
|
|
||||||
val testKeys = (0 until 100).map { "test$it" }
|
var secondWriteAcquired = false
|
||||||
|
val job = launch {
|
||||||
for (i in testKeys.indices) {
|
locker.lockWrite(key)
|
||||||
val it = testKeys[i]
|
secondWriteAcquired = true
|
||||||
locker.lockWrite(it)
|
locker.unlockWrite(key)
|
||||||
val previous = testKeys.take(i)
|
|
||||||
val next = testKeys.drop(i + 1)
|
|
||||||
|
|
||||||
previous.forEach {
|
|
||||||
assertTrue { locker.writeMutexOrNull(it) ?.isLocked == true }
|
|
||||||
}
|
}
|
||||||
next.forEach {
|
|
||||||
assertTrue { locker.writeMutexOrNull(it) ?.isLocked != true }
|
delay(200.milliseconds)
|
||||||
|
assertFalse(secondWriteAcquired, "Second write for same key should be blocked")
|
||||||
|
|
||||||
|
locker.unlockWrite(key)
|
||||||
|
job.join()
|
||||||
|
|
||||||
|
assertTrue(secondWriteAcquired)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun testKeyWriteAllowsOperationsOnDifferentKeys() = runTest {
|
||||||
|
locker.lockWrite("key1")
|
||||||
|
|
||||||
|
val results = mutableMapOf<String, Boolean>()
|
||||||
|
|
||||||
|
val jobs = listOf(
|
||||||
|
launch {
|
||||||
|
locker.acquireRead("key2")
|
||||||
|
results["read-key2"] = true
|
||||||
|
locker.releaseRead("key2")
|
||||||
|
},
|
||||||
|
launch {
|
||||||
|
locker.lockWrite("key3")
|
||||||
|
results["write-key3"] = true
|
||||||
|
locker.unlockWrite("key3")
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
jobs.joinAll()
|
||||||
|
assertEquals(2, results.size, "Operations on different keys should succeed")
|
||||||
|
|
||||||
|
locker.unlockWrite("key1")
|
||||||
|
}
|
||||||
|
|
||||||
|
// ==================== Complex Scenarios ====================
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun testMultipleReadersThenWriter() = runTest {
|
||||||
|
val key = "testKey"
|
||||||
|
val readCount = 5
|
||||||
|
val readers = mutableListOf<Job>()
|
||||||
|
|
||||||
|
repeat(readCount) {
|
||||||
|
readers.add(launch {
|
||||||
|
locker.acquireRead(key)
|
||||||
|
delay(100.milliseconds)
|
||||||
|
locker.releaseRead(key)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
delay(50.milliseconds) // Let readers acquire
|
||||||
|
|
||||||
|
var writerExecuted = false
|
||||||
|
val writer = launch {
|
||||||
|
locker.lockWrite(key)
|
||||||
|
writerExecuted = true
|
||||||
|
locker.unlockWrite(key)
|
||||||
|
}
|
||||||
|
|
||||||
|
delay(50.milliseconds)
|
||||||
|
assertFalse(writerExecuted, "Writer should wait for all readers")
|
||||||
|
|
||||||
|
readers.joinAll()
|
||||||
|
writer.join()
|
||||||
|
|
||||||
|
assertTrue(writerExecuted, "Writer should execute after all readers done")
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun testWriterThenMultipleReaders() = runTest {
|
||||||
|
val key = "testKey"
|
||||||
|
|
||||||
|
locker.lockWrite(key)
|
||||||
|
|
||||||
|
val readerFlags = mutableListOf<Boolean>()
|
||||||
|
val readers = List(5) {
|
||||||
|
launch {
|
||||||
|
locker.acquireRead(key)
|
||||||
|
readerFlags.add(true)
|
||||||
|
locker.releaseRead(key)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
for (i in testKeys.indices) {
|
delay(200.milliseconds)
|
||||||
val it = testKeys[i]
|
assertTrue(readerFlags.isEmpty(), "Readers should be blocked by writer")
|
||||||
assertFails { realWithTimeout(13.milliseconds) { locker.lockWrite() } }
|
|
||||||
|
|
||||||
val readPermitsBeforeLock = locker.readSemaphore().freePermits
|
locker.unlockWrite(key)
|
||||||
assertFails { realWithTimeout(13.milliseconds) { locker.acquireRead() } }
|
readers.joinAll()
|
||||||
assertEquals(readPermitsBeforeLock, locker.readSemaphore().freePermits)
|
|
||||||
|
|
||||||
locker.unlockWrite(it)
|
assertEquals(5, readerFlags.size, "All readers should succeed after writer")
|
||||||
}
|
}
|
||||||
|
|
||||||
assertTrue { locker.readSemaphore().freePermits == Int.MAX_VALUE }
|
@Test
|
||||||
realWithTimeout(1.seconds) { locker.lockWrite() }
|
fun testCascadingLocksWithDifferentKeys() = runTest {
|
||||||
assertFails {
|
val executed = mutableMapOf<String, Boolean>()
|
||||||
realWithTimeout(13.milliseconds) { locker.acquireRead() }
|
|
||||||
|
launch {
|
||||||
|
locker.lockWrite("key1")
|
||||||
|
executed["write-key1-start"] = true
|
||||||
|
delay(100.milliseconds)
|
||||||
|
locker.unlockWrite("key1")
|
||||||
|
executed["write-key1-end"] = true
|
||||||
}
|
}
|
||||||
assertTrue { locker.unlockWrite() }
|
|
||||||
assertTrue { locker.readSemaphore().freePermits == Int.MAX_VALUE }
|
delay(50.milliseconds)
|
||||||
|
|
||||||
|
launch {
|
||||||
|
locker.acquireRead("key2")
|
||||||
|
executed["read-key2"] = true
|
||||||
|
delay(100.milliseconds)
|
||||||
|
locker.releaseRead("key2")
|
||||||
|
}
|
||||||
|
|
||||||
|
delay(200.milliseconds)
|
||||||
|
|
||||||
|
assertTrue(executed["write-key1-start"] == true)
|
||||||
|
assertTrue(executed["read-key2"] == true)
|
||||||
|
assertTrue(executed["write-key1-end"] == true)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun testReleaseWithoutAcquireReturnsFalse() = runTest {
|
||||||
|
assertFalse(locker.releaseRead(), "Release without acquire should return false")
|
||||||
|
assertFalse(locker.releaseRead("key1"), "Release without acquire should return false")
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun testUnlockWithoutLockReturnsFalse() = runTest {
|
||||||
|
assertFalse(locker.unlockWrite(), "Unlock without lock should return false")
|
||||||
|
assertFalse(locker.unlockWrite("key1"), "Unlock without lock should return false")
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun testProperReleaseReturnsTrue() = runTest {
|
||||||
|
locker.acquireRead()
|
||||||
|
assertTrue(locker.releaseRead(), "Release after acquire should return true")
|
||||||
|
|
||||||
|
locker.acquireRead("key1")
|
||||||
|
assertTrue(locker.releaseRead("key1"), "Release after acquire should return true")
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun testProperUnlockReturnsTrue() = runTest {
|
||||||
|
locker.lockWrite()
|
||||||
|
assertTrue(locker.unlockWrite(), "Unlock after lock should return true")
|
||||||
|
|
||||||
|
locker.lockWrite("key1")
|
||||||
|
assertTrue(locker.unlockWrite("key1"), "Unlock after lock should return true")
|
||||||
|
}
|
||||||
|
|
||||||
|
// ==================== Stress Tests ====================
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun stressTestWithMixedOperations() = runTest(timeout = 10.seconds) {
|
||||||
|
val operations = 100
|
||||||
|
val keys = listOf("key1", "key2", "key3", "key4", "key5")
|
||||||
|
val jobs = mutableListOf<Job>()
|
||||||
|
|
||||||
|
repeat(operations) { i ->
|
||||||
|
val key = keys[i % keys.size]
|
||||||
|
|
||||||
|
when (i % 4) {
|
||||||
|
0 -> jobs.add(launch {
|
||||||
|
locker.acquireRead(key)
|
||||||
|
delay(10.milliseconds)
|
||||||
|
locker.releaseRead(key)
|
||||||
|
})
|
||||||
|
1 -> jobs.add(launch {
|
||||||
|
locker.lockWrite(key)
|
||||||
|
delay(10.milliseconds)
|
||||||
|
locker.unlockWrite(key)
|
||||||
|
})
|
||||||
|
2 -> jobs.add(launch {
|
||||||
|
locker.acquireRead()
|
||||||
|
delay(10.milliseconds)
|
||||||
|
locker.releaseRead()
|
||||||
|
})
|
||||||
|
3 -> jobs.add(launch {
|
||||||
|
locker.lockWrite()
|
||||||
|
delay(10.milliseconds)
|
||||||
|
locker.unlockWrite()
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
jobs.joinAll()
|
||||||
|
// If we reach here without deadlock or exceptions, test passes
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun testFairnessReadersDontStarveWriters() = runTest(timeout = 5.seconds) {
|
||||||
|
val key = "testKey"
|
||||||
|
var writerExecuted = false
|
||||||
|
|
||||||
|
// Start continuous readers
|
||||||
|
val readers = List(10) {
|
||||||
|
launch {
|
||||||
|
repeat(5) {
|
||||||
|
locker.acquireRead(key)
|
||||||
|
delay(50.milliseconds)
|
||||||
|
locker.releaseRead(key)
|
||||||
|
delay(10.milliseconds)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
delay(100.milliseconds)
|
||||||
|
|
||||||
|
// Try to acquire write lock
|
||||||
|
val writer = launch {
|
||||||
|
locker.lockWrite(key)
|
||||||
|
writerExecuted = true
|
||||||
|
locker.unlockWrite(key)
|
||||||
|
}
|
||||||
|
|
||||||
|
readers.joinAll()
|
||||||
|
writer.join()
|
||||||
|
|
||||||
|
assertTrue(writerExecuted, "Writer should eventually execute")
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -18,5 +18,5 @@ crypto_js_version=4.1.1
|
|||||||
# Project data
|
# Project data
|
||||||
|
|
||||||
group=dev.inmo
|
group=dev.inmo
|
||||||
version=0.26.4
|
version=0.26.7
|
||||||
android_code_version=303
|
android_code_version=306
|
||||||
|
|||||||
@@ -1,31 +1,32 @@
|
|||||||
[versions]
|
[versions]
|
||||||
|
|
||||||
kt = "2.2.10"
|
kt = "2.2.20"
|
||||||
kt-serialization = "1.9.0"
|
kt-serialization = "1.9.0"
|
||||||
kt-coroutines = "1.10.2"
|
kt-coroutines = "1.10.2"
|
||||||
|
|
||||||
kotlinx-browser = "0.3"
|
kotlinx-browser = "0.5.0"
|
||||||
|
|
||||||
kslog = "1.5.0"
|
kslog = "1.5.1"
|
||||||
|
|
||||||
jb-compose = "1.8.2"
|
jb-compose = "1.8.2"
|
||||||
jb-exposed = "0.61.0"
|
jb-exposed = "0.61.0"
|
||||||
jb-dokka = "2.0.0"
|
jb-dokka = "2.1.0"
|
||||||
|
|
||||||
|
# 3.50.3.0 contains bug https://github.com/InsanusMokrassar/MicroUtils/actions/runs/18138301958/job/51629588088
|
||||||
sqlite = "3.50.1.0"
|
sqlite = "3.50.1.0"
|
||||||
|
|
||||||
korlibs = "5.4.0"
|
korlibs = "5.4.0"
|
||||||
uuid = "0.8.4"
|
uuid = "0.8.4"
|
||||||
|
|
||||||
ktor = "3.2.3"
|
ktor = "3.3.1"
|
||||||
|
|
||||||
gh-release = "2.5.2"
|
gh-release = "2.5.2"
|
||||||
|
|
||||||
koin = "4.1.0"
|
koin = "4.1.1"
|
||||||
|
|
||||||
okio = "3.16.0"
|
okio = "3.16.2"
|
||||||
|
|
||||||
ksp = "2.2.10-2.0.2"
|
ksp = "2.2.20-2.0.3"
|
||||||
kotlin-poet = "2.2.0"
|
kotlin-poet = "2.2.0"
|
||||||
|
|
||||||
versions = "0.52.0"
|
versions = "0.52.0"
|
||||||
|
|||||||
2
gradle/wrapper/gradle-wrapper.properties
vendored
2
gradle/wrapper/gradle-wrapper.properties
vendored
@@ -1,5 +1,5 @@
|
|||||||
distributionBase=GRADLE_USER_HOME
|
distributionBase=GRADLE_USER_HOME
|
||||||
distributionPath=wrapper/dists
|
distributionPath=wrapper/dists
|
||||||
distributionUrl=https\://services.gradle.org/distributions/gradle-8.13-bin.zip
|
distributionUrl=https\://services.gradle.org/distributions/gradle-8.14.3-bin.zip
|
||||||
zipStoreBase=GRADLE_USER_HOME
|
zipStoreBase=GRADLE_USER_HOME
|
||||||
zipStorePath=wrapper/dists
|
zipStorePath=wrapper/dists
|
||||||
|
|||||||
@@ -1,10 +1,81 @@
|
|||||||
package dev.inmo.micro_utils.repos.exposed
|
package dev.inmo.micro_utils.repos.exposed
|
||||||
|
|
||||||
import org.jetbrains.exposed.sql.*
|
import org.jetbrains.exposed.sql.*
|
||||||
|
import org.jetbrains.exposed.sql.SchemaUtils.addMissingColumnsStatements
|
||||||
|
import org.jetbrains.exposed.sql.SchemaUtils.checkMappingConsistence
|
||||||
|
import org.jetbrains.exposed.sql.SchemaUtils.createStatements
|
||||||
|
import org.jetbrains.exposed.sql.transactions.TransactionManager
|
||||||
import org.jetbrains.exposed.sql.transactions.transaction
|
import org.jetbrains.exposed.sql.transactions.transaction
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Code in this function mostly duplicates Exposed [SchemaUtils.createMissingTablesAndColumns]. It made due to deprecation
|
||||||
|
* status of the last one and potential lost of it in future updates.
|
||||||
|
*
|
||||||
|
* Code doing:
|
||||||
|
*
|
||||||
|
* * Creating missed tables
|
||||||
|
* * Altering missed tables (where possible)
|
||||||
|
* * Calculate problems with [checkMappingConsistence] and add them in execution too
|
||||||
|
*
|
||||||
|
* All changes made in [transaction] with [database] as its `db` argument
|
||||||
|
*/
|
||||||
|
fun initTablesInTransaction(vararg tables: Table, database: Database, inBatch: Boolean = false, withLogs: Boolean = true) {
|
||||||
|
transaction(database) {
|
||||||
|
fun <R> logTimeSpent(message: String, withLogs: Boolean, block: () -> R): R {
|
||||||
|
return if (withLogs) {
|
||||||
|
val start = System.currentTimeMillis()
|
||||||
|
val answer = block()
|
||||||
|
exposedLogger.info(message + " took " + (System.currentTimeMillis() - start) + "ms")
|
||||||
|
answer
|
||||||
|
} else {
|
||||||
|
block()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
fun Transaction.execStatements(inBatch: Boolean, statements: List<String>) {
|
||||||
|
if (inBatch) {
|
||||||
|
execInBatch(statements)
|
||||||
|
} else {
|
||||||
|
for (statement in statements) {
|
||||||
|
exec(statement)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
with(TransactionManager.current()) {
|
||||||
|
db.dialect.resetCaches()
|
||||||
|
val createStatements = logTimeSpent("Preparing create tables statements", withLogs) {
|
||||||
|
createStatements(*tables)
|
||||||
|
}
|
||||||
|
logTimeSpent("Executing create tables statements", withLogs) {
|
||||||
|
execStatements(inBatch, createStatements)
|
||||||
|
commit()
|
||||||
|
}
|
||||||
|
|
||||||
|
val alterStatements = logTimeSpent("Preparing alter table statements", withLogs) {
|
||||||
|
addMissingColumnsStatements(tables = tables, withLogs)
|
||||||
|
}
|
||||||
|
logTimeSpent("Executing alter table statements", withLogs) {
|
||||||
|
execStatements(inBatch, alterStatements)
|
||||||
|
commit()
|
||||||
|
}
|
||||||
|
val executedStatements = createStatements + alterStatements
|
||||||
|
logTimeSpent("Checking mapping consistence", withLogs) {
|
||||||
|
val modifyTablesStatements = checkMappingConsistence(
|
||||||
|
tables = tables,
|
||||||
|
withLogs
|
||||||
|
).filter { it !in executedStatements }
|
||||||
|
execStatements(inBatch, modifyTablesStatements)
|
||||||
|
commit()
|
||||||
|
}
|
||||||
|
db.dialect.resetCaches()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fun Table.initTable(database: Database, inBatch: Boolean, withLogs: Boolean) {
|
||||||
|
initTablesInTransaction(this, database = database, inBatch = inBatch, withLogs = withLogs)
|
||||||
|
}
|
||||||
fun Table.initTable(database: Database) {
|
fun Table.initTable(database: Database) {
|
||||||
transaction(database) { SchemaUtils.createMissingTablesAndColumns(this@initTable) }
|
initTable(database = database, inBatch = false, withLogs = true)
|
||||||
}
|
}
|
||||||
|
|
||||||
fun <T> T.initTable() where T: ExposedRepo, T: Table = initTable(database)
|
fun <T> T.initTable() where T: ExposedRepo, T: Table = initTable(database)
|
||||||
|
|||||||
Reference in New Issue
Block a user