mirror of https://github.com/k3s-io/k3s
30 lines
789 B
Go
30 lines
789 B
Go
package cluster
|
|
|
|
import (
|
|
"net/http"
|
|
)
|
|
|
|
// getHandler returns a basic request handler that processes requests through
|
|
// the cluster's request router chain.
|
|
func (c *Cluster) getHandler(handler http.Handler) (http.Handler, error) {
|
|
next := c.router()
|
|
|
|
return http.HandlerFunc(func(rw http.ResponseWriter, req *http.Request) {
|
|
handler.ServeHTTP(rw, req)
|
|
next.ServeHTTP(rw, req)
|
|
}), nil
|
|
}
|
|
|
|
// router is a stub request router that returns a Service Unavailable response
|
|
// if no additional handlers are available.
|
|
func (c *Cluster) router() http.Handler {
|
|
return http.HandlerFunc(func(rw http.ResponseWriter, req *http.Request) {
|
|
if c.runtime.Handler == nil {
|
|
http.Error(rw, "starting", http.StatusServiceUnavailable)
|
|
return
|
|
}
|
|
|
|
c.runtime.Handler.ServeHTTP(rw, req)
|
|
})
|
|
}
|