Combine自定义操作符：赋予Publisher和Subscriber更多灵活性

IOS

2024-01-21 19:03:57

在了解Combine自定义操作符之前，我们有必要回顾一下Combine框架的基本概念。

Combine简介

Combine是一个发布-订阅框架，允许开发人员在Swift中创建反应式数据流。它建立在以下关键概念之上：

Publisher： 数据源，发出一个值流。
Subscriber： 接收并处理Publisher发出的值的实体。
Operator： 修改和转换Publisher发出的值的中间步骤。

自定义操作符

自定义操作符允许开发人员创建自己的操作符，从而扩展Combine的功能。这对于处理复杂的数据转换或需要对Publisher和Subscriber的行为进行细粒度控制的情况特别有用。

Combine之Operator

Combine中内置了丰富的操作符，用于执行各种数据处理操作，例如映射、过滤和聚合。然而，对于那些需要更高级别定制的情况，自定义操作符就派上了用场。

自定义操作符是通过继承Operator基类创建的。该基类提供了执行操作所需的基本框架，包括：

init(subscriber:)： 初始化操作符并接收一个Subscriber实例。
receive(subscription:)： 当订阅发生时调用。
receive(_ input:)： 每次Publisher发出值时调用。
receive(completion:)： 当Publisher完成或失败时调用。

自定义Operator示例：mapMany()

为了展示自定义操作符如何工作，我们来创建一个mapMany()操作符。该操作符将Publisher发出的每个集合元素映射到一个新的集合。

struct MapManyOperator<Input, Output>: Operator {
  typealias Failure = Error

  let transform: (Input) -> [Output]

  init(transform: @escaping (Input) -> [Output]) {
    self.transform = transform
  }

  func makeSubscriber(context: inout Context<Input, Output, Failure>) -> AnyCancellable {
    return context.upstream.subscribe(AnySubscriber(receiveSubscription: { subscription in
      context.subscription = subscription
      subscription.request(.unlimited)
    }, receiveValue: { input in
      let outputs = self.transform(input)
      for output in outputs {
        context.publish(output)
      }
    }, receiveCompletion: { completion in
      context.subscription = nil
      context.finish(completion: completion)
    }))
  }
}

extension Publisher {
  func mapMany<Output>(_ transform: @escaping (Input) -> [Output]) -> Publishers.MapManyOperator<Input, Output> {
    return Publishers.MapManyOperator(transform: transform).eraseToAnyPublisher()
  }
}

使用mapMany()操作符

要使用mapMany()操作符，我们首先需要创建一个Publisher，发出一个集合元素的流。然后，我们可以使用mapMany()操作符来映射每个集合元素到一个新的集合。

let publisher = Just([1, 2, 3])

let mappedPublisher = publisher
  .mapMany { $0.map { $0 * 2 } }

mappedPublisher
  .sink(receiveCompletion: { completion in
    print("Completion: \(completion)")
  }, receiveValue: { value in
    print("Value: \(value)")
  })