在Golang Echo中间件中，如何在执行next()后添加/设置标头？

哈喽！大家好，很高兴又见面了，我是golang学习网的一名作者，今天由我给大家带来一篇《在Golang Echo中间件中，如何在执行next()后添加/设置标头？》，本文主要会讲到等等知识点，希望大家一起学习进步，也欢迎大家关注、点赞、收藏、转发! 下面就一起来看看吧！

主要目的： 我想简单地计算任何 api/路由的执行时间，并将其添加到带有 key:executiontime 的响应标头中。

我对另一种选择非常开放，因为我的方法可能是错误的，因为我对 go-echo 框架还是新手。

我一直在做什么： 这是我的中间件代码：

func (s *stats) exectime(next echo.handlerfunc) echo.handlerfunc {
    return func(c echo.context) error {
        c.response().header().set("executionstartedat", time.now().string())
        if err := next(c); err != nil { //exec main process
            c.error(err)
        }
        c.response().header().set("executiondoneat", time.now().string())
        return nil
    }
}

当前结果：

< http/1.1 200 ok
< content-type: application/json; charset=utf-8
< executionstartedat: 2022-09-11 16:32:15.612045 +0700 wib m=+3.619364986
< server: echo/3.0
< date: sun, 11 sep 2022 09:32:21 gmt
< content-length: 856

请注意：添加了一个 executionstartedat。但不幸的是，没有像预期的那样 executiondoneat。

预期结果：

< http/1.1 200 ok
< content-type: application/json; charset=utf-8
< executionstartedat: 2022-09-11 16:32:15.612045 +0700 wib m=+3.619364986
< executiondoneat: 2022-09-11 16:32:18.612045 +0700 wib m=+3.619364986
< server: echo/3.0
< date: sun, 11 sep 2022 09:32:21 gmt
< content-length: 856

我正在关注此文档：https://echo.labstack.com/cookbook/middleware/#response

任何帮助、任何想法、任何替代方法来实现相同的结果将非常非常感激。非常感谢。

编辑#1：

我还尝试过什么：

我尝试过使用另一个中间件，时机很好，正如我预期的那样。但不幸的是标题仍然是空的。

e.use(middleware.bodydump(func(c echo.context, reqbody, resbody []byte) {
        fmt.println("this is from bodydump 7: ", time.now())
        c.response().header().set("executiontime7", "exec time: " + time.now().string())
    }))

结果和上面一模一样，没有任何变化。

已编辑#2

这是工作代码示例：

// ExecTime is the middleware function.
func (s *Stats) ExecTime(next echo.HandlerFunc) echo.HandlerFunc {
    return func(c echo.Context) error {
        before := time.Now()
        c.Response().Header().Set("ExecutionStartedAt", before.String())

        c.Response().Before(func() {
            after := time.Now()
            elapsed := time.Since(before)

            c.Response().Header().Set("ExecutionDoneAt", after.String())
            c.Response().Header().Set("ExecutionTime", elapsed.String())
        })

        if err := next(c); err != nil { //exec main process
            c.Error(err)
        }
        return nil
    }
}

正确答案

在 http 中，标头位于正文之前，正如名称“标头”所暗示的那样。所以一旦你写了正文，你就不能写标题了。您可以尝试使用 http/2 trailers。或者执行该进程，但在将其输出写入正文之前，计算执行时间，添加标头，然后写入正文。

我不使用 echo，所以我不知道正确的方法是什么，但在我看来，如果您使用响应 before hook 会更好。

@JackySupit 的示例：

// ExecTime is the middleware function.
func (s *Stats) ExecTime(next echo.HandlerFunc) echo.HandlerFunc {
    return func(c echo.Context) error {
        before := time.Now()
        c.Response().Header().Set("ExecutionStartedAt", before.String())

        c.Response().Before(func() {
            after := time.Now()
            elapsed := time.Since(before)

            c.Response().Header().Set("ExecutionDoneAt", after.String())
            c.Response().Header().Set("ExecutionTime", elapsed.String())
        })

        if err := next(c); err != nil { //exec main process
            c.Error(err)
        }
        return nil
    }
}

理论要掌握，实操不能落！以上关于《在Golang Echo中间件中，如何在执行next()后添加/设置标头？》的详细介绍，大家都掌握了吧！如果想要继续提升自己的能力，那么就来关注golang学习网公众号吧！