Kotlin数据类初始化：已知非空场景下处理可空返回类型

问题场景代码 #

data class EndObject(
    val neverNullDataSet: DataSet<String>,
    val nullableDataSet: DataSet<String>?
) {
    companion object {
        fun from(startObject: StartObject): EndObject{
            return EndObject (
                neverNullDataSet = createDataSet(startObject.neverNullValue), // 编译错误：返回DataSet<String>?，但属性要求非空
                nullableDataSet = createDataSet(startObject.nullableValue) // 正常
            )
        }
        fun createDataSet(value: String?): DataSet<String>? {
            return if (value == null) null
            else DataSet(value)
        }
    }
}

data class StartObject(
    val neverNullValue: String,
    val nullableValue: String?
)

data class DataSet<T>( 
    val importantValue: T // 仅当importantValue非空时才会创建实例
)

核心问题 #

createDataSet函数接收可空String?时返回DataSet<String>?，但传入**非空的startObject.neverNullValue**时，开发者明确知道返回值不可能为null，可编译器无法自动识别这一逻辑，同时又不想修改neverNullDataSet的非空属性、不愿使用!!断言。

解决方案 #

方案1：重载createDataSet函数

新增一个接收非空String参数的重载函数，直接返回非空的DataSet<String>：

fun createDataSet(value: String): DataSet<String> {
    return DataSet(value)
}

// 原可空参数的函数保留，内部复用新重载的逻辑
fun createDataSet(value: String?): DataSet<String>? {
    return value?.let { createDataSet(it) }
}

此时调用createDataSet(startObject.neverNullValue)会自动匹配非空参数的重载，编译器能直接推断返回值为非空，完美适配neverNullDataSet的属性要求。这种方式无需依赖实验性API，代码逻辑清晰，是最稳妥的方案。

方案2：使用Kotlin契约（Contracts）

通过契约注解告诉编译器函数的输入输出关联逻辑，即当输入value非空时，返回值一定非空：

import kotlin.contracts.ExperimentalContracts
import kotlin.contracts.contract

@ExperimentalContracts
fun createDataSet(value: String?): DataSet<String>? {
    contract {
        // 声明：返回值非空时，输入value一定非空；反向推导：输入value非空时，返回值也非空
        returnsNotNull() implies (value != null)
    }
    return value?.let { DataSet(it) }
}

添加契约后，编译器能根据startObject.neverNullValue的非空性，自动推断createDataSet的返回值非空，无需修改函数结构。但注意该特性属于实验性API，需要添加@ExperimentalContracts注解，且需确保项目Kotlin版本支持该特性。

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