specify type and renaming

SchwarzIT · Jan 8, 2025 · 34a28ec · 34a28ec
1 parent 1ea8c29
commit 34a28ec
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/...-map-processor/src/main/java/com/schwarz/crystalprocessor/processing/model/ModelWorker.kt b/...-map-processor/src/main/java/com/schwarz/crystalprocessor/processing/model/ModelWorker.kt
@@ -20,6 +20,7 @@ import com.schwarz.crystalprocessor.generation.model.TypeConverterObjectGenerati
 import com.schwarz.crystalprocessor.generation.model.WrapperGeneration
 import com.schwarz.crystalprocessor.meta.SchemaGenerator
 import com.schwarz.crystalprocessor.model.entity.BaseEntityHolder
+import com.schwarz.crystalprocessor.model.entity.SchemaClassHolder
 import com.schwarz.crystalprocessor.model.typeconverter.TypeConverterHolderForEntityGeneration
 import com.schwarz.crystalprocessor.processing.Worker
 import com.squareup.kotlinpoet.FileSpec
@@ -87,11 +88,8 @@ class ModelWorker(override val logger: Logger, override val codeGenerator: CodeG
         schemaGenerator?.generate()
     }
 
-    private fun <T : BaseEntityHolder> process(
-        models: List<T>,
-        generate: (T) -> FileSpec
-    ) {
-        for (model in models) {
+    private fun process(schemaModels: List<SchemaClassHolder>, generate: (SchemaClassHolder) -> FileSpec) {
+        for (model in schemaModels) {
             documentationGenerator?.addEntitySegments(model)
             schemaGenerator?.addEntity(model)
             entityRelationshipGenerator?.addEntityNodes(model)