如何在KSP中获取带注解数据类属性的默认值？

嘿，这个问题我刚好处理过！在KSP里确实没有直接给属性默认值的API，得绕个弯子从主构造函数的参数入手——因为数据类的属性默认值本质上就是主构造参数的默认值。给你一步步拆解：

核心思路 #

数据类的每个属性都对应主构造函数的一个参数，属性的默认值完全继承自对应参数的默认值。所以我们的操作路径是：ClassDeclaration → 主构造函数 → 构造参数 → 参数默认值表达式 → 解析表达式得到具体值。

具体步骤 #

1. 获取数据类的主构造函数
   从你的ClassDeclaration实例中调用primaryConstructor方法，它会返回对应的KSFunctionDeclaration?（主构造函数在KSP里也被视为一种函数声明）。

2. 遍历构造参数，解析默认值
   每个构造参数对应数据类的一个属性，调用参数的defaultValue方法会得到KSExpression?——这就是默认值的语法树节点。接下来根据表达式的类型提取具体值：

   • 布尔字面量：KSBooleanLiteralExpression，直接取value字段
   • 数字字面量：KSNumberLiteralExpression，取value字段
   • 字符串字面量：KSStringLiteralExpression，取value字段（如果需要保留引号可以自行拼接）
   • Null值：KSNullLiteralExpression，直接返回"null"

代码示例 #

// 假设你已经通过KSP拿到了目标数据类的ClassDeclaration实例
val dataClassDeclaration: ClassDeclaration = ... // 你的MyDataClass对应的实例

dataClassDeclaration.primaryConstructor?.let { constructor ->
    constructor.parameters.forEach { param ->
        val propertyName = param.simpleName.asString()
        param.defaultValue?.let { expr ->
            val defaultValue = when (expr) {
                is KSBooleanLiteralExpression -> expr.value.toString()
                is KSNumberLiteralExpression -> expr.value
                is KSStringLiteralExpression -> "\"${expr.value}\""
                is KSNullLiteralExpression -> "null"
                // 若需要处理复杂默认值（比如函数调用），可以在这里扩展分支
                else -> "无法解析的默认值类型"
            }
            println("属性 $propertyName 的默认值是：$defaultValue")
        } ?: run {
            println("属性 $propertyName 没有默认值")
        }
    }
}

注意事项 #

• 如果默认值是复杂表达式（比如mutableListOf(1,2,3)或自定义函数调用），上面的基础解析逻辑就不够用了，需要递归遍历KSExpression的子节点来解析——不过这种场景在数据类里相对少见。
• 确保只处理主构造函数的参数，因为数据类的属性默认值只和主构造参数绑定；如果是类体里手动定义的属性，默认值不会关联到构造参数，这种情况需要单独处理（但一般数据类不会这么写）。

内容的提问来源于stack exchange，提问作者Roberto Leinardi