2016-06-06 11:51:56 +08:00
|
|
|
package proxy
|
|
|
|
|
|
|
|
import (
|
|
|
|
"github.com/mholt/caddy"
|
|
|
|
"github.com/mholt/caddy/caddyhttp/httpserver"
|
|
|
|
)
|
|
|
|
|
|
|
|
func init() {
|
2016-06-07 05:31:03 +08:00
|
|
|
caddy.RegisterPlugin("proxy", caddy.Plugin{
|
2016-06-06 11:51:56 +08:00
|
|
|
ServerType: "http",
|
|
|
|
Action: setup,
|
|
|
|
})
|
|
|
|
}
|
|
|
|
|
|
|
|
// setup configures a new Proxy middleware instance.
|
|
|
|
func setup(c *caddy.Controller) error {
|
2017-04-17 23:58:47 +08:00
|
|
|
upstreams, err := NewStaticUpstreams(c.Dispenser, httpserver.GetConfig(c).Host())
|
2016-06-06 11:51:56 +08:00
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
2016-06-21 01:44:20 +08:00
|
|
|
httpserver.GetConfig(c).AddMiddleware(func(next httpserver.Handler) httpserver.Handler {
|
2016-06-06 11:51:56 +08:00
|
|
|
return Proxy{Next: next, Upstreams: upstreams}
|
|
|
|
})
|
2017-04-03 04:58:15 +08:00
|
|
|
|
|
|
|
// Register shutdown handlers.
|
|
|
|
for _, upstream := range upstreams {
|
|
|
|
c.OnShutdown(upstream.Stop)
|
|
|
|
}
|
|
|
|
|
2016-06-06 11:51:56 +08:00
|
|
|
return nil
|
|
|
|
}
|