Go | 閉包的使用

甜點cc發表於2022-12-06
Go

閉包基本介紹

閉包就是 一個函式 和其相關的 引用環境 組合的一個整體

好處： 儲存引用的變數，下次繼續使用，不會銷燬

下面透過閉包的方式，寫一個數字累加器，體驗一下閉包的妙處?

閉包實現數字累加

package main

import "fmt"

// 累加器
// 閉包 - 函式柯里化
// 返回值型別： 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是一個函式，返回的資料型別 func(int) int

  2. 閉包的說明

返回的是一個匿名函式，但是這個匿名函式引用到了函式外的變數 n ，因此這個匿名函式就和 n 形成一個整體，構成閉包

  1. 當反覆呼叫 f 函式時，因為 n 只初始化一次，儲存了變數的值，因此每呼叫一次就相當於進行了累加。

  2. 我們要搞清楚閉包，關鍵就是要分析返回的函式使用到哪些變數

程式碼分析

這裡我引入了一個字串變數str，來幫助分析閉包是怎麼儲存變數的。

package main

import (
	"fmt"
)

// 累加器
// 閉包 - 函式柯里化
// 返回值型別： 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並沒有在呼叫函式的時候重複宣告，而是保留了下次函式呼叫後更新的值。

閉包案例

需求：

  1. 編寫一個函式 makeSuffix(suffix string) ，可以接收一個檔案字尾名，並返回一個閉包

  2. 呼叫閉包，可以傳入一個檔名，如果該檔名沒有指定字尾，則返回 檔名.jpg ，如果已經有.jpg，則返回原檔名。

strings.HasSuffix，該函式可以判斷某個字串是否有指定的字尾。

上程式碼

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
	}
}

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

func main() {
	// 閉包呼叫
	f := makeSuffix(".jpg")
	fmt.Println(f("xiao"))         // xiao.jpg
	fmt.Println(f("xiaoxiao.jpg")) // xiaoxiao.jpg
	fmt.Println(f("xiaoxiao.666")) // xiaoxiao.666.jpg

	// 傳統寫法呼叫
	fmt.Println("makeSuffixV2=", makeSuffixV2(".jpg", "allblue"))  // makeSuffixV2= allblue.jpg
	fmt.Println("makeSuffixV2=", makeSuffixV2(".jpg", "all.blue")) // makeSuffixV2= all.blue.jpg
}

程式碼說明

返回的匿名函式和 makeSuffix(suffix string) 的 suffix 變數組合成一個閉包

傳統寫法和閉包寫法實現效果一樣，但是，傳統寫法需要重複寫變數， 比如上面的

makeSuffixV2(".jpg", "all.blue"))

閉包則解決了這個問題，是程式碼看起來更加的簡潔

閉包的好處之一： 引數複用

好處： 儲存引用的變數，下次繼續使用，不會銷燬

函式柯里化（閉包）

參考我的另一篇文章： 甜點cc的語雀知識庫

我是 甜點cc☭

微信公眾號：【看見另一種可能】

專注前端開發，也喜歡專研各種跟本職工作關係不大的技術，技術、產品興趣廣泛且濃厚。本號主要致力於分享個人經驗總結，希望可以給一小部分人一些微小幫助。

相關文章