Go语言入门教程系列——函数、循环与分支

本文始发于个人公众号：TechFlow，原创不易，求个关注

今天是Golang专题的第四篇，这一篇文章将会介绍golang当中的函数、循环以及选择判断的具体用法。

函数

在之前的文章当中其实我们已经接触过函数了，因为我们写的main函数本质上也是一个函数。只不过由于main函数没有返回值，也没有传参，所以省略了很多信息。

func main() {
 fmt.Println("Hello World")
}

下面，我们来看看一个完整的函数是怎样的，这是golang官网上的例子。

func add(x int, y int) int {
    return x + y
}

这是一个非常简单的a+b的函数，我想大家应该都能看懂。我们来重点关注一下函数的格式。首先是func关键字，我们使用这个关键字定义一个函数，之后跟着的是函数名，然后是函数的传参，最后是函数的返回值。

这个顺序可能和我们之前普遍接触的语法不太一样，例如C++当中是把函数返回类型写在最前面，然后是函数名和传参。再比如Python当中则是没有返回值的任何信息，只有def关键字和函数名以及传入的参数。

golang有些像是Python和C++的综合体，总体来说我觉得内涵上更接近C++，但是写法上和Python更接近一些。

我们理解了函数的定义之后，下面来看看golang当中支持的一些特性。

变量简写

在变量声明的时候，我们如果定义两个相同类型的变量是可以把它们进行缩写的。比如我们定义两个int类型的变量，分别叫做a和b。那么可以简写成这样：

var a, b int

同样，在函数当中，如果传入的参数类型相同，也一样是可以简写的。我们可以把x和y两个参数缩写在一起，用逗号分开，共享变量类型。

func add(x, y int) int {
    return x + y
}

多值返回

在前面介绍golang特性的时候曾经提到过，golang作为一个看起来很守旧的语言，但是却支持很多新鲜的特性。其中最知名的一个特性就是函数支持多值返回，即使是现在，也只有少量的语言支持这一特性。

在许多语言当中，如果需要返回多个值，往往需要用一个结构体或者是tuple、list等数据结构将它们包装起来。但是在golang当中支持同时返回多个结果，这将会极大地方便我们的编码。

func sample() (string, string) {
    return "sample1", "sample2"
}

多值返回也会有一个小小的问题，就是如果我们要返回的值过多，会导致这个return会写得很长，或者是组装的逻辑变得很复杂。或者是很容易产生遗漏、搞混顺序之类的问题，golang当中针对这个问题也进行优化，支持我们对返回值进行命名。当命名的变量赋值完成之后，我们就可以直接用return关键字返回所有数据。

这个操作很难用语言描述很清楚，我们来看下面的例子：

func sample(x, y, z int) (xPrime, yPrime, zPrime int) {
    xPrime, yPrime, zPrime = x-1, y+1, z-2
    return 
}

在上面的代码当中，在返回之前，我们先给要返回的值起好了名字，我们在函数体当中对这些值进行赋值完成之后，我们就可以直接return了，golang会自动将它们的值填充进行返回。这样不但可以简化一定的编码过程，也可以增加可读性。

defer

golang的函数当中有一个特殊的用法，就是defer。这个用法据说其他语言也有，但是我暂时没有见到过。defer是一个关键字，用它修饰的语句会被存入栈中，直到函数退出的时候执行。

比如：

func main() {
 defer fmt.Println("world")

 fmt.Println("hello")
}

上面这两行代码虽然defer的那一行在先，但是并不会被先执行，而是等main函数执行退出之前才会执行。

看起来这个用法有一点点怪，但是它的用处很大，经常用到。比如当我们打开一个文件的时候，不管文件有没有打开成功，我们都需要记得关闭文件。但如果文件打开不成功可能就会有异常或者是报错，如果我们把这些情况全部都考虑到，会变得非常复杂。所以这个时候我们通常都会用defer来执行文件的关闭。

要注意的是，defer修饰的代码会被放入栈中。所以最后会按照先进后出的原则进行执行。比如：

func main() {
 for i := 0; i < 10; i++ {
  defer fmt.Println(i)
 }

 fmt.Println("done")
}

最后执行的结果是9876543210，而不是相反。这一点蛮重要的，有的时候如果搞混了，很容易出现问题。

循环

和其他语言不同，Golang当中只有一种循环，就是for循环。没有while，更没有do while循环。在golang的设计中设想当中，只需要一种循环，就可以实现所有的功能。从某种程度上来说，也的确如此，golang中的循环有点像是C++和Python循环的结合体，集合两种所长。

