mirror of
https://github.com/woodpecker-ci/woodpecker.git
synced 2024-12-12 08:23:48 +02:00
25 lines
529 B
Go
25 lines
529 B
Go
|
package middleware
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/drone/drone/cache"
|
||
|
|
|
||
|
|
"github.com/codegangsta/cli"
|
||
|
|
"github.com/gin-gonic/gin"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Cache is a middleware function that initializes the Cache and attaches to
|
||
|
|
// the context of every http.Request.
|
||
|
|
func Cache(cli *cli.Context) gin.HandlerFunc {
|
||
|
|
v := setupCache(cli)
|
||
|
|
return func(c *gin.Context) {
|
||
|
|
cache.ToContext(c, v)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// helper function to create the cache from the CLI context.
|
||
|
|
func setupCache(c *cli.Context) cache.Cache {
|
||
|
|
return cache.NewTTL(
|
||
|
|
c.Duration("cache-ttl"),
|
||
|
|
)
|
||
|
|
}
|