mirror of
				https://github.com/InsanusMokrassar/MicroUtils.git
				synced 2025-10-29 19:20:26 +00:00 
			
		
		
		
	Compare commits
	
		
			33 Commits
		
	
	
		
	
	| Author | SHA1 | Date | |
|---|---|---|---|
| d46cc3b09c | |||
| dfd23f8d60 | |||
| 28eb1a11e6 | |||
| 04f82a03bf | |||
| aac545074b | |||
| 87a3a925ee | |||
| 5447bf9691 | |||
| 761070b9b7 | |||
| 4c9e435df8 | |||
| 4b7d65e8b4 | |||
| 0515b49b98 | |||
| edb97215ef | |||
| 5577a24548 | |||
| cfaa2a8927 | |||
| 78494b6036 | |||
| f9ea7eca61 | |||
| d69fee1732 | |||
| 178518db5e | |||
| 6fb20fb973 | |||
| 831bf44e34 | |||
| a4c6c367e3 | |||
| bc98e59709 | |||
| ef287bc331 | |||
| 3437f4c712 | |||
| 618f2dcd79 | |||
| 6df8ad3095 | |||
| eda6221288 | |||
| a9859f6a0d | |||
| 0db88bac25 | |||
| daa3d9c0dd | |||
| b343b33594 | |||
| 46e435a448 | |||
| 7fe62b4ffa | 
							
								
								
									
										29
									
								
								CHANGELOG.md
									
									
									
									
									
								
							
							
						
						
									
										29
									
								
								CHANGELOG.md
									
									
									
									
									
								
							| @@ -1,5 +1,34 @@ | |||||||
