From eab7fb853464db00bc605f905d64655331cb4eb0 Mon Sep 17 00:00:00 2001 From: Mike Crute Date: Sat, 21 May 2022 19:31:04 -0700 Subject: echo: add server header middleware --- echo/middleware/add_server_header.go | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) create mode 100644 echo/middleware/add_server_header.go diff --git a/echo/middleware/add_server_header.go b/echo/middleware/add_server_header.go new file mode 100644 index 0000000..ef8345f --- /dev/null +++ b/echo/middleware/add_server_header.go @@ -0,0 +1,16 @@ +package middleware + +import ( + "fmt" + + "github.com/labstack/echo/v4" +) + +func AddServerHeader(app, version string) echo.MiddlewareFunc { + return func(next echo.HandlerFunc) echo.HandlerFunc { + return func(c echo.Context) error { + c.Response().Header().Add("Server", fmt.Sprintf("%s/%s", app, version)) + return next(c) + } + } +} -- cgit v1.2.3