go语言中利用匿名函数和闭包实现文件名修改

151 阅读1分钟
package main

import (
	"fmt"
	"strings"
)

//利用匿名函数和闭包实现文件后缀修改
func makeSuffix(suffix string) func (string) string{
	return func (fileName string) string{
		//如果文件后缀不是以suffix结尾,则加上后缀,否则返回原文件名
		if !strings.HasSuffix(fileName, suffix) {
			return fileName + suffix
		}
		return fileName
	}
}

func main(){

	f := makeSuffix(".jpg")
	fileName1 := "file.jpg"
	fileName2 := "file"
	fmt.Println(f(fileName1))
	fmt.Println(f(fileName2))
}

执行结果

file.jpg
file.jpg

说明

这里makeSuffix返回的匿名函数f和作为外部传入的参数".jpg"共同构成了一个闭包,使得在传入一次".jpg"的情况下可以重复使用,这就巧妙的用到了闭包中保留上次使用的值的特性替代了反复传值的问题。