ganfra
12 months ago
committed by
GitHub
8 changed files with 124 additions and 29 deletions
@ -0,0 +1 @@
@@ -0,0 +1 @@
|
||||
Room : makes subscribeToSync/unsubscribeFromSync suspendable. |
@ -0,0 +1,84 @@
@@ -0,0 +1,84 @@
|
||||
/* |
||||
* Copyright (c) 2023 New Vector Ltd |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
* you may not use this file except in compliance with the License. |
||||
* You may obtain a copy of the License at |
||||
* |
||||
* http://www.apache.org/licenses/LICENSE-2.0 |
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
*/ |
||||
|
||||
package io.element.android.libraries.matrix.impl.room |
||||
|
||||
import io.element.android.libraries.core.coroutine.CoroutineDispatchers |
||||
import io.element.android.libraries.matrix.api.core.RoomId |
||||
import io.element.android.libraries.matrix.api.timeline.item.event.EventType |
||||
import kotlinx.coroutines.sync.Mutex |
||||
import kotlinx.coroutines.sync.withLock |
||||
import kotlinx.coroutines.withContext |
||||
import org.matrix.rustcomponents.sdk.RequiredState |
||||
import org.matrix.rustcomponents.sdk.RoomListService |
||||
import org.matrix.rustcomponents.sdk.RoomSubscription |
||||
import timber.log.Timber |
||||
|
||||
class RoomSyncSubscriber( |
||||
private val roomListService: RoomListService, |
||||
private val dispatchers: CoroutineDispatchers, |
||||
) { |
||||
|
||||
private val subscriptionCounts = HashMap<RoomId, Int>() |
||||
private val mutex = Mutex() |
||||
|
||||
private val settings = RoomSubscription( |
||||
requiredState = listOf( |
||||
RequiredState(key = EventType.STATE_ROOM_CANONICAL_ALIAS, value = ""), |
||||
RequiredState(key = EventType.STATE_ROOM_TOPIC, value = ""), |
||||
RequiredState(key = EventType.STATE_ROOM_JOIN_RULES, value = ""), |
||||
RequiredState(key = EventType.STATE_ROOM_POWER_LEVELS, value = ""), |
||||
), |
||||
timelineLimit = null |
||||
) |
||||
|
||||
suspend fun subscribe(roomId: RoomId) = mutex.withLock { |
||||
withContext(dispatchers.io) { |
||||
try { |
||||
val currentSubscription = subscriptionCounts.getOrElse(roomId) { 0 } |
||||
if (currentSubscription == 0) { |
||||
Timber.d("Subscribing to room $roomId}") |
||||
roomListService.room(roomId.value).use { roomListItem -> |
||||
roomListItem.subscribe(settings) |
||||
} |
||||
} |
||||
subscriptionCounts[roomId] = currentSubscription + 1 |
||||
} catch (exception: Exception) { |
||||
Timber.e("Failed to subscribe to room $roomId") |
||||
} |
||||
} |
||||
} |
||||
|
||||
suspend fun unsubscribe(roomId: RoomId) = mutex.withLock { |
||||
withContext(dispatchers.io) { |
||||
try { |
||||
val currentSubscription = subscriptionCounts.getOrElse(roomId) { 0 } |
||||
when (currentSubscription) { |
||||
0 -> return@withContext |
||||
1 -> { |
||||
Timber.d("Unsubscribe from room $roomId") |
||||
roomListService.room(roomId.value).use { roomListItem -> |
||||
roomListItem.unsubscribe() |
||||
} |
||||
} |
||||
} |
||||
subscriptionCounts[roomId] = currentSubscription - 1 |
||||
} catch (exception: Exception) { |
||||
Timber.e("Failed to unsubscribe from room $roomId") |
||||
} |
||||
} |
||||
} |
||||
} |
Loading…
Reference in new issue