git/lancet
1
0
Fork 0
mirror of https://github.com/duke-git/lancet.git synced 2026-03-01 00:35:28 +08:00
Code Issues Projects Releases Wiki Activity

refactor: update Compose func in function.go

Browse Source
This commit is contained in:
dudaodong
2021-12-14 10:29:53 +08:00
parent d1b74cfcfb
commit 60f3a72c88
2 changed files with 6 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
function/function.go
View File

@@ -48,8 +48,8 @@ func (f Fn) Curry(i interface{}) func(...interface{}) interface{} {
} }
// Compose compose the functions from right to left // Compose compose the functions from right to left
func Compose(fnList ...func(...string) string) func(...string) string { func Compose(fnList ...func(...interface{}) interface{}) func(...interface{}) interface{} {
return func(s... string) string { return func(s... interface{}) interface{} {
f := fnList[0] f := fnList[0]
restFn := fnList[1:] restFn := fnList[1:]

10
function/function_test.go
View File

@@ -71,19 +71,17 @@ func TestCurry(t *testing.T) {
} }
func TestCompose(t *testing.T) { func TestCompose(t *testing.T) {
toUpper := func(a... string) string { toUpper := func(a ...interface{}) interface{} {
return strings.ToUpper(a[0]) return strings.ToUpper(a[0].(string))
} }
toLower := func(a ...interface{}) interface{} {
toLower := func(a... string) string { return strings.ToLower(a[0].(string))
return strings.ToLower(a[0])
} }
expect := toUpper(toLower("aBCde")) expect := toUpper(toLower("aBCde"))
cf := Compose(toUpper, toLower) cf := Compose(toUpper, toLower)
res := cf("aBCde") res := cf("aBCde")
fmt.Println(res, expect)
if res != expect { if res != expect {
t.FailNow() t.FailNow()
} }