Swift Codable - Как кодировать пользовательский массив - PullRequest
Новая
       8

Swift Codable - Как кодировать пользовательский массив

1 голос
/ 14 февраля 2020

Это мой JSON кейс 

      {
                 "image_id": 11101,
                 "image_source_id": 9,
                 "image_author": "",
                 "image_copyright": "",
                 "image_format_list": [{
                         "image_format": {
                             "image_url": "https://static.musixmatch.com/images-storage/mxmimages/1/0/1/1/1/11101_2.jpg",
                             "image_format_id": 2,
                             "width": 150,
                             "height": 150
                         }
                     },
                     {
                         "image_format": {
                             "image_url": "https://static.musixmatch.com/images-storage/mxmimages/1/0/1/1/1/11101_16.jpg",
                             "image_format_id": 16,
                             "width": 451,
                             "height": 500
                         }
                     }
                 ]
             }

Я правильно декодирую свой пользовательский объект в двух разных классах: MXMImage и MXMImageFormat. Но я не могу понять, как перекодировать мой объект, чтобы восстановить тот же JSON

Это мой код: 

    struct MXMImage : Decodable, Encodable, Equatable {
        let imageId: Int
        let imageSourceId: Int
        let imageAuthor: String?
        let imageCopyright: String?
        let imageFormatList: [MXMImageFormat]?

        enum CodingKeys: String, Swift.CodingKey {
            case imageId
            case imageSourceId
            case imageAuthor
            case imageCopyright
            case imageFormatList

            enum ImageFormatListKey: String, CodingKey {
                case imageFormat
            }
        }

        public init(from decoder: Decoder) throws {
            let container = try decoder.container(keyedBy: CodingKeys.self)
            imageId = try (container.decodeIfPresent(Int.self, forKey: .imageId) ?? 0)
            imageSourceId = try (container.decodeIfPresent(Int.self, forKey: .imageSourceId) ?? 0)
            imageAuthor = try? container.decodeIfPresent(String.self, forKey: .imageAuthor)
            imageCopyright = try? container.decodeIfPresent(String.self, forKey: .imageCopyright)

            var imagesFormatListContainer = try container.nestedUnkeyedContainer(forKey: .imageFormatList)
            var imagesList:[MXMImageFormat] = []
            while !imagesFormatListContainer.isAtEnd {
                let imageFormatContainer = try imagesFormatListContainer.nestedContainer(keyedBy: CodingKeys.ImageFormatListKey.self)
                let imageFormat = try? imageFormatContainer.decode(MXMImageFormat.self, forKey: .imageFormat)
                if let imageFormat = imageFormat {
                    imagesList.append(imageFormat)
                }
            }
            self.imageFormatList = imagesList
        }

        public func encode(to encoder: Encoder) throws {
            var container = encoder.container(keyedBy: CodingKeys.self)

            try container.encodeIfPresent(imageId, forKey: .imageId)
            try container.encodeIfPresent(imageSourceId, forKey: .imageSourceId)
            try container.encodeIfPresent(imageAuthor, forKey: .imageAuthor)
            try container.encodeIfPresent(imageCopyright, forKey: .imageCopyright)

            var imageContainer = container.nestedUnkeyedContainer(forKey: .imageFormatList)
            try imageFormatList?.forEach { imgFormat in
                var nested = imageContainer.nestedContainer(keyedBy: CodingKeys.ImageFormatListKey.self)
                let data = try imgFormat.encoded()
                try nested.encode(data, forKey: .imageFormat)

            }
        }
    }

В частности, я не знаю, как переопределите мои объекты MXMImageFormat внутри ключа image_format, а затем закодируйте пользовательский массив. Возможно ли это сделать? Заранее спасибо

Ответы [ 2 ]

2 голосов
/ 14 февраля 2020

Вместо nestedContainers вы можете декодировать / кодировать массив [[String:MXMImageFormat]] и отобразить его 

struct MXMImage : Codable, Equatable {
    let imageId: Int
    let imageSourceId: Int
    let imageAuthor: String?
    let imageCopyright: String?
    let imageFormatList: [MXMImageFormat]?

