scala – groupBy的子流是否可以依赖于它们生成的密钥？

我想用每个用户的一个子流分离我的流,并在实现子流时为每个用户加载状态,以便可以相对于该状态处理子流的元素.

如果我不想合并下游的子流,我可以使用groupBy和Sink.lazyInit做一些事情：

def getState(userId: UserId): Future[UserState] = ...
def getUserId(element: Element): UserId = ...
def treatUser(state: UserState): Sink[Element,_] = ...

val treatByUser: Sink[Element] = Flow[Element].groupBy(
  Int.MaxValue,getUserId
).to(
  Sink.lazyInit(
    elt => getState(getUserId(elt)).map(treatUser),??? // this is never called,since the subflow is created when an element comes
  )
)

但是,如果treatUser成为Flow,则这不起作用,因为Sink.lazyInit没有等价物.

由于groupBy的子流仅在推送新元素时才具体化,因此应该可以使用此元素来实现子流,但是我无法调整groupBy的源代码以使此工作始终如一.同样,Sink.lazyInitdo似乎不容易转换为Flow案例.

有关如何解决这个问题的任何想法？