From 34a28ec2d324f3fa0e25df11b7e11e6535a67e69 Mon Sep 17 00:00:00 2001 From: Jan Sigrist Date: Wed, 8 Jan 2025 15:09:10 +0100 Subject: [PATCH] specify type and renaming --- .../crystalprocessor/processing/model/ModelWorker.kt | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/crystal-map-processor/src/main/java/com/schwarz/crystalprocessor/processing/model/ModelWorker.kt b/crystal-map-processor/src/main/java/com/schwarz/crystalprocessor/processing/model/ModelWorker.kt index 19c4799..5eebfb4 100644 --- a/crystal-map-processor/src/main/java/com/schwarz/crystalprocessor/processing/model/ModelWorker.kt +++ b/crystal-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 process( - models: List, - generate: (T) -> FileSpec - ) { - for (model in models) { + private fun process(schemaModels: List, generate: (SchemaClassHolder) -> FileSpec) { + for (model in schemaModels) { documentationGenerator?.addEntitySegments(model) schemaGenerator?.addEntity(model) entityRelationshipGenerator?.addEntityNodes(model)