如何使用 Kotlin 符号处理 （KSP） 填充现有列表

Question

我最近一直在玩KSP，并设法开发了有趣的功能（例如基于布局ID的自动回收器视图和视图持有人生成器），到目前为止一切都很好。

但是，我现在正在尝试做一些不同于根据我设计的注释创建文件的事情。与其创建文件，我只想用我注释的类/对象填充一个列表。

示例：

ClassA.kt

@MyAnnotation
class ClassA(context: Context): SomeBaseClass(context) {
    override fun baseClassFunction() {
        // custom code goes here
    }
}

ClassB.kt代码

@MyAnnotation
class ClassB(context: Context): SomeBaseClass(context) {
    override fun baseClassFunction() {
        // custom code goes here
    }
}

myListGregator.kt

object MyListAgregator {
    const val classList: List<SomeBaseClass> = mutableListOf()
}

我的自定义KSP将执行以下操作

1. 收集所有由my@MyAnnoection
注释的类/对象（通常）
2. 使用适当的参数创建它们的实例（在本例中只是一个上下文）
3. 将每一个添加到MyListAssocigator
中的 class List

我总能找到类集合的要点(步骤1)，但其余的对我来说完全是个谜，感觉KSP总是期望创建代码，而不是执行它？也许我对后者的理解是不正确的，但是我在可用的文档中找不到任何相关的解释。

我考虑过的另一种选择，但出于对任何负面影响（例如性能）的考虑而宁愿避免，那就是直接修改MyListAssocigator文件以包含列表中的所有实例，就好像我自己写的一样。但是，如果可能的话，我仍然更喜欢使用前面的选项。

Answer 1

首先，您需要建立一个规则集，该规则集将应用于使用您的注释（KSP词汇表中的符号）注释的类。例如，它们必须包含一个参数，该参数必须是成员属性并且类型为Context并且必须是的子类一些BaseClass。我建议首先查找正确的继承，然后查找参数计数和类型。

您仍在阅读和探索所有带有此符号的文件。基于这个规则集进行过滤，您将在第2点获得一组类。

在这里，KSP可以为您提供生成代码的接口。然而，KSP不会让你编辑源文件，而是根据你的条件生成一个新的。在这里，您必须通过访问重写的函数来编写它的实现

您可以保留输出(在步骤2中新生成的类)并生成您的< code>MyListAggregator对象。