echo: add server header middleware

1 files changed, 16 insertions, 0 deletions

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)
+                }
+        }
+}