FreezeJ' Blog

Go函数链式处理

2022-03-14

代码来自http://c.biancheng.net/view/56.html
挺好的写法,把操作与数据分离。

package main

import (
	"fmt"
	"strings"
)

// 字符串处理函数,传入字符串切片和处理链
func StringProccess(list []string, chain []func(string) string) {
	// 遍历每一个字符串
	for index, str := range list {
		// 第一个需要处理的字符串
		result := str
		// 遍历每一个处理链
		for _, proc := range chain {
			// 输入一个字符串进行处理,返回数据作为下一个处理链的输入。
			result = proc(result)
		}
		// 将结果放回切片
		list[index] = result
	}
}

// 自定义的移除前缀的处理函数
func removePrefix(str string) string {
	return strings.TrimPrefix(str, "go")
}
func main() {
	// 待处理的字符串列表
	list := []string{
		"go scanner",
		"go parser",
		"go compiler",
		"go printer",
		"go formater",
	}
	// 处理函数链
	chain := []func(string) string{
		removePrefix,
		strings.TrimSpace,
		strings.ToUpper,
	}
	// 处理字符串
	StringProccess(list, chain)
	// 输出处理好的字符串
	for _, str := range list {
		fmt.Println(str)
	}
}

chain声明为一个数组类型的变量,其中数组元素为相同参数、相同返回的函数。使用range循环执行每一个函数,并且把返回值作为下一个函数的参数调用。

标签: GO
使用支付宝打赏
使用微信打赏

若你觉得我的文章对你有帮助,欢迎点击上方按钮对我打赏