| # Changelog | # Changelog | ||||||
|  |  | ||||||
|  | ## 0.25.3 | ||||||
|  |  | ||||||
|  | * `Coroutines`: | ||||||
|  |   * Add extensions `SmartRWLocker.alsoWithUnlockingOnSuccessAsync` and `SmartRWLocker.alsoWithUnlockingOnSuccess` | ||||||
|  |   * Fix of `SmartRWLocker.lockWrite` issue when it could lock write mutex without unlocking | ||||||
|  |   * Add tool `SmartKeyRWLocker` | ||||||
|  |   * `SmartSemaphore` got new property `maxPermits` | ||||||
|  |   * New extension `SmartSemaphore.waitReleaseAll()` | ||||||
|  | * `Transactions`: | ||||||
|  |   * Add `TransactionsDSL` | ||||||
|  |  | ||||||
|  | ## 0.25.2 | ||||||
|  |  | ||||||
|  | * `Versions`: | ||||||
|  |   * `Exposed`: `0.59.0` -> `0.60.0` | ||||||
|  | * `Repo`: | ||||||
|  |   * `Cache`: | ||||||
|  |     * Add extensions `alsoInvalidate`, `alsoInvalidateAsync`, `alsoInvalidateSync` and `alsoInvalidateSyncLogging` | ||||||
|  | * `Koin`: | ||||||
|  |   * Add extensions `singleSuspend` and `factorySuspend` for defining of dependencies with suspendable blocks | ||||||
|  |  | ||||||
|  | ## 0.25.1 | ||||||
|  |  | ||||||
|  | * `Coroutines`: | ||||||
|  |   * Add `SortedMapLikeBinaryTreeNode` | ||||||
|  | * `Pagination`: | ||||||
|  |   * `Compose`: | ||||||
|  |     * One more rework of `InfinityPagedComponent` and `PagedComponent` | ||||||
|  |  | ||||||
| ## 0.25.0 | ## 0.25.0 | ||||||
|  |  | ||||||
| * `Repos`: | * `Repos`: | ||||||
|   | |||||||
| @@ -13,7 +13,7 @@ import kotlin.math.floor | |||||||
|  * |  * | ||||||
|  * Anyway it is recommended to use |  * Anyway it is recommended to use | ||||||
|  * |  * | ||||||
|  * @param hexaUInt rgba [UInt] in format `0xFFEEBBAA` where FF - red, EE - green, BB - blue` and AA - alpha |  * @param hexaUInt rgba [UInt] in format `0xRRGGBBAA` where RR - red, GG - green, BB - blue` and AA - alpha | ||||||
|  */ |  */ | ||||||
| @Serializable | @Serializable | ||||||
| @JvmInline | @JvmInline | ||||||
| @@ -21,18 +21,18 @@ value class HEXAColor ( | |||||||
|     val hexaUInt: UInt |     val hexaUInt: UInt | ||||||
| ) : Comparable<HEXAColor> { | ) : Comparable<HEXAColor> { | ||||||
|     /** |     /** | ||||||
|      * @returns [hexaUInt] as a string with format `#FFEEBBAA` where FF - red, EE - green, BB - blue and AA - alpha |      * @returns [hexaUInt] as a string with format `#RRGGBBAA` where RR - red, GG - green, BB - blue and AA - alpha | ||||||
|      */ |      */ | ||||||
|     val hexa: String |     val hexa: String | ||||||
|         get() = "#${hexaUInt.toString(16).padStart(8, '0')}" |         get() = "#${hexaUInt.toString(16).padStart(8, '0')}" | ||||||
|  |  | ||||||
|     /** |     /** | ||||||
|      * @returns [hexaUInt] as a string with format `#FFEEBB` where FF - red, EE - green and BB - blue |      * @returns [hexaUInt] as a string with format `#RRGGBB` where RR - red, GG - green and BB - blue | ||||||
|      */ |      */ | ||||||
|     val hex: String |     val hex: String | ||||||
|         get() = hexa.take(7) |         get() = hexa.take(7) | ||||||
|     /** |     /** | ||||||
|      * @returns [hexaUInt] as a string with format `#AAFFEEBB` where AA - alpha, FF - red, EE - green and BB - blue |      * @returns [hexaUInt] as a string with format `#AARRGGBB` where AA - alpha, RR - red, GG - green and BB - blue | ||||||
|      */ |      */ | ||||||
|     val ahex: String |     val ahex: String | ||||||
|         get() = "#${a.toString(16).padStart(2, '2')}${hex.drop(1)}" |         get() = "#${a.toString(16).padStart(2, '2')}${hex.drop(1)}" | ||||||
| @@ -140,12 +140,12 @@ value class HEXAColor ( | |||||||
|         }.lowercase().toUInt(16).let(::HEXAColor) |         }.lowercase().toUInt(16).let(::HEXAColor) | ||||||
|  |  | ||||||
|         /** |         /** | ||||||
|          * Creates [HEXAColor] from [uint] presume it is in format `0xFFEEBBAA` where FF - red, EE - green, BB - blue` and AA - alpha |          * Creates [HEXAColor] from [uint] presume it is in format `0xRRGGBBAA` where RR - red, GG - green, BB - blue` and AA - alpha | ||||||
|          */ |          */ | ||||||
|         fun fromHexa(uint: UInt) = HEXAColor(uint) |         fun fromHexa(uint: UInt) = HEXAColor(uint) | ||||||
|  |  | ||||||
|         /** |         /** | ||||||
|          * Creates [HEXAColor] from [uint] presume it is in format `0xAAFFEEBB` where AA - alpha, FF - red, EE - green and BB - blue` |          * Creates [HEXAColor] from [uint] presume it is in format `0xAARRGGBB` where AA - alpha, RR - red, GG - green and BB - blue` | ||||||
|          */ |          */ | ||||||
|         fun fromAhex(uint: UInt) = HEXAColor( |         fun fromAhex(uint: UInt) = HEXAColor( | ||||||
|             a = ((uint and 0xff000000u) / 0x1000000u).toInt(), |             a = ((uint and 0xff000000u) / 0x1000000u).toInt(), | ||||||
|   | |||||||
| @@ -71,7 +71,7 @@ fun <T, M> Flow<T>.subscribeAsync( | |||||||
|         it.invoke(markersMap) |         it.invoke(markersMap) | ||||||
|     } |     } | ||||||
|  |  | ||||||
|     val job = subscribeSafelyWithoutExceptions(subscope) { data -> |     val job = subscribeLoggingDropExceptions(subscope) { data -> | ||||||
|         val dataCommand = AsyncSubscriptionCommandData(data, subscope, markerFactory, block) { marker -> |         val dataCommand = AsyncSubscriptionCommandData(data, subscope, markerFactory, block) { marker -> | ||||||
|             actor.send( |             actor.send( | ||||||
|                 AsyncSubscriptionCommandClearReceiver(marker) |                 AsyncSubscriptionCommandClearReceiver(marker) | ||||||
|   | |||||||
| @@ -0,0 +1,224 @@ | |||||||
|  | package dev.inmo.micro_utils.coroutines | ||||||
|  |  | ||||||
|  | import kotlinx.coroutines.CancellationException | ||||||
|  | import kotlinx.coroutines.sync.Mutex | ||||||
|  | import kotlinx.coroutines.sync.withLock | ||||||
|  | import kotlin.contracts.ExperimentalContracts | ||||||
|  | import kotlin.contracts.InvocationKind | ||||||
|  | import kotlin.contracts.contract | ||||||
|  |  | ||||||
|  | /** | ||||||
|  |  * Combining [globalRWLocker] and internal map of [SmartRWLocker] associated by [T] to provide next logic: | ||||||
|  |  * | ||||||
|  |  * * Locker by key, for reading: waiting for [globalRWLocker] unlock write by acquiring read permit in it and then | ||||||
|  |  * taking or creating locker for key [T] and lock its reading too | ||||||
|  |  * * Locker by key, for writing: waiting for [globalRWLocker] unlock write by acquiring read permit in it and then | ||||||
|  |  * taking or creating locker for key [T] and lock its writing | ||||||
|  |  * * [globalRWLocker], for reading: using [SmartRWLocker.acquireRead], will be suspended until its | ||||||
|  |  * [SmartRWLocker.lockWrite] will not be unlocked | ||||||
|  |  * * [globalRWLocker], for writing: using [SmartRWLocker.lockWrite], will be paused by other reading and writing | ||||||
|  |  * operations and will pause other operations until the end of operation (calling of [unlockWrite]) | ||||||
|  |  * | ||||||
|  |  * You may see, that lockers by key still will use global locker permits - it is required to prevent [globalRWLocker] | ||||||
|  |  * write locking during all other operations are not done. In fact, all the keys works like a simple [SmartRWLocker] as | ||||||
|  |  * well, as [globalRWLocker], but they are linked with [globalRWLocker] [SmartRWLocker.acquireRead] permissions | ||||||
|  |  */ | ||||||
|  | class SmartKeyRWLocker<T>( | ||||||
|  |     globalLockerReadPermits: Int = Int.MAX_VALUE, | ||||||
|  |     globalLockerWriteIsLocked: Boolean = false, | ||||||
|  |     private val perKeyReadPermits: Int = Int.MAX_VALUE | ||||||
|  | ) { | ||||||
|  |     private val globalRWLocker: SmartRWLocker = SmartRWLocker( | ||||||
|  |         readPermits = globalLockerReadPermits, | ||||||
|  |         writeIsLocked = globalLockerWriteIsLocked | ||||||
|  |     ) | ||||||
|  |     private val lockers = mutableMapOf<T, SmartRWLocker>() | ||||||
|  |     private val lockersMutex = Mutex() | ||||||
|  |     private val lockersWritingLocker = SmartSemaphore.Mutable(Int.MAX_VALUE) | ||||||
|  |     private val globalWritingLocker = SmartSemaphore.Mutable(Int.MAX_VALUE) | ||||||
|  |  | ||||||
|  |     private fun allocateLockerWithoutLock(key: T) = lockers.getOrPut(key) { | ||||||
|  |         SmartRWLocker(perKeyReadPermits) | ||||||
|  |     } | ||||||
|  |     private suspend fun allocateLocker(key: T) = lockersMutex.withLock { | ||||||
|  |         lockers.getOrPut(key) { | ||||||
|  |             SmartRWLocker(perKeyReadPermits) | ||||||
|  |         } | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     suspend fun writeMutex(key: T): SmartMutex.Immutable = globalRWLocker.withReadAcquire { | ||||||
|  |         allocateLockerWithoutLock(key).writeMutex | ||||||
|  |     } | ||||||
|  |     suspend fun readSemaphore(key: T): SmartSemaphore.Immutable = globalRWLocker.withReadAcquire { | ||||||
|  |         allocateLockerWithoutLock(key).readSemaphore | ||||||
|  |     } | ||||||
|  |     fun writeMutexOrNull(key: T): SmartMutex.Immutable? = lockers[key] ?.writeMutex | ||||||
|  |     fun readSemaphoreOrNull(key: T): SmartSemaphore.Immutable? = lockers[key] ?.readSemaphore | ||||||
|  |  | ||||||
|  |     fun writeMutex(): SmartMutex.Immutable = globalRWLocker.writeMutex | ||||||
|  |     fun readSemaphore(): SmartSemaphore.Immutable = globalRWLocker.readSemaphore | ||||||
|  |  | ||||||
|  |     suspend fun acquireRead() { | ||||||
|  |         globalWritingLocker.acquire() | ||||||
|  |         try { | ||||||
|  |             lockersWritingLocker.waitReleaseAll() | ||||||
|  |             globalRWLocker.acquireRead() | ||||||
|  |         } catch (e: CancellationException) { | ||||||
|  |             globalWritingLocker.release() | ||||||
|  |             throw e | ||||||
|  |         } | ||||||
|  |     } | ||||||
|  |     suspend fun releaseRead(): Boolean { | ||||||
|  |         globalWritingLocker.release() | ||||||
|  |         return globalRWLocker.releaseRead() | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     suspend fun lockWrite() { | ||||||
|  |         globalRWLocker.lockWrite() | ||||||
|  |     } | ||||||
|  |     suspend fun unlockWrite(): Boolean { | ||||||
|  |         return globalRWLocker.unlockWrite() | ||||||
|  |     } | ||||||
|  |     fun isWriteLocked(): Boolean = globalRWLocker.writeMutex.isLocked == true | ||||||
|  |  | ||||||
|  |  | ||||||
|  |     suspend fun acquireRead(key: T) { | ||||||
|  |         globalRWLocker.acquireRead() | ||||||
|  |         val locker = allocateLocker(key) | ||||||
|  |         try { | ||||||
|  |             locker.acquireRead() | ||||||
|  |         } catch (e: CancellationException) { | ||||||
|  |             globalRWLocker.releaseRead() | ||||||
|  |             throw e | ||||||
|  |         } | ||||||
|  |     } | ||||||
|  |     suspend fun releaseRead(key: T): Boolean { | ||||||
|  |         val locker = allocateLocker(key) | ||||||
|  |         return locker.releaseRead() && globalRWLocker.releaseRead() | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     suspend fun lockWrite(key: T) { | ||||||
|  |         globalWritingLocker.withAcquire(globalWritingLocker.maxPermits) { | ||||||
|  |             lockersWritingLocker.acquire() | ||||||
|  |         } | ||||||
|  |         try { | ||||||
|  |             globalRWLocker.acquireRead() | ||||||
|  |             try { | ||||||
|  |                 val locker = allocateLocker(key) | ||||||
|  |                 locker.lockWrite() | ||||||
|  |             } catch (e: CancellationException) { | ||||||
|  |                 globalRWLocker.releaseRead() | ||||||
|  |                 throw e | ||||||
|  |             } | ||||||
|  |         } catch (e: CancellationException) { | ||||||
|  |             lockersWritingLocker.release() | ||||||
|  |             throw e | ||||||
|  |         } | ||||||
|  |     } | ||||||
|  |     suspend fun unlockWrite(key: T): Boolean { | ||||||
|  |         val locker = allocateLocker(key) | ||||||
|  |         return (locker.unlockWrite() && globalRWLocker.releaseRead()).also { | ||||||
|  |             if (it) { | ||||||
|  |                 lockersWritingLocker.release() | ||||||
|  |             } | ||||||
|  |         } | ||||||
|  |     } | ||||||
|  |     fun isWriteLocked(key: T): Boolean = lockers[key] ?.writeMutex ?.isLocked == true | ||||||
|  | } | ||||||
|  |  | ||||||
|  | @OptIn(ExperimentalContracts::class) | ||||||
|  | suspend inline fun <T, R> SmartKeyRWLocker<T>.withReadAcquire(action: () -> R): R { | ||||||
|  |     contract { | ||||||
|  |         callsInPlace(action, InvocationKind.EXACTLY_ONCE) | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     acquireRead() | ||||||
|  |     try { | ||||||
|  |         return action() | ||||||
|  |     } finally { | ||||||
|  |         releaseRead() | ||||||
|  |     } | ||||||
|  | } | ||||||
|  |  | ||||||
|  | @OptIn(ExperimentalContracts::class) | ||||||
|  | suspend inline fun <T, R> SmartKeyRWLocker<T>.withWriteLock(action: () -> R): R { | ||||||
|  |     contract { | ||||||
|  |         callsInPlace(action, InvocationKind.EXACTLY_ONCE) | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     lockWrite() | ||||||
|  |     try { | ||||||
|  |         return action() | ||||||
|  |     } finally { | ||||||
|  |         unlockWrite() | ||||||
|  |     } | ||||||
|  | } | ||||||
|  |  | ||||||
|  | @OptIn(ExperimentalContracts::class) | ||||||
|  | suspend inline fun <T, R> SmartKeyRWLocker<T>.withReadAcquire(key: T, action: () -> R): R { | ||||||
|  |     contract { | ||||||
|  |         callsInPlace(action, InvocationKind.EXACTLY_ONCE) | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     acquireRead(key) | ||||||
|  |     try { | ||||||
|  |         return action() | ||||||
|  |     } finally { | ||||||
|  |         releaseRead(key) | ||||||
|  |     } | ||||||
|  | } | ||||||
|  |  | ||||||
|  | @OptIn(ExperimentalContracts::class) | ||||||
|  | suspend inline fun <T, R> SmartKeyRWLocker<T>.withReadAcquires(keys: Iterable<T>, action: () -> R): R { | ||||||
|  |     contract { | ||||||
|  |         callsInPlace(action, InvocationKind.EXACTLY_ONCE) | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     val acquired = mutableSetOf<T>() | ||||||
|  |     try { | ||||||
|  |         keys.forEach { | ||||||
|  |             acquireRead(it) | ||||||
|  |             acquired.add(it) | ||||||
|  |         } | ||||||
|  |         return action() | ||||||
|  |     } finally { | ||||||
|  |         acquired.forEach { | ||||||
|  |             releaseRead(it) | ||||||
|  |         } | ||||||
|  |     } | ||||||
|  | } | ||||||
|  | suspend inline fun <T, R> SmartKeyRWLocker<T>.withReadAcquires(vararg keys: T, action: () -> R): R = withReadAcquires(keys.asIterable(), action) | ||||||
|  |  | ||||||
|  | @OptIn(ExperimentalContracts::class) | ||||||
|  | suspend inline fun <T, R> SmartKeyRWLocker<T>.withWriteLock(key: T, action: () -> R): R { | ||||||
|  |     contract { | ||||||
|  |         callsInPlace(action, InvocationKind.EXACTLY_ONCE) | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     lockWrite(key) | ||||||
|  |     try { | ||||||
|  |         return action() | ||||||
|  |     } finally { | ||||||
|  |         unlockWrite(key) | ||||||
|  |     } | ||||||
|  | } | ||||||
|  |  | ||||||
|  | @OptIn(ExperimentalContracts::class) | ||||||
|  | suspend inline fun <T, R> SmartKeyRWLocker<T>.withWriteLocks(keys: Iterable<T>, action: () -> R): R { | ||||||
|  |     contract { | ||||||
|  |         callsInPlace(action, InvocationKind.EXACTLY_ONCE) | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     val locked = mutableSetOf<T>() | ||||||
|  |     try { | ||||||
|  |         keys.forEach { | ||||||
|  |             lockWrite(it) | ||||||
|  |             locked.add(it) | ||||||
|  |         } | ||||||
|  |         return action() | ||||||
|  |     } finally { | ||||||
|  |         locked.forEach { | ||||||
|  |             unlockWrite(it) | ||||||
|  |         } | ||||||
|  |     } | ||||||
|  | } | ||||||
| @@ -1,5 +1,6 @@ | |||||||
| package dev.inmo.micro_utils.coroutines | package dev.inmo.micro_utils.coroutines | ||||||
|  |  | ||||||
|  | import kotlinx.coroutines.CancellationException | ||||||
| import kotlin.contracts.ExperimentalContracts | import kotlin.contracts.ExperimentalContracts | ||||||
| import kotlin.contracts.InvocationKind | import kotlin.contracts.InvocationKind | ||||||
| import kotlin.contracts.contract | import kotlin.contracts.contract | ||||||
| @@ -7,9 +8,10 @@ import kotlin.contracts.contract | |||||||
| /** | /** | ||||||
|  * Composite mutex which works with next rules: |  * Composite mutex which works with next rules: | ||||||
|  * |  * | ||||||
|  * * [acquireRead] require to [writeMutex] be free. Then it will take one lock from [readSemaphore] |  * * [acquireRead] require to [writeMutex] to be free. Then it will take one lock from [readSemaphore] | ||||||
|  * * [releaseRead] will just free up one permit in [readSemaphore] |  * * [releaseRead] will just free up one permit in [readSemaphore] | ||||||
|  * * [lockWrite] will lock [writeMutex] and then await while all [readSemaphore] will be freed |  * * [lockWrite] will lock [writeMutex] and then await while all [readSemaphore] will be freed. If coroutine will be | ||||||
|  |  * cancelled during read semaphore freeing, locking will be cancelled too with [SmartMutex.Mutable.unlock]ing of [writeMutex] | ||||||
|  * * [unlockWrite] will just unlock [writeMutex] |  * * [unlockWrite] will just unlock [writeMutex] | ||||||
|  */ |  */ | ||||||
| class SmartRWLocker(private val readPermits: Int = Int.MAX_VALUE, writeIsLocked: Boolean = false) { | class SmartRWLocker(private val readPermits: Int = Int.MAX_VALUE, writeIsLocked: Boolean = false) { | ||||||
| @@ -39,7 +41,12 @@ class SmartRWLocker(private val readPermits: Int = Int.MAX_VALUE, writeIsLocked: | |||||||
|      */ |      */ | ||||||
|     suspend fun lockWrite() { |     suspend fun lockWrite() { | ||||||
|         _writeMutex.lock() |         _writeMutex.lock() | ||||||
|  |         try { | ||||||
|             _readSemaphore.acquire(readPermits) |             _readSemaphore.acquire(readPermits) | ||||||
|  |         } catch (e: CancellationException) { | ||||||
|  |             _writeMutex.unlock() | ||||||
|  |             throw e | ||||||
|  |         } | ||||||
|     } |     } | ||||||
|  |  | ||||||
|     /** |     /** | ||||||
|   | |||||||
| @@ -0,0 +1,23 @@ | |||||||
|  | package dev.inmo.micro_utils.coroutines | ||||||
|  |  | ||||||
|  | import kotlinx.coroutines.CoroutineScope | ||||||
|  | import kotlinx.coroutines.Job | ||||||
|  |  | ||||||
|  | suspend inline fun alsoWithUnlockingOnSuccess( | ||||||
|  |     vararg lockers: SmartRWLocker, | ||||||
|  |     block: suspend () -> Unit | ||||||
|  | ): Result<Unit> { | ||||||
|  |     return runCatching { | ||||||
|  |         block() | ||||||
|  |     }.onSuccess { | ||||||
|  |         lockers.forEach { it.unlockWrite() } | ||||||
|  |     } | ||||||
|  | } | ||||||
|  |  | ||||||
|  | fun alsoWithUnlockingOnSuccessAsync( | ||||||
|  |     scope: CoroutineScope, | ||||||
|  |     vararg lockers: SmartRWLocker, | ||||||
|  |     block: suspend () -> Unit | ||||||
|  | ): Job = scope.launchLoggingDropExceptions { | ||||||
|  |     alsoWithUnlockingOnSuccess(*lockers, block = block) | ||||||
|  | } | ||||||
| @@ -24,6 +24,7 @@ import kotlin.contracts.contract | |||||||
|  * [Mutable] creator |  * [Mutable] creator | ||||||
|  */ |  */ | ||||||
| sealed interface SmartSemaphore { | sealed interface SmartSemaphore { | ||||||
|  |     val maxPermits: Int | ||||||
|     val permitsStateFlow: StateFlow<Int> |     val permitsStateFlow: StateFlow<Int> | ||||||
|  |  | ||||||
|     /** |     /** | ||||||
| @@ -36,7 +37,7 @@ sealed interface SmartSemaphore { | |||||||
|     /** |     /** | ||||||
|      * Immutable variant of [SmartSemaphore]. In fact will depend on the owner of [permitsStateFlow] |      * Immutable variant of [SmartSemaphore]. In fact will depend on the owner of [permitsStateFlow] | ||||||
|      */ |      */ | ||||||
|     class Immutable(override val permitsStateFlow: StateFlow<Int>) : SmartSemaphore |     class Immutable(override val permitsStateFlow: StateFlow<Int>, override val maxPermits: Int) : SmartSemaphore | ||||||
|  |  | ||||||
|     /** |     /** | ||||||
|      * Mutable variant of [SmartSemaphore]. With that variant you may [lock] and [unlock]. Besides, you may create |      * Mutable variant of [SmartSemaphore]. With that variant you may [lock] and [unlock]. Besides, you may create | ||||||
| @@ -44,15 +45,16 @@ sealed interface SmartSemaphore { | |||||||
|      * |      * | ||||||
|      * @param locked Preset state of [freePermits] and its internal [_freePermitsStateFlow] |      * @param locked Preset state of [freePermits] and its internal [_freePermitsStateFlow] | ||||||
|      */ |      */ | ||||||
|     class Mutable(private val permits: Int, acquiredPermits: Int = 0) : SmartSemaphore { |     class Mutable(permits: Int, acquiredPermits: Int = 0) : SmartSemaphore { | ||||||
|  |         override val maxPermits: Int = permits | ||||||
|         private val _freePermitsStateFlow = SpecialMutableStateFlow<Int>(permits - acquiredPermits) |         private val _freePermitsStateFlow = SpecialMutableStateFlow<Int>(permits - acquiredPermits) | ||||||
|         override val permitsStateFlow: StateFlow<Int> = _freePermitsStateFlow.asStateFlow() |         override val permitsStateFlow: StateFlow<Int> = _freePermitsStateFlow.asStateFlow() | ||||||
|  |  | ||||||
|         private val internalChangesMutex = Mutex(false) |         private val internalChangesMutex = Mutex(false) | ||||||
|  |  | ||||||
|         fun immutable() = Immutable(permitsStateFlow) |         fun immutable() = Immutable(permitsStateFlow, maxPermits) | ||||||
|  |  | ||||||
|         private fun checkedPermits(permits: Int) = permits.coerceIn(1 .. this.permits) |         private fun checkedPermits(permits: Int) = permits.coerceIn(1 .. this.maxPermits) | ||||||
|  |  | ||||||
|         /** |         /** | ||||||
|          * Holds call until this [SmartSemaphore] will be re-locked. That means that current method will |          * Holds call until this [SmartSemaphore] will be re-locked. That means that current method will | ||||||
| @@ -126,10 +128,10 @@ sealed interface SmartSemaphore { | |||||||
|          */ |          */ | ||||||
|         suspend fun release(permits: Int = 1): Boolean { |         suspend fun release(permits: Int = 1): Boolean { | ||||||
|             val checkedPermits = checkedPermits(permits) |             val checkedPermits = checkedPermits(permits) | ||||||
|             return if (_freePermitsStateFlow.value < this.permits) { |             return if (_freePermitsStateFlow.value < this.maxPermits) { | ||||||
|                 internalChangesMutex.withLock { |                 internalChangesMutex.withLock { | ||||||
|                     if (_freePermitsStateFlow.value < this.permits) { |                     if (_freePermitsStateFlow.value < this.maxPermits) { | ||||||
|                         _freePermitsStateFlow.value = minOf(_freePermitsStateFlow.value + checkedPermits, this.permits) |                         _freePermitsStateFlow.value = minOf(_freePermitsStateFlow.value + checkedPermits, this.maxPermits) | ||||||
|                         true |                         true | ||||||
|                     } else { |                     } else { | ||||||
|                         false |                         false | ||||||
| @@ -166,3 +168,4 @@ suspend inline fun <T> SmartSemaphore.Mutable.withAcquire(permits: Int = 1, acti | |||||||
|  * the fact that some other parties may lock it again |  * the fact that some other parties may lock it again | ||||||
|  */ |  */ | ||||||
| suspend fun SmartSemaphore.waitRelease(permits: Int = 1) = permitsStateFlow.first { it >= permits } | suspend fun SmartSemaphore.waitRelease(permits: Int = 1) = permitsStateFlow.first { it >= permits } | ||||||
|  | suspend fun SmartSemaphore.waitReleaseAll() = permitsStateFlow.first { it == maxPermits } | ||||||
|   | |||||||
| @@ -1,7 +1,6 @@ | |||||||
| package dev.inmo.micro_utils.coroutines.collections | package dev.inmo.micro_utils.coroutines.collections | ||||||
|  |  | ||||||
| import dev.inmo.micro_utils.coroutines.SmartRWLocker | import dev.inmo.micro_utils.coroutines.SmartRWLocker | ||||||
| import dev.inmo.micro_utils.coroutines.waitReadRelease |  | ||||||
| import dev.inmo.micro_utils.coroutines.withReadAcquire | import dev.inmo.micro_utils.coroutines.withReadAcquire | ||||||
| import dev.inmo.micro_utils.coroutines.withWriteLock | import dev.inmo.micro_utils.coroutines.withWriteLock | ||||||
| import kotlinx.coroutines.job | import kotlinx.coroutines.job | ||||||
| @@ -93,7 +92,7 @@ class SortedBinaryTreeNode<T>( | |||||||
|  * This process will continue until function will not find place to put [SortedBinaryTreeNode] with data or |  * This process will continue until function will not find place to put [SortedBinaryTreeNode] with data or | ||||||
|  * [SortedBinaryTreeNode] with [SortedBinaryTreeNode.data] same as [newData] will be found |  * [SortedBinaryTreeNode] with [SortedBinaryTreeNode.data] same as [newData] will be found | ||||||
|  */ |  */ | ||||||
| private suspend fun <T> SortedBinaryTreeNode<T>.addSubNode( | private suspend fun <T> SortedBinaryTreeNode<T>.upsertSubNode( | ||||||
|     subNode: SortedBinaryTreeNode<T>, |     subNode: SortedBinaryTreeNode<T>, | ||||||
|     skipLockers: Set<SmartRWLocker> = emptySet() |     skipLockers: Set<SmartRWLocker> = emptySet() | ||||||
| ): SortedBinaryTreeNode<T> { | ): SortedBinaryTreeNode<T> { | ||||||
| @@ -149,7 +148,7 @@ private suspend fun <T> SortedBinaryTreeNode<T>.addSubNode( | |||||||
|  * [SortedBinaryTreeNode] with [SortedBinaryTreeNode.data] same as [newData] will be found |  * [SortedBinaryTreeNode] with [SortedBinaryTreeNode.data] same as [newData] will be found | ||||||
|  */ |  */ | ||||||
| suspend fun <T> SortedBinaryTreeNode<T>.addSubNode(newData: T): SortedBinaryTreeNode<T> { | suspend fun <T> SortedBinaryTreeNode<T>.addSubNode(newData: T): SortedBinaryTreeNode<T> { | ||||||
|     return addSubNode( |     return upsertSubNode( | ||||||
|         SortedBinaryTreeNode(newData, comparator) |         SortedBinaryTreeNode(newData, comparator) | ||||||
|     ) |     ) | ||||||
| } | } | ||||||
| @@ -198,8 +197,8 @@ suspend fun <T> SortedBinaryTreeNode<T>.findParentNode(data: T): SortedBinaryTre | |||||||
|  */ |  */ | ||||||
| suspend fun <T> SortedBinaryTreeNode<T>.removeSubNode(data: T): Pair<SortedBinaryTreeNode<T>, SortedBinaryTreeNode<T>>? { | suspend fun <T> SortedBinaryTreeNode<T>.removeSubNode(data: T): Pair<SortedBinaryTreeNode<T>, SortedBinaryTreeNode<T>>? { | ||||||
|     val onFoundToRemoveCallback: suspend SortedBinaryTreeNode<T>.(left: SortedBinaryTreeNode<T>?, right: SortedBinaryTreeNode<T>?) -> Unit = { left, right -> |     val onFoundToRemoveCallback: suspend SortedBinaryTreeNode<T>.(left: SortedBinaryTreeNode<T>?, right: SortedBinaryTreeNode<T>?) -> Unit = { left, right -> | ||||||
|         left ?.also { leftNode -> addSubNode(leftNode, setOf(locker)) } |         left ?.also { leftNode -> upsertSubNode(leftNode, setOf(locker)) } | ||||||
|         right ?.also { rightNode -> addSubNode(rightNode, setOf(locker)) } |         right ?.also { rightNode -> upsertSubNode(rightNode, setOf(locker)) } | ||||||
|     } |     } | ||||||
|     while (coroutineContext.job.isActive) { |     while (coroutineContext.job.isActive) { | ||||||
|         val foundParentNode = findParentNode(data) ?: return null |         val foundParentNode = findParentNode(data) ?: return null | ||||||
|   | |||||||
| @@ -0,0 +1,401 @@ | |||||||
|  | package dev.inmo.micro_utils.coroutines.collections | ||||||
|  |  | ||||||
|  | import dev.inmo.micro_utils.coroutines.SmartRWLocker | ||||||
|  | import dev.inmo.micro_utils.coroutines.withReadAcquire | ||||||
|  | import dev.inmo.micro_utils.coroutines.withWriteLock | ||||||
|  | import kotlinx.coroutines.job | ||||||
|  | import kotlinx.serialization.Serializable | ||||||
|  | import kotlin.coroutines.coroutineContext | ||||||
|  |  | ||||||
|  | /** | ||||||
|  |  * Creates simple [Comparator] which will use [compareTo] of [T] for both objects | ||||||
|  |  */ | ||||||
|  | private fun <T : Comparable<C>, C : T> T.createComparator() = Comparator<C> { o1, o2 -> o1.compareTo(o2) } | ||||||
|  |  | ||||||
|  | @Serializable | ||||||
|  | class SortedMapLikeBinaryTreeNode<K, V>( | ||||||
|  |     val key: K, | ||||||
|  |     val value: V, | ||||||
|  |     internal val comparator: Comparator<K>, | ||||||
|  | ) : Iterable<SortedMapLikeBinaryTreeNode<K, V>> { | ||||||
|  |     internal var leftNode: SortedMapLikeBinaryTreeNode<K, V>? = null | ||||||
|  |     internal var rightNode: SortedMapLikeBinaryTreeNode<K, V>? = null | ||||||
|  |     internal val locker: SmartRWLocker by lazy { | ||||||
|  |         SmartRWLocker() | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     suspend fun getLeftNode() = locker.withReadAcquire { | ||||||
|  |         leftNode | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     suspend fun getRightNode() = locker.withReadAcquire { | ||||||
|  |         rightNode | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     suspend fun getLeftKey() = getLeftNode() ?.key | ||||||
|  |     suspend fun getLeftValue() = getLeftNode() ?.value | ||||||
|  |  | ||||||
|  |     suspend fun getRightKey() = getRightNode() ?.value | ||||||
|  |     suspend fun getRightValue() = getRightNode() ?.value | ||||||
|  |  | ||||||
|  |     override fun equals(other: Any?): Boolean { | ||||||
|  |         return other === this || (other is SortedMapLikeBinaryTreeNode<*, *> && other.key == key && other.rightNode == rightNode && other.leftNode == leftNode) | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     override fun hashCode(): Int { | ||||||
|  |         return key.hashCode() * 31 + rightNode.hashCode() + leftNode.hashCode() | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     suspend fun size(): Int { | ||||||
|  |         return locker.withReadAcquire { | ||||||
|  |             1 + (leftNode ?.size() ?: 0) + (rightNode ?.size() ?: 0) | ||||||
|  |         } | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     /** | ||||||
|  |      * This [Iterator] will run from less to greater values of nodes starting the | ||||||
|  |      * [dev.inmo.micro_utils.coroutines.collections.SortedMapLikeBinaryTreeNode]-receiver. Due to non-suspending | ||||||
|  |      * nature of [iterator] builder, this [Iterator] **DO NOT** guarantee consistent content due to iterations. It | ||||||
|  |      * means, that tree can be changed during to iteration process | ||||||
|  |      */ | ||||||
|  |     override fun iterator(): Iterator<SortedMapLikeBinaryTreeNode<K, V>> = iterator { | ||||||
|  |         leftNode ?.let { | ||||||
|  |             it.iterator().forEach { yield(it) } | ||||||
|  |         } | ||||||
|  |         yield(this@SortedMapLikeBinaryTreeNode) | ||||||
|  |         rightNode ?.let { | ||||||
|  |             it.iterator().forEach { yield(it) } | ||||||
|  |         } | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     override fun toString(): String { | ||||||
|  |         return "$key($leftNode;$rightNode)" | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     companion object { | ||||||
|  |         operator fun <K : Comparable<K>, V> invoke( | ||||||
|  |             key: K, | ||||||
|  |             value: V | ||||||
|  |         ) = SortedMapLikeBinaryTreeNode( | ||||||
|  |             key, | ||||||
|  |             value, | ||||||
|  |             key.createComparator() | ||||||
|  |         ) | ||||||
|  |     } | ||||||
|  | } | ||||||
|  |  | ||||||
|  | /** | ||||||
|  |  * Will add subnode in tree if there are no any node with [newData] | ||||||
|  |  * | ||||||
|  |  * * If [newData] is greater than [SortedMapLikeBinaryTreeNode.key] of currently checking node, | ||||||
|  |  * will be used [SortedMapLikeBinaryTreeNode.rightNode] | ||||||
|  |  * * If [newData] is equal to [SortedMapLikeBinaryTreeNode.key] of currently | ||||||
|  |  * checking node - will be returned currently checking node | ||||||
|  |  * * If [newData] is less than [SortedMapLikeBinaryTreeNode.key] of currently | ||||||
|  |  * checking node - will be used [SortedMapLikeBinaryTreeNode.leftNode] | ||||||
|  |  * | ||||||
|  |  * This process will continue until function will not find place to put [SortedMapLikeBinaryTreeNode] with data or | ||||||
|  |  * [SortedMapLikeBinaryTreeNode] with [SortedMapLikeBinaryTreeNode.key] same as [newData] will be found | ||||||
|  |  * | ||||||
|  |  * @param replaceMode Will replace only value if node already exists | ||||||
|  |  */ | ||||||
|  | private suspend fun <K, V> SortedMapLikeBinaryTreeNode<K, V>.upsertSubNode( | ||||||
|  |     subNode: SortedMapLikeBinaryTreeNode<K, V>, | ||||||
|  |     skipLockers: Set<SmartRWLocker> = emptySet(), | ||||||
|  |     replaceMode: Boolean | ||||||
|  | ): SortedMapLikeBinaryTreeNode<K, V> { | ||||||
|  |     var currentlyChecking = this | ||||||
|  |     var latestParent: SortedMapLikeBinaryTreeNode<K, V>? = null | ||||||
|  |     val lockedLockers = mutableSetOf<SmartRWLocker>() | ||||||
|  |     try { | ||||||
|  |         while (coroutineContext.job.isActive) { | ||||||
|  |             if (currentlyChecking.locker !in lockedLockers && currentlyChecking.locker !in skipLockers) { | ||||||
|  |                 currentlyChecking.locker.lockWrite() | ||||||
|  |                 lockedLockers.add(currentlyChecking.locker) | ||||||
|  |             } | ||||||
|  |             val left = currentlyChecking.leftNode | ||||||
|  |             val right = currentlyChecking.rightNode | ||||||
|  |             val comparingResult = currentlyChecking.comparator.compare(subNode.key, currentlyChecking.key) | ||||||
|  |             val isGreater = comparingResult > 0 | ||||||
|  |             when { | ||||||
|  |                 comparingResult == 0 -> { | ||||||
|  |                     val resultNode = if (replaceMode) { | ||||||
|  |                         subNode | ||||||
|  |                     } else { | ||||||
|  |                         val newNode = SortedMapLikeBinaryTreeNode( | ||||||
|  |                             subNode.key, | ||||||
|  |                             subNode.value, | ||||||
|  |                             currentlyChecking.comparator, | ||||||
|  |                         ) | ||||||
|  |                         newNode.leftNode = currentlyChecking.leftNode | ||||||
|  |                         newNode.rightNode = currentlyChecking.rightNode | ||||||
|  |                         newNode | ||||||
|  |                     } | ||||||
|  |  | ||||||
|  |                     latestParent ?.let { | ||||||
|  |                         when { | ||||||
|  |                             it.leftNode === currentlyChecking -> it.leftNode = resultNode | ||||||
|  |                             it.rightNode === currentlyChecking -> it.rightNode = resultNode | ||||||
|  |                         } | ||||||
|  |                     } | ||||||
|  |  | ||||||
|  |                     return resultNode | ||||||
|  |                 } | ||||||
|  |                 isGreater && right == null -> { | ||||||
|  |                     currentlyChecking.rightNode = subNode | ||||||
|  |                     return subNode | ||||||
|  |                 } | ||||||
|  |                 isGreater && right != null -> { | ||||||
|  |                     latestParent = currentlyChecking | ||||||
|  |                     currentlyChecking = right | ||||||
|  |                 } | ||||||
|  |                 left == null -> { | ||||||
|  |                     currentlyChecking.leftNode = subNode | ||||||
|  |                     return subNode | ||||||
|  |                 } | ||||||
|  |                 else -> { | ||||||
|  |                     latestParent = currentlyChecking | ||||||
|  |                     currentlyChecking = left | ||||||
|  |                 } | ||||||
|  |             } | ||||||
|  |         } | ||||||
|  |     } finally { | ||||||
|  |         lockedLockers.forEach { | ||||||
|  |             runCatching { it.unlockWrite() } | ||||||
|  |         } | ||||||
|  |     } | ||||||
|  |     error("Unable to add node") | ||||||
|  | } | ||||||
|  |  | ||||||
|  | /** | ||||||
|  |  * Will add subnode in tree if there are no any node with [key] | ||||||
|  |  * | ||||||
|  |  * * If [key] is greater than [SortedMapLikeBinaryTreeNode.key] of currently checking node, | ||||||
|  |  * will be used [SortedMapLikeBinaryTreeNode.rightNode] | ||||||
|  |  * * If [key] is equal to [SortedMapLikeBinaryTreeNode.key] of currently | ||||||
|  |  * checking node - will be returned currently checking node | ||||||
|  |  * * If [key] is less than [SortedMapLikeBinaryTreeNode.key] of currently | ||||||
|  |  * checking node - will be used [SortedMapLikeBinaryTreeNode.leftNode] | ||||||
|  |  * | ||||||
|  |  * This process will continue until function will not find place to put [SortedMapLikeBinaryTreeNode] with data or | ||||||
|  |  * [SortedMapLikeBinaryTreeNode] with [SortedMapLikeBinaryTreeNode.key] same as [key] will be found | ||||||
|  |  */ | ||||||
|  | suspend fun <K, V> SortedMapLikeBinaryTreeNode<K, V>.upsertSubNode( | ||||||
|  |     key: K, | ||||||
|  |     value: V | ||||||
|  | ): SortedMapLikeBinaryTreeNode<K, V> { | ||||||
|  |     return upsertSubNode( | ||||||
|  |         SortedMapLikeBinaryTreeNode(key, value, comparator), | ||||||
|  |         replaceMode = false | ||||||
|  |     ) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | suspend fun <K, V> SortedMapLikeBinaryTreeNode<K, V>.findParentNode(data: K): SortedMapLikeBinaryTreeNode<K, V>? { | ||||||
|  |     var currentParent: SortedMapLikeBinaryTreeNode<K, V>? = null | ||||||
|  |     var currentlyChecking: SortedMapLikeBinaryTreeNode<K, V>? = this | ||||||
|  |     val lockedLockers = mutableSetOf<SmartRWLocker>() | ||||||
|  |     try { | ||||||
|  |         while (coroutineContext.job.isActive) { | ||||||
|  |             if (currentlyChecking == null) { | ||||||
|  |                 return null | ||||||
|  |             } | ||||||
|  |             if (currentlyChecking.locker !in lockedLockers) { | ||||||
|  |                 currentlyChecking.locker.acquireRead() | ||||||
|  |                 lockedLockers.add(currentlyChecking.locker) | ||||||
|  |             } | ||||||
|  |             val comparingResult = currentlyChecking.comparator.compare(data, currentlyChecking.key) | ||||||
|  |             when { | ||||||
|  |                 comparingResult > 0 -> { | ||||||
|  |                     currentParent = currentlyChecking | ||||||
|  |                     currentlyChecking = currentlyChecking.rightNode | ||||||
|  |                     continue | ||||||
|  |                 } | ||||||
|  |                 comparingResult < 0 -> { | ||||||
|  |                     currentParent = currentlyChecking | ||||||
|  |                     currentlyChecking = currentlyChecking.leftNode | ||||||
|  |                     continue | ||||||
|  |                 } | ||||||
|  |                 else -> return currentParent | ||||||
|  |             } | ||||||
|  |         } | ||||||
|  |     } finally { | ||||||
|  |         lockedLockers.forEach { | ||||||
|  |             runCatching { it.releaseRead() } | ||||||
|  |         } | ||||||
|  |     } | ||||||
|  |     error("Unable to find node") | ||||||
|  | } | ||||||
|  |  | ||||||
|  | /** | ||||||
|  |  * Will remove (detach) node from tree starting with [this] [SortedMapLikeBinaryTreeNode] | ||||||
|  |  * | ||||||
|  |  * @return If data were found, [Pair] where [Pair.first] is the parent node where from [Pair.second] has been detached; | ||||||
|  |  *         null otherwise | ||||||
|  |  */ | ||||||
|  | suspend fun <K, V> SortedMapLikeBinaryTreeNode<K, V>.removeSubNode(data: K): Pair<SortedMapLikeBinaryTreeNode<K, V>, SortedMapLikeBinaryTreeNode<K, V>>? { | ||||||
|  |     val onFoundToRemoveCallback: suspend SortedMapLikeBinaryTreeNode<K, V>.(left: SortedMapLikeBinaryTreeNode<K, V>?, right: SortedMapLikeBinaryTreeNode<K, V>?) -> Unit = { left, right -> | ||||||
|  |         left ?.also { leftNode -> upsertSubNode(leftNode, setOf(locker), replaceMode = true) } | ||||||
|  |         right ?.also { rightNode -> upsertSubNode(rightNode, setOf(locker), replaceMode = true) } | ||||||
|  |     } | ||||||
|  |     while (coroutineContext.job.isActive) { | ||||||
|  |         val foundParentNode = findParentNode(data) ?: return null | ||||||
|  |         foundParentNode.locker.withWriteLock { | ||||||
|  |             val left = foundParentNode.leftNode | ||||||
|  |             val right = foundParentNode.rightNode | ||||||
|  |             when { | ||||||
|  |                 left != null && left.comparator.compare(data, left.key) == 0 -> { | ||||||
|  |                     foundParentNode.leftNode = null | ||||||
|  |                     foundParentNode.onFoundToRemoveCallback(left.leftNode, left.rightNode) | ||||||
|  |                     return foundParentNode to left | ||||||
|  |                 } | ||||||
|  |                 right != null && right.comparator.compare(data, right.key) == 0 -> { | ||||||
|  |                     foundParentNode.rightNode = null | ||||||
|  |                     foundParentNode.onFoundToRemoveCallback(right.leftNode, right.rightNode) | ||||||
|  |                     return foundParentNode to right | ||||||
|  |                 } | ||||||
|  |                 else -> { | ||||||
|  |                     return@withWriteLock // data has been changed, new search required | ||||||
|  |                 } | ||||||
|  |             } | ||||||
|  |         } | ||||||
|  |     } | ||||||
|  |     error("Unable to remove node") | ||||||
|  | } | ||||||
|  | suspend fun <K, V> SortedMapLikeBinaryTreeNode<K, V>.findNode(key: K): SortedMapLikeBinaryTreeNode<K, V>? { | ||||||
|  |     var currentlyChecking: SortedMapLikeBinaryTreeNode<K, V>? = this | ||||||
|  |     val lockedLockers = mutableSetOf<SmartRWLocker>() | ||||||
|  |     try { | ||||||
|  |         while (coroutineContext.job.isActive) { | ||||||
|  |             if (currentlyChecking == null) { | ||||||
|  |                 return null | ||||||
|  |             } | ||||||
|  |             if (currentlyChecking.locker !in lockedLockers) { | ||||||
|  |                 currentlyChecking.locker.acquireRead() | ||||||
|  |                 lockedLockers.add(currentlyChecking.locker) | ||||||
|  |             } | ||||||
|  |             val comparingResult = currentlyChecking.comparator.compare(key, currentlyChecking.key) | ||||||
|  |             when { | ||||||
|  |                 comparingResult > 0 -> { | ||||||
|  |                     currentlyChecking = currentlyChecking.rightNode | ||||||
|  |                     continue | ||||||
|  |                 } | ||||||
|  |                 comparingResult < 0 -> { | ||||||
|  |                     currentlyChecking = currentlyChecking.leftNode | ||||||
|  |                     continue | ||||||
|  |                 } | ||||||
|  |                 else -> return currentlyChecking | ||||||
|  |             } | ||||||
|  |         } | ||||||
|  |     } finally { | ||||||
|  |         lockedLockers.forEach { | ||||||
|  |             runCatching { it.releaseRead() } | ||||||
|  |         } | ||||||
|  |     } | ||||||
|  |     error("Unable to find node") | ||||||
|  | } | ||||||
|  | suspend fun <K, V> SortedMapLikeBinaryTreeNode<K, V>.contains(data: K): Boolean = findNode(data) != null | ||||||
|  |  | ||||||
|  | suspend fun <K, V> SortedMapLikeBinaryTreeNode<K, V>.findNodesInRange(from: K, to: K, fromInclusiveMode: Boolean, toInclusiveMode: Boolean): Set<SortedMapLikeBinaryTreeNode<K, V>> { | ||||||
|  |     val results = mutableSetOf<SortedMapLikeBinaryTreeNode<K, V>>() | ||||||
|  |     val leftToCheck = mutableSetOf(this) | ||||||
|  |     val lockedLockers = mutableSetOf<SmartRWLocker>() | ||||||
|  |     val fromComparingFun: (SortedMapLikeBinaryTreeNode<K, V>) -> Boolean = if (fromInclusiveMode) { | ||||||
|  |         { it.comparator.compare(from, it.key) <= 0 } | ||||||
|  |     } else { | ||||||
|  |         { it.comparator.compare(from, it.key) < 0 } | ||||||
|  |     } | ||||||
|  |     val toComparingFun: (SortedMapLikeBinaryTreeNode<K, V>) -> Boolean = if (toInclusiveMode) { | ||||||
|  |         { it.comparator.compare(to, it.key) >= 0 } | ||||||
|  |     } else { | ||||||
|  |         { it.comparator.compare(to, it.key) > 0 } | ||||||
|  |     } | ||||||
|  |     try { | ||||||
|  |         while (coroutineContext.job.isActive && leftToCheck.isNotEmpty()) { | ||||||
|  |             val currentlyChecking = leftToCheck.first() | ||||||
|  |             leftToCheck.remove(currentlyChecking) | ||||||
|  |             if (currentlyChecking in results) { | ||||||
|  |                 continue | ||||||
|  |             } | ||||||
|  |             currentlyChecking.locker.acquireRead() | ||||||
|  |             lockedLockers.add(currentlyChecking.locker) | ||||||
|  |             if (fromComparingFun(currentlyChecking) && toComparingFun(currentlyChecking)) { | ||||||
|  |                 results.add(currentlyChecking) | ||||||
|  |                 currentlyChecking.leftNode ?.let { leftToCheck.add(it) } | ||||||
|  |                 currentlyChecking.rightNode ?.let { leftToCheck.add(it) } | ||||||
|  |                 continue | ||||||
|  |             } | ||||||
|  |             when { | ||||||
|  |                 currentlyChecking.comparator.compare(to, currentlyChecking.key) < 0 -> currentlyChecking.leftNode ?.let { leftToCheck.add(it) } | ||||||
|  |                 currentlyChecking.comparator.compare(from, currentlyChecking.key) > 0 -> currentlyChecking.rightNode ?.let { leftToCheck.add(it) } | ||||||
|  |             } | ||||||
|  |         } | ||||||
|  |         return results.toSet() | ||||||
|  |     } finally { | ||||||
|  |         lockedLockers.forEach { | ||||||
|  |             runCatching { it.releaseRead() } | ||||||
|  |         } | ||||||
|  |     } | ||||||
|  |     error("Unable to find nodes range") | ||||||
|  | } | ||||||
|  | suspend fun <K, V> SortedMapLikeBinaryTreeNode<K, V>.deepEquals(other: SortedMapLikeBinaryTreeNode<K, V>): Boolean { | ||||||
|  |     val leftToCheck = mutableSetOf(this) | ||||||
|  |     val othersToCheck = mutableSetOf(other) | ||||||
|  |     val lockedLockers = mutableSetOf<SmartRWLocker>() | ||||||
|  |     try { | ||||||
|  |         while (leftToCheck.isNotEmpty() && othersToCheck.isNotEmpty()) { | ||||||
|  |             val thisToCheck = leftToCheck.first() | ||||||
|  |             leftToCheck.remove(thisToCheck) | ||||||
|  |  | ||||||
|  |             val otherToCheck = othersToCheck.first() | ||||||
|  |             othersToCheck.remove(otherToCheck) | ||||||
|  |  | ||||||
|  |             if (thisToCheck.locker !in lockedLockers) { | ||||||
|  |                 thisToCheck.locker.acquireRead() | ||||||
|  |                 lockedLockers.add(thisToCheck.locker) | ||||||
|  |             } | ||||||
|  |             if (otherToCheck.locker !in lockedLockers) { | ||||||
|  |                 otherToCheck.locker.acquireRead() | ||||||
|  |                 lockedLockers.add(otherToCheck.locker) | ||||||
|  |             } | ||||||
|  |  | ||||||
|  |             if (thisToCheck.key != otherToCheck.key || thisToCheck.value != otherToCheck.value) { | ||||||
|  |                 return false | ||||||
|  |             } | ||||||
|  |  | ||||||
|  |             if ((thisToCheck.leftNode == null).xor(otherToCheck.leftNode == null)) { | ||||||
|  |                 return false | ||||||
|  |             } | ||||||
|  |             if ((thisToCheck.rightNode == null).xor(otherToCheck.rightNode == null)) { | ||||||
|  |                 return false | ||||||
|  |             } | ||||||
|  |  | ||||||
|  |             thisToCheck.leftNode?.let { leftToCheck.add(it) } | ||||||
|  |             thisToCheck.rightNode?.let { leftToCheck.add(it) } | ||||||
|  |  | ||||||
|  |             otherToCheck.leftNode?.let { othersToCheck.add(it) } | ||||||
|  |             otherToCheck.rightNode?.let { othersToCheck.add(it) } | ||||||
|  |         } | ||||||
|  |     } finally { | ||||||
|  |         lockedLockers.forEach { | ||||||
|  |             runCatching { it.releaseRead() } | ||||||
|  |         } | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     return leftToCheck.isEmpty() && othersToCheck.isEmpty() | ||||||
|  | } | ||||||
|  | suspend fun <K, V> SortedMapLikeBinaryTreeNode<K, V>.findNodesInRange(from: K, to: K): Set<SortedMapLikeBinaryTreeNode<K, V>> = findNodesInRange( | ||||||
|  |     from = from, | ||||||
|  |     to = to, | ||||||
|  |     fromInclusiveMode = true, | ||||||
|  |     toInclusiveMode = true | ||||||
|  | ) | ||||||
|  | suspend fun <K, V> SortedMapLikeBinaryTreeNode<K, V>.findNodesInRangeExcluding(from: K, to: K): Set<SortedMapLikeBinaryTreeNode<K, V>> = findNodesInRange( | ||||||
|  |     from = from, | ||||||
|  |     to = to, | ||||||
|  |     fromInclusiveMode = false, | ||||||
|  |     toInclusiveMode = false | ||||||
|  | ) | ||||||
|  | suspend fun <K : Comparable<K>, V> SortedMapLikeBinaryTreeNode<K, V>.findNodesInRange(range: ClosedRange<K>): Set<SortedMapLikeBinaryTreeNode<K, V>> = findNodesInRange( | ||||||
|  |     from = range.start, | ||||||
|  |     to = range.endInclusive, | ||||||
|  | ) | ||||||
							
								
								
									
										12
									
								
								coroutines/src/commonTest/kotlin/RealTimeOut.kt
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										12
									
								
								coroutines/src/commonTest/kotlin/RealTimeOut.kt
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,12 @@ | |||||||
|  | import kotlinx.coroutines.Dispatchers | ||||||
|  | import kotlinx.coroutines.withContext | ||||||
|  | import kotlinx.coroutines.withTimeout | ||||||
|  | import kotlin.time.Duration | ||||||
|  |  | ||||||
|  | suspend fun <T> realWithTimeout(time: Duration, block: suspend () -> T): T { | ||||||
|  |     return withContext(Dispatchers.Default.limitedParallelism(1)) { | ||||||
|  |         withTimeout(time) { | ||||||
|  |             block() | ||||||
|  |         } | ||||||
|  |     } | ||||||
|  | } | ||||||
							
								
								
									
										196
									
								
								coroutines/src/commonTest/kotlin/SmartKeyRWLockerTests.kt
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										196
									
								
								coroutines/src/commonTest/kotlin/SmartKeyRWLockerTests.kt
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,196 @@ | |||||||
|  | import dev.inmo.micro_utils.coroutines.* | ||||||
|  | import kotlinx.coroutines.* | ||||||
|  | import kotlinx.coroutines.flow.first | ||||||
|  | import kotlinx.coroutines.sync.Mutex | ||||||
|  | import kotlinx.coroutines.sync.withLock | ||||||
|  | import kotlinx.coroutines.test.runTest | ||||||
|  | import kotlin.test.Test | ||||||
|  | import kotlin.test.assertEquals | ||||||
|  | import kotlin.test.assertFails | ||||||
|  | import kotlin.test.assertFalse | ||||||
|  | import kotlin.test.assertTrue | ||||||
|  | import kotlin.time.Duration.Companion.milliseconds | ||||||
|  | import kotlin.time.Duration.Companion.seconds | ||||||
|  |  | ||||||
|  | class SmartKeyRWLockerTests { | ||||||
|  |     @Test | ||||||
|  |     fun writeLockKeyFailedOnGlobalWriteLockTest() = runTest { | ||||||
|  |         val locker = SmartKeyRWLocker<String>() | ||||||
|  |         val testKey = "test" | ||||||
|  |         locker.lockWrite() | ||||||
|  |  | ||||||
|  |         assertTrue { locker.isWriteLocked() } | ||||||
|  |  | ||||||
|  |         assertFails { | ||||||
|  |             realWithTimeout(1.seconds) { | ||||||
|  |                 locker.lockWrite(testKey) | ||||||
|  |             } | ||||||
|  |         } | ||||||
|  |         assertFalse { locker.isWriteLocked(testKey) } | ||||||
|  |  | ||||||
|  |         locker.unlockWrite() | ||||||
|  |         assertFalse { locker.isWriteLocked() } | ||||||
|  |  | ||||||
|  |         realWithTimeout(1.seconds) { | ||||||
|  |             locker.lockWrite(testKey) | ||||||
|  |         } | ||||||
|  |         assertTrue { locker.isWriteLocked(testKey) } | ||||||
|  |         assertTrue { locker.unlockWrite(testKey) } | ||||||
|  |         assertFalse { locker.isWriteLocked(testKey) } | ||||||
|  |     } | ||||||
|  |     @Test | ||||||
|  |     fun writeLockKeyFailedOnGlobalReadLockTest() = 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() | ||||||
|  |             } | ||||||
|  |         } | ||||||
|  |         assertFalse(locker.isWriteLocked()) | ||||||
|  |  | ||||||
|  |         locker.unlockWrite(testKey) | ||||||
|  |         assertFalse { locker.isWriteLocked(testKey) } | ||||||
|  |  | ||||||
|  |         realWithTimeout(1.seconds) { | ||||||
|  |             locker.lockWrite() | ||||||
|  |         } | ||||||
|  |         assertTrue(locker.isWriteLocked()) | ||||||
|  |         assertTrue { locker.unlockWrite() } | ||||||
|  |         assertFalse(locker.isWriteLocked()) | ||||||
|  |     } | ||||||
|  |     @Test | ||||||
|  |     fun readsBlockingGlobalWrite() = runTest { | ||||||
|  |         val locker = SmartKeyRWLocker<String>() | ||||||
|  |  | ||||||
|  |         val testKeys = (0 until 100).map { "test$it" } | ||||||
|  |  | ||||||
|  |         for (i in testKeys.indices) { | ||||||
|  |             val it = testKeys[i] | ||||||
|  |             locker.acquireRead(it) | ||||||
|  |             val previous = testKeys.take(i) | ||||||
|  |             val next = testKeys.drop(i + 1) | ||||||
|  |  | ||||||
|  |             previous.forEach { | ||||||
|  |                 assertTrue { locker.readSemaphoreOrNull(it) ?.freePermits == Int.MAX_VALUE - 1 } | ||||||
|  |             } | ||||||
|  |             next.forEach { | ||||||
|  |                 assertTrue { locker.readSemaphoreOrNull(it) ?.freePermits == null } | ||||||
|  |             } | ||||||
|  |         } | ||||||
|  |  | ||||||
|  |         for (i in testKeys.indices) { | ||||||
|  |             val it = testKeys[i] | ||||||
|  |             assertFails { | ||||||
|  |                 realWithTimeout(13.milliseconds) { locker.lockWrite() } | ||||||
|  |             } | ||||||
|  |             val readPermitsBeforeLock = locker.readSemaphore().freePermits | ||||||
|  |             realWithTimeout(1.seconds) { locker.acquireRead() } | ||||||
|  |             locker.releaseRead() | ||||||
|  |             assertEquals(readPermitsBeforeLock, locker.readSemaphore().freePermits) | ||||||
|  |  | ||||||
|  |             locker.releaseRead(it) | ||||||
|  |         } | ||||||
|  |  | ||||||
|  |         assertTrue { locker.readSemaphore().freePermits == Int.MAX_VALUE } | ||||||
|  |         realWithTimeout(1.seconds) { locker.lockWrite() } | ||||||
|  |         assertFails { | ||||||
|  |             realWithTimeout(13.milliseconds) { locker.acquireRead() } | ||||||
|  |         } | ||||||
|  |         assertTrue { locker.unlockWrite() } | ||||||
|  |         assertTrue { locker.readSemaphore().freePermits == Int.MAX_VALUE } | ||||||
|  |     } | ||||||
|  |     @Test | ||||||
|  |     fun writesBlockingGlobalWrite() = runTest { | ||||||
|  |         val locker = SmartKeyRWLocker<String>() | ||||||
|  |  | ||||||
|  |         val testKeys = (0 until 100).map { "test$it" } | ||||||
|  |  | ||||||
|  |         for (i in testKeys.indices) { | ||||||
|  |             val it = testKeys[i] | ||||||
|  |             locker.lockWrite(it) | ||||||
|  |             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 } | ||||||
|  |             } | ||||||
|  |         } | ||||||
|  |  | ||||||
|  |         for (i in testKeys.indices) { | ||||||
|  |             val it = testKeys[i] | ||||||
|  |             assertFails { realWithTimeout(13.milliseconds) { locker.lockWrite() } } | ||||||
|  |  | ||||||
|  |             val readPermitsBeforeLock = locker.readSemaphore().freePermits | ||||||
|  |             assertFails { realWithTimeout(13.milliseconds) { locker.acquireRead() } } | ||||||
|  |             assertEquals(readPermitsBeforeLock, locker.readSemaphore().freePermits) | ||||||
|  |  | ||||||
|  |             locker.unlockWrite(it) | ||||||
|  |         } | ||||||
|  |  | ||||||
|  |         assertTrue { locker.readSemaphore().freePermits == Int.MAX_VALUE } | ||||||
|  |         realWithTimeout(1.seconds) { locker.lockWrite() } | ||||||
|  |         assertFails { | ||||||
|  |             realWithTimeout(13.milliseconds) { locker.acquireRead() } | ||||||
|  |         } | ||||||
|  |         assertTrue { locker.unlockWrite() } | ||||||
|  |         assertTrue { locker.readSemaphore().freePermits == Int.MAX_VALUE } | ||||||
|  |     } | ||||||
|  | } | ||||||
| @@ -6,7 +6,10 @@ import kotlinx.coroutines.sync.withLock | |||||||
| import kotlinx.coroutines.test.runTest | import kotlinx.coroutines.test.runTest | ||||||
| import kotlin.test.Test | import kotlin.test.Test | ||||||
| import kotlin.test.assertEquals | import kotlin.test.assertEquals | ||||||
|  | import kotlin.test.assertFails | ||||||
|  | import kotlin.test.assertFalse | ||||||
| import kotlin.test.assertTrue | import kotlin.test.assertTrue | ||||||
|  | import kotlin.time.Duration.Companion.seconds | ||||||
|  |  | ||||||
| class SmartRWLockerTests { | class SmartRWLockerTests { | ||||||
|     @Test |     @Test | ||||||
| @@ -148,4 +151,17 @@ class SmartRWLockerTests { | |||||||
|             assertEquals(false, locker.writeMutex.isLocked) |             assertEquals(false, locker.writeMutex.isLocked) | ||||||
|         } |         } | ||||||
|     } |     } | ||||||
|  |  | ||||||
|  |     @Test | ||||||
|  |     fun exceptionOnLockingWillNotLockLocker() = runTest { | ||||||
|  |         val locker = SmartRWLocker() | ||||||
|  |  | ||||||
|  |         locker.acquireRead() | ||||||
|  |         assertFails { | ||||||
|  |             realWithTimeout(1.seconds) { | ||||||
|  |                 locker.lockWrite() | ||||||
|  |             } | ||||||
|  |         } | ||||||
|  |         assertFalse { locker.writeMutex.isLocked } | ||||||
|  |     } | ||||||
| } | } | ||||||
|   | |||||||
| @@ -7,7 +7,9 @@ fun <T> CoroutineScope.launchSynchronously(block: suspend CoroutineScope.() -> T | |||||||
|     val objectToSynchronize = Object() |     val objectToSynchronize = Object() | ||||||
|     synchronized(objectToSynchronize) { |     synchronized(objectToSynchronize) { | ||||||
|         launch(start = CoroutineStart.UNDISPATCHED) { |         launch(start = CoroutineStart.UNDISPATCHED) { | ||||||
|             result = safelyWithResult(block) |             result = runCatching { | ||||||
|  |                 block() | ||||||
|  |             } | ||||||
|         }.invokeOnCompletion { |         }.invokeOnCompletion { | ||||||
|             synchronized(objectToSynchronize) { |             synchronized(objectToSynchronize) { | ||||||
|                 objectToSynchronize.notifyAll() |                 objectToSynchronize.notifyAll() | ||||||
|   | |||||||
| @@ -0,0 +1,118 @@ | |||||||
|  | package dev.inmo.micro_utils.coroutines | ||||||
|  |  | ||||||
|  | import dev.inmo.micro_utils.coroutines.collections.* | ||||||
|  | import kotlinx.coroutines.test.runTest | ||||||
|  | import kotlin.test.Test | ||||||
|  | import kotlin.test.assertEquals | ||||||
|  | import kotlin.test.assertTrue | ||||||
|  | import kotlin.time.Duration.Companion.seconds | ||||||
|  |  | ||||||
|  | class SortedMapLikeBinaryTreeNodeTests { | ||||||
|  |     @Test | ||||||
|  |     fun insertOnZeroLevelWorks() = runTest { | ||||||
|  |         val zeroNode = SortedMapLikeBinaryTreeNode(0, 0) | ||||||
|  |         zeroNode.upsertSubNode(1, 1) | ||||||
|  |         zeroNode.upsertSubNode(-1, -1) | ||||||
|  |  | ||||||
|  |         assertEquals(0, zeroNode.key) | ||||||
|  |         assertEquals(1, zeroNode.getRightNode() ?.key) | ||||||
|  |         assertEquals(-1, zeroNode.getLeftNode() ?.key) | ||||||
|  |  | ||||||
|  |         assertEquals(0, zeroNode.findNode(0) ?.value) | ||||||
|  |         assertEquals(1, zeroNode.findNode(1) ?.value) | ||||||
|  |         assertEquals(-1, zeroNode.findNode(-1) ?.value) | ||||||
|  |     } | ||||||
|  |     @Test | ||||||
|  |     fun searchOnZeroLevelWorks() = runTest { | ||||||
|  |         val zeroNode = SortedMapLikeBinaryTreeNode(0, 0) | ||||||
|  |         val oneNode = zeroNode.upsertSubNode(1, 1) | ||||||
|  |         val minusOneNode = zeroNode.upsertSubNode(-1, -1) | ||||||
|  |  | ||||||
|  |         val assertingNodesToSearchQuery = mapOf( | ||||||
|  |             setOf(oneNode) to (1 .. 1), | ||||||
|  |             setOf(zeroNode, oneNode) to (0 .. 1), | ||||||
|  |             setOf(minusOneNode, zeroNode, oneNode) to (-1 .. 1), | ||||||
|  |             setOf(minusOneNode, zeroNode) to (-1 .. 0), | ||||||
|  |             setOf(minusOneNode) to (-1 .. -1), | ||||||
|  |             setOf(zeroNode) to (0 .. 0), | ||||||
|  |         ) | ||||||
|  |  | ||||||
|  |         assertingNodesToSearchQuery.forEach { | ||||||
|  |             val foundData = zeroNode.findNodesInRange(it.value) | ||||||
|  |             assertTrue(foundData.containsAll(it.key)) | ||||||
|  |             assertTrue(it.key.containsAll(foundData)) | ||||||
|  |         } | ||||||
|  |     } | ||||||
|  |     @Test | ||||||
|  |     fun deepReInsertOnWorks() = runTest(timeout = 300.seconds) { | ||||||
|  |         var zeroNode = SortedMapLikeBinaryTreeNode(0, 0) | ||||||
|  |         val rangeRadius = 500 | ||||||
|  |         val nodes = mutableMapOf<Int, SortedMapLikeBinaryTreeNode<Int, Int>>() | ||||||
|  |         for (i in -rangeRadius .. rangeRadius) { | ||||||
|  |             nodes[i] = zeroNode.upsertSubNode(i, i) | ||||||
|  |             if (i == zeroNode.key) { | ||||||
|  |                 zeroNode = nodes.getValue(i) | ||||||
|  |             } | ||||||
|  |         } | ||||||
|  |  | ||||||
|  |         for (i in -rangeRadius .. rangeRadius) { | ||||||
|  |             val expectedNode = nodes.getValue(i) | ||||||
|  |             val foundNode = zeroNode.findNode(i) | ||||||
|  |  | ||||||
|  |             assertEquals(expectedNode, foundNode) | ||||||
|  |  | ||||||
|  |             if (expectedNode === zeroNode) continue | ||||||
|  |  | ||||||
|  |             val parentNode = zeroNode.findParentNode(i) | ||||||
|  |             assertTrue( | ||||||
|  |                 parentNode ?.getLeftNode() === expectedNode || parentNode ?.getRightNode() === expectedNode, | ||||||
|  |                 "It is expected, that parent node with data ${parentNode ?.key} will be parent of ${expectedNode.key}, but its left subnode is ${parentNode ?.getLeftNode() ?.key} and right one is ${parentNode ?.getRightNode() ?.key}" | ||||||
|  |             ) | ||||||
|  |             assertTrue( | ||||||
|  |                 foundNode != null && expectedNode.deepEquals(foundNode) | ||||||
|  |             ) | ||||||
|  |  | ||||||
|  |             zeroNode.upsertSubNode(i, -i) | ||||||
|  |             val foundModifiedNode = zeroNode.findNode(i) | ||||||
|  |             assertEquals(foundNode ?.value, foundModifiedNode ?.value ?.times(-1)) | ||||||
|  |         } | ||||||
|  |     } | ||||||
|  |     @Test | ||||||
|  |     fun deepInsertOnWorks() = runTest(timeout = 240.seconds) { | ||||||
|  |         val zeroNode = SortedMapLikeBinaryTreeNode(0, 0) | ||||||
|  |         val rangeRadius = 500 | ||||||
|  |         val nodes = mutableMapOf<Int, SortedMapLikeBinaryTreeNode<Int, Int>>() | ||||||
|  |         for (i in -rangeRadius .. rangeRadius) { | ||||||
|  |             if (zeroNode.key != i) { | ||||||
|  |                 nodes[i] = zeroNode.upsertSubNode(i, i) | ||||||
|  |             } | ||||||
|  |         } | ||||||
|  |         nodes[zeroNode.key] = zeroNode | ||||||
|  |  | ||||||
|  |         for (i in -rangeRadius .. rangeRadius) { | ||||||
|  |             val expectedNode = nodes.getValue(i) | ||||||
|  |             val foundNode = zeroNode.findNode(i) | ||||||
|  |  | ||||||
|  |             assertTrue(expectedNode === foundNode) | ||||||
|  |  | ||||||
|  |             if (expectedNode === zeroNode) continue | ||||||
|  |  | ||||||
|  |             val parentNode = zeroNode.findParentNode(i) | ||||||
|  |             assertTrue( | ||||||
|  |                 parentNode ?.getLeftNode() === expectedNode || parentNode ?.getRightNode() === expectedNode, | ||||||
|  |                 "It is expected, that parent node with data ${parentNode ?.key} will be parent of ${expectedNode.key}, but its left subnode is ${parentNode ?.getLeftNode() ?.key} and right one is ${parentNode ?.getRightNode() ?.key}" | ||||||
|  |             ) | ||||||
|  |         } | ||||||
|  |  | ||||||
|  |         val sourceTreeSize = zeroNode.size() | ||||||
|  |  | ||||||
|  |         var previousData = -rangeRadius - 1 | ||||||
|  |         for (node in zeroNode) { | ||||||
|  |             assertTrue(nodes[node.key] === node) | ||||||
|  |             assertTrue(previousData == node.key - 1) | ||||||
|  |             previousData = node.key | ||||||
|  |         } | ||||||
|  |  | ||||||
|  |         assertTrue(sourceTreeSize == zeroNode.size()) | ||||||
|  |     } | ||||||
|  | } | ||||||
| @@ -15,5 +15,5 @@ crypto_js_version=4.1.1 | |||||||
| # Project data | # Project data | ||||||
|  |  | ||||||
| group=dev.inmo | group=dev.inmo | ||||||
| version=0.25.0 | version=0.25.3 | ||||||
| android_code_version=290 | android_code_version=293 | ||||||
|   | |||||||
| @@ -1,13 +1,13 @@ | |||||||
| [versions] | [versions] | ||||||
|  |  | ||||||
| kt = "2.1.10" | kt = "2.1.20" | ||||||
| kt-serialization = "1.8.0" | kt-serialization = "1.8.0" | ||||||
| kt-coroutines = "1.10.1" | kt-coroutines = "1.10.1" | ||||||
|  |  | ||||||
| kslog = "1.4.1" | kslog = "1.4.1" | ||||||
|  |  | ||||||
| jb-compose = "1.7.3" | jb-compose = "1.7.3" | ||||||
| jb-exposed = "0.59.0" | jb-exposed = "0.60.0" | ||||||
| jb-dokka = "2.0.0" | jb-dokka = "2.0.0" | ||||||
|  |  | ||||||
| sqlite = "3.49.1.0" | sqlite = "3.49.1.0" | ||||||
| @@ -23,7 +23,7 @@ koin = "4.0.2" | |||||||
|  |  | ||||||
| okio = "3.10.2" | okio = "3.10.2" | ||||||
|  |  | ||||||
| ksp = "2.1.10-1.0.31" | ksp = "2.1.20-1.0.31" | ||||||
| kotlin-poet = "1.18.1" | kotlin-poet = "1.18.1" | ||||||
|  |  | ||||||
| versions = "0.51.0" | versions = "0.51.0" | ||||||
|   | |||||||
| @@ -17,11 +17,13 @@ kotlin { | |||||||
|         jvmMain { |         jvmMain { | ||||||
|             dependencies { |             dependencies { | ||||||
|                 api libs.kt.reflect |                 api libs.kt.reflect | ||||||
|  |                 api project(":micro_utils.coroutines") | ||||||
|             } |             } | ||||||
|         } |         } | ||||||
|         androidMain { |         androidMain { | ||||||
|             dependencies { |             dependencies { | ||||||
|                 api libs.kt.reflect |                 api libs.kt.reflect | ||||||
|  |                 api project(":micro_utils.coroutines") | ||||||
|             } |             } | ||||||
|         } |         } | ||||||
|     } |     } | ||||||
|   | |||||||
							
								
								
									
										32
									
								
								koin/src/jvmMain/kotlin/FactorySuspend.kt
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										32
									
								
								koin/src/jvmMain/kotlin/FactorySuspend.kt
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,32 @@ | |||||||
|  | package dev.inmo.micro_utils.koin | ||||||
|  |  | ||||||
|  | import dev.inmo.micro_utils.coroutines.doSynchronously | ||||||
|  | import kotlinx.coroutines.CoroutineScope | ||||||
|  | import org.koin.core.module.Module | ||||||
|  | import org.koin.core.parameter.ParametersHolder | ||||||
|  | import org.koin.core.qualifier.Qualifier | ||||||
|  | import org.koin.core.qualifier.StringQualifier | ||||||
|  | import org.koin.core.scope.Scope | ||||||
|  | import kotlin.reflect.KClass | ||||||
|  |  | ||||||
|  | inline fun <reified T : Any> Module.factorySuspend( | ||||||
|  |     qualifier: Qualifier? = null, | ||||||
|  |     coroutineScope: CoroutineScope? = null, | ||||||
|  |     noinline definition: suspend Scope.(ParametersHolder) -> T | ||||||
|  | ) = factory( | ||||||
|  |     qualifier, | ||||||
|  |     if (coroutineScope == null) { | ||||||
|  |         { | ||||||
|  |             doSynchronously { | ||||||
|  |                 definition(it) | ||||||
|  |             } | ||||||
|  |         } | ||||||
|  |     } else { | ||||||
|  |         { | ||||||
|  |             coroutineScope.doSynchronously { | ||||||
|  |                 definition(it) | ||||||
|  |             } | ||||||
|  |         } | ||||||
|  |     } | ||||||
|  | ) | ||||||
|  |  | ||||||
							
								
								
									
										32
									
								
								koin/src/jvmMain/kotlin/SingleSuspend.kt
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										32
									
								
								koin/src/jvmMain/kotlin/SingleSuspend.kt
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,32 @@ | |||||||
|  | package dev.inmo.micro_utils.koin | ||||||
|  |  | ||||||
|  | import dev.inmo.micro_utils.coroutines.doSynchronously | ||||||
|  | import kotlinx.coroutines.CoroutineScope | ||||||
|  | import org.koin.core.module.Module | ||||||
|  | import org.koin.core.parameter.ParametersHolder | ||||||
|  | import org.koin.core.qualifier.StringQualifier | ||||||
|  | import org.koin.core.scope.Scope | ||||||
|  |  | ||||||
|  | inline fun <reified T : Any> Module.singleSuspend( | ||||||
|  |     qualifier: StringQualifier, | ||||||
|  |     createdAtStart: Boolean = false, | ||||||
|  |     coroutineScope: CoroutineScope? = null, | ||||||
|  |     noinline definition: suspend Scope.(ParametersHolder) -> T | ||||||
|  | ) = single( | ||||||
|  |     qualifier, | ||||||
|  |     createdAtStart, | ||||||
|  |     if (coroutineScope == null) { | ||||||
|  |         { | ||||||
|  |             doSynchronously { | ||||||
|  |                 definition(it) | ||||||
|  |             } | ||||||
|  |         } | ||||||
|  |     } else { | ||||||
|  |         { | ||||||
|  |             coroutineScope.doSynchronously { | ||||||
|  |                 definition(it) | ||||||
|  |             } | ||||||
|  |         } | ||||||
|  |     } | ||||||
|  | ) | ||||||
|  |  | ||||||
| @@ -2,7 +2,13 @@ package dev.inmo.micro_utils.pagination.compose | |||||||
|  |  | ||||||
| import androidx.compose.runtime.* | import androidx.compose.runtime.* | ||||||
| import dev.inmo.micro_utils.coroutines.SpecialMutableStateFlow | import dev.inmo.micro_utils.coroutines.SpecialMutableStateFlow | ||||||
|  | import dev.inmo.micro_utils.coroutines.launchLoggingDropExceptions | ||||||
|  | import dev.inmo.micro_utils.coroutines.runCatchingLogging | ||||||
| import dev.inmo.micro_utils.pagination.* | import dev.inmo.micro_utils.pagination.* | ||||||
|  | import kotlinx.coroutines.CoroutineScope | ||||||
|  | import kotlinx.coroutines.Job | ||||||
|  | import kotlinx.coroutines.sync.Mutex | ||||||
|  | import kotlinx.coroutines.sync.withLock | ||||||
|  |  | ||||||
| /** | /** | ||||||
|  * Context for managing infinite pagination in a Compose UI. |  * Context for managing infinite pagination in a Compose UI. | ||||||
| @@ -16,30 +22,46 @@ import dev.inmo.micro_utils.pagination.* | |||||||
|  */ |  */ | ||||||
| class InfinityPagedComponentContext<T> internal constructor( | class InfinityPagedComponentContext<T> internal constructor( | ||||||
|     page: Int, |     page: Int, | ||||||
|     size: Int |     size: Int, | ||||||
|  |     private val scope: CoroutineScope, | ||||||
|  |     private val loader: suspend InfinityPagedComponentContext<T>.(Pagination) -> PaginationResult<T> | ||||||
| ) { | ) { | ||||||
|     internal val startPage = SimplePagination(page, size) |     internal val startPage = SimplePagination(page, size) | ||||||
|     internal val currentlyLoadingPage = SpecialMutableStateFlow<Pagination?>(startPage) |  | ||||||
|     internal val latestLoadedPage = SpecialMutableStateFlow<PaginationResult<T>?>(null) |     internal val latestLoadedPage = SpecialMutableStateFlow<PaginationResult<T>?>(null) | ||||||
|     internal val dataState = SpecialMutableStateFlow<List<T>?>(null) |     internal val dataState = SpecialMutableStateFlow<List<T>?>(null) | ||||||
|  |     internal var loadingJob: Job? = null | ||||||
|  |     internal val loadingMutex = Mutex() | ||||||
|  |  | ||||||
|     /** |     /** | ||||||
|      * Loads the next page of data. If the current page is the last one, the function returns early. |      * Loads the next page of data. If the current page is the last one, the function returns early. | ||||||
|      */ |      */ | ||||||
|     fun loadNext() { |     fun loadNext(): Job { | ||||||
|         if (latestLoadedPage.value ?.isLastPage == true) return |         return scope.launchLoggingDropExceptions { | ||||||
|         if (currentlyLoadingPage.value != null) return // Data loading has been inited but not loaded yet |             loadingMutex.withLock { | ||||||
|  |                 if (latestLoadedPage.value ?.isLastPage == true) return@launchLoggingDropExceptions | ||||||
|         currentlyLoadingPage.value = latestLoadedPage.value ?.nextPage() ?: startPage |                 loadingJob = loadingJob ?: scope.launchLoggingDropExceptions { | ||||||
|  |                     runCatching { | ||||||
|  |                         loader(latestLoadedPage.value ?.nextPage() ?: startPage) | ||||||
|  |                     }.onSuccess { | ||||||
|  |                         latestLoadedPage.value = it | ||||||
|  |                         dataState.value = (dataState.value ?: emptyList()) + it.results | ||||||
|  |                     } | ||||||
|  |                     loadingMutex.withLock { | ||||||
|  |                         loadingJob = null | ||||||
|  |                     } | ||||||
|  |                 } | ||||||
|  |                 loadingJob | ||||||
|  |             } ?.join() | ||||||
|  |         } | ||||||
|     } |     } | ||||||
|  |  | ||||||
|     /** |     /** | ||||||
|      * Reloads the pagination from the first page, clearing previously loaded data. |      * Reloads the pagination from the first page, clearing previously loaded data. | ||||||
|      */ |      */ | ||||||
|     fun reload() { |     fun reload(): Job { | ||||||
|         latestLoadedPage.value = null |         latestLoadedPage.value = null | ||||||
|         currentlyLoadingPage.value = null |         dataState.value = null | ||||||
|         loadNext() |         return loadNext() | ||||||
|     } |     } | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -58,17 +80,13 @@ internal fun <T> InfinityPagedComponent( | |||||||
|     page: Int, |     page: Int, | ||||||
|     size: Int, |     size: Int, | ||||||
|     loader: suspend InfinityPagedComponentContext<T>.(Pagination) -> PaginationResult<T>, |     loader: suspend InfinityPagedComponentContext<T>.(Pagination) -> PaginationResult<T>, | ||||||
|  |     predefinedScope: CoroutineScope? = null, | ||||||
|     block: @Composable InfinityPagedComponentContext<T>.(List<T>?) -> Unit |     block: @Composable InfinityPagedComponentContext<T>.(List<T>?) -> Unit | ||||||
| ) { | ) { | ||||||
|     val context = remember { InfinityPagedComponentContext<T>(page, size) } |     val scope = predefinedScope ?: rememberCoroutineScope() | ||||||
|  |     val context = remember { InfinityPagedComponentContext<T>(page, size, scope, loader) } | ||||||
|     val currentlyLoadingState = context.currentlyLoadingPage.collectAsState() |     remember { | ||||||
|     LaunchedEffect(currentlyLoadingState.value) { |         context.reload() | ||||||
|         val paginationResult = loader(context, currentlyLoadingState.value ?: return@LaunchedEffect) |  | ||||||
|         context.latestLoadedPage.value = paginationResult |  | ||||||
|         context.currentlyLoadingPage.value = null |  | ||||||
|  |  | ||||||
|         context.dataState.value = (context.dataState.value ?: emptyList()) + paginationResult.results |  | ||||||
|     } |     } | ||||||
|  |  | ||||||
|     val dataState = context.dataState.collectAsState() |     val dataState = context.dataState.collectAsState() | ||||||
| @@ -88,12 +106,14 @@ internal fun <T> InfinityPagedComponent( | |||||||
| fun <T> InfinityPagedComponent( | fun <T> InfinityPagedComponent( | ||||||
|     pageInfo: Pagination, |     pageInfo: Pagination, | ||||||
|     loader: suspend InfinityPagedComponentContext<T>.(Pagination) -> PaginationResult<T>, |     loader: suspend InfinityPagedComponentContext<T>.(Pagination) -> PaginationResult<T>, | ||||||
|  |     predefinedScope: CoroutineScope? = null, | ||||||
|     block: @Composable InfinityPagedComponentContext<T>.(List<T>?) -> Unit |     block: @Composable InfinityPagedComponentContext<T>.(List<T>?) -> Unit | ||||||
| ) { | ) { | ||||||
|     InfinityPagedComponent( |     InfinityPagedComponent( | ||||||
|         pageInfo.page, |         pageInfo.page, | ||||||
|         pageInfo.size, |         pageInfo.size, | ||||||
|         loader, |         loader, | ||||||
|  |         predefinedScope, | ||||||
|         block |         block | ||||||
|     ) |     ) | ||||||
| } | } | ||||||
| @@ -111,7 +131,8 @@ fun <T> InfinityPagedComponent( | |||||||
| fun <T> InfinityPagedComponent( | fun <T> InfinityPagedComponent( | ||||||
|     size: Int, |     size: Int, | ||||||
|     loader: suspend InfinityPagedComponentContext<T>.(Pagination) -> PaginationResult<T>, |     loader: suspend InfinityPagedComponentContext<T>.(Pagination) -> PaginationResult<T>, | ||||||
|  |     predefinedScope: CoroutineScope? = null, | ||||||
|     block: @Composable InfinityPagedComponentContext<T>.(List<T>?) -> Unit |     block: @Composable InfinityPagedComponentContext<T>.(List<T>?) -> Unit | ||||||
| ) { | ) { | ||||||
|     InfinityPagedComponent(0, size, loader, block) |     InfinityPagedComponent(0, size, loader, predefinedScope, block) | ||||||
| } | } | ||||||
|   | |||||||
| @@ -2,7 +2,12 @@ package dev.inmo.micro_utils.pagination.compose | |||||||
|  |  | ||||||
| import androidx.compose.runtime.* | import androidx.compose.runtime.* | ||||||
| import dev.inmo.micro_utils.coroutines.SpecialMutableStateFlow | import dev.inmo.micro_utils.coroutines.SpecialMutableStateFlow | ||||||
|  | import dev.inmo.micro_utils.coroutines.launchLoggingDropExceptions | ||||||
| import dev.inmo.micro_utils.pagination.* | import dev.inmo.micro_utils.pagination.* | ||||||
|  | import kotlinx.coroutines.CoroutineScope | ||||||
|  | import kotlinx.coroutines.Job | ||||||
|  | import kotlinx.coroutines.sync.Mutex | ||||||
|  | import kotlinx.coroutines.sync.withLock | ||||||
|  |  | ||||||
| /** | /** | ||||||
|  * Context for managing paginated data in a Compose UI. |  * Context for managing paginated data in a Compose UI. | ||||||
| @@ -18,39 +23,73 @@ import dev.inmo.micro_utils.pagination.* | |||||||
|  */ |  */ | ||||||
| class PagedComponentContext<T> internal constructor( | class PagedComponentContext<T> internal constructor( | ||||||
|     initialPage: Int, |     initialPage: Int, | ||||||
|     size: Int |     size: Int, | ||||||
|  |     private val scope: CoroutineScope, | ||||||
|  |     private val loader: suspend PagedComponentContext<T>.(Pagination) -> PaginationResult<T> | ||||||
| ) { | ) { | ||||||
|     internal val startPage = SimplePagination(initialPage, size) |     internal val startPage = SimplePagination(initialPage, size) | ||||||
|     internal val currentlyLoadingPageState = SpecialMutableStateFlow<Pagination?>(startPage) |  | ||||||
|     internal val latestLoadedPage = SpecialMutableStateFlow<PaginationResult<T>?>(null) |     internal val latestLoadedPage = SpecialMutableStateFlow<PaginationResult<T>?>(null) | ||||||
|  |     internal val dataState = SpecialMutableStateFlow<PaginationResult<T>?>(null) | ||||||
|  |     internal var loadingJob: Job? = null | ||||||
|  |     internal val loadingMutex = Mutex() | ||||||
|  |  | ||||||
|  |     private fun initLoadingJob( | ||||||
|  |         skipCheckerInLock: () -> Boolean, | ||||||
|  |         pageGetter: () -> Pagination | ||||||
|  |     ): Job { | ||||||
|  |         return scope.launchLoggingDropExceptions { | ||||||
|  |             loadingMutex.withLock { | ||||||
|  |                 if (skipCheckerInLock()) return@launchLoggingDropExceptions | ||||||
|  |                 loadingJob = loadingJob ?: scope.launchLoggingDropExceptions { | ||||||
|  |                     runCatching { | ||||||
|  |                         loader(pageGetter()) | ||||||
|  |                     }.onSuccess { | ||||||
|  |                         latestLoadedPage.value = it | ||||||
|  |                         dataState.value = it | ||||||
|  |                     } | ||||||
|  |                     loadingMutex.withLock { | ||||||
|  |                         loadingJob = null | ||||||
|  |                     } | ||||||
|  |                 } | ||||||
|  |                 loadingJob | ||||||
|  |             } ?.join() | ||||||
|  |         } | ||||||
|  |     } | ||||||
|  |  | ||||||
|     /** |     /** | ||||||
|      * Loads the next page of data. If the last page is reached, this function returns early. |      * Loads the next page of data. If the last page is reached, this function returns early. | ||||||
|      */ |      */ | ||||||
|     fun loadNext() { |     fun loadNext(): Job { | ||||||
|         when { |         return initLoadingJob( | ||||||
|             currentlyLoadingPageState.value != null -> return |             { latestLoadedPage.value ?.isLastPage == true } | ||||||
|             latestLoadedPage.value ?.isLastPage == true -> return |         ) { | ||||||
|             else -> currentlyLoadingPageState.value = (latestLoadedPage.value ?.nextPage()) ?: startPage |             latestLoadedPage.value ?.nextPage() ?: startPage | ||||||
|         } |         } | ||||||
|     } |     } | ||||||
|  |  | ||||||
|     /** |     /** | ||||||
|      * Loads the previous page of data if available. |      * Loads the previous page of data if available. | ||||||
|      */ |      */ | ||||||
|     fun loadPrevious() { |     fun loadPrevious(): Job { | ||||||
|         when { |         return initLoadingJob( | ||||||
|             currentlyLoadingPageState.value != null -> return |             { latestLoadedPage.value ?.isFirstPage == true } | ||||||
|             latestLoadedPage.value ?.isFirstPage == true -> return |         ) { | ||||||
|             else -> currentlyLoadingPageState.value = (latestLoadedPage.value ?.previousPage()) ?: startPage |             latestLoadedPage.value ?.previousPage() ?: startPage | ||||||
|         } |         } | ||||||
|     } |     } | ||||||
|  |  | ||||||
|     /** |     /** | ||||||
|      * Reloads the current page, refreshing the data. |      * Reloads the current page, refreshing the data. | ||||||
|      */ |      */ | ||||||
|     fun reload() { |     fun reload(): Job { | ||||||
|         currentlyLoadingPageState.value = latestLoadedPage.value |         return initLoadingJob( | ||||||
|  |             { | ||||||
|  |                 latestLoadedPage.value = null | ||||||
|  |                 true | ||||||
|  |             } | ||||||
|  |         ) { | ||||||
|  |             startPage | ||||||
|  |         } | ||||||
|     } |     } | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -69,18 +108,16 @@ internal fun <T> PagedComponent( | |||||||
|     initialPage: Int, |     initialPage: Int, | ||||||
|     size: Int, |     size: Int, | ||||||
|     loader: suspend PagedComponentContext<T>.(Pagination) -> PaginationResult<T>, |     loader: suspend PagedComponentContext<T>.(Pagination) -> PaginationResult<T>, | ||||||
|  |     predefinedScope: CoroutineScope? = null, | ||||||
|     block: @Composable PagedComponentContext<T>.(PaginationResult<T>) -> Unit |     block: @Composable PagedComponentContext<T>.(PaginationResult<T>) -> Unit | ||||||
| ) { | ) { | ||||||
|     val context = remember { PagedComponentContext<T>(initialPage, size) } |     val scope = predefinedScope ?: rememberCoroutineScope() | ||||||
|  |     val context = remember { PagedComponentContext<T>(initialPage, size, scope, loader) } | ||||||
|     val currentlyLoadingState = context.currentlyLoadingPageState.collectAsState() |     remember { | ||||||
|     LaunchedEffect(currentlyLoadingState.value) { |         context.reload() | ||||||
|         val paginationResult = loader(context, currentlyLoadingState.value ?: return@LaunchedEffect) |  | ||||||
|         context.latestLoadedPage.value = paginationResult |  | ||||||
|         context.currentlyLoadingPageState.value = null |  | ||||||
|     } |     } | ||||||
|  |  | ||||||
|     val pageState = context.latestLoadedPage.collectAsState() |     val pageState = context.dataState.collectAsState() | ||||||
|     pageState.value ?.let { |     pageState.value ?.let { | ||||||
|         context.block(it) |         context.block(it) | ||||||
|     } |     } | ||||||
| @@ -98,12 +135,14 @@ internal fun <T> PagedComponent( | |||||||
| fun <T> PagedComponent( | fun <T> PagedComponent( | ||||||
|     pageInfo: Pagination, |     pageInfo: Pagination, | ||||||
|     loader: suspend PagedComponentContext<T>.(Pagination) -> PaginationResult<T>, |     loader: suspend PagedComponentContext<T>.(Pagination) -> PaginationResult<T>, | ||||||
|  |     predefinedScope: CoroutineScope? = null, | ||||||
|     block: @Composable PagedComponentContext<T>.(PaginationResult<T>) -> Unit |     block: @Composable PagedComponentContext<T>.(PaginationResult<T>) -> Unit | ||||||
| ) { | ) { | ||||||
|     PagedComponent( |     PagedComponent( | ||||||
|         pageInfo.page, |         pageInfo.page, | ||||||
|         pageInfo.size, |         pageInfo.size, | ||||||
|         loader, |         loader, | ||||||
|  |         predefinedScope, | ||||||
|         block |         block | ||||||
|     ) |     ) | ||||||
| } | } | ||||||
| @@ -120,7 +159,8 @@ fun <T> PagedComponent( | |||||||
| fun <T> PagedComponent( | fun <T> PagedComponent( | ||||||
|     size: Int, |     size: Int, | ||||||
|     loader: suspend PagedComponentContext<T>.(Pagination) -> PaginationResult<T>, |     loader: suspend PagedComponentContext<T>.(Pagination) -> PaginationResult<T>, | ||||||
|  |     predefinedScope: CoroutineScope? = null, | ||||||
|     block: @Composable PagedComponentContext<T>.(PaginationResult<T>) -> Unit |     block: @Composable PagedComponentContext<T>.(PaginationResult<T>) -> Unit | ||||||
| ) { | ) { | ||||||
|     PagedComponent(0, size, loader, block) |     PagedComponent(0, size, loader, predefinedScope, block) | ||||||
| } | } | ||||||
|   | |||||||
| @@ -30,13 +30,13 @@ class InfinityPagedComponentTests { | |||||||
|                 } |                 } | ||||||
|             ) { |             ) { | ||||||
|                 if (it == null) { |                 if (it == null) { | ||||||
|                     assertEquals(0, this.currentlyLoadingPage.value ?.page) |                     assertEquals(null, it) | ||||||
|                 } else { |                 } else { | ||||||
|                     assertEquals(expectedList, it) |                     assertEquals(expectedList, it) | ||||||
|                 } |                 } | ||||||
|  |  | ||||||
|                 LaunchedEffect(it ?.size) { |                 LaunchedEffect(it ?.size) { | ||||||
|                     loadNext() |                     loadNext().join() | ||||||
|                 } |                 } | ||||||
|             } |             } | ||||||
|         } |         } | ||||||
|   | |||||||
| @@ -0,0 +1,14 @@ | |||||||
|  | package dev.inmo.micro_utils.repos.annotations | ||||||
|  |  | ||||||
|  |  | ||||||
|  | @RequiresOptIn( | ||||||
|  |     "Overriding of this invalidate message requires manual launching of invalidation on class initialization process", | ||||||
|  |     RequiresOptIn.Level.WARNING | ||||||
|  | ) | ||||||
|  | @Target( | ||||||
|  |     AnnotationTarget.CONSTRUCTOR, | ||||||
|  |     AnnotationTarget.FIELD, | ||||||
|  |     AnnotationTarget.PROPERTY, | ||||||
|  |     AnnotationTarget.FUNCTION, | ||||||
|  | ) | ||||||
|  | annotation class OverrideRequireManualInvalidation | ||||||
| @@ -4,6 +4,7 @@ import dev.inmo.micro_utils.coroutines.SmartRWLocker | |||||||
| import dev.inmo.micro_utils.coroutines.withReadAcquire | import dev.inmo.micro_utils.coroutines.withReadAcquire | ||||||
| import dev.inmo.micro_utils.coroutines.withWriteLock | import dev.inmo.micro_utils.coroutines.withWriteLock | ||||||
| import dev.inmo.micro_utils.repos.* | import dev.inmo.micro_utils.repos.* | ||||||
|  | import dev.inmo.micro_utils.repos.annotations.OverrideRequireManualInvalidation | ||||||
| import dev.inmo.micro_utils.repos.cache.cache.KVCache | import dev.inmo.micro_utils.repos.cache.cache.KVCache | ||||||
| import dev.inmo.micro_utils.repos.cache.util.ActualizeAllClearMode | import dev.inmo.micro_utils.repos.cache.util.ActualizeAllClearMode | ||||||
| import dev.inmo.micro_utils.repos.cache.util.actualizeAll | import dev.inmo.micro_utils.repos.cache.util.actualizeAll | ||||||
| @@ -39,6 +40,7 @@ open class ReadCRUDCacheRepo<ObjectType, IdType>( | |||||||
|         kvCache.contains(id) |         kvCache.contains(id) | ||||||
|     } || parentRepo.contains(id) |     } || parentRepo.contains(id) | ||||||
|  |  | ||||||
|  |     @OverrideRequireManualInvalidation | ||||||
|     override suspend fun invalidate() = locker.withWriteLock { |     override suspend fun invalidate() = locker.withWriteLock { | ||||||
|         kvCache.clear() |         kvCache.clear() | ||||||
|     } |     } | ||||||
| @@ -117,6 +119,7 @@ open class WriteCRUDCacheRepo<ObjectType, IdType, InputValueType>( | |||||||
|         return created |         return created | ||||||
|     } |     } | ||||||
|  |  | ||||||
|  |     @OverrideRequireManualInvalidation | ||||||
|     override suspend fun invalidate() = locker.withWriteLock { |     override suspend fun invalidate() = locker.withWriteLock { | ||||||
|         kvCache.clear() |         kvCache.clear() | ||||||
|     } |     } | ||||||
| @@ -150,6 +153,7 @@ WriteCRUDRepo<ObjectType, IdType, InputValueType> by WriteCRUDCacheRepo( | |||||||
|     idGetter |     idGetter | ||||||
| ), | ), | ||||||
| CRUDRepo<ObjectType, IdType, InputValueType> { | CRUDRepo<ObjectType, IdType, InputValueType> { | ||||||
|  |     @OverrideRequireManualInvalidation | ||||||
|     override suspend fun invalidate() = kvCache.actualizeAll(parentRepo, locker = locker) |     override suspend fun invalidate() = kvCache.actualizeAll(parentRepo, locker = locker) | ||||||
| } | } | ||||||
|  |  | ||||||
|   | |||||||
| @@ -1,5 +1,10 @@ | |||||||
| package dev.inmo.micro_utils.repos.cache | package dev.inmo.micro_utils.repos.cache | ||||||
|  |  | ||||||
|  | import dev.inmo.micro_utils.coroutines.SmartRWLocker | ||||||
|  | import dev.inmo.micro_utils.coroutines.launchLoggingDropExceptions | ||||||
|  | import kotlinx.coroutines.CoroutineScope | ||||||
|  | import kotlinx.coroutines.Job | ||||||
|  |  | ||||||
| interface InvalidatableRepo { | interface InvalidatableRepo { | ||||||
|     /** |     /** | ||||||
|      * Invalidates its internal data. It __may__ lead to autoreload of data. In case when repo makes autoreload, |      * Invalidates its internal data. It __may__ lead to autoreload of data. In case when repo makes autoreload, | ||||||
| @@ -8,4 +13,14 @@ interface InvalidatableRepo { | |||||||
|     suspend fun invalidate() |     suspend fun invalidate() | ||||||
| } | } | ||||||
|  |  | ||||||
|  | suspend fun <T : InvalidatableRepo> T.alsoInvalidate() = also { | ||||||
|  |     invalidate() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | fun <T : InvalidatableRepo> T.alsoInvalidateAsync(scope: CoroutineScope) = also { | ||||||
|  |     scope.launchLoggingDropExceptions { | ||||||
|  |         invalidate() | ||||||
|  |     } | ||||||
|  | } | ||||||
|  |  | ||||||
| typealias CacheRepo = InvalidatableRepo | typealias CacheRepo = InvalidatableRepo | ||||||
|   | |||||||
| @@ -5,6 +5,7 @@ import dev.inmo.micro_utils.coroutines.withReadAcquire | |||||||
| import dev.inmo.micro_utils.coroutines.withWriteLock | import dev.inmo.micro_utils.coroutines.withWriteLock | ||||||
| import dev.inmo.micro_utils.pagination.* | import dev.inmo.micro_utils.pagination.* | ||||||
| import dev.inmo.micro_utils.repos.* | import dev.inmo.micro_utils.repos.* | ||||||
|  | import dev.inmo.micro_utils.repos.annotations.OverrideRequireManualInvalidation | ||||||
| import dev.inmo.micro_utils.repos.cache.cache.KVCache | import dev.inmo.micro_utils.repos.cache.cache.KVCache | ||||||
| import dev.inmo.micro_utils.repos.cache.util.actualizeAll | import dev.inmo.micro_utils.repos.cache.util.actualizeAll | ||||||
| import kotlinx.coroutines.CoroutineScope | import kotlinx.coroutines.CoroutineScope | ||||||
| @@ -49,6 +50,7 @@ open class ReadKeyValueCacheRepo<Key,Value>( | |||||||
|         } |         } | ||||||
|     } |     } | ||||||
|  |  | ||||||
|  |     @OverrideRequireManualInvalidation | ||||||
|     override suspend fun invalidate() = kvCache.actualizeAll(parentRepo, locker = locker) |     override suspend fun invalidate() = kvCache.actualizeAll(parentRepo, locker = locker) | ||||||
| } | } | ||||||
|  |  | ||||||
|   | |||||||
| @@ -6,6 +6,7 @@ import dev.inmo.micro_utils.coroutines.withWriteLock | |||||||
| import dev.inmo.micro_utils.pagination.* | import dev.inmo.micro_utils.pagination.* | ||||||
| import dev.inmo.micro_utils.pagination.utils.* | import dev.inmo.micro_utils.pagination.utils.* | ||||||
| import dev.inmo.micro_utils.repos.* | import dev.inmo.micro_utils.repos.* | ||||||
|  | import dev.inmo.micro_utils.repos.annotations.OverrideRequireManualInvalidation | ||||||
| import dev.inmo.micro_utils.repos.cache.cache.KVCache | import dev.inmo.micro_utils.repos.cache.cache.KVCache | ||||||
| import dev.inmo.micro_utils.repos.cache.util.actualizeAll | import dev.inmo.micro_utils.repos.cache.util.actualizeAll | ||||||
| import kotlinx.coroutines.CoroutineScope | import kotlinx.coroutines.CoroutineScope | ||||||
| @@ -48,6 +49,7 @@ open class ReadKeyValuesCacheRepo<Key,Value>( | |||||||
|         kvCache.contains(k) |         kvCache.contains(k) | ||||||
|     } || parentRepo.contains(k) |     } || parentRepo.contains(k) | ||||||
|  |  | ||||||
|  |     @OverrideRequireManualInvalidation | ||||||
|     override suspend fun invalidate() = kvCache.actualizeAll(parentRepo, locker = locker) |     override suspend fun invalidate() = kvCache.actualizeAll(parentRepo, locker = locker) | ||||||
| } | } | ||||||
|  |  | ||||||
|   | |||||||
| @@ -6,6 +6,7 @@ import dev.inmo.micro_utils.pagination.PaginationResult | |||||||
| import dev.inmo.micro_utils.repos.KeyValueRepo | import dev.inmo.micro_utils.repos.KeyValueRepo | ||||||
| import dev.inmo.micro_utils.repos.MapKeyValueRepo | import dev.inmo.micro_utils.repos.MapKeyValueRepo | ||||||
| import dev.inmo.micro_utils.repos.ReadCRUDRepo | import dev.inmo.micro_utils.repos.ReadCRUDRepo | ||||||
|  | import dev.inmo.micro_utils.repos.annotations.OverrideRequireManualInvalidation | ||||||
| import dev.inmo.micro_utils.repos.cache.fallback.ActionWrapper | import dev.inmo.micro_utils.repos.cache.fallback.ActionWrapper | ||||||
| import dev.inmo.micro_utils.repos.cache.util.actualizeAll | import dev.inmo.micro_utils.repos.cache.util.actualizeAll | ||||||
| import dev.inmo.micro_utils.repos.cache.FallbackCacheRepo | import dev.inmo.micro_utils.repos.cache.FallbackCacheRepo | ||||||
| @@ -90,6 +91,7 @@ open class AutoRecacheReadCRUDRepo<RegisteredObject, Id>( | |||||||
|         kvCache.set(idGetter(it), it) |         kvCache.set(idGetter(it), it) | ||||||
|     } ?: kvCache.get(id) |     } ?: kvCache.get(id) | ||||||
|  |  | ||||||
|  |     @OverrideRequireManualInvalidation | ||||||
|     override suspend fun invalidate() { |     override suspend fun invalidate() { | ||||||
|         actualizeAll() |         actualizeAll() | ||||||
|     } |     } | ||||||
|   | |||||||
| @@ -2,6 +2,7 @@ package dev.inmo.micro_utils.repos.cache.fallback.crud | |||||||
|  |  | ||||||
| import dev.inmo.micro_utils.coroutines.subscribeSafelyWithoutExceptions | import dev.inmo.micro_utils.coroutines.subscribeSafelyWithoutExceptions | ||||||
| import dev.inmo.micro_utils.repos.* | import dev.inmo.micro_utils.repos.* | ||||||
|  | import dev.inmo.micro_utils.repos.annotations.OverrideRequireManualInvalidation | ||||||
| import dev.inmo.micro_utils.repos.cache.FallbackCacheRepo | import dev.inmo.micro_utils.repos.cache.FallbackCacheRepo | ||||||
| import kotlinx.coroutines.CoroutineScope | import kotlinx.coroutines.CoroutineScope | ||||||
| import kotlinx.coroutines.flow.Flow | import kotlinx.coroutines.flow.Flow | ||||||
| @@ -53,6 +54,7 @@ open class AutoRecacheWriteCRUDRepo<RegisteredObject, Id, InputObject>( | |||||||
|         kvCache.set(idGetter(it), it) |         kvCache.set(idGetter(it), it) | ||||||
|     } |     } | ||||||
|  |  | ||||||
|  |     @OverrideRequireManualInvalidation | ||||||
|     override suspend fun invalidate() { |     override suspend fun invalidate() { | ||||||
|         kvCache.clear() |         kvCache.clear() | ||||||
|     } |     } | ||||||
|   | |||||||
| @@ -6,6 +6,7 @@ import dev.inmo.micro_utils.pagination.PaginationResult | |||||||
| import dev.inmo.micro_utils.repos.KeyValueRepo | import dev.inmo.micro_utils.repos.KeyValueRepo | ||||||
| import dev.inmo.micro_utils.repos.MapKeyValueRepo | import dev.inmo.micro_utils.repos.MapKeyValueRepo | ||||||
| import dev.inmo.micro_utils.repos.ReadKeyValueRepo | import dev.inmo.micro_utils.repos.ReadKeyValueRepo | ||||||
|  | import dev.inmo.micro_utils.repos.annotations.OverrideRequireManualInvalidation | ||||||
| import dev.inmo.micro_utils.repos.cache.fallback.ActionWrapper | import dev.inmo.micro_utils.repos.cache.fallback.ActionWrapper | ||||||
| import dev.inmo.micro_utils.repos.cache.util.actualizeAll | import dev.inmo.micro_utils.repos.cache.util.actualizeAll | ||||||
| import dev.inmo.micro_utils.repos.cache.FallbackCacheRepo | import dev.inmo.micro_utils.repos.cache.FallbackCacheRepo | ||||||
| @@ -100,6 +101,7 @@ open class AutoRecacheReadKeyValueRepo<Id, RegisteredObject>( | |||||||
|         originalRepo.keys(v, pagination, reversed) |         originalRepo.keys(v, pagination, reversed) | ||||||
|     }.getOrElse { kvCache.keys(v, pagination, reversed) } |     }.getOrElse { kvCache.keys(v, pagination, reversed) } | ||||||
|  |  | ||||||
|  |     @OverrideRequireManualInvalidation | ||||||
|     override suspend fun invalidate() { |     override suspend fun invalidate() { | ||||||
|         actualizeAll() |         actualizeAll() | ||||||
|     } |     } | ||||||
|   | |||||||
| @@ -2,6 +2,7 @@ package dev.inmo.micro_utils.repos.cache.fallback.keyvalue | |||||||
|  |  | ||||||
| import dev.inmo.micro_utils.coroutines.subscribeSafelyWithoutExceptions | import dev.inmo.micro_utils.coroutines.subscribeSafelyWithoutExceptions | ||||||
| import dev.inmo.micro_utils.repos.* | import dev.inmo.micro_utils.repos.* | ||||||
|  | import dev.inmo.micro_utils.repos.annotations.OverrideRequireManualInvalidation | ||||||
| import dev.inmo.micro_utils.repos.cache.FallbackCacheRepo | import dev.inmo.micro_utils.repos.cache.FallbackCacheRepo | ||||||
| import kotlinx.coroutines.CoroutineScope | import kotlinx.coroutines.CoroutineScope | ||||||
| import kotlinx.coroutines.flow.Flow | import kotlinx.coroutines.flow.Flow | ||||||
| @@ -44,6 +45,7 @@ open class AutoRecacheWriteKeyValueRepo<Id, RegisteredObject>( | |||||||
|         kvCache.set(toSet) |         kvCache.set(toSet) | ||||||
|     } |     } | ||||||
|  |  | ||||||
|  |     @OverrideRequireManualInvalidation | ||||||
|     override suspend fun invalidate() { |     override suspend fun invalidate() { | ||||||
|         kvCache.clear() |         kvCache.clear() | ||||||
|     } |     } | ||||||
|   | |||||||
| @@ -13,6 +13,7 @@ import dev.inmo.micro_utils.pagination.utils.paginate | |||||||
| import dev.inmo.micro_utils.repos.KeyValueRepo | import dev.inmo.micro_utils.repos.KeyValueRepo | ||||||
| import dev.inmo.micro_utils.repos.MapKeyValueRepo | import dev.inmo.micro_utils.repos.MapKeyValueRepo | ||||||
| import dev.inmo.micro_utils.repos.ReadKeyValuesRepo | import dev.inmo.micro_utils.repos.ReadKeyValuesRepo | ||||||
|  | import dev.inmo.micro_utils.repos.annotations.OverrideRequireManualInvalidation | ||||||
| import dev.inmo.micro_utils.repos.cache.fallback.ActionWrapper | import dev.inmo.micro_utils.repos.cache.fallback.ActionWrapper | ||||||
| import dev.inmo.micro_utils.repos.cache.util.actualizeAll | import dev.inmo.micro_utils.repos.cache.util.actualizeAll | ||||||
| import dev.inmo.micro_utils.repos.cache.FallbackCacheRepo | import dev.inmo.micro_utils.repos.cache.FallbackCacheRepo | ||||||
| @@ -140,6 +141,7 @@ open class AutoRecacheReadKeyValuesRepo<Id, RegisteredObject>( | |||||||
|         }) ?: (kvCache.get(k) ?.contains(v) == true) |         }) ?: (kvCache.get(k) ?.contains(v) == true) | ||||||
|     } |     } | ||||||
|  |  | ||||||
|  |     @OverrideRequireManualInvalidation | ||||||
|     override suspend fun invalidate() { |     override suspend fun invalidate() { | ||||||
|         actualizeAll() |         actualizeAll() | ||||||
|     } |     } | ||||||
|   | |||||||
| @@ -3,6 +3,7 @@ package dev.inmo.micro_utils.repos.cache.fallback.keyvalues | |||||||
| import dev.inmo.micro_utils.coroutines.subscribeSafelyWithoutExceptions | import dev.inmo.micro_utils.coroutines.subscribeSafelyWithoutExceptions | ||||||
| import dev.inmo.micro_utils.pagination.utils.doForAllWithNextPaging | import dev.inmo.micro_utils.pagination.utils.doForAllWithNextPaging | ||||||
| import dev.inmo.micro_utils.repos.* | import dev.inmo.micro_utils.repos.* | ||||||
|  | import dev.inmo.micro_utils.repos.annotations.OverrideRequireManualInvalidation | ||||||
| import dev.inmo.micro_utils.repos.cache.FallbackCacheRepo | import dev.inmo.micro_utils.repos.cache.FallbackCacheRepo | ||||||
| import dev.inmo.micro_utils.repos.pagination.maxPagePagination | import dev.inmo.micro_utils.repos.pagination.maxPagePagination | ||||||
| import kotlinx.coroutines.CoroutineScope | import kotlinx.coroutines.CoroutineScope | ||||||
| @@ -89,6 +90,7 @@ open class AutoRecacheWriteKeyValuesRepo<Id, RegisteredObject>( | |||||||
|         } |         } | ||||||
|     } |     } | ||||||
|  |  | ||||||
|  |     @OverrideRequireManualInvalidation | ||||||
|     override suspend fun invalidate() { |     override suspend fun invalidate() { | ||||||
|         kvCache.clear() |         kvCache.clear() | ||||||
|     } |     } | ||||||
|   | |||||||
| @@ -8,6 +8,7 @@ import dev.inmo.micro_utils.coroutines.withWriteLock | |||||||
| import dev.inmo.micro_utils.pagination.Pagination | import dev.inmo.micro_utils.pagination.Pagination | ||||||
| import dev.inmo.micro_utils.pagination.PaginationResult | import dev.inmo.micro_utils.pagination.PaginationResult | ||||||
| import dev.inmo.micro_utils.repos.* | import dev.inmo.micro_utils.repos.* | ||||||
|  | import dev.inmo.micro_utils.repos.annotations.OverrideRequireManualInvalidation | ||||||
| import dev.inmo.micro_utils.repos.cache.* | import dev.inmo.micro_utils.repos.cache.* | ||||||
| import dev.inmo.micro_utils.repos.cache.util.ActualizeAllClearMode | import dev.inmo.micro_utils.repos.cache.util.ActualizeAllClearMode | ||||||
| import dev.inmo.micro_utils.repos.cache.util.actualizeAll | import dev.inmo.micro_utils.repos.cache.util.actualizeAll | ||||||
| @@ -133,6 +134,7 @@ open class FullCRUDCacheRepo<ObjectType, IdType, InputValueType>( | |||||||
|             locker.unlockWrite() |             locker.unlockWrite() | ||||||
|         } |         } | ||||||
|     } |     } | ||||||
|  |     @OverrideRequireManualInvalidation | ||||||
|     override suspend fun invalidate() { |     override suspend fun invalidate() { | ||||||
|         actualizeAll() |         actualizeAll() | ||||||
|     } |     } | ||||||
|   | |||||||
| @@ -8,6 +8,7 @@ import dev.inmo.micro_utils.coroutines.withWriteLock | |||||||
| import dev.inmo.micro_utils.pagination.Pagination | import dev.inmo.micro_utils.pagination.Pagination | ||||||
| import dev.inmo.micro_utils.pagination.PaginationResult | import dev.inmo.micro_utils.pagination.PaginationResult | ||||||
| import dev.inmo.micro_utils.repos.* | import dev.inmo.micro_utils.repos.* | ||||||
|  | import dev.inmo.micro_utils.repos.annotations.OverrideRequireManualInvalidation | ||||||
| import dev.inmo.micro_utils.repos.cache.util.ActualizeAllClearMode | import dev.inmo.micro_utils.repos.cache.util.ActualizeAllClearMode | ||||||
| import dev.inmo.micro_utils.repos.cache.util.actualizeAll | import dev.inmo.micro_utils.repos.cache.util.actualizeAll | ||||||
| import kotlinx.coroutines.CoroutineScope | import kotlinx.coroutines.CoroutineScope | ||||||
| @@ -86,6 +87,7 @@ open class FullReadKeyValueCacheRepo<Key,Value>( | |||||||
|         { if (it.results.isNotEmpty()) actualizeAll() } |         { if (it.results.isNotEmpty()) actualizeAll() } | ||||||
|     ) |     ) | ||||||
|  |  | ||||||
|  |     @OverrideRequireManualInvalidation | ||||||
|     override suspend fun invalidate() { |     override suspend fun invalidate() { | ||||||
|         actualizeAll() |         actualizeAll() | ||||||
|     } |     } | ||||||
| @@ -160,6 +162,7 @@ open class FullKeyValueCacheRepo<Key,Value>( | |||||||
|             locker.unlockWrite() |             locker.unlockWrite() | ||||||
|         } |         } | ||||||
|     } |     } | ||||||
|  |     @OverrideRequireManualInvalidation | ||||||
|     override suspend fun invalidate() { |     override suspend fun invalidate() { | ||||||
|         kvCache.actualizeAll(parentRepo, locker) |         kvCache.actualizeAll(parentRepo, locker) | ||||||
|     } |     } | ||||||
|   | |||||||
| @@ -8,9 +8,9 @@ import dev.inmo.micro_utils.coroutines.withWriteLock | |||||||
| import dev.inmo.micro_utils.pagination.* | import dev.inmo.micro_utils.pagination.* | ||||||
| import dev.inmo.micro_utils.pagination.utils.* | import dev.inmo.micro_utils.pagination.utils.* | ||||||
| import dev.inmo.micro_utils.repos.* | import dev.inmo.micro_utils.repos.* | ||||||
|  | import dev.inmo.micro_utils.repos.annotations.OverrideRequireManualInvalidation | ||||||
| import dev.inmo.micro_utils.repos.cache.util.ActualizeAllClearMode | import dev.inmo.micro_utils.repos.cache.util.ActualizeAllClearMode | ||||||
| import dev.inmo.micro_utils.repos.cache.util.actualizeAll | import dev.inmo.micro_utils.repos.cache.util.actualizeAll | ||||||
| import dev.inmo.micro_utils.repos.pagination.maxPagePagination |  | ||||||
| import kotlinx.coroutines.CoroutineScope | import kotlinx.coroutines.CoroutineScope | ||||||
| import kotlinx.coroutines.Dispatchers | import kotlinx.coroutines.Dispatchers | ||||||
| import kotlinx.coroutines.flow.* | import kotlinx.coroutines.flow.* | ||||||
| @@ -153,6 +153,7 @@ open class FullReadKeyValuesCacheRepo<Key,Value>( | |||||||
|         { if (it.results.isNotEmpty()) actualizeAll() } |         { if (it.results.isNotEmpty()) actualizeAll() } | ||||||
|     ) |     ) | ||||||
|  |  | ||||||
|  |     @OverrideRequireManualInvalidation | ||||||
|     override suspend fun invalidate() { |     override suspend fun invalidate() { | ||||||
|         actualizeAll() |         actualizeAll() | ||||||
|     } |     } | ||||||
| @@ -235,6 +236,7 @@ open class FullKeyValuesCacheRepo<Key,Value>( | |||||||
|             locker.unlockWrite() |             locker.unlockWrite() | ||||||
|         } |         } | ||||||
|     } |     } | ||||||
|  |     @OverrideRequireManualInvalidation | ||||||
|     override suspend fun invalidate() { |     override suspend fun invalidate() { | ||||||
|         kvCache.actualizeAll(parentRepo, locker = locker) |         kvCache.actualizeAll(parentRepo, locker = locker) | ||||||
|     } |     } | ||||||
|   | |||||||
| @@ -1,11 +1,13 @@ | |||||||
| package dev.inmo.micro_utils.repos.cache.full.direct | package dev.inmo.micro_utils.repos.cache.full.direct | ||||||
|  |  | ||||||
|  | import dev.inmo.micro_utils.common.Warning | ||||||
| import dev.inmo.micro_utils.coroutines.SmartRWLocker | import dev.inmo.micro_utils.coroutines.SmartRWLocker | ||||||
| import dev.inmo.micro_utils.coroutines.launchLoggingDropExceptions | import dev.inmo.micro_utils.coroutines.launchLoggingDropExceptions | ||||||
| import dev.inmo.micro_utils.coroutines.withReadAcquire | import dev.inmo.micro_utils.coroutines.withReadAcquire | ||||||
| import dev.inmo.micro_utils.pagination.Pagination | import dev.inmo.micro_utils.pagination.Pagination | ||||||
| import dev.inmo.micro_utils.pagination.PaginationResult | import dev.inmo.micro_utils.pagination.PaginationResult | ||||||
| import dev.inmo.micro_utils.repos.* | import dev.inmo.micro_utils.repos.* | ||||||
|  | import dev.inmo.micro_utils.repos.annotations.OverrideRequireManualInvalidation | ||||||
| import dev.inmo.micro_utils.repos.cache.* | import dev.inmo.micro_utils.repos.cache.* | ||||||
| import dev.inmo.micro_utils.repos.cache.util.actualizeAll | import dev.inmo.micro_utils.repos.cache.util.actualizeAll | ||||||
| import kotlinx.coroutines.CoroutineScope | import kotlinx.coroutines.CoroutineScope | ||||||
| @@ -96,6 +98,8 @@ open class DirectFullCRUDCacheRepo<ObjectType, IdType, InputValueType>( | |||||||
|             locker.unlockWrite() |             locker.unlockWrite() | ||||||
|         } |         } | ||||||
|     } |     } | ||||||
|  |  | ||||||
|  |     @OverrideRequireManualInvalidation | ||||||
|     override suspend fun invalidate() { |     override suspend fun invalidate() { | ||||||
|         actualizeAll() |         actualizeAll() | ||||||
|     } |     } | ||||||
|   | |||||||
| @@ -7,9 +7,7 @@ import dev.inmo.micro_utils.coroutines.withWriteLock | |||||||
| import dev.inmo.micro_utils.pagination.Pagination | import dev.inmo.micro_utils.pagination.Pagination | ||||||
| import dev.inmo.micro_utils.pagination.PaginationResult | import dev.inmo.micro_utils.pagination.PaginationResult | ||||||
| import dev.inmo.micro_utils.repos.* | import dev.inmo.micro_utils.repos.* | ||||||
| import dev.inmo.micro_utils.repos.cache.full.FullKeyValueCacheRepo | import dev.inmo.micro_utils.repos.annotations.OverrideRequireManualInvalidation | ||||||
| import dev.inmo.micro_utils.repos.cache.full.FullReadKeyValueCacheRepo |  | ||||||
| import dev.inmo.micro_utils.repos.cache.full.FullWriteKeyValueCacheRepo |  | ||||||
| import dev.inmo.micro_utils.repos.cache.util.actualizeAll | import dev.inmo.micro_utils.repos.cache.util.actualizeAll | ||||||
| import kotlinx.coroutines.CoroutineScope | import kotlinx.coroutines.CoroutineScope | ||||||
| import kotlinx.coroutines.Dispatchers | import kotlinx.coroutines.Dispatchers | ||||||
| @@ -54,6 +52,7 @@ open class DirectFullReadKeyValueCacheRepo<Key, Value>( | |||||||
|         kvCache.keys(v, pagination, reversed) |         kvCache.keys(v, pagination, reversed) | ||||||
|     } |     } | ||||||
|  |  | ||||||
|  |     @OverrideRequireManualInvalidation | ||||||
|     override suspend fun invalidate() { |     override suspend fun invalidate() { | ||||||
|         actualizeAll() |         actualizeAll() | ||||||
|     } |     } | ||||||
| @@ -86,6 +85,7 @@ open class DirectFullWriteKeyValueCacheRepo<Key, Value>( | |||||||
|         } |         } | ||||||
|     }.launchIn(scope) |     }.launchIn(scope) | ||||||
|  |  | ||||||
|  |     @OverrideRequireManualInvalidation | ||||||
|     override suspend fun invalidate() { |     override suspend fun invalidate() { | ||||||
|         locker.withWriteLock { |         locker.withWriteLock { | ||||||
|             kvCache.clear() |             kvCache.clear() | ||||||
| @@ -135,6 +135,7 @@ open class DirectFullKeyValueCacheRepo<Key, Value>( | |||||||
|             locker.unlockWrite() |             locker.unlockWrite() | ||||||
|         } |         } | ||||||
|     } |     } | ||||||
|  |     @OverrideRequireManualInvalidation | ||||||
|     override suspend fun invalidate() { |     override suspend fun invalidate() { | ||||||
|         kvCache.actualizeAll(parentRepo, locker) |         kvCache.actualizeAll(parentRepo, locker) | ||||||
|     } |     } | ||||||
|   | |||||||
| @@ -1,6 +1,5 @@ | |||||||
| package dev.inmo.micro_utils.repos.cache.full.direct | package dev.inmo.micro_utils.repos.cache.full.direct | ||||||
|  |  | ||||||
| import dev.inmo.micro_utils.common.* |  | ||||||
| import dev.inmo.micro_utils.coroutines.SmartRWLocker | import dev.inmo.micro_utils.coroutines.SmartRWLocker | ||||||
| import dev.inmo.micro_utils.coroutines.launchLoggingDropExceptions | import dev.inmo.micro_utils.coroutines.launchLoggingDropExceptions | ||||||
| import dev.inmo.micro_utils.coroutines.withReadAcquire | import dev.inmo.micro_utils.coroutines.withReadAcquire | ||||||
| @@ -8,6 +7,7 @@ import dev.inmo.micro_utils.coroutines.withWriteLock | |||||||
| import dev.inmo.micro_utils.pagination.* | import dev.inmo.micro_utils.pagination.* | ||||||
| import dev.inmo.micro_utils.pagination.utils.* | import dev.inmo.micro_utils.pagination.utils.* | ||||||
| import dev.inmo.micro_utils.repos.* | import dev.inmo.micro_utils.repos.* | ||||||
|  | import dev.inmo.micro_utils.repos.annotations.OverrideRequireManualInvalidation | ||||||
| import dev.inmo.micro_utils.repos.cache.util.ActualizeAllClearMode | import dev.inmo.micro_utils.repos.cache.util.ActualizeAllClearMode | ||||||
| import dev.inmo.micro_utils.repos.cache.util.actualizeAll | import dev.inmo.micro_utils.repos.cache.util.actualizeAll | ||||||
| import kotlinx.coroutines.CoroutineScope | import kotlinx.coroutines.CoroutineScope | ||||||
| @@ -88,6 +88,7 @@ open class DirectFullReadKeyValuesCacheRepo<Key,Value>( | |||||||
|         return result ?: emptyPaginationResult() |         return result ?: emptyPaginationResult() | ||||||
|     } |     } | ||||||
|  |  | ||||||
|  |     @OverrideRequireManualInvalidation | ||||||
|     override suspend fun invalidate() { |     override suspend fun invalidate() { | ||||||
|         actualizeAll() |         actualizeAll() | ||||||
|     } |     } | ||||||
| @@ -121,6 +122,7 @@ open class DirectFullWriteKeyValuesCacheRepo<Key,Value>( | |||||||
|         } |         } | ||||||
|     }.launchIn(scope) |     }.launchIn(scope) | ||||||
|  |  | ||||||
|  |     @OverrideRequireManualInvalidation | ||||||
|     override suspend fun invalidate() { |     override suspend fun invalidate() { | ||||||
|         locker.withWriteLock { |         locker.withWriteLock { | ||||||
|             kvCache.clear() |             kvCache.clear() | ||||||
| @@ -170,6 +172,7 @@ open class DirectFullKeyValuesCacheRepo<Key,Value>( | |||||||
|             locker.unlockWrite() |             locker.unlockWrite() | ||||||
|         } |         } | ||||||
|     } |     } | ||||||
|  |     @OverrideRequireManualInvalidation | ||||||
|     override suspend fun invalidate() { |     override suspend fun invalidate() { | ||||||
|         kvCache.actualizeAll(parentRepo, locker = locker) |         kvCache.actualizeAll(parentRepo, locker = locker) | ||||||
|     } |     } | ||||||
|   | |||||||
							
								
								
									
										54
									
								
								repos/cache/src/jvmMain/kotlin/InvalidateSynchronously.kt
									
									
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										54
									
								
								repos/cache/src/jvmMain/kotlin/InvalidateSynchronously.kt
									
									
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,54 @@ | |||||||
|  | package dev.inmo.micro_utils.repos.cache | ||||||
|  |  | ||||||
|  | import dev.inmo.kslog.common.KSLog | ||||||
|  | import dev.inmo.micro_utils.coroutines.doSynchronously | ||||||
|  | import dev.inmo.micro_utils.coroutines.runCatchingLogging | ||||||
|  | import kotlinx.coroutines.CoroutineScope | ||||||
|  |  | ||||||
|  | fun <T : InvalidatableRepo> T.alsoInvalidateSync( | ||||||
|  |     scope: CoroutineScope, | ||||||
|  |     onFailure: suspend (Throwable) -> Unit = {}, | ||||||
|  | ) = also { | ||||||
|  |     scope.doSynchronously { | ||||||
|  |         runCatching { | ||||||
|  |             invalidate() | ||||||
|  |         }.onFailure { | ||||||
|  |             onFailure(it) | ||||||
|  |         } | ||||||
|  |     } | ||||||
|  | } | ||||||
|  |  | ||||||
|  | fun <T : InvalidatableRepo> T.alsoInvalidateSync( | ||||||
|  |     onFailure: suspend (Throwable) -> Unit = {}, | ||||||
|  | ) = also { | ||||||
|  |     doSynchronously { | ||||||
|  |         runCatching { | ||||||
|  |             invalidate() | ||||||
|  |         }.onFailure { | ||||||
|  |             onFailure(it) | ||||||
|  |         } | ||||||
|  |     } | ||||||
|  | } | ||||||
|  |  | ||||||
|  | fun <T : InvalidatableRepo> T.alsoInvalidateSyncLogging( | ||||||
|  |     scope: CoroutineScope, | ||||||
|  |     errorMessageBuilder: CoroutineScope.(Throwable) -> Any = { "Something web wrong" }, | ||||||
|  |     logger: KSLog = KSLog, | ||||||
|  | ) = also { | ||||||
|  |     scope.doSynchronously { | ||||||
|  |         runCatchingLogging(errorMessageBuilder, logger) { | ||||||
|  |             invalidate() | ||||||
|  |         } | ||||||
|  |     } | ||||||
|  | } | ||||||
|  |  | ||||||
|  | fun <T : InvalidatableRepo> T.alsoInvalidateSyncLogging( | ||||||
|  |     errorMessageBuilder: CoroutineScope.(Throwable) -> Any = { "Something web wrong" }, | ||||||
|  |     logger: KSLog = KSLog, | ||||||
|  | ) = also { | ||||||
|  |     doSynchronously { | ||||||
|  |         runCatchingLogging(errorMessageBuilder, logger) { | ||||||
|  |             invalidate() | ||||||
|  |         } | ||||||
|  |     } | ||||||
|  | } | ||||||
| @@ -73,7 +73,7 @@ class KtorCRUDRepoTests : CommonCRUDRepoTests() { | |||||||
|             } |             } | ||||||
|             val server = io.ktor.server.engine.embeddedServer( |             val server = io.ktor.server.engine.embeddedServer( | ||||||
|                 CIO, |                 CIO, | ||||||
|                 34567, |                 34568, | ||||||
|                 "127.0.0.1" |                 "127.0.0.1" | ||||||
|             ) { |             ) { | ||||||
|                 install(ContentNegotiation) { |                 install(ContentNegotiation) { | ||||||
| @@ -100,7 +100,7 @@ class KtorCRUDRepoTests : CommonCRUDRepoTests() { | |||||||
|                 } |                 } | ||||||
|             } |             } | ||||||
|             val crudClient = KtorCRUDRepoClient<ComplexData, Int, SimpleData>( |             val crudClient = KtorCRUDRepoClient<ComplexData, Int, SimpleData>( | ||||||
|                 "http://127.0.0.1:34567", |                 "http://127.0.0.1:34568", | ||||||
|                 client, |                 client, | ||||||
|                 ContentType.Application.Json |                 ContentType.Application.Json | ||||||
|             ) { |             ) { | ||||||
|   | |||||||
| @@ -63,7 +63,7 @@ class KtorKeyValueRepoTests : CommonKeyValueRepoTests() { | |||||||
|             val repo = MapKeyValueRepo<Int, ComplexData>(map) |             val repo = MapKeyValueRepo<Int, ComplexData>(map) | ||||||
|             val server = io.ktor.server.engine.embeddedServer( |             val server = io.ktor.server.engine.embeddedServer( | ||||||
|                 CIO, |                 CIO, | ||||||
|                 34567, |                 34569, | ||||||
|                 "127.0.0.1" |                 "127.0.0.1" | ||||||
|             ) { |             ) { | ||||||
|                 install(ContentNegotiation) { |                 install(ContentNegotiation) { | ||||||
| @@ -91,7 +91,7 @@ class KtorKeyValueRepoTests : CommonKeyValueRepoTests() { | |||||||
|                 } |                 } | ||||||
|             } |             } | ||||||
|             val crudClient = KtorKeyValueRepoClient<Int, ComplexData>( |             val crudClient = KtorKeyValueRepoClient<Int, ComplexData>( | ||||||
|                 "http://127.0.0.1:34567", |                 "http://127.0.0.1:34569", | ||||||
|                 client, |                 client, | ||||||
|                 ContentType.Application.Json, |                 ContentType.Application.Json, | ||||||
|                 Int.serializer(), |                 Int.serializer(), | ||||||
|   | |||||||
| @@ -3,6 +3,7 @@ rootProject.name='micro_utils' | |||||||
| String[] includes = [ | String[] includes = [ | ||||||
|     ":common", |     ":common", | ||||||
|     ":common:compose", |     ":common:compose", | ||||||
|  |     ":transactions", | ||||||
|     ":matrix", |     ":matrix", | ||||||
|     ":safe_wrapper", |     ":safe_wrapper", | ||||||
|     ":crypto", |     ":crypto", | ||||||
|   | |||||||
| @@ -1,13 +1,19 @@ | |||||||
| plugins { | plugins { | ||||||
|     id "org.jetbrains.kotlin.multiplatform" |     id "org.jetbrains.kotlin.multiplatform" | ||||||
|     id "org.jetbrains.kotlin.plugin.serialization" |     id "org.jetbrains.kotlin.plugin.serialization" | ||||||
|     id "application" |  | ||||||
|     id "com.google.devtools.ksp" |     id "com.google.devtools.ksp" | ||||||
| } | } | ||||||
|  |  | ||||||
| apply from: "$mppJvmJsLinuxMingwProject" | apply from: "$mppJvmJsLinuxMingwProject" | ||||||
|  |  | ||||||
| kotlin { | kotlin { | ||||||
|  |     jvm { | ||||||
|  |         binaries { | ||||||
|  |             executable { | ||||||
|  |                 mainClass.set("dev.inmo.micro_utils.startup.launcher.MainKt") | ||||||
|  |             } | ||||||
|  |         } | ||||||
|  |     } | ||||||
|     sourceSets { |     sourceSets { | ||||||
|         commonMain { |         commonMain { | ||||||
|             dependencies { |             dependencies { | ||||||
| @@ -23,10 +29,6 @@ kotlin { | |||||||
|     } |     } | ||||||
| } | } | ||||||
|  |  | ||||||
| application { |  | ||||||
|     mainClassName = "dev.inmo.micro_utils.startup.launcher.MainKt" |  | ||||||
| } |  | ||||||
|  |  | ||||||
| java { | java { | ||||||
|     sourceCompatibility = JavaVersion.VERSION_17 |     sourceCompatibility = JavaVersion.VERSION_17 | ||||||
|     targetCompatibility = JavaVersion.VERSION_17 |     targetCompatibility = JavaVersion.VERSION_17 | ||||||
|   | |||||||
							
								
								
									
										7
									
								
								transactions/build.gradle
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										7
									
								
								transactions/build.gradle
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,7 @@ | |||||||
|  | plugins { | ||||||
|  |     id "org.jetbrains.kotlin.multiplatform" | ||||||
|  |     id "org.jetbrains.kotlin.plugin.serialization" | ||||||
|  |     id "com.android.library" | ||||||
|  | } | ||||||
|  |  | ||||||
|  | apply from: "$mppJvmJsAndroidLinuxMingwLinuxArm64Project" | ||||||
							
								
								
									
										82
									
								
								transactions/src/commonMain/kotlin/TransactionsDSL.kt
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										82
									
								
								transactions/src/commonMain/kotlin/TransactionsDSL.kt
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,82 @@ | |||||||
|  | package dev.inmo.micro_utils.transactions | ||||||
|  |  | ||||||
|  | typealias TransactionDSLRollbackLambda = suspend (Throwable) -> Unit | ||||||
|  | class TransactionsDSL internal constructor() { | ||||||
|  |     internal val rollbackActions = LinkedHashSet<TransactionDSLRollbackLambda>() | ||||||
|  |  | ||||||
|  |     internal fun addRollbackAction(rollbackAction: TransactionDSLRollbackLambda) { | ||||||
|  |         rollbackActions.add(rollbackAction) | ||||||
|  |     } | ||||||
|  | } | ||||||
|  | class RollbackContext<T> internal constructor ( | ||||||
|  |     val actionResult: T, | ||||||
|  |     val error: Throwable | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | /** | ||||||
|  |  * Calls [action] and, if it succeeded - saving [rollback] action for future usage for cases when some other | ||||||
|  |  * action or even main one throwing an error | ||||||
|  |  * | ||||||
|  |  * @param rollback Will be called if | ||||||
|  |  */ | ||||||
|  | suspend fun <T> TransactionsDSL.rollableBackOperation( | ||||||
|  |     rollback: suspend RollbackContext<T>.() -> Unit, | ||||||
|  |     action: suspend () -> T | ||||||
|  | ): T { | ||||||
|  |     return runCatching { action() } | ||||||
|  |         .onSuccess { | ||||||
|  |             addRollbackAction { e -> | ||||||
|  |                 val context = RollbackContext(it, e) | ||||||
|  |                 context.rollback() | ||||||
|  |             } | ||||||
|  |         } | ||||||
|  |         .getOrThrow() | ||||||
|  | } | ||||||
|  |  | ||||||
|  | /** | ||||||
|  |  * Starts transaction with opportunity to add actions [rollableBackOperation]. How to use: | ||||||
|  |  * | ||||||
|  |  * ```kotlin | ||||||
|  |  * doSuspendTransaction { | ||||||
|  |  *     println("start of action") | ||||||
|  |  * | ||||||
|  |  *     withRollback({ // it - result of action | ||||||
|  |  *         // some rollback action | ||||||
|  |  *     }) { | ||||||
|  |  *         // Some action with rollback | ||||||
|  |  *     } | ||||||
|  |  * | ||||||
|  |  *     withRollback({ | ||||||
|  |  *          repository.delete(it) // it - result of createSomething, if it completes successfully | ||||||
|  |  *     }) { | ||||||
|  |  *          repository.createSomething() | ||||||
|  |  *     } | ||||||
|  |  * | ||||||
|  |  *     withRollback({ | ||||||
|  |  *         // will not be triggered due to error in action | ||||||
|  |  *     }) { | ||||||
|  |  *         error("It is just a simple error") // Will trigger rolling back previously successfully completed actions | ||||||
|  |  *     } | ||||||
|  |  * } | ||||||
|  |  * ``` | ||||||
|  |  * | ||||||
|  |  * @param onRollbackStepError Will be called if rollback action throwing some error | ||||||
|  |  */ | ||||||
|  | suspend fun <T> doSuspendTransaction( | ||||||
|  |     onRollbackStepError: suspend (Throwable) -> Unit = { }, | ||||||
|  |     block: suspend TransactionsDSL.() -> T | ||||||
|  | ): Result<T> { | ||||||
|  |     val transactionsDSL = TransactionsDSL() | ||||||
|  |  | ||||||
|  |     return runCatching { | ||||||
|  |         transactionsDSL.block() | ||||||
|  |     }.onFailure { e -> | ||||||
|  |         transactionsDSL.rollbackActions.forEach { | ||||||
|  |             runCatching { | ||||||
|  |                 it.invoke(e) | ||||||
|  |             }.onFailure { ee -> | ||||||
|  |                 onRollbackStepError(ee) | ||||||
|  |             } | ||||||
|  |         } | ||||||
|  |     } | ||||||
|  | } | ||||||
							
								
								
									
										93
									
								
								transactions/src/commonTest/kotlin/TransactionsDSLTests.kt
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										93
									
								
								transactions/src/commonTest/kotlin/TransactionsDSLTests.kt
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,93 @@ | |||||||
|  | import dev.inmo.micro_utils.transactions.doSuspendTransaction | ||||||
|  | import dev.inmo.micro_utils.transactions.rollableBackOperation | ||||||
|  | import kotlinx.coroutines.test.runTest | ||||||
|  | import kotlin.test.Test | ||||||
|  | import kotlin.test.assertEquals | ||||||
|  | import kotlin.test.assertFalse | ||||||
|  | import kotlin.test.assertTrue | ||||||
|  |  | ||||||
|  | class TransactionsDSLTests { | ||||||
|  |     @Test | ||||||
|  |     fun successfulTest() = runTest { | ||||||
|  |         val dataCollections = Array(100) { | ||||||
|  |             Triple( | ||||||
|  |                 it, // expected data | ||||||
|  |                 false, // has rollback happen or not | ||||||
|  |                 -1 // actual data | ||||||
|  |             ) | ||||||
|  |         } | ||||||
|  |  | ||||||
|  |         val actionResult = doSuspendTransaction { | ||||||
|  |             dataCollections.forEachIndexed { i, _ -> | ||||||
|  |                 val resultData = rollableBackOperation({ | ||||||
|  |                     dataCollections[i] = actionResult.copy(second = true) | ||||||
|  |                 }) { | ||||||
|  |                     val result = dataCollections[i] | ||||||
|  |                     dataCollections[i] = result.copy( | ||||||
|  |                         third = i | ||||||
|  |                     ) | ||||||
|  |                     dataCollections[i] | ||||||
|  |                 } | ||||||
|  |                 assertEquals(dataCollections[i], resultData) | ||||||
|  |                 assertTrue(dataCollections[i] === resultData) | ||||||
|  |             } | ||||||
|  |             true | ||||||
|  |         }.getOrThrow() | ||||||
|  |  | ||||||
|  |         dataCollections.forEachIndexed { i, triple -> | ||||||
|  |             assertFalse(triple.second) | ||||||
|  |             assertEquals(triple.first, i) | ||||||
|  |             assertEquals(i, triple.third) | ||||||
|  |         } | ||||||
|  |         assertTrue(actionResult) | ||||||
|  |     } | ||||||
|  |     @Test | ||||||
|  |     fun fullTest() = runTest { | ||||||
|  |         val testsCount = 100 | ||||||
|  |         for (testNumber in 0 until testsCount) { | ||||||
|  |             val error = IllegalStateException("Test must fail at $testNumber") | ||||||
|  |             val dataCollections = Array(testsCount) { | ||||||
|  |                 Triple( | ||||||
|  |                     it, // expected data | ||||||
|  |                     false, // has rollback happen or not | ||||||
|  |                     -1 // actual data | ||||||
|  |                 ) | ||||||
|  |             } | ||||||
|  |  | ||||||
|  |             val actionResult = doSuspendTransaction { | ||||||
|  |                 dataCollections.forEachIndexed { i, _ -> | ||||||
|  |                     val resultData = rollableBackOperation({ | ||||||
|  |                         assertTrue(error === this.error) | ||||||
|  |                         dataCollections[i] = actionResult.copy(second = true) | ||||||
|  |                     }) { | ||||||
|  |                         if (i == testNumber) throw error | ||||||
|  |                         val result = dataCollections[i] | ||||||
|  |                         dataCollections[i] = result.copy( | ||||||
|  |                             third = i | ||||||
|  |                         ) | ||||||
|  |                         dataCollections[i] | ||||||
|  |                     } | ||||||
|  |                     assertEquals(dataCollections[i], resultData) | ||||||
|  |                     assertTrue(dataCollections[i] === resultData) | ||||||
|  |                 } | ||||||
|  |                 true | ||||||
|  |             }.getOrElse { | ||||||
|  |                 assertTrue(it === error) | ||||||
|  |                 true | ||||||
|  |             } | ||||||
|  |  | ||||||
|  |             dataCollections.forEachIndexed { i, triple -> | ||||||
|  |                 if (i < testNumber) { | ||||||
|  |                     assertTrue(triple.second) | ||||||
|  |                     assertEquals(triple.first, i) | ||||||
|  |                     assertEquals(i, triple.third) | ||||||
|  |                 } else { | ||||||
|  |                     assertFalse(triple.second) | ||||||
|  |                     assertEquals(triple.first, i) | ||||||
|  |                     assertEquals(-1, triple.third) | ||||||
|  |                 } | ||||||
|  |             } | ||||||
|  |             assertTrue(actionResult) | ||||||
|  |         } | ||||||
|  |     } | ||||||
|  | } | ||||||
		Reference in New Issue
	
	Block a user