在我当前的 Golang 项目中，我在将日志发送到我的日志库中的 Elasticsearch 之前使用日志缓冲。我想引入类似 C atexit() 函数的东西，以在意外退出时刷新所有待处理的日志。

我找到了 atexit库，但在我的例子中它是不够的，因为它不允许将参数传递给处理函数。我决定编写我的版本并以大致相似的结构结束。我有模块 atexit:

package atexit

import (
    "fmt"
    "os"
    "reflect"
)

var handlers []interface{}
var params []interface{}

func runHandler(handler interface{}, p interface{}) {
    defer func() {
        if err := recover(); err != nil {
            fmt.Fprintln(os.Stderr, "error: atexit handler error:", err)
        }
    }()

    f := reflect.ValueOf(handler)
    s := reflectSlice(p)
    fmt.Printf("%#v", s)
    f.Call(s)
}

func reflectSlice(slice interface{}) []reflect.Value {
    s := reflect.ValueOf(slice)
    if s.Kind() != reflect.Slice {
        panic("InterfaceSlice() given a non-slice type")
    }

    ret := make([]reflect.Value, s.Len())

    for i:=0; i<s.Len(); i++ {
        ret[i] = reflect.ValueOf(s.Index(i))
    }

    return ret
}

func runHandlers() {
    for i, handler := range handlers {
        runHandler(handler, params[i])
    }
}

func Exit(code int) {
    runHandlers()
    os.Exit(code)
}

func Register(handler interface{}, p interface{}) {
    f := reflect.TypeOf(handler)
    if f.Kind() != reflect.Func {
        panic("Register() given a non-function type")
    }
    handlers = append(handlers, handler)
    params = append(params, p)
}

我从主程序中调用它:

package main
import (
    "fmt"
    "./atexit"
    "encoding/json"
    "reflect"
)

type Batch struct {
    Index string        `json:"index"`
    Type string     `json:"_Type"`
    Content interface{} `json:"Content"`
}

func flush(b ...interface{}) {
    for _, entry := range(b) {
        fmt.Printf("%#v\n", entry)
        fmt.Println(reflect.TypeOf(reflect.ValueOf(entry)))
        fmt.Println(reflect.TypeOf(entry))
        a, err  := json.Marshal(entry)
        if err != nil {
            fmt.Println("error!")
        }
        fmt.Println(string(a))
    }
}

func handler(v ...interface{}) {
    fmt.Println("Exiting")
    for _, batch := range(v) {
        fmt.Printf("%#v\n", batch)
    }
}

func main() {
    type ColorGroup struct {
        ID     int
        Name   string
        Colors []string
    }
    group := ColorGroup{
        ID:     1,
        Name:   "Reds",
        Colors: []string{"Crimson", "Red", "Ruby", "Maroon"},
    }
    fmt.Printf("%#v\n", group)
    r, err := json.Marshal(group)
    if err != nil {
        fmt.Println("error:", err)
    }
    fmt.Println(string(r))


    b := []Batch{Batch{"index", "type", "content1"},Batch{"index", "type", "content2"}}
    atexit.Register(handler, b)
    atexit.Register(flush, []ColorGroup{group})
    atexit.Exit(0)
}

如您所见，通过调用 reflect.ValueOf()，我得到结构 reflect.Value，然后将其传递给回调函数。问题似乎是此结构不包含有关 json 导出的元数据或未使用 json.Marshal() 正确处理，然后输出空 json。有什么方法可以将正确的 []interface{} 结构传递给回调函数或一些类似的机制，这些机制可以大致完成我想要完成的任务？请注意，我想要一个通用的回调机制，它应该独立于传递给它的类型。到目前为止，这似乎是不可能的，或者至少受到 f.Call(s) 的限制，它在 runHandler() 中调用，并以 reflect.Value 作为参数。

最佳答案

你不应该为此使用反射。你应该让你的处理程序成为 func() 类型，然后传入一个匿名函数(可能是一个闭包)来直接处理任何必要的参数。示例:

atexit.Register(func() {
    handler(b)
})

关于go - 如何在 Golang 中重新输入一个函数？，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/50132870/