manually modify codegenerated models

Author: lawmicha

AmplifyPlugins/DataStore/Tests/DataStoreHostApp/AWSDataStorePluginLazyLoadTests/Models/Comment4V2.swift

@@ -3,30 +3,53 @@ import Amplify
 import Foundation
 
 public struct Comment4V2: Model {
-  public let id: String
-  public var content: String
-  public var post: Post4V2?
-  public var createdAt: Temporal.DateTime?
-  public var updatedAt: Temporal.DateTime?
-  
-  public init(id: String = UUID().uuidString,
-      content: String,
-      post: Post4V2? = nil) {
-    self.init(id: id,
-      content: content,
-      post: post,
-      createdAt: nil,
-      updatedAt: nil)
-  }
-  internal init(id: String = UUID().uuidString,
-      content: String,
-      post: Post4V2? = nil,
-      createdAt: Temporal.DateTime? = nil,
-      updatedAt: Temporal.DateTime? = nil) {
-      self.id = id
-      self.content = content
-      self.post = post
-      self.createdAt = createdAt
-      self.updatedAt = updatedAt
-  }
-}
+    public let id: String
+    public var content: String
+    internal var _post: LazyModel<Post4V2>
+    public var post: Post4V2? {
+        get async throws {
+            try await _post.get()
+        }
+    }
+    public var createdAt: Temporal.DateTime?
+    public var updatedAt: Temporal.DateTime?
+    
+    public init(id: String = UUID().uuidString,
+                content: String,
+                post: Post4V2? = nil) {
+        self.init(id: id,
+                  content: content,
+                  post: post,
+                  createdAt: nil,
+                  updatedAt: nil)
+    }
+    internal init(id: String = UUID().uuidString,
+                  content: String,
+                  post: Post4V2? = nil,
+                  createdAt: Temporal.DateTime? = nil,
+                  updatedAt: Temporal.DateTime? = nil) {
+        self.id = id
+        self.content = content
+        self._post = LazyModel(element: post)
+        self.createdAt = createdAt
+        self.updatedAt = updatedAt
+    }
+    
+    public mutating func setPost(_ post: Post4V2) {
+        self._post = LazyModel(element: post)
+    }
+    
+    public init(from decoder: Decoder) throws {
+        let values = try decoder.container(keyedBy: CodingKeys.self)
+        id = try values.decode(String.self, forKey: .id)
+        content = try values.decode(String.self, forKey: .content)
+        _post = try values.decode(LazyModel<Post4V2>.self, forKey: .post)
+    }
+    
+    public func encode(to encoder: Encoder) throws {
+        var container = encoder.container(keyedBy: CodingKeys.self)
+        try container.encode(id, forKey: .id)
+        try container.encode(content, forKey: .content)
+        try container.encode(_post, forKey: .post)
+    }
+}