Add a static TYPE variable to java models

pinterest · May 11, 2019 · 47a9616 · 47a9616
1 parent d467436
commit 47a9616
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/Sources/Core/JavaModelRenderer.swift b/Sources/Core/JavaModelRenderer.swift
@@ -30,6 +30,10 @@ public struct JavaModelRenderer: JavaFileRenderer {
         }
     }
 
+    func renderStaticTypeString() -> JavaIR.Property {
+        return JavaIR.Property(annotations: [], modifiers: [.public, .static, .final], type: "String", name: "TYPE", initialValue: "\"" + className.lowercased() + "\"")
+    }
+
     func renderModelHashCode() -> JavaIR.Method {
         let bodyHashCode = transitiveProperties.map { param, _ in
             Languages.java.snakeCaseToPropertyName(param)
@@ -73,7 +77,7 @@ public struct JavaModelRenderer: JavaFileRenderer {
             index += 1
         }
 
-        return [props, bitmasks, [bits]]
+        return [[renderStaticTypeString()], props, bitmasks, [bits]]
     }
 
     func propertyGetterForParam(param: String, schemaObj: SchemaObjectProperty) -> JavaIR.Method {