func (p *dockerProvisioner) runRestartAfterHooks(cont *container.Container, w io.Writer) error {
yamlData, err := getImageTsuruYamlData(cont.Image)
if err != nil {
return err
}
cmds := yamlData.Hooks.Restart.After
for _, cmd := range cmds {
err := cont.Exec(p, w, w, cmd)
if err != nil {
return fmt.Errorf("couldn't execute restart:after hook %q(%s): %s", cmd, cont.ShortID(), err.Error())
}
}
return nil
}
func runHealthcheck(cont *container.Container, w io.Writer) error {
yamlData, err := getImageTsuruYamlData(cont.Image)
if err != nil {
return err
}
path := yamlData.Healthcheck.Path
method := yamlData.Healthcheck.Method
match := yamlData.Healthcheck.Match
status := yamlData.Healthcheck.Status
allowedFailures := yamlData.Healthcheck.AllowedFailures
if path == "" {
return nil
}
path = strings.TrimSpace(strings.TrimLeft(path, "/"))
if method == "" {
method = "get"
}
method = strings.ToUpper(method)
if status == 0 && match == "" {
status = 200
}
var matchRE *regexp.Regexp
if match != "" {
match = "(?s)" + match
matchRE, err = regexp.Compile(match)
if err != nil {
return err
}
}
maxWaitTime, _ := config.GetInt("docker:healthcheck:max-time")
if maxWaitTime == 0 {
maxWaitTime = 120
}
maxWaitTime = maxWaitTime * int(time.Second)
sleepTime := 3 * time.Second
startedTime := time.Now()
url := fmt.Sprintf("http://%s:%s/%s", cont.HostAddr, cont.HostPort, path)
for {
var lastError error = nil
req, err := http.NewRequest(method, url, nil)
if err != nil {
return err
}
rsp, err := timeoutHttpClient.Do(req)
if err != nil {
lastError = fmt.Errorf("healthcheck fail(%s): %s", cont.ShortID(), err.Error())
} else {
defer rsp.Body.Close()
if status != 0 && rsp.StatusCode != status {
lastError = fmt.Errorf("healthcheck fail(%s): wrong status code, expected %d, got: %d", cont.ShortID(), status, rsp.StatusCode)
} else if matchRE != nil {
result, err := ioutil.ReadAll(rsp.Body)
if err != nil {
lastError = err
}
if !matchRE.Match(result) {
lastError = fmt.Errorf("healthcheck fail(%s): unexpected result, expected %q, got: %s", cont.ShortID(), match, string(result))
}
}
if lastError != nil {
if allowedFailures == 0 {
return lastError
}
allowedFailures--
}
}
if lastError == nil {
fmt.Fprintf(w, " ---> healthcheck successful(%s)\n", cont.ShortID())
return nil
}
if time.Now().Sub(startedTime) > time.Duration(maxWaitTime) {
return lastError
}
fmt.Fprintf(w, " ---> %s. Trying again in %s\n", lastError.Error(), sleepTime)
time.Sleep(sleepTime)
}
}
