使具有原始字符串类型的枚举可编码为 Objective-C

Question

我有一个可编码的枚举：

public enum MyEnum: String, Codable, Hashable, Sendable {
  case one = "ONE"
  case two = "TWO"

  public init(from decoder: Decoder) throws {
    let container = try decoder.singleValueContainer()
    let rawValue = try container.decode(RawValue.self)
    self = MyEnum(rawValue: rawValue) ?? .one
  }
}

但是，我现在需要使其与 objective-c 兼容。我知道我不能拥有String原始值，它必须是Int 。 我仍然需要它像以前一样兼容，因为它是从 JSON 创建的，它是一个字符串而不是一个 Int。 我该怎么做呢？

Answer 1

与其修改您的纯 Swift 类型以将其公开给 Obj-C 并因此限制您可以使用的语言功能，我建议创建一个您公开给 Obj-C 的单独桥接类型。

这种方法可以更轻松地使用 Swift 的全部功能来 model 您的数据，并且一旦您的代码库完全迁移到 Swift（或至少它的所有需要访问此的部分），也可以更轻松地删除您的 Obj-C 兼容类型类型）。

@objc public enum MyEnumBridge: Int {
  case one
  case two
}

现在向您的 Obj-C 兼容类型添加一个扩展，以从您的纯 Swift 类型初始化它：

extension MyEnumBridge {
  init(wrapped: MyEnum) {
    switch wrapped {
    case .one:
      self = .one
    case .two:
      self = .two
    }
  }
}

您可以从 Obj-C 类型的 Swift 扩展中调用 init，但仍然在 Obj-C 中存储/使用枚举。

并将 Obj-C 类型转换回 Swift：

extension MyEnumBridge {
  var wrapped: MyEnum {
    switch self {
    case .one:
      self = .one
    case .two:
      self = .two
    }
  }
}

Answer 2

好的，我想我想出了一个解决方案（如果我错了请纠正我！）

public enum MyEnum: Int, Codable, Hashable, Sendable {
  case one
  case two

  enum InternalMyEnum: String, Codable {
    case one = "ONE"
    case two = "TWO"
  }

  public func encode(to encoder: Encoder) throws {
    var container = encoder.singleValueContainer()

    switch self {
    case .one:
      try container.encode(InternalMyEnum.one.rawValue)
    case .two:
      try container.encode(InternalMyEnum.two.rawValue)
    }
  }

  public init(from decoder: Decoder) throws {
    let container = try decoder.singleValueContainer()
    let rawValue = try container.decode(String.self)
    let internalMyEnum = InternalMyEnum(rawValue: rawValue) ?? .one
    switch internalVariantType {
    case .one:
      self = .one
    case .one:
      self = .one
    }
  }
}