现在我们首先定义一个支持4个参数相加的函数:
(defn add [ v1 v2 v3 v4]
(+ v1 v2
(if v3
v3
0)
(if v4
v4
0)
))
我们想达到这样一种效果。如果我们调用(add 1 2 3 4),则正常返回10。如果我们调用(add 1 2 3)能得到结果6。或者调用(add 1 2)能得到3。
我们执行后发现:
=> (add 1 2 3 4)
10
=> (add 1 2)
... ArityException Wrong number of args (2) passed to: user$add ...
当参数数量和我们定义函数时的参数数量一致时能得到正确的值。否则会抛出参数数量不匹配异常。看来这个和我们预想的不一样啊。
固定参数
我们之前定义函数时,使用的都是固定参数方式。调用函数时,只要我们传入参数的数量不符合定义时的参数数量,clojure都会提出抗议(抛出参数数量不匹配异常)。像上面我们可以这么调用(add 1 2 3 nil),这样会返回6。 (if v4 v4 0)的含义是,如果v4 不为nil或false,就返回v4,否则返回0。我们传入的正是nil。所以函数会正确返回6。当然(add 1 2 nil nil )也会正确返回3。
难道clojure就这么点能力?当然不是,clojure的固定参数可不止这点能耐。我们在上篇列出的定义函数时的脚手架其实并没有列完整。请看下面一个更加完整的脚手架:
(defn variable-name
"description"
{metadata}
([argument-pattern #1] ;;第一种参数形式
body-of-expressions)
([argument-pattern #2] ;;第二种参数形式
body-of-expressions)
more-patterns... )
先让我们用上面方式重构一下之前的add函数,看看怎么来支持2个、3个或4个参数的相加:
(defn add
( [v1 v2] ( + v1 v2))
( [v1 v2 v3] (+ v1 v2 v3))
( [v1 v2 v3 v4] (+ v1 v2 v3 v4))
)
上面我们总共定义了三种不同参数形式的add函数。这样调用的话(add 1 2)会匹配第一种形式,正确返回3。(add 1 2 3)会匹配第二种参数模式,返回结果6。这其实就是lisp中模式匹配的一种应用。是不是比java中的重载方式更加灵活呀。这只是固定参数的一种用法,下面再来看看clojure的可变参数用法。
可变参数
如果我们的参数模式就两三种(例如上面只需要支持3种不同个数的数字相加),我们可以采用固定参数+模式匹配来实现函数。但是如果我们的参数模式个数不固定(例如支持任意个数字相加),固定参数+模式匹配也拯救不了我们了。不过不用担心,clojure还给我们提供了可变参数 。
我们看一下如何使用可变参数来实现任意个数的数字相加:
(defn add [v1 v2 & others] ;;&后面的是可变参数
(+ v1 v2
(if others ;;判断可变参数列表是否是空,如果不是累加列表中的值,否则返回0
(reduce + 0 others) ;;使用reduce函数计算others的数字之和。
0
)
)
)
这里有几点需要注意一下,固定参数要写在一个“&”之后,只能有一个可变参数。&后的可变参数名代表的是由可变参数组成的列表。这里的reduce函数,先理解成对列表中的数字进行累加就行,本文目的主要是理解可变参数的用法。
现在我们执行一下上面定义的函数:
=> (add 1 2)
3
=> (add 1 2 3)
6
=> (add 1 2 3 4 5 6)
21
现在add已经支持任意个数字参数相加了。