淺析Go語言中閉包的使用

更新時間：2022年12月06日 16:06:11 作者：甜點cc

閉包是一個函數(shù)和其相關(guān)的引用環(huán)境組合的一個整體。本文主要為大家介紹一下Go語言中閉包的使用，文中的示例代碼講解詳細(xì)，對我們學(xué)習(xí)Go語言有一定幫助，需要的可以參考一下

閉包基本介紹

閉包就是 一個函數(shù) 和其相關(guān)的 引用環(huán)境 組合的一個整體

好處： 保存引用的變量，下次繼續(xù)使用，不會銷毀

下面通過閉包的方式，寫一個數(shù)字累加器，體驗一下閉包的妙處

閉包實現(xiàn)數(shù)字累加

package main

import "fmt"

// 累加器
// 閉包 - 函數(shù)柯里化
// 返回值類型： func(int) int
func AddUpper() func(int) int {
	var n int = 100
	return func(i int) int {
		n = n + i
		return n
	}
}

func main() {
	f := AddUpper()
	fmt.Println(f(1)) // 101
	fmt.Println(f(2)) // 103
	fmt.Println(f(3)) // 106
}

代碼說明

1.AddUpper是一個函數(shù)，返回的數(shù)據(jù)類型 func(int) int

2.閉包的說明

返回的是一個匿名函數(shù)，但是這個匿名函數(shù)引用到了函數(shù)外的變量 n ，因此這個匿名函數(shù)就和 n 形成一個整體，構(gòu)成閉包

3.當(dāng)反復(fù)調(diào)用 f 函數(shù)時，因為 n 只初始化一次，保存了變量的值，因此每調(diào)用一次就相當(dāng)于進(jìn)行了累加。

4.我們要搞清楚閉包，關(guān)鍵就是要分析返回的函數(shù)使用到哪些變量

代碼分析

這里我引入了一個字符串變量str，來幫助分析閉包是怎么保存變量的。

package main

import (
	"fmt"
)

// 累加器
// 閉包 - 函數(shù)柯里化
// 返回值類型： func(int) int
func AddUpper() func(int) int {
	var n int = 100
	var str = "hello"
	return func(i int) int {
		n = n + i
		fmt.Println("i=", i)
		str += string(36) // ascii 36 = '$'
		fmt.Printf("str==%s\n", str)
		return n
	}
}

func main() {
	f := AddUpper()
	// fmt.Println(AddUpper()(1)) // 101
	fmt.Println("f(1)=", f(1)) // 101
	fmt.Println("f(2)=", f(2)) // 103
	fmt.Println("f(3)=", f(3)) // 106
}


i= 1
str==hello$
f(1)= 101
i= 2
str==hello$$
f(2)= 103
i= 3
str==hello$$$
f(3)= 106

從輸出可以看出來，閉包引用的變量n和str并沒有在調(diào)用函數(shù)的時候重復(fù)聲明，而是保留了下次函數(shù)調(diào)用后更新的值。

閉包案例

需求：

編寫一個函數(shù) makeSuffix(suffix string) ，可以接收一個文件后綴名，并返回一個閉包
調(diào)用閉包，可以傳入一個文件名，如果該文件名沒有指定后綴，則返回文件名.jpg ，如果已經(jīng)有.jpg，則返回原文件名。

strings.HasSuffix，該函數(shù)可以判斷某個字符串是否有指定的后綴。

上代碼

package main

import (
	"fmt"
	"strings"
)

func makeSuffix(suffix string) func(string) string {
	return func(name string) string {
		if !strings.HasSuffix(name, suffix) {
			return name + suffix
		}
		return name
	}
}

// 傳統(tǒng)寫法
func makeSuffixV2(suffix string, name string) string {
	if !strings.HasSuffix(name, suffix) {
		return name + suffix
	}
	return name
}

func main() {
	// 閉包調(diào)用
	f := makeSuffix(".jpg")
	fmt.Println(f("xiao"))         // xiao.jpg
	fmt.Println(f("xiaoxiao.jpg")) // xiaoxiao.jpg
	fmt.Println(f("xiaoxiao.666")) // xiaoxiao.666.jpg

	// 傳統(tǒng)寫法調(diào)用
	fmt.Println("makeSuffixV2=", makeSuffixV2(".jpg", "allblue"))  // makeSuffixV2= allblue.jpg
	fmt.Println("makeSuffixV2=", makeSuffixV2(".jpg", "all.blue")) // makeSuffixV2= all.blue.jpg
}