Go 中的一流函数

Question

我来自 JavaScript，它具有一流的函数支持。例如，您可以：

• 将一个函数作为参数传递给另一个函数
• 从一个函数返回一个函数。

有人可以给我一个例子来说明如何在 Go 中做到这一点吗？

Answer 1

只是一个带有递归函数定义的脑筋急转弯，用于在 Web 应用程序中链接中间件。

首先，工具箱：

func MakeChain() (Chain, http.Handler) {
    nop := http.HandlerFunc(func(res http.ResponseWriter, req *http.Request) {})
    var list []Middleware
    var final http.Handler = nop
    var f Chain
    f = func(m Middleware) Chain {
        if m != nil {
            list = append(list, m)
        } else {
            for i := len(list) - 1; i >= 0; i-- {
                mid := list[i]

                if mid == nil {
                    continue
                }

                if next := mid(final); next != nil {
                    final = next
                } else {
                    final = nop
                }
            }

            if final == nil {
                final = nop
            }
            return nil
        }
        return f
    }
    return f, final
}

type (
    Middleware func(http.Handler) http.Handler
    Chain      func(Middleware) Chain
)

如您所见，Chain 类型是一个函数，它返回相同类型 Chain 的另一个函数（这真是一流！）。

现在进行一些测试来看看它的实际效果：

func TestDummy(t *testing.T) {
    c, final := MakeChain()
    c(mw1(`OK!`))(mw2(t, `OK!`))(nil)
    log.Println(final)

    w1 := httptest.NewRecorder()
    r1, err := http.NewRequest("GET", "/api/v1", nil)
    if err != nil {
        t.Fatal(err)
    }
    final.ServeHTTP(w1, r1)
}

func mw2(t *testing.T, expectedState string) func(next http.Handler) http.Handler {
    return func(next http.Handler) http.Handler {
        return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
            val := r.Context().Value(contextKey("state"))
            sval := fmt.Sprintf("%v", val)
            assert.Equal(t, sval, expectedState)
        })
    }
}

func mw1(initialState string) func(next http.Handler) http.Handler {
    return func(next http.Handler) http.Handler {
        return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
            ctx := context.WithValue(r.Context(), contextKey("state"), initialState)
            next.ServeHTTP(w, r.WithContext(ctx))
        })
    }
}

type contextKey string

同样，这只是一个脑筋急转弯，表明我们可以以不同的方式在 Go 中使用第一类函数。就我个人而言，我现在使用 chi 作为路由器和处理中间件。

Answer 2

虽然您可以使用 var 或声明类型，但您不需要这样做。
你可以非常简单地做到这一点：

package main

import "fmt"

var count int

func increment(i int) int {
    return i + 1
}

func decrement(i int) int {
    return i - 1
}

func execute(f func(int) int) int {
    return f(count)
}

func main() {
    count = 2
    count = execute(increment)
    fmt.Println(count)
    count = execute(decrement)
    fmt.Println(count)
}

//The output is:
3
2

Answer 3

规范中的相关部分：函数类型。

这里的所有其他答案首先声明一个新类型，这很好（实践）并且使您的代码更易于阅读，但要知道这不是必需的。

您可以使用函数值，而无需为其声明新类型，如下例所示。

声明一个函数类型的变量，该变量有 2 个 float64 类型的参数，并且有一个 float64 类型的返回值，如下所示：

// Create a var of the mentioned function type:
var f func(float64, float64) float64

让我们编写一个返回加法器函数的函数。此加法器函数应采用 2 个 float64 类型的参数，并在调用时返回这 2 个数字的总和：

func CreateAdder() func(float64, float64) float64 {
    return func(x, y float64) float64 {
        return x + y
    }
}

让我们编写一个具有 3 个参数的函数，前 2 个为 float64 类型>，第三个是函数值，该函数采用 2 个 float64 类型的输入参数并生成 float64 类型的值。我们编写的函数将调用作为参数传递给它的函数值，并使用前 2 个 float64 值作为函数值的参数，并返回传递的函数值的结果返回：

func Execute(a, b float64, op func(float64, float64) float64) float64 {
    return op(a, b)
}

让我们看看之前的示例：

var adder func(float64, float64) float64 = CreateAdder()
result := Execute(1.5, 2.5, adder)
fmt.Println(result) // Prints 4

请注意，当然您可以使用 短变量声明 创建 adder 时：

adder := CreateAdder() // adder is of type: func(float64, float64) float64

在 Go Playground 上尝试这些示例。

使用现有函数

当然，如果您已经在包中声明了具有相同函数类型的函数，您也可以使用它。

例如 math.Mod() 具有相同的功能类型：

func Mod(x, y float64) float64

因此您可以将此值传递给我们的 Execute() 函数：

fmt.Println(Execute(12, 10, math.Mod)) // Prints 2

打印 2 因为 12 mod 10 = 2。请注意，现有函数的名称充当函数值。

在 Go Playground 上尝试一下。

注意：

请注意，参数名称不是类型的一部分，具有相同参数和结果类型的两个函数的类型是相同的，无论参数名称如何。但要知道，在参数或结果列表中，名称必须全部存在或全部不存在。

例如，您还可以写：

func CreateAdder() func(P float64, Q float64) float64 {
    return func(x, y float64) float64 {
        return x + y
    }
}

或者：

var adder func(x1, x2 float64) float64 = CreateAdder()

Answer 4

Go 语言和函数式编程可能会有所帮助。来自这篇博文：

package main
import fmt "fmt"
type Stringy func() string
func foo() string{
        return "Stringy function"
}
func takesAFunction(foo Stringy){
    fmt.Printf("takesAFunction: %v\n", foo())
}
func returnsAFunction()Stringy{
    return func()string{
        fmt.Printf("Inner stringy function\n");
        return "bar" // have to return a string to be stringy
    }
}
func main(){
    takesAFunction(foo);
    var f Stringy = returnsAFunction();
    f();
    var baz Stringy = func()string{
        return "anonymous stringy\n"
    };
    fmt.Printf(baz());
}

作者是博客所有者：Dethe Elza（不是我）

Answer 5

package main

import (
    "fmt"
)

type Lx func(int) int

func cmb(f, g Lx) Lx {
    return func(x int) int {
        return g(f(x))
    }
}

func inc(x int) int {
    return x + 1
}

func sum(x int) int {
    result := 0

    for i := 0; i < x; i++ {
        result += i
    }

    return result
}

func main() {
    n := 666

    fmt.Println(cmb(inc, sum)(n))
    fmt.Println(n * (n + 1) / 2)
}

输出：

222111
222111