首先，我们先来看下for循环的语法，在for循环当中，我们使用分号分开循环条件。循环条件分为三个部分，第一个部分是初始化部分，我们对循环体进行初始化，第二个部分是判断部分，判断循环结束的终止条件，第三个部分是循环变量的改变部分。

写出来大概是这样的：

for i := 0; i < 10; i++ {
    fmt.Println(i)
}

这个语法是不是和C++中的循环很像呢？可以说除了没有括号之外，基本上就是一样的。golang当中同样支持++的自增操作，不过golang中只支持i++，而不支持++i。

和C++一样，这三段当中的任何一段都是可以省略的，比如我们可以省略判断条件：

for i := 0; ; i++ {
    fmt.Println(i)
    if i > 10 {
        break
    }
}

我们也可以省略循环变量的自增条件：

for i := 0; i < 10; {
    i += 2
    fmt.Println(i)
}

甚至可以全部省略，如果全部省略的话，等价于C++中的while(true)循环，也就是死循环。

range的用法

如果我们用循环遍历一个数组或者是map，它的这个用法和Python中的用法非常类似。我们来看下，假如我们有一个数组是：

nums := []int{2, 3, 4}
sum := 0
for i, v := range nums {
    sum += v
    fmt.Println(i)
}

这个用法等价于Python中的for i, v in enumerate(nums)。也就是通过range会同时返回数组和map中的下标与对应的值，我们再来看下map，其实也是一样的。

kvs := map[string]string{"a": "apple", "b": "banana"}
for k, v := range kvs {
    fmt.Printf("%s -> %s\n", k, v)
}

如果你看不懂map和数组的定义没有关系，我们会在之后的文章当中再来详细讲解，这篇的主要内容是循环。我们只需要看得懂for循环的range操作即可。

判断

golang当中支持if与switch进行条件判断。我们先来看if，在golang当中的if和Python比较接近，在if的判断条件外面不需要加上小括号()，但是if的执行条件当中必须要大括号{}，即使只有一行代码。

比如刚才我们写的循环中的那个break。

for i := 0; ; i++ {
    fmt.Println(i)
    if i > 10 {
        break
    }
}

在判断中初始化

上面的逻辑在各个语言中都大同小异，很多语言都是这么写的。但是golang对于if还有特殊的支持，golang支持在if条件当中加上初始化信息。

比如：

if v := sample(); v < 10 {
    fmt.Println(v)
}

上面当中的v是在if执行的时候才进行的初始化，也就是说我们将变量的初始化和if判断结合在了一起。这个用法非常重要，在golang当中也大规模使用，所以我们一定要学会这个用法。

switch

golang当中也支持switch用法，它的基本套路和C++一样，但是在细微的地方又做了优化。

比如和if一样，switch也支持在执行的时候初始化。比如：

switch flag := sample(); flag {
case "a":
    fmt.Println(flag)
case "b":
    fmt.Println(flag)
default:
    fmt.Println(flag)
}

看明白了吗，代码当中的flag是我们执行switch的时候才创建出来的。分号之前的都是初始化的代码，分号之后的表达式才是switch进行判断的内容。

还有一个小细节需要注意，在golang当中使用switch的时候，每个case的判断条件后面不需要再加上break。我们在写其他语言的时候，如果用到switch要么就是忘记了case的执行条件后面要加上break，要么就是写很多break非常麻烦。golang的设计者觉得每个case都加上break太二了，因为大家基本上都只用switch执行一个case，所以就去掉了必须要加上break这个设定。

switch执行顺序

在golang当中，switch的判断条件按照顺序执行。

为什么要强调这个呢？因为你很有可能会看到有些人的代码里的switch没有判断条件，比如：

switch a := sample();{
case a < 5:
    fmt.Println(a)
case a > 5:
    fmt.Println(a)
default:
    fmt.Println("end")
}

在上面这段代码当中，我们根本没有为switch设置判断的根据，这段逻辑完全等同于若干个if-else条件的罗列，它在golang当中同样是允许的。

题外话

今天本来是分布式专题，但实在是没有想到什么很好的题目，我也不喜欢强求，干脆就换个主题吧。以后分布式专题还会更新，不过可能要改成间歇式的了，后面想少写点理论，能够分享一点可以实际用上的东西（所以需要的时间比较久）。

不知道大家从今天的内容当中有没有感受到golang这门语言的个性，很多地方看起来中规中矩，却又能创造出新的用法来，至少我是很佩服设计者的想法的。golang当中这些新特性初见的时候往往会觉得不喜欢和排斥，怎么看怎么怪异，但是写多了之后还是蛮香的。

今天的文章就到这里，扫码关注，获取更多优质文章。

各位看官大大，赏赐个关注再走吧~