    private enum CodingKeys : String, CodingKey { case imageId,  imageSourceId,  imageAuthor, imageCopyright, imageFormatList}

    public init(from decoder: Decoder) throws {
        let container = try decoder.container(keyedBy: CodingKeys.self)
        imageId = try container.decode(Int.self, forKey: .imageId)
        imageSourceId = try container.decode(Int.self, forKey: .imageSourceId)
        imageAuthor = try container.decodeIfPresent(String.self, forKey: .imageAuthor)
        imageCopyright = try container.decodeIfPresent(String.self, forKey: .imageCopyright)
        if let imageFormatListData = try container.decodeIfPresent([[String:MXMImageFormat]].self, forKey: .imageFormatList) {
            imageFormatList = imageFormatListData.compactMap{$0["image_format"]}
        } else {
            imageFormatList = nil
        }
    }

    public func encode(to encoder: Encoder) throws {
        var container = encoder.container(keyedBy: CodingKeys.self)
        try container.encode(imageId, forKey: .imageId)
        try container.encode(imageSourceId, forKey: .imageSourceId)
        try container.encodeIfPresent(imageAuthor, forKey: .imageAuthor)
        try container.encodeIfPresent(imageCopyright, forKey: .imageCopyright)
        if let imageFormatListData = imageFormatList {
            try container.encode(imageFormatListData.map{["image_format":$0]}, forKey: .imageFormatList)
        }
    }
}

struct MXMImageFormat : Codable, Equatable {
    let imageUrl : URL
    let imageFormatId, width, height : Int
}
0 голосов
/ 14 февраля 2020

Предположим, что MXMImageFormat выглядит следующим образом: 

struct MXMImageFormat : Codable {
    let imageUrl: String
    let imageFormatId: Int
    let width: Int
    let height: Int
}

Я думаю, что вы обдумываете это. Вы можете просто сделать: 

try imageFormatList?.forEach { imgFormat in
    var nested = imageContainer.nestedContainer(keyedBy: CodingKeys.ImageFormatListKey.self)
    try nested.encode(imgFormat, forKey: .imageFormat)
}

Так как encode принимает любые Encodable, включая imgFormat. На самом деле вам не нужно сначала преобразовывать в Data (по крайней мере, именно поэтому вы, похоже, пытаетесь это сделать).

Некоторые тестовые коды: 

let json = """
{
   "image_id": 11101,
   "image_source_id": 9,
   "image_author": "",
   "image_copyright": "",
   "image_format_list": [{
           "image_format": {
               "image_url": "https://static.musixmatch.com/images-storage/mxmimages/1/0/1/1/1/11101_2.jpg",
               "image_format_id": 2,
               "width": 150,
               "height": 150
           }
       },
       {
           "image_format": {
               "image_url": "https://static.musixmatch.com/images-storage/mxmimages/1/0/1/1/1/11101_16.jpg",
               "image_format_id": 16,
               "width": 451,
               "height": 500
           }
       }
   ]
}
""".data(using: .utf8)!

let decoder = JSONDecoder()
decoder.keyDecodingStrategy = .convertFromSnakeCase
let model = try! decoder.decode(MXMImage.self, from: json)
let encoder = JSONEncoder()
encoder.keyEncodingStrategy = .convertToSnakeCase
let string = String(data: try! encoder.encode(model), encoding: .utf8)!
print(string) // this should be the same JSON as the one in the string literal

Также обратите внимание, что в encode вам не нужно использовать try? здесь: 

while !imagesFormatListContainer.isAtEnd {
    let imageFormatContainer = try imagesFormatListContainer.nestedContainer(keyedBy: CodingKeys.ImageFormatListKey.self)
    // here vvvvvvv
    let imageFormat = try? imageFormatContainer.decode(MXMImageFormat.self, forKey: .imageFormat)
    if let imageFormat = imageFormat {
        imagesList.append(imageFormat)
    }
}
...