如何使用Golang的函数?一篇文章带你了解

63 阅读1分钟

函数

  • 可以有多个返回值
  • 所有的参数都是值传递:slice、map、channel会有传引用的错觉
  • 函数可以作为变量的值
  • 函数可以作为参数和返回值

多返回值

// 多返回值
func returnMultiValues() (int, int) {
   return rand.Intn(10), rand.Intn(20)
}
func TestFn(t *testing.T) {
   a, b := returnMultiValues()
   c, _ := returnMultiValues()//忽略一个返回值
   t.Log(a, b, c)
}

函数作为参数和返回值实现装饰器模式

// 装饰器模式
func timeSpent(inner func(op int) int) func(op int) int {
   //对原有的方法进行装饰
   return func(op int) int {
      //计时
      start := time.Now()
      //执行原有方法
      ans := inner(op)
      //取结果
      fmt.Println("time spelt", time.Since(start).Seconds())
      return ans
   }
}

func slowFun(op int) int {
   //睡1s
   time.Sleep(time.Second * 1)
   return op
}

func TestFn(t *testing.T) {
   //给slowFun函数装饰计时功能
   tsSF := timeSpent(slowFun)
   t.Log(tsSF(10))
   /*
      time spelt 1.0067761
      10
   */
}

可变参数

func Sum(ops ...int) int {
   //可变参数就相当于一个数组,直接数组遍历即可
   ans := 0
   for _, op := range ops {
      ans += op
   }
   return ans
}
func TestVarParam(t *testing.T) {
   t.Log(Sum(1, 2, 3, 4, 5))    //15
   t.Log(Sum(1, 2, 3, 4, 5, 6)) //21
}

defer函数

延迟函数,即使发生错误,也会被执行,基于这种特性,可用来清理回收资源。

func TestDefer(t *testing.T) {
   defer Clear()
   fmt.Println("Start")
   panic("err") //即使发生错误,defer函数也会被执行
   /*
      Start
      Clear resources.
   */
}

image.png