C# 函数式编程:LINQ
一直以来,我以为 LINQ 是专门用来对不同数据源进行查询的工具,直到我看了这篇十多年前的文章,才发现 LINQ 的功能远不止 Query。这篇文章的内容比较高级,主要写了用 C# 3.0 推出的 LINQ 语法实现了一套“解析器组合子(Parser Combinator)”的过程。那么这个组合子是用来干什么的呢?简单来说,就是把一个个小型的语法解析器组装成一个大的语法解析器。当然了,我本身水平有限,暂时还写不出来这么高级的代码,不过这篇文章中的一段话引起了我的注意:
Any type which implements Select, SelectMany and Where methods supports (part of) the "query pattern" which means we can write C#3.0 queries including multiple froms, an optional where clause and a select clause to process objects of this type.
大意就是,任何实现了 Select
,SelectMany
等方法的类型,都是支持类似于 from x in y select x.z
这样的 LINQ 语法的。比如说,如果我们为 Task
类型实现了上面提到的两个方法,那么我们就可以不借助 async/await
来对 Task 进行操作:
// 请在 Xamarin WorkBook 中执行
var taskA = Task.FromResult(12);
var taskB = Task.FromResult(12);
// 使用 async/await 计算 taskA 跟 taskB 的和
var a = await taskA;
var b = await taskB;
var r = a + b;
// 如果为 Task 实现了 LINQ 拓展方法,就可以这么写:
var r = from a in taskA
from b in taskB
select a + b;
那么我们就来看看如何实现一个非常简单的 LINQ to Task 吧。
LINQ to Task
首先我们要定义一个 Select
拓展方法,用来实现通过一个 Func<TValue, TResult>
将 Task<TValue>
转换成 Task<TResult>
的功能。
static async Task<TR> Select<TV,TR>(this Task<TV> task, Func<TV, TR> selector) {
var value = await task; // 取出 task 中的值
return selector(value); // 使用 selector 对取出的值进行变换
}
这个函数非常简单,甚至可以简化为一行代码,不过仅仅这是这样就可以让我们写出一个非常简单的 LINQ 语句了:
var taskA = Task.FromResult(12);
var r = from a in taskA select a * a;
那么实际上 C# 编译器是如何工作的呢?我们可以借助下面这个有趣的函数来一探究竟:
void PrintExpr<T1,T2>(Expression<Func<T1, T2>> expr) {
Console.WriteLine(expr.ToString());
}
熟悉 LINQ 的人肯定对 Expression 不陌生,Expressing 给了我们在运行时解析代码结构的能力。在 C# 里面,我们可以非常轻松地把一个 Lambda 转换成一个 Expression,然后调用转换后的 Expression 对象的 ToString()
方法,我们就可以在运行时以字符串的形式获取到 Lambda 的源码。例如:
var taskA = Task.FromResult(12);
PrintExpr((int _) => from a in taskA select a * a);
// 输出: _ => taskA.Select(a => (a * a))
可以看到,Expression 把这段 LINQ 的真面目给我们揭示出来了。那么,更加复杂一点的 LINQ 呢?
var taskA = Task.FromResult(12);
var taskB = Task.FromResult(12);
PrintExpr((int _) =>
from a in taskA
from b in taskB
select a * b
);
如果你尝试运行这段代码,你应该会遇到一个错误——缺少对应的 SelectMany
方法,下面给出的就是这个 SelectMany
方法的实现:
static async Task<TR> SelectMany<TV, TS, TR>(this Task<TV> task, Func<TV, Task<TS>> selector, Func<TV,TS, TR> projector){
var value = await task;
var selected = await selector(value);
return projector(value, selected);
}
这个 SelectMany
实现的功能就是,通过一个 Func<TValue, Task<TResult>>
将 Task<TValue>
转换成 Task<TResult>
。有了这个之后,你就可以看到上面的那个较为复杂的 LINQ to Task 语句编译后的结果:
_ => taskA.SelectMany(a => taskB, (a, b) => (a * b))
可以看到,当出现了两个 Task 之后,LINQ 就会使用 SelectMany
来代替 Select
。可是我想为什么 LINQ 不像之前那样,用两个 Select
分别处理两个 Task 呢?为了弄清楚这个问题,我试着推导了一番:
// 首先简单粗暴的用两个 Select 来实现这个功能
Task<Task<int>> r = taskA.Select(a => b.Select(b => a + b));
// r 被包裹了两层 Task,我们可以用 SelectMany 来去掉一层 Task 包装
// 这时 TValue 是 Task<int>, TResult 是 int
//
// 那么 Task<Task<int>>
// 将通过 Func<Task<int>, Task<int>>
// 转换成 Task<int>
Task<int> result = r.SelectMany(x => x, (_, x) => x);
结果比 LINQ 还多调用了两次 Select
。仔细看的话,就会发现,我们所写的第二个 Select
其实就是 SelectMany
,的第二个参数,而对于第一个 Select
来说,因为 b 是一个 Task,所以 b.Select(xxx)
的返回值肯定是一个 Task,而这又恰好符合 SelectMany
函数的第一个参数的特征。
有了上面的经验,我们不难推断出,当 from x in y
语句的个数超过 2 个的时候,LINQ 仍然会只使用 SelectMany
来进行翻译。因为 SelectMany
可以被看作为把两层 Task 转换成单层 Task,例如:
var taskA = Task.FromResult(12);
var taskB = Task.FromResult(12);
var taskC = Task.FromResult(12);
PrintExpr((int _) =>
from a in taskA
from b in taskB
from c in taskC
select a * b + c
);
// 我的推断:
var r = taskA.SelectMany(a => taskB, (a, b) => new {a, b}).SelectMany(temp => taskC, (temp, c) => temp.a * temp.b + c);
// 实际的输出:
// _ => taskA.SelectMany(a => taskB, (a, b) => new <>f__AnonymousType0#1`2(a = a, b = b)).SelectMany(<>h__TransparentIdentifier0 => taskC, (<>h__TransparentIdentifier0, c) => ((<>h__TransparentIdentifier0.a * <>h__TransparentIdentifier0.b) + c))
这里 LINQ 为第一个 SelectMany
的结果生成了一个匿名的中间类型,将 taskA 跟 taskB 的结果组合成了 Task<{a, b}>,方便在第二个 SelectMany
中使用。
至此,一个非常简单的 LINQ to Task 就完成了,通过这个小工具,我们可以实现不使用 async/await
就对类型进行操作。然而这并没有什么卵用,因为 async/await
确实要比 from x in y
这种语法要来的更加简单。不过举一反三,我们可以根据上面的经验来实现一个更加使用的小功能。
LINQ to Result
在一些比较函数式的语言(如 F#,Rust)中,会使用一种叫做 Result<TValue, TError>
的类型来进行异常处理。这个类型通常用来描述一个操作结果以及错误信息,帮助我们远离 Exception 的同时,还能保证我们全面的处理可能出现的错误。如果使用 C# 实现的话,一个 Result 类型可以被这么来定义:
class Result<TValue, TError>
{
public TValue Value {get; private set;}
public TError ErrorMsg {get; private set;}
public bool IsSuccess {get; private set;}
public override string ToString()
{
if(this.IsSuccess)
return "Success: " + Value.ToString();
return "Error: " + ErrorMsg.ToString();
}
public static Result<TValue, TError> OK(TValue value)
{
return new Result<TValue, TError> {Value = value, ErrorMsg = default(TError), IsSuccess = true};
}
public static Result<TValue, TError> Error(TError error)
{
return new Result<TValue, TError> {Value = default(TValue), ErrorMsg = error, IsSuccess = false};
}
}
接着仿照上面为 Task 定义 LINQ 拓展方法,为了 Result 设计 Select
跟 SelectMany
:
static Result<TR, TE> Select<TV,TR, TE>(this Result<TV, TE> result, Func<TV, TR> selector) =>
result.IsSuccess
? Result<TR, TE>.OK(selector(result.Value))
: Result<TR, TE>.Error(result.ErrorMsg);
static Result<TR, TE> SelectMany<TV, TS, TR, TE>(this Result<TV, TE> result, Func<TV, Result<TS, TE>> selector, Func<TV, TS, TR> projector){
if (result.IsSuccess)
{
var tempResult = selector(result.Value);
if (tempResult.IsSuccess)
{
return Result<TR, TE>.OK(projector(tempResult.Value, tempResult.Value));
}
return Result<TR, TE>.Error(tempResult.ErrorMsg);
}
return Result<TR, TE>.Error(result.ErrorMsg);
}
那么 LINQ to Result 在实际中的应用是什么样子的呢,接下来我用一个小例子来说明:
某公司为感谢广大新老用户对 “5 元 30 M”流量包的支持,准备给余额在 350 元用户的以上的用户送 10% 话费。但是呢,如果用户在收到赠送的话费后余额会超出 600 元,就不送话费了。
using Money = Result<double, string>;
// 查找指定 Id 的用户是否存在
Result<int, string> GetUserById(int id)
{
if(id % 7 == 0)
{
// 正常的用户
return Result<int,string>.OK(id);
}
if(id % 2 == 0)
{
return Result<int, string>.Error("用户已被冻结");
}
return Result<int, string>.Error("用户不存在");
}
// 查找指定用户的余额
Money GetMoneyFromUser(int id)
{
if (id >= 35)
{
return Money.OK(id * 10);
}
return Money.Error("穷逼用户不参与这次活动");
}
// 给用户转账
Money Transfer(double money, double amount)
{
return from canTransfer in CheckForTransfer(money, amount)
select canTransfer ? money + amount : money;
}
// 检查用户是否满足转账条件,如果转账后的余额超过了 600 元,则终止转账
Result<bool, string> CheckForTransfer(double a, double b)
{
if (a + b >= 600) {
return Result<bool,string>.Error("超出余额限制");
}
return Result<bool,string>.OK(true);
}
Money SendGift(int userId)
{
return // 查询用户信息
from user in GetUserById(userId)
// 获取该用户的余额
from money in GetMoneyFromUser(user)
// 给这个用户转账
from transfer in Transfer(money, money * 0.1)
// 获取结果
select transfer;
}
SendGift(42)
// Success: 462
SendGift(56)
// Error: 超出余额限制
SendGift(1)
// Error: 用户不存在
SendGift(14)
// Error: 穷逼用户不参与这次活动
SendGift(16)
// Error: 用户已被冻结
可以看到,使用 Result 能够让我们更加清晰地用代码描述业务逻辑,而且如果我们需要向现有流程中添加新的验证逻辑,只需要在合适地地方插入 from result in validate(xxx)
就可以了,换句话说,我们的代码变得更加“声明式”了。
函数式编程
细心的你可能已经发现了,不管是 LINQ to Task 还是 LINQ to Result,我们都使用了某种特殊的类型(如:Task,Result)对值进行了包装,然后编写了特定的拓展方法 —— SelectMany
,为这种类型定义了一个重要的基本操作。在函数式编程的里面,我们把这种特殊的类型统称为“Monad”,所谓“Monad”,不过是自函子范畴上的半幺群而已。
范畴(Category)与函子(Functor)
在高中数学,我们学习了一个概念——集合,这是范畴的一种。
对于我们程序员来说,int
类型的全部实例构成了一个集合(范畴),如果我们为其定义了一些函数,而且它们之间的复合运算满足结合律的话,我们就可以把这种函数叫做 int
类型范畴上的“态射”,态射讲的是范畴内部元素间的映射关系,例如:
// f(x) = x * 2
Func<int, int> f = (int x) => x * 2;
// g(x) = x + 1
Func<int, int> g = (int x) => x + 1;
// h(x) = x + 10
Func<int, int> h = (int x) => x + 10;
// 将函数 g 与 f 复合,(g ∘ f)(x) = g(f(x))
Func<X, Z> Compose<X, Y, Z>(Func<Y, Z> g, Func<X, Y> f) => (X x) => g(f(x));
Compose(h, Compose(g, f))(42) == Compose(Compose(h, g), f)(42)
// true
f
,g
,h
都是 int
类型范畴上的态射,因为函数的复合运算是满足结合律的。
我们还可以定义一种范畴间进行元素映射的函数,例如:
Func<int, double> ToDouble = x => Convert.ToDouble(x);
这里的函数 Select
实现了 int
范畴到 double
范畴的一个映射,不过光映射元素是不够的,要是有一种方法能够帮我们把 int
中的态射(f
,g
,h
),映射到 double
范畴中,那该多好。那么下面的函数 F
就帮助我们实现了这了功能。
// 为了方便使用 Compose 进行演示,故定义了一个比较函数式的 ToInt 函数
Func<double, int> ToInt = x => Convert.ToInt32(x);
// 一个将 int -> int 转换为 double -> double 的函数
Func<double, double> F(Func<int, int> selector) => x => Compose(Compose(ToDouble, selector), ToInt)(x);
// 在范畴间映射 f
var Ff = F(f);
Ff(42.0);
// 84.00
// 在范畴间映射 g
var Fg = F(g);
Fg(42.0);
// 43.00
// 在范畴间映射 h
var Fh = F(h);
Fh(42.0);
// 52.00
// Ff, Fg, Fh 之间仍然保持结合律,因为他们是 `double` 范畴上的态射
Compose(Fh, Compose(Fg, Ff))(42) == Compose(Compose(Fh, Fg), Ff)(42)
因为 F
能够将一个范畴内的态射映射为另一个范畴内的态射,ToDouble
可以将一个范畴内的元素映射为另一个范畴内的元素,所以,我们可以把 F
与 ToDouble
的组合称作“函子”。函子体现了两个范畴间元素的抽象结构上的相似性。
相信看到这里你应该对范畴跟函子这两个概念有了一定的了解,现在让我们更进一步,看看 C# 中泛型与范畴之间的关系。
类型与范畴
在之前,我们是以数值为基础来理解范畴这个概念的,那么现在我们从类型的层面来理解范畴。
泛型是我们非常熟悉的 C# 语言特性了,泛型类型与普通类型不一样,泛型类型可以接受一个类型参数,看起来就像是类型的函数。我们把接受函数作为参数的函数称为高阶函数,依此类推,我们就把接受类型作为参数的类型叫做高阶类型吧。这样,我们就可以从这个层面把 C# 的类型分为两类:普通类型(非泛型)和高阶类型(泛型)。
前面的例子中,我列出的 f
,g
,h
能够完成 int -> int
的转换,因为它们是 int
范畴内的态射。而 ToDouble
能够完成 int -> double
的转换,那我们就可以将他看作是普通类型范畴的态射,类似的,我们还可以定义出 ToInt32
,ToString
这样的函数,它们都能完成两个普通类型之间的转换,所以也都可以看作是普通类型范畴的态射。
那么对于高阶类型(也就是泛型)范畴来说,是不是也存在态射这样的东西呢?答案是肯定的,举个例子,用 LINQ 把 List<int>
转换成 List<double>
:
Func<List<int>, List<double>> ToDoubleList = x => x.Select(ToDouble).ToList();
不难发现,这里的 ToDoubleList
是 List<T>
类型范畴内的一个态射。不过你可能已经注意到了我们使用的 ToDouble
函数,它是普通类型范畴内的一个态射,我们仅仅通过一个 Select
函数就把普通类型范畴内的一个态射映射成了 List<T>
范畴内的一个态射(上面的例子中,是把 (int -> double)
转换成了 (List<int> -> List<double>)
),而且 List<T>
还提供了能够把 int
类型转换成 List<int>
类型(type)的方法:new List<int>{ intValue }
,那么我们就可以把 List<T>
类(class)称为“函子”。事情变得有趣了起来。
自函子
List<T>
还有一个构造函数可以允许我们使用另一个 List 对象创建一个新的 List 对象:new List<T>(list)
,这完成了 List<T> -> List<T>
转换,这看起来像是把 List<T>
范畴中的元素重新映射到了 List<T>
范畴中。有了这个构造函数的帮助,我们就可以试着使用 Select
来映射 List<T>
中的态射(比如,ToDoubleList
):
// 这个映射后的 ToDoubleListAgain 仍然能够正常的工作
Func<List<int>, List<List<double>>> ToDoubleListAgain = x => x.Select(e => ToDoubleList(new List<int>(){e})).ToList();
这里的返回值类型看起来有些奇怪,我们得到了一个嵌套两层的 List
,如果你熟悉 LINQ 的话,马上就会想到 SelectMany
函数——它能够把嵌套的 List
拍扁:
Func<List<TV>, List<TR>> FF<TV, TR>(Func<List<TV>, List<TR>> selector)
{
return xl => xl.SelectMany(x => selector(new List<int>() {x})).ToList();
}
var ToDoubleListAgain = FF(ToDoubleList);
ToDoubleListAgain(new List<int>{1})
这样,我们就实现了 (List<T1> -> List<T2>) -> (List<T1> -> List<T2>)
的映射,虽然功能上并没有什么卵用,但是却实现了把 List<T>
范畴中的态射映射到了 List<T>
范畴中的功能。现在看来,List<T>
类不仅是普通类型映射到 List<T>
的一个函子,它也是 List<T>
映射到 List<T>
的一个函子。这种能够把一个范畴映射到该范畴本畴上的函子也被称为“自函子”。
我们可以发现,C# 中大部分的自函子都通过 LINQ 拓展方法实现了 SelectMany
函数,其签名是:
SomeType<TR> SelectMany<TV, TR>(SomeType<TV> source, Func<TV, SomeType<TR>> selector);
List<T>
还有一个不接受任何参数的构造函数,它会创建出一个空的列表,我们可以把这个函数称作 unit
,因为它的返回值在 List<T>
相关的一些二元运算中起到了单位 1 的作用。比如,concat(unit(), someList)
与 concat(someList, unit())
得到的列表,在结构上是等价的。拥有这种性质的元素被称为“单位元”。
在函数式编程中,我们把拥有 SelectMany
(也被叫做 bind
),unit
函数的自函子称为“Monad”。
但是 C# 中并不是所有的泛型类是自函子,例如 Task<T>
,如果我们不为它添加 Select
拓展方法,它连函子都算不上。所以如果把 C# 中全部的自函子类型放在一个集合中,然后把这些自函子类型之间用来做类型转换的全部函数(例如,list.ToArray()
等)看作是态射,那么我们就构建出来了一个 C# 中的“自函子范畴”。在这个范畴上,我们只能对 Monad 类型使用 LINQ 语法进行复合运算,例如上面的:
// 原版
var result =
from a in taskA
from b in taskB
from c in taskC
select a * b + c;
// 1. 满足结合律
var left =
from a in taskA
from t in (
from b in taskB
from c in taskC
select new {b, c}
)
select a * t.b + t.c;
var left =
from t in (
from a in taskA
from b in taskB
select new {a, b}
)
from c in taskC
select t.a * t.b + c;
left == right
// true
// 2. 存在单位元
var left = from a in Task.FromException(null)
from b in taskB
select a + b;
var right = from b in taskB
from a in Task.FromException(null)
select a + b;
// 因为 left right 得到的都是 Task.FromException(null) 的返回值,故 Task.FromException(null) 是单位元
由于这种作用在两个 Monad 上面的二元运算满足交换律且 Monad 中存在单位元,与群论中幺半群的定义比较类似,所以,我们也把 Monad 称为“自函子范畴上的幺半群”。尽管这句话听起来十分的高大上,但是却并没有说明 Monad 的特征所在。就好比别人跟你介绍手机运营商,说这是一个提供短信、电话业务的公司,你肯定不知道他到底再说哪一家,不过他要是说,这是一个提供 5 元 30 M 流量包的手机运营商,那你就知道了他指的是中国移动。
个人体会
其实我一开始想写的内容只有 LINQ to Result 跟 LINQ to Task 的,但是在编写代码的过程中,种种迹象都表明着 LINQ 跟函数式编程中的 Monad 有不少关系,所以就把剩下的函数式编程这一部分给写出来了。
Monad 作为函数式编程中一种重要的数据类型,可以用来表达计算中的每一小步的功能,通过 Monad 之间的复合运算,我们可以灵活的将这些小的功能片段以一种统一的方式重组、复用,除此之外,我们还可以针对特定的需求(异步、错误处理、懒惰计算)定义专门的 Monad 类型,帮助我们以一种统一的形式将这些特别的功能嵌入到代码之中。在传统的面向对象的编程语言中 Monad 这个概念确实是不太好表达的,不过有了 LINQ 的帮助,我们可以比较优雅地将各种 Monad 组合起来。
用 LINQ 来对 Monad 进行运算的缺点,主要就是除了 SelectMany
之外的,我们没办法定义其他的能在 Query 语法中使用的函数了,要解决这个问题,请关注我的下一篇文章:“F# 函数式编程:Computational Expression”(挖坑预备)。