aboutsummaryrefslogtreecommitdiff
path: root/echo/middleware/cache_headers_middleware.go
blob: f168dfd563552d30598d21e085776b790a89ee15 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
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)
		}
	}
}