Update SwiftYYModelBuilder

Angelnil · Jun 12, 2023 · 2c0c56e · 2c0c56e
1 parent 7f6ecc7
commit 2c0c56e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/JSONConverter/Classes/Builder/SwiftYYModelBuilder.swift b/JSONConverter/Classes/Builder/SwiftYYModelBuilder.swift
@@ -60,9 +60,9 @@ class SwiftYYModelBuilder: BuilderProtocol {
         guard structType == .class else { fatalError() }
         if let propertiesInitText = propertiesInitText, !propertiesInitText.isEmpty {
             let tempPropertiesInitText = StringUtils.removeLastChar(StringUtils.removeLastChar(propertiesInitText))
-            return "\nclass \(clsName)\(parentClsName) {\n\(propertiesText)\n\tstatic func modelContainerPropertyGenericClass() -> [String : Any]? {\n\t\t[\(tempPropertiesInitText)]\n\t}\n}\n"
+            return "\n@objcMembers\nclass \(clsName)\(parentClsName) {\n\(propertiesText)\n\tstatic func modelContainerPropertyGenericClass() -> [String : Any]? {\n\t\t[\(tempPropertiesInitText)]\n\t}\n}\n"
         }
-        return "\nclass \(clsName)\(parentClsName) {\n\(tempPropertiesText)\n}\n"
+        return "\n@objcMembers\nclass \(clsName)\(parentClsName) {\n\(tempPropertiesText)\n}\n"
     }
 
     func fileSuffix() -> String {