Skip to content

Commit

Permalink
remove log
Browse files Browse the repository at this point in the history
  • Loading branch information
@lwj1994
lwj1994 committed Jul 20, 2022
1 parent 04c71b9 commit dc9520a
Show file tree
Hide file tree
Showing 2 changed files with 28 additions and 31 deletions.
25 changes: 13 additions & 12 deletions compiler/src/main/java/com/lwjlol/ormkv/annotation/compiler/KaptOrmkvProcessor.kt
View file
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@ import javax.tools.Diagnostic
/**
* @author luwenjie on 2019-08-11 19:17:41
*/
//@AutoService(value = [Process::class])
// @AutoService(value = [Process::class])
@Suppress("DEPRECATION")
@SupportedSourceVersion(value = SourceVersion.RELEASE_8)
@SupportedAnnotationTypes(value = ["com.lwjlol.ormkv.annotation.Entity", "com.lwjlol.ormkv.annotation.ColumnInfo", "com.lwjlol.ormkv.annotation.Ignore"])
Expand All @@ -35,7 +35,7 @@ class KaptOrmkvProcessor : AbstractProcessor() {

override fun process(
annotations: MutableSet<out TypeElement>,
roundEnv: RoundEnvironment,
roundEnv: RoundEnvironment
): Boolean {
val elementUtils = processingEnv.elementUtils
print("process")
Expand Down Expand Up @@ -70,9 +70,9 @@ class KaptOrmkvProcessor : AbstractProcessor() {
} else {
packageName
}
val fileName = if (entity.className.isNotEmpty()) name else "${name}_${END_FIX}"
val fileName = if (entity.className.isNotEmpty()) name else "${name}_$END_FIX"
val typeSpec = TypeSpec.objectBuilder(fileName)
.addKdoc("this class is generated by https://github.com/lwj1994/ormkv for [${packageName}.${className}], Please don't modify it!")
.addKdoc("this class is generated by https://github.com/lwj1994/ormkv for [$packageName.$className], Please don't modify it!")

typeSpec.addProperty(
PropertySpec.builder(
Expand Down Expand Up @@ -114,7 +114,6 @@ class KaptOrmkvProcessor : AbstractProcessor() {

print("propertyName = $propertyName, memberTypeName = $memberTypeName, typeName = $typeName")


val paramType = typeName.toString()
val defValue = when {
paramType.contains("String") -> if (defInitValue.isNotEmpty()) defInitValue else ""
Expand Down Expand Up @@ -153,7 +152,7 @@ class KaptOrmkvProcessor : AbstractProcessor() {
}

val keyUnitName = columnName.ifEmpty { propertyName }
val prefix = "$packageName.${className}"
val prefix = "$packageName.$className"
val keyName = if (prefixKeyWithClassName) "${prefix}_$keyUnitName" else keyUnitName

val getName = when {
Expand Down Expand Up @@ -185,7 +184,8 @@ class KaptOrmkvProcessor : AbstractProcessor() {
| $valueName = $HANDLER.$getName
|}
|return $valueName!!
|""".trimMargin()
|
""".trimMargin()
)
.build()
)
Expand All @@ -194,15 +194,16 @@ class KaptOrmkvProcessor : AbstractProcessor() {
"""
|$valueName = value
|$HANDLER.$setName
|""".trimMargin()
|
""".trimMargin()
).build()
)
.build()
)
toStringCode.append("|$propertyName = $$propertyName\n")
toModelCode.append("|$propertyName = $propertyName, \n")
updateCode.append("|$propertyName = model.$propertyName\n")
initCode.append("|$valueName = ${HANDLER}.$getName\n")
initCode.append("|$valueName = $HANDLER.$getName\n")
if (clear) {
if (typeName.toString().contains("String")) {
clearCode.append("$propertyName = \"\"\"$defValue\"\"\" \n")
Expand All @@ -218,7 +219,7 @@ class KaptOrmkvProcessor : AbstractProcessor() {
.addCode(
"""
$initCode
""".trimMargin()
""".trimMargin()
).build()
)

Expand All @@ -228,7 +229,7 @@ class KaptOrmkvProcessor : AbstractProcessor() {
.addCode(
"""
|$clearCode
""".trimMargin()
""".trimMargin()
)
.build()
).addToString("return \"\"\"$toStringCode\"\"\".trimMargin()")
Expand Down Expand Up @@ -261,7 +262,7 @@ class KaptOrmkvProcessor : AbstractProcessor() {
"""
|return $className(
${code.trim('\n').trim(' ').trim(',')})
""".trimMargin()
""".trimMargin()
)
.build()
)
Expand Down
34 changes: 15 additions & 19 deletions compiler/src/main/java/com/lwjlol/ormkv/annotation/compiler/KspOrmkvProvider.kt
View file
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,6 @@ fun OutputStream.appendText(str: String) {
try {
this.write(str.toByteArray())
} catch (e: IOException) {

}
}

Expand All @@ -37,7 +36,6 @@ class KspOrmkvProcessor(
file.appendText("$indent$s\n")
}


override fun process(resolver: Resolver): List<KSAnnotated> {
if (invoked) return emptyList()
if (LOG && !invoked) {
Expand Down Expand Up @@ -84,9 +82,9 @@ class KspOrmkvProcessor(
if (name.isEmpty()) {
name = className
}
val fileName = if (classNameArg.isNotEmpty()) name else "${name}${END_FIX}"
val fileName = if (classNameArg.isNotEmpty()) name else "${name}$END_FIX"
val typeSpec = TypeSpec.objectBuilder(fileName)
.addKdoc("this class is generated by https://github.com/lwj1994/ormkv for [${packageName}.${className}], Please don't modify it!")
.addKdoc("this class is generated by https://github.com/lwj1994/ormkv for [$packageName.$className], Please don't modify it!")

if (handlerCodeReferenceArg.isEmpty()) {
return
Expand Down Expand Up @@ -138,7 +136,7 @@ class KspOrmkvProcessor(
val keyUnitName = columnName.ifEmpty { member.name?.asString() ?: "" }
var getName = ""

val prefix = "$packageName.${className}"
val prefix = "$packageName.$className"
val keyName = if (prefixKeyArg) "${prefix}_$keyUnitName" else keyUnitName
when (memberTypeName) {
"kotlin.String" -> {
Expand Down Expand Up @@ -205,27 +203,29 @@ class KspOrmkvProcessor(
.addCode(
"""
|if ($valueName == null) {
| $valueName = ${HANDLER}.$getName
| $valueName = $HANDLER.$getName
|}
|return $valueName!!
|""".trimMargin()
|
""".trimMargin()
)
.build()
)
.setter(
FunSpec.setterBuilder().addParameter("value", typeName).addCode(
"""
|$valueName = value
|${HANDLER}.$setName
|""".trimMargin()
|$HANDLER.$setName
|
""".trimMargin()
).build()
)
.build()
)
toStringCode.append("|$propertyName = $$propertyName\n")
toModelCode.append("|$propertyName = $propertyName, \n")
updateCode.append("|$propertyName = model.$propertyName\n")
initCode.append("|$valueName = ${HANDLER}.$getName\n")
initCode.append("|$valueName = $HANDLER.$getName\n")
if (enableReset) {
if (memberTypeName.contains("String")) {
clearCode.append("$propertyName = \"\"\"$defaultValue\"\"\" \n")
Expand All @@ -240,7 +240,7 @@ class KspOrmkvProcessor(
.addCode(
"""
$initCode
""".trimMargin()
""".trimMargin()
).build()
)

Expand All @@ -250,7 +250,7 @@ class KspOrmkvProcessor(
.addCode(
"""
|$clearCode
""".trimMargin()
""".trimMargin()
)
.build()
).addToString("return \"\"\"$toStringCode\"\"\".trimMargin()")
Expand All @@ -264,7 +264,6 @@ class KspOrmkvProcessor(
file.writeTo(codeGenerator = codeGenerator, dependencies = Dependencies(false, source))
}


private fun TypeSpec.Builder.addToString(code: String): TypeSpec.Builder {
return addFunction(
FunSpec.builder("toString")
Expand All @@ -283,7 +282,7 @@ class KspOrmkvProcessor(
"""
|return $className(
${code.trim('\n').trim(' ').trim(',')})
""".trimMargin()
""".trimMargin()
)
.build()
)
Expand Down Expand Up @@ -325,7 +324,6 @@ class KspOrmkvProcessor(
)
}


generateClass(
className = className,
packageName = classDeclaration.packageName.asString(),
Expand All @@ -345,10 +343,8 @@ class KspOrmkvProcessor(
}
}


class KspOrmkvProvider : SymbolProcessorProvider {


companion object {
}

Expand All @@ -358,11 +354,11 @@ class KspOrmkvProvider : SymbolProcessorProvider {
}

private const val END_FIX = "Registry"
private const val LOG = true
private const val LOG = false
private const val HANDLER = "kvHandler"
private const val COROUTINE_HANDLER = "coroutineKvHandler"
private const val COROUTINE_Scpoe = "coroutineKvScope"
private const val TAG = "KspOrmkv"
private const val DEFAULT_VALUE = "defaultValue"
private const val NAME = "name"
private const val ENABLE_RESET = "enableReset"
private const val ENABLE_RESET = "enableReset"

0 comments on commit dc9520a

Please sign in to comment.