如何在不阻塞的情况下过滤一个存储库返回的另一个存储库的流？

Question

我有一个服务可以获取用户的所有朋友，它应该过滤掉被用户阻止或阻止用户的用户。

这是我目前的方法。有两个存储库。

友情信息：

interface FriendshipRepository : CoroutineCrudRepository<Friendship, Long> {
  fun findByUserId(userId: Long): Flow<Friendship>
}

还有一个用于块信息：

interface BlockRepository : CoroutineCrudRepository<Block, Long> {
  fun findByUserIdAndFriendIdIn(userId: Long, friendId: Set<Long>): Flow<Block>
  fun findByUserIdInAndFriendId(userId: Set<Long>, friendId: Long): Flow<Block>
}

问题

我有两个问题。

1. 在查询blockRepository之前，我需要阻止收集朋友ID。
2. 我不知道如何过滤原始流而不阻塞blockRepository的结果，因为我需要知道所有块。

此实现有效 - 由于 asFlux 和 block()!! 操作，我觉得它看起来很奇怪，但我想不出更好的解决方案：

override suspend fun findFriendIdsByUserId(userId: Long): Flow<Long> {
  val friends = friendshipRepository.findByUserId(userId)
    .filter { it.status == FRIEND }
    .map { it.friendId }
    .asFlux()
    .collectList()
    .block()!!
    .toMutableSet()

  val userBlocks = blockRepository.findByUserIdAndFriendIdIn(userId, friends)
  val userIsBlockedBy = blockRepository.findByUserIdInAndFriendId(friends, userId)

  val blocks =  userBlocks.asFlux().mergeWith(userIsBlockedBy.asFlux())
    .map { it.friendId }
    .collectList()
    .block()!!
    .toSet()
  friends.removeAll(blocks)
  return friends.asFlow()
}

有更好的方法吗？

Answer 1

我认为至少您可以通过依赖Flow 的挂起运算符（如toSet() 而不是Flux 的block）来避免阻塞：

override suspend fun findFriendIdsByUserId(userId: Long): Flow<Long> {
  val friends = friendshipRepository.findByUserId(userId)
    .filter { it.status == FRIEND }
    .map { it.friendId }
    .toSet()

  val userBlocks = blockRepository.findByUserIdAndFriendIdIn(userId, friends)
  val userIsBlockedBy = blockRepository.findByUserIdInAndFriendId(friends, userId)

  val blocks = merge(userBlocks, userIsBlockedBy)
    .map { it.friendId }
    .toSet()

  return (friends - blocks).asFlow()
}

但是由于您需要知道blocks 的所有值才能开始过滤，并且您需要所有friends 才能知道blocks，我不确定您是否可以在不重新设计的情况下做得更好数据库。