1 files changed, 29 insertions, 0 deletions

diff --git a/echo/middleware/cache_headers_middleware.go b/echo/middleware/cache_headers_middleware.go
new file mode 100644
index 0000000..f168dfd
--- /dev/null
+++ b/echo/middleware/cache_headers_middleware.go
@@ -0,0 +1,29 @@
+package middleware
+
+import (
+        "fmt"
+        "time"
+
+        gltime "code.crute.us/mcrute/golib/time"
+
+        "github.com/labstack/echo/v4"
+)
+
+var (
+        CacheOneHourMiddleware  = CacheHeadersMiddleware(1 * time.Hour)
+        CacheOneDayMiddleware   = CacheHeadersMiddleware(1 * gltime.Day)
+        CacheOneMonthMiddleware = CacheHeadersMiddleware(30 * gltime.Day)
+)
+
+func CacheHeadersMiddleware(d time.Duration) echo.MiddlewareFunc {
+        ds := int(d.Seconds())
+
+        return func(next echo.HandlerFunc) echo.HandlerFunc {
+                return func(c echo.Context) error {
+                        c.Response().Header().Set("Vary", "Accept-Encoding")
+                        c.Response().Header().Set("Cache-Control", fmt.Sprintf("public, max-age=%d", ds))
+                        c.Response().Header().Set("Expires", time.Now().Add(d).Format(time.RFC1123))
+                        return next(c)
+                }
+        }
+}