函数初探
在前面的,你已经见到了 Go 语言中一个非常重要的函数:main 函数,它是一个 Go 语言程序的入口函数,我在演示代码示例的时候,会一遍遍地使用它。
下面的示例就是一个 main 函数:
func main() {
}
它由以下几部分构成:
-
任何一个函数的定义,都有一个 func 关键字,用于声明一个函数,就像使用 var 关键字声明一个变量一样
-
然后紧跟的 main 是函数的名字,命名符合 Go 语言的规范即可,比如不能以数字开头
-
main 函数名字后面的一对括号 () 是不能省略的,括号里可以定义函数使用的参数,这里的 main 函数没有参数,所以是空括号 ()
-
括号 () 后还可以有函数的返回值,因为 main 函数没有返回值,所以这里没有定义
-
最后就是大括号 {} 函数体了,你可以在函数体里书写代码,写该函数自己的业务逻辑
函数声明
经过上一小节的介绍,相信你已经对 Go 语言函数的构成有一个比较清晰的了解了,现在让我们一起总结出函数的声明格式,如下面的代码所示:
func funcName(params) result {
body
}
这就是一个函数的签名定义,它包含以下几个部分:
-
关键字 func
-
函数名字 funcName
-
函数的参数 params,用来定义形参的变量名和类型,可以有一个参数,也可以有多个,也可以没有
-
result 是返回的函数值,用于定义返回值的类型,如果没有返回值,省略即可,也可以有多个返回值
-
body 就是函数体,可以在这里写函数的代码逻辑
现在,我们一起根据上面的函数声明格式,自定义一个函数,如下所示:
func sum(a int,b int) int{
return a+b
}
这是一个计算两数之和的函数,函数的名字是 sum,它有两个参数 a、b,参数的类型都是 int。sum 函数的返回值也是 int 类型,函数体部分就是把 a 和 b 相加,然后通过 return 关键字返回,如果函数没有返回值,可以不用使用 return 关键字。
终于可以声明自己的函数了,恭喜你迈出了一大步!
函数中形参的定义和我们定义变量是一样的,都是变量名称在前,变量类型在后,只不过在函数里,变量名称叫作参数名称,也就是函数的形参,形参只能在该函数体内使用。函数形参的值由调用者提供,这个值也称为函数的实参,现在我们传递实参给 sum 函数,演示函数的调用,如下面的代码所示:
func main() {
result:=sum(1,2)
fmt.Println(result)
}
我们自定义的 sum 函数,在 main 函数中直接调用,调用的时候需要提供真实的参数,也就是实参 1 和 2。
函数的返回值被赋值给变量 result,然后把这个结果打印出来。你可以自己运行一下,能看到结果是 3,这样我们就通过函数 sum 达到了两数相加的目的,如果其他业务逻辑也需要两数相加,那么就可以直接使用这个 sum 函数,不用再定义了。
在以上函数定义中,a 和 b 形参的类型是一样的,这个时候我们可以省略其中一个类型的声明,如下所示:
func sum(a, b int) int {
return a + b
}
像这样使用逗号分隔变量,后面统一使用 int 类型,这和变量的声明是一样的,多个相同类型的变量都可以这么声明。
多值返回
同有的编程语言不一样,Go 语言的函数可以返回多个值,也就是多值返回。在 Go 语言的标准库中,你可以看到很多这样的函数:第一个值返回函数的结果,第二个值返回函数出错的信息,这种就是多值返回的经典应用。
对于 sum 函数,假设我们不允许提供的实参是负数,可以这样改造:在实参是负数的时候,通过多值返回,返回函数的错误信息,如下面的代码所示:
func sum(a, b int) (int,error){
if a<0 || b<0 {
return 0,errors.New("a或者b不能是负数")
}
return a + b,nil
}
这里需要注意的是,如果函数有多个返回值,返回值部分的类型定义需要使用小括号括起来,也就是 (int,error)。这代表函数 sum 有两个返回值,第一个是 int 类型,第二个是 error 类型,我们在函数体中使用 return 返回结果的时候,也要符合这个类型顺序。
在函数体中,可以使用 return 返回多个值,返回的多个值通过逗号分隔即可,返回多个值的类型顺序要和函数声明的返回类型顺序一致,比如下面的例子:
return 0,errors.New("a或者b不能是负数")
返回的第一个值 0 是 int 类型,第二个值是 error 类型,和函数定义的返回类型完全一致。定义好了多值返回的函数,现在我们用如下代码尝试调用:
func main() {
result,err := sum(1, 2)
if err!=nil {
fmt.Println(err)
}else {
fmt.Println(result)
}
}
函数有多值返回的时候,需要有多个变量接收它的值,示例中使用 result 和 err 变量,使用逗号分开。
如果有的函数的返回值不需要,可以使用下划线 _ 丢弃它,这种方式我在 for range 循环那节课里也使用过,如下所示:
result,_ := sum(1, 2)
这样即可忽略函数 sum 返回的错误信息,也不用再做判断。
提示:这里使用的 error 是 Go 语言内置的一个接口,用于表示程序的错误信息,后续课程我会详细介绍。