11package com.querydsl.ksp.codegen
22
33import com.google.devtools.ksp.processing.CodeGenerator
4- import com.google.devtools.ksp.processing.Dependencies
54import com.google.devtools.ksp.processing.Resolver
65import com.google.devtools.ksp.processing.SymbolProcessor
76import com.google.devtools.ksp.symbol.*
@@ -13,10 +12,8 @@ class QueryDslProcessor(
1312 private val codeGenerator : CodeGenerator
1413) : SymbolProcessor {
1514 val typeProcessor = QueryModelExtractor (settings)
16- lateinit var typeResolver: Resolver
1715
1816 override fun process (resolver : Resolver ): List <KSAnnotated > {
19- typeResolver = resolver
2017 if (settings.enable) {
2118 QueryModelType .entries.forEach { type ->
2219 resolver.getSymbolsWithAnnotation(type.associatedAnnotation)
@@ -32,22 +29,14 @@ class QueryDslProcessor(
3229 val models = typeProcessor.process()
3330 models.forEach { model ->
3431 val typeSpec = QueryModelRenderer .render(model)
35-
36- val sources = if (model.originatingFile != null ) {
37- arrayOf(model.originatingFile)
38- } else if (this ::typeResolver.isInitialized) {
39- typeResolver.getAllFiles().toList().toTypedArray()
40- } else {
41- emptyArray()
42- }
43-
4432 FileSpec .builder(model.className)
4533 .indent(settings.indent)
4634 .addType(typeSpec)
4735 .build()
4836 .writeTo(
49- codeGenerator,
50- Dependencies (false , * sources)
37+ codeGenerator = codeGenerator,
38+ aggregating = false ,
39+ originatingKSFiles = listOfNotNull(model.originatingFile)
5140 )
5241 }
5342 }
0 commit comments