Erlang那些事儿第3回之我是函数(fun),万物之源MFA
Erlang代码到处都是模式匹配，这把屠龙刀可是Erlang的看家本领、独家绝学，之前在《Erlang那些事儿第1回之我是变量,一次赋值永不改变》文章提到过，Erlang一切皆是模式匹配。从变量的赋值、函数的形参传递、重载函数的应用，到处都有模式匹配的影子，刚开始写代码会感觉不习惯，但是当你用习惯之后，会觉得这个武林秘籍是多么的好用。但是本回书重点讲函数，毕竟以后写代码都会应用到函数fun，早点讲方便后面的使用。
Erlang语言中的函数很强大，同一个逻辑可以用多种写法，因为一个函数有形参，也有函数内部实现，这2个地方只要合理地应用模式匹配，那么可以发挥出非常大的作用。本回书会使用模式匹配、关卡、递归、apply、参数传递在函数中的用法。
知识点1：函数重载
特点1：形参数量不同；
特点2：函数之间用点号（.）分隔。
C++一个很重要的特性就是函数重载，这个特性的必要条件是函数的形参数量必须不一样。在Erlang代码中，这个规则同样适用，一个同名函数可以有多个不同参数数量的版本，导出列表也要相应地体现出来，来买个水果试试：
创建文件fruit_price01.erl，代码如下：
1 -module(fruit_price01). 2 -author("snowcicada"). 3 4 %% API 5 -export([fruit_price/0, fruit_price/1]). 6 7 %% 买1个水果的价格 8 fruit_price() -> 9 fruit_price(1). 10 11 %% 买多个水果的价格 12 fruit_price(Count) -> 13 Count * 10.
上述代码存在2个fruit_price函数，一个是0参，一个是1参，如果有需要给其他模块使用的情况下，那么就要添加到export导出列表，来运行试一下：
Eshell V11.1.3 (abort with ^G) 1> c(fruit_price01). {ok,fruit_price01} 2> fruit_price01:fruit_price(). 10 3> fruit_price01:fruit_price(10). 100
在Erlang终端执行函数，不管是0参还是1参都能正常工作。
知识点2：函数形参的模式匹配
特点1：形参数量相同；
特点2：函数之间用分号;分隔；
特点3：从上往下匹配。
函数的每个形参都可以用一个表达式进行匹配，当传入的参数匹配上提前写好的表达式，那么就进入这个函数执行；如果不匹配的话，那么要么报错，要么会进入一个能够匹配的函数分支。
假设买1个水果没打折，买2个打8折，买3个以上打5折。创建文件fruit_price01.erl，代码如下：
1 -module(fruit_price02). 2 -author("snowcicada"). 3 4 %% API 5 -export([fruit_price/1, discount/2]). 6 7 fruit_price(Count) -> 8 io:format("~p~n", [discount(Count, 10)]). 9 10 %% 函数形参进行模式匹配 11 %% 买1个没打折，买2个打8折，买3个以上打5折 12 discount(1 = Count, Price) -> 13 Count * Price; 14 discount(2 = Count, Price) -> 15 Count * Price * 0.8; 16 discount(Count, Price) -> 17 Count * Price * 0.5.
在Erlang语言中，等于号（=）并不是赋值，而是进行了一次模式匹配。所以第12行里面写的1 = Count是在匹配，匹配Count是否等于1，如果匹配成功，那么就会执行第13行的代码。第14行同理。
有趣的是第16行，只写了Count表示对任何数据都可以匹配成功，既然第12行、14行已经匹配了1和2，那么当Count等于3或3以上的时候就会执行第17行的代码。
知识点3：case...of...end表达式的模式匹配
特点1：不同的匹配表达式末尾用分号（;）分隔，最后一个匹配表达式不需要加分号（;）；
特点2：下划线或者一个普通变量可以匹配任何情况；
特点3：从上往下匹配。
写代码总不能为了处理不同的情况而每次都写多个函数匹配，这样写起来不一定方便，所以Erlang还提供了case...of...end表达式。接下来使用case表达式来重写上面的discount函数。
新增函数discount_case，代码如下：
1 %% 参数模式匹配 2 %% 买1个没打折，买2个打8折，买3个以上打5折 3 discount_case(Count, Price) -> 4 case Count of 5 1 -> 6 Count * Price; 7 2 -> 8 Count * Price * 0.8; 9 _ -> %% 下划线也可以替换成一个变量，比如N，Cnt，都可以，只要是变量就行 10 Count * Price * 0.5 11 end.
case...of中间写的是表达式，of后面可以写入不一样的匹配表达式，匹配成功就会执行箭头后面的语句。
知识点4：函数关卡
函数外部可以对形参添加一些条件，指定不同的条件执行不同的函数，这里称为关卡。
新增函数discount_guard，代码如下：
1 %% 函数形参关卡判断 2 %% 买1个没打折，买2个打8折，买3个以上打5折 3 discount_guard(Count, Price) when Count =:= 1 -> 4 Count * Price; 5 discount_guard(Count, Price) when Count =:= 2 -> 6 Count * Price * 0.8; 7 discount_guard(Count, Price) -> 8 Count * Price * 0.5.
在箭头（->）前面，使用when关键字对形参进行判断，第3行显示，当Count等于1的时候，会执行这个函数。
知识点5：if关卡
同样还有更方便的关卡方式，就是使用if...end表达式。
新增函数discount_if，代码如下：
1 %% 参数关卡判断 2 %% 买1个没打折，买2个打8折，买3个以上打5折 3 discount_if(Count, Price) -> 4 if 5 Count =:= 1 -> 6 Count * Price; 7 Count =:= 2 -> 8 Count * Price * 0.8; 9 true -> 10 Count * Price * 0.5 11 end.
case和if的差别在于表达式写的是不是模式匹配，if中间那些表达式是用来判断是否相等，这种是很明确的相等比较。但是case中间的表达式放的是匹配表达式，而且case...of中间可以写复杂的表达式。
以下列出的关卡判断函数和关卡内置函数，可用于if关卡或者函数外的when关卡。
关卡判断函数：
关卡内置函数：
介绍以上4种版本的discount，我们调整下fruit_price函数，
1 fruit_price(Count) -> 2 io:format("~p~n", [discount(Count, 10)]), 3 io:format("~p~n", [discount_case(Count, 10)]), 4 io:format("~p~n", [discount_guard(Count, 10)]), 5 io:format("~p~n", [discount_if(Count, 10)]).
运行结果：
Erlang/OTP 23 [erts-11.1.3] [source] [64-bit] [smp:8:8] [ds:8:8:10] [async-threads:1] [hipe] [dtrace] Eshell V11.1.3 (abort with ^G) 1> c(fruit_price02). {ok,fruit_price01} 2> fruit_price02:fruit_price(10). 50.0 50.0 50.0 50.0 ok
4个版本的discount运行结果都一样。
知识点6：函数作为参数传递
这个特性很常见，很多语言都可以把函数作为参数进行传递，只是语法有些小差异罢了。不啰嗦，写个例子吧，建个fruit_price03.erl文件，代码如下：
1 -module(fruit_price03). 2 -author("snowcicada"). 3 4 %% API 5 -export([fruit_price/1, discount/2, get_discount_func/0]). 6 7 fruit_price(Count) -> 8 Discount = get_discount_func(), 9 io:format("~p~n", [Discount(Count, 10)]). 10 11 get_discount_func() -> 12 fun discount/2. 13 14 %% 函数形参进行模式匹配 15 %% 买1个没打折，买2个打8折，买3个以上打5折 16 discount(1 = Count, Price) -> 17 Count * Price; 18 discount(2 = Count, Price) -> 19 Count * Price * 0.8; 20 discount(Count, Price) -> 21 Count * Price * 0.5.
discount函数有2个形参，所以Erlang要返回一个函数，就如你所见的第12行，fun discount/2。
知识点7：递归函数
Erlang函数在处理模式匹配或者关卡的时候，可以有多个分支，就如同知识点2和知识点4的形式。通过这个方式，可以灵活的写出递归函数，对一些临界情况的处理，这里写个简单的例子就好，以后讲到列表的时候会使用到，用得很灵活有趣。
创建div_three.erl文件，代码如下：
1 -module(div_three). 2 -author("snowcicada"). 3 4 %% API 5 -export([print/1]). 6 7 print(N) when N =:= 0 -> 8 io:format("~n"); 9 print(N) when N rem 3 =:= 0 -> 10 io:format("~p ", [N]), 11 print(N - 1); 12 print(N) -> 13 print(N - 1).
当N等于0的时候，会运行第7行，函数输出换行立马结束；当N对3取余等于0的时候，执行第9行，可以被3整除的数字将会打印出来，然后继续调用print(N-1)，这里就是递归调用。
当执行过了第7、第9行的关卡，剩下的都会执行第12行，这里什么都没处理，直接递归调用print即可。
执行结果：
Erlang/OTP 23 [erts-11.1.3] [source] [64-bit] [smp:8:8] [ds:8:8:10] [async-threads:1] [hipe] [dtrace] Eshell V11.1.3 (abort with ^G) 1> c(div_three). {ok,div_three} 2> div_three:print(100). 99 96 93 90 87 84 81 78 75 72 69 66 63 60 57 54 51 48 45 42 39 36 33 30 27 24 21 18 15 12 9 6 3 ok
知识点8：使用apply调用函数
MFA是Module、Function、Arguments的缩写，指模块调用函数，传入形参，格式如：M:F(A)，也可以这样：apply(M, F, A)。在Erlang自带的标准库中，MFA的调用方式很常见，也是Erlang实现热更新屡试不爽的步骤之一。其中的A很容易出现低级错误，大部分模块的参数支持传入列表，所以通常的调用方式如：M:F([A1, A2, A3])。
Erlang提供了apply函数，可通过指定模块名、函数名和参数进行调用，这里贴下apply的实现源码：
1 %% Shadowed by erl_bif_types: erlang:apply/2 2 -spec apply(Fun, Args) -> term() when 3 Fun :: function(), 4 Args :: [term()]. 5 apply(Fun, Args) -> 6 erlang:apply(Fun, Args). 7 8 %% Shadowed by erl_bif_types: erlang:apply/3 9 -spec apply(Module, Function, Args) -> term() when 10 Module :: module(), 11 Function :: atom(), 12 Args :: [term()]. 13 apply(Mod, Name, Args) -> 14 erlang:apply(Mod, Name, Args).
apply分别有2个形参和3个形参，2个形参的版本是apply(F,A)，不用传入模块名，3个形参的版本是apply(M,F,A)，需要指定模块名。
打开Erlang终端做个试验就行，使用io:format来测试打印信息：
Erlang/OTP 23 [erts-11.1.3] [source] [64-bit] [smp:8:8] [ds:8:8:10] [async-threads:1] [hipe] [dtrace] Eshell V11.1.3 (abort with ^G) 1> io:format("Name:~page:~p~n",["Lucy", 16]). Name:"Lucy"age:16 ok 2> apply(io, format, ["Name:~p age:~p~n", [lucy, 16]]). Name:lucy age:16 ok 3> erlang:apply(io, format, ["Name:~p age:~p~n", [lucy, 16]]). %%也可以指定erlang模块，这样写的好处是会有智能提示 Name:lucy age:16 ok
函数的应用大概就这些了，虽然简单，但是这些都是日常很经典的技巧。
本回使用的代码已上传Github：https://github.com/snowcicada/erlang-story/tree/main/story003
下一回将介绍原子（Atom）的使用，且听下回分解。
