程序员的快乐如此简单

最近在GitHub上发起了一个关于Beego框架的小插件的开源仓库，这一举动虽然看似微小，但其中的快乐和意义却是无法用言语表达的。

Beego是一个开源的Go语言Web框架，它采用了MVC架构模式，并集成了很多常用的功能和中间件。小插件是指与Beego框架配套使用的、可扩展的、独立的软件模块或组件。通过开发小插件，可以扩展Beego框架的功能，提高开发效率和代码可维护性。

Beego-Requestid是一种中间件（Middleware），用于在处理HTTP请求时，为每个请求生成一个唯一的ID，并将其附加到请求上下文中。这样，您可以在应用程序的其他地方方便地访问这个唯一的请求ID，从而更好地跟踪和调试应用程序。

使用RequestId中间件可以帮助您解决以下问题：

跟踪请求：通过在每个请求中添加唯一的ID，您可以轻松地在应用程序中跟踪请求的来源和路径。这对于调试和性能优化非常有用。
日志记录：您可以使用请求ID将日志记录与特定请求关联起来。这样，当您在日志中查找特定请求的信息时，可以更方便地定位相关的日志条目。
错误处理：如果应用程序中出现错误，请求ID可以帮助您识别是哪个请求引发了错误。这对于故障排查和问题报告非常有帮助。

代码

话不多说直接看源码：

package beego_requestidimport ("github.com/beego/beego""github.com/beego/beego/context""github.com/google/uuid"
)const DefaultHeaderReqIdKey = "X-Request-Id"type Option func(config *Config)type GenRequestIdFunc func() stringtype Config struct {genRequestIdFunc               GenRequestIdFuncheaderReqIdKey, customReqIdKey string
}func NewFilter(opts ...Option) beego.FilterFunc {cnf := &Config{genRequestIdFunc: DefaultGenRequestIdFunc,headerReqIdKey:   DefaultHeaderReqIdKey,}for _, opt := range opts {opt(cnf)}return func(c *context.Context) {reqId := c.Request.Header.Get(cnf.headerReqIdKey)if reqId == "" {reqId = cnf.genRequestIdFunc()c.Request.Header.Add(cnf.headerReqIdKey, reqId)}if cnf.customReqIdKey != "" {c.Input.SetData(cnf.customReqIdKey, reqId)}}
}func WithGenRequestIdFunc(genFunc GenRequestIdFunc) Option {return func(config *Config) {config.genRequestIdFunc = genFunc}
}func WithHeaderReqIdKey(key string) Option {return func(config *Config) {config.headerReqIdKey = key}
}func WithCustomReqIdKey(key string) Option {return func(config *Config) {config.customReqIdKey = key}
}func DefaultGenRequestIdFunc() string {return uuid.NewString()
}

使用示例：

package mainimport ("log""time""github.com/spf13/cast""github.com/beego/beego""github.com/beego/beego/context"beego_requestid "github.com/ibarryyan/beego-requestid"
)func example1() {beego.InsertFilter("/*", beego.BeforeRouter, beego_requestid.NewFilter())beego.Get("/hello", func(c *context.Context) {reqId := c.Request.Header.Get("X-Request-Id")log.Printf("reqestid = %s", reqId)_, _ = c.ResponseWriter.Write([]byte("hello..."))return})beego.Run(":9900")
}func example2() {beego.InsertFilter("/*", beego.BeforeRouter, beego_requestid.NewFilter(beego_requestid.WithGenRequestIdFunc(func() string {return cast.ToString(time.Now().Unix())}),beego_requestid.WithHeaderReqIdKey("my_header_reqid"),beego_requestid.WithCustomReqIdKey("my_reqid"),))beego.Get("/hello", func(c *context.Context) {reqId := c.Request.Header.Get("my_header_reqid")log.Printf("reqestid = %s", reqId)cReqId := c.Input.GetData("my_reqid")log.Printf("my reqestid = %s", cReqId)_, _ = c.ResponseWriter.Write([]byte("hello..."))return})beego.Run(":9900")
}