如何在Kotlin中实现类似C# LINQ的简洁集成查询？

当然可以！Kotlin的函数式编程特性和灵活的语法，能帮你摆脱繁琐的元组处理，写出更简洁、贴近C# LINQ风格的代码。下面分享两种实用方案：

方案1：原生嵌套flatMap写法（最直接） #

Kotlin的flatMap对应C#的SelectMany，但得益于lambda的作用域捕获特性，你不需要像C#那样创建匿名类型来传递中间变量。针对你的例子，等价的Kotlin代码可以写成：

A.flatMap { a ->
    B.flatMap { b ->
        C.map { c ->
            fun(a, b, c) // 直接使用外层作用域的a、b、c
        }
    }
}

这个写法比C#的SelectMany链式调用更易读：每一层flatMap的lambda都会自动捕获上一级的变量（比如内层的b能直接用外层的a），完全不需要显式处理元组或匿名对象。

方案2：自定义DSL模拟LINQ查询语法（最贴近风格） #

如果想要完全复刻C# LINQ的from...in...select语法，可以自定义一个简单的DSL（领域特定语言），让代码风格和原C#查询几乎一致：

首先定义DSL的构建器类：

// 单集合查询构建器
class SingleQuery<A>(private val source: Iterable<A>) {
    infix fun <B> from(other: Iterable<B>): PairQuery<A, B> {
        return PairQuery(source.flatMap { a -> other.map { b -> a to b } })
    }

    fun <R> select(transform: (A) -> R): List<R> = source.map(transform)
}

// 双集合查询构建器
class PairQuery<A, B>(private val pairs: Iterable<Pair<A, B>>) {
    infix fun <C> from(other: Iterable<C>): TripleQuery<A, B, C> {
        return TripleQuery(pairs.flatMap { (a, b) -> other.map { c -> Triple(a, b, c) } })
    }

    fun <R> select(transform: (A, B) -> R): List<R> = pairs.map { (a, b) -> transform(a, b) }
}

// 三集合查询构建器
class TripleQuery<A, B, C>(private val triples: Iterable<Triple<A, B, C>>) {
    fun <R> select(transform: (A, B, C) -> R): List<R> = triples.map { (a, b, c) -> transform(a, b, c) }
}

// 扩展函数，给集合添加查询入口
fun <A> Iterable<A>.query(): SingleQuery<A> = SingleQuery(this)

然后你就可以像写LINQ一样调用了：

A.query() from B from C select { a, b, c -> fun(a, b, c) }

这个DSL完全屏蔽了元组的处理细节，select方法直接接收对应数量的参数，写法和你原来的C#查询几乎一模一样。如果需要支持更多层级的集合嵌套，只需要继续扩展对应的查询构建器类即可。

方案选择建议 #

• 如果只是偶尔需要这类查询，方案1的原生写法足够简洁，不需要额外代码。
• 如果项目中经常用到多集合的笛卡尔积查询，方案2的DSL能大幅提升代码的可读性和一致性。

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