committed by
							
								 GitHub
								GitHub
							
						
					
				
				 1 changed files with 122 additions and 0 deletions
			
			
		| @ -0,0 +1,122 @@ | |||
| // Copyright 2015 Matthew Holt and The Caddy Authors
 | |||
| //
 | |||
| // Licensed under the Apache License, Version 2.0 (the "License");
 | |||
| // you may not use this file except in compliance with the License.
 | |||
| // You may obtain a copy of the License at
 | |||
| //
 | |||
| //     http://www.apache.org/licenses/LICENSE-2.0
 | |||
| //
 | |||
| // Unless required by applicable law or agreed to in writing, software
 | |||
| // distributed under the License is distributed on an "AS IS" BASIS,
 | |||
| // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 | |||
| // See the License for the specific language governing permissions and
 | |||
| // limitations under the License.
 | |||
| 
 | |||
| package reverseproxy | |||
| 
 | |||
| import ( | |||
| 	"encoding/json" | |||
| 	"fmt" | |||
| 	"net/http" | |||
| 
 | |||
| 	"github.com/caddyserver/caddy/v2" | |||
| ) | |||
| 
 | |||
| func init() { | |||
| 	caddy.RegisterModule(adminUpstreams{}) | |||
| } | |||
| 
 | |||
| // adminUpstreams is a module that provides the
 | |||
| // /reverse_proxy/upstreams endpoint for the Caddy admin
 | |||
| // API. This allows for checking the health of configured
 | |||
| // reverse proxy upstreams in the pool.
 | |||
| type adminUpstreams struct{} | |||
| 
 | |||
| // upstreamResults holds the status of a particular upstream
 | |||
| type upstreamStatus struct { | |||
| 	Address     string `json:"address"` | |||
| 	Healthy     bool   `json:"healthy"` | |||
| 	NumRequests int    `json:"num_requests"` | |||
| 	Fails       int    `json:"fails"` | |||
| } | |||
| 
 | |||
| // CaddyModule returns the Caddy module information.
 | |||
| func (adminUpstreams) CaddyModule() caddy.ModuleInfo { | |||
| 	return caddy.ModuleInfo{ | |||
| 		ID:  "admin.api.reverse_proxy", | |||
| 		New: func() caddy.Module { return new(adminUpstreams) }, | |||
| 	} | |||
| } | |||
| 
 | |||
| // Routes returns a route for the /reverse_proxy/upstreams endpoint.
 | |||
| func (al adminUpstreams) Routes() []caddy.AdminRoute { | |||
| 	return []caddy.AdminRoute{ | |||
| 		{ | |||
| 			Pattern: "/reverse_proxy/upstreams", | |||
| 			Handler: caddy.AdminHandlerFunc(al.handleUpstreams), | |||
| 		}, | |||
| 	} | |||
| } | |||
| 
 | |||
| // handleUpstreams reports the status of the reverse proxy
 | |||
| // upstream pool.
 | |||
| func (adminUpstreams) handleUpstreams(w http.ResponseWriter, r *http.Request) error { | |||
| 	if r.Method != http.MethodGet { | |||
| 		return caddy.APIError{ | |||
| 			HTTPStatus: http.StatusMethodNotAllowed, | |||
| 			Err:        fmt.Errorf("method not allowed"), | |||
| 		} | |||
| 	} | |||
| 
 | |||
| 	// Prep for a JSON response
 | |||
| 	w.Header().Set("Content-Type", "application/json") | |||
| 	enc := json.NewEncoder(w) | |||
| 
 | |||
| 	// Collect the results to respond with
 | |||
| 	results := []upstreamStatus{} | |||
| 
 | |||
| 	// Iterate over the upstream pool (needs to be fast)
 | |||
| 	var rangeErr error | |||
| 	hosts.Range(func(key, val interface{}) bool { | |||
| 		address, ok := key.(string) | |||
| 		if !ok { | |||
| 			rangeErr = caddy.APIError{ | |||
| 				HTTPStatus: http.StatusInternalServerError, | |||
| 				Err:        fmt.Errorf("could not type assert upstream address"), | |||
| 			} | |||
| 			return false | |||
| 		} | |||
| 
 | |||
| 		upstream, ok := val.(*upstreamHost) | |||
| 		if !ok { | |||
| 			rangeErr = caddy.APIError{ | |||
| 				HTTPStatus: http.StatusInternalServerError, | |||
| 				Err:        fmt.Errorf("could not type assert upstream struct"), | |||
| 			} | |||
| 			return false | |||
| 		} | |||
| 
 | |||
| 		results = append(results, upstreamStatus{ | |||
| 			Address:     address, | |||
| 			Healthy:     !upstream.Unhealthy(), | |||
| 			NumRequests: upstream.NumRequests(), | |||
| 			Fails:       upstream.Fails(), | |||
| 		}) | |||
| 		return true | |||
| 	}) | |||
| 
 | |||
| 	// If an error happened during the range, return it
 | |||
| 	if rangeErr != nil { | |||
| 		return rangeErr | |||
| 	} | |||
| 
 | |||
| 	err := enc.Encode(results) | |||
| 	if err != nil { | |||
| 		return caddy.APIError{ | |||
| 			HTTPStatus: http.StatusInternalServerError, | |||
| 			Err:        err, | |||
| 		} | |||
| 	} | |||
| 
 | |||
| 	return nil | |||
| } | |||
					Loading…
					
					
				
		Reference in new issue