Как вручную реализовать Codable для нескольких структур по типу? - PullRequest
Новая
       7

Как вручную реализовать Codable для нескольких структур по типу?

0 голосов
/ 14 сентября 2018

Рассмотрим следующий json: 

{
    "from": "Guille",
    "text": "Look what I just found!",
    "attachments": [
        {
            "type": "image",
            "payload": {
                "url": "http://via.placeholder.com/640x480",
                "width": 640,
                "height": 480
            }
        },
        {
            "type": "audio",
            "payload": {
                "title": "Never Gonna Give You Up",
                "url": "https://audio.com/NeverGonnaGiveYouUp.mp3",
                "shouldAutoplay": true,
            }
        }
    ]
}

И следующую структуру Swift: 

struct ImageAttachment: Codable {
    let url: URL
    let width: Int
    let height: Int
}
struct AudioAttachment: Codable {
    let title: String
    let url: URL
    let shouldAutoplay: Bool
}

enum Attachment {
  case image(ImageAttachment)
  case audio(AudioAttachment)
  case unsupported
}

extension Attachment: Codable {
  private enum CodingKeys: String, CodingKey {
    case type
    case payload
  }
  init(from decoder: Decoder) throws {
    let container = try decoder.container(keyedBy: CodingKeys.self)
    let type = try container.decode(String.self, forKey: .type)
    switch type {
    case "image":
      let payload = try container.decode(ImageAttachment.self, forKey: .payload)
      self = .image(payload)
    case "audio":
      let payload = try container.decode(AudioAttachment.self, forKey: .payload)
      self = .audio(payload)
    default:
      self = .unsupported
    }
  }
  ...
}

Как мне поступить с аналогичным сценарием использования, если бы ключевые параметры полезной нагрузки былиflat (он же «без полезной нагрузки») вроде: 

{
  "type": "image",
  "url": "http://via.placeholder.com/640x480",
  "width": 640,
  "height": 480
}

{
  "type": "audio",
  "title": "Never Gonna Give You Up",
  "url": "https://audio.com/NeverGonnaGiveYouUp.mp3",
  "shouldAutoplay": true,
 }

Я не могу понять, как правильно реализовать init-декодер для плоского случая, так как при сохранении структур Attachment и учете будущей гибкости (добавлениябольше типов вложений).

1 Ответ

0 голосов
/ 15 сентября 2018

Вам нужно только сделать небольшое изменение 

extension Attachment: Codable {
    private enum CodingKeys: String, CodingKey {
        case type
        case payload
    }

    init(from decoder: Decoder) throws {
        let container = try decoder.container(keyedBy: CodingKeys.self)
        let type = try container.decode(String.self, forKey: .type)

        // The attachment data is nested if it has the `payload` key
        let isNested = container.allKeys.contains(.payload)

        switch type {
        case "image":
            // If the attachment data is nested inside the `payload` property, decode
            // it from that property. Otherwise, decode it from the current decoder
            let payload = try isNested ? container.decode(ImageAttachment.self, forKey: .payload) : ImageAttachment(from: decoder)
            self = .image(payload)
        case "audio":
            // Same as image attachment above
            let payload = try isNested ? container.decode(AudioAttachment.self, forKey: .payload) : AudioAttachment(from: decoder)
            self = .audio(payload)
        default:
            self = .unsupported
        }
    }
}
...