Add a section to the config file called "health". Within this section, "filecheckers" and "httpcheckers" list checks to run. Each check specifies a file or URI, a time interval for the check, and a threshold specifying how many times the check must fail to reach an unhealthy state. Document the new options in docs/configuration.md. Add unit testing for both types of checkers. Add an UnregisterAll function in the health package to support the unit tests, and an Unregister function for consistency with Register. Fix a string conversion problem in the health package's HTTP checker. Signed-off-by: Aaron Lehmann <aaron.lehmann@docker.com>
37 lines
934 B
Go
37 lines
934 B
Go
package checks
|
|
|
|
import (
|
|
"errors"
|
|
"net/http"
|
|
"os"
|
|
"strconv"
|
|
|
|
"github.com/docker/distribution/health"
|
|
)
|
|
|
|
// FileChecker checks the existence of a file and returns and error
|
|
// if the file exists, taking the application out of rotation
|
|
func FileChecker(f string) health.Checker {
|
|
return health.CheckFunc(func() error {
|
|
if _, err := os.Stat(f); err == nil {
|
|
return errors.New("file exists")
|
|
}
|
|
return nil
|
|
})
|
|
}
|
|
|
|
// HTTPChecker does a HEAD request and verifies if the HTTP status
|
|
// code return is a 200, taking the application out of rotation if
|
|
// otherwise
|
|
func HTTPChecker(r string) health.Checker {
|
|
return health.CheckFunc(func() error {
|
|
response, err := http.Head(r)
|
|
if err != nil {
|
|
return errors.New("error while checking: " + r)
|
|
}
|
|
if response.StatusCode != http.StatusOK {
|
|
return errors.New("downstream service returned unexpected status: " + strconv.Itoa(response.StatusCode))
|
|
}
|
|
return nil
|
|
})
|
|
}
|