1 files changed, 24 insertions, 0 deletions

diff --git a/echo/middleware/vary.go b/echo/middleware/vary.go
new file mode 100644
index 0000000..8f87d29
--- /dev/null
+++ b/echo/middleware/vary.go
@@ -0,0 +1,24 @@
+package middleware
+
+import (
+        "github.com/labstack/echo/v4"
+)
+
+type VaryConfig struct {
+        Vary []string
+}
+
+func VaryCookie() echo.MiddlewareFunc {
+        return VaryWithConfig(VaryConfig{Vary: []string{"Cookie"}})
+}
+
+func VaryWithConfig(cfg VaryConfig) echo.MiddlewareFunc {
+        return func(next echo.HandlerFunc) echo.HandlerFunc {
+                return func(c echo.Context) error {
+                        for _, v := range cfg.Vary {
+                                c.Response().Header().Add(echo.HeaderVary, v)
+                        }
+                        return next(c)
+                }
+        }
+}