HGKeeper

* Convey

* This program is free software: you can redistribute it and/or modify

* it under the terms of the GNU General Public License as published by

* the Free Software Foundation, either version 3 of the License, or

* (at your option) any later version.

* This program is distributed in the hope that it will be useful,

* but WITHOUT ANY WARRANTY; without even the implied warranty of

* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the

* GNU General Public License for more details.

* You should have received a copy of the GNU General Public License

* along with this program. If not, see <http://www.gnu.org/licenses/>.

package docker

import (

"strings"

"time"

"github.com/aphistic/gomol"

"bitbucket.org/rw_grim/convey/state"

)

type HealthCheck struct {

Command string `yaml:"command"`

Interval time.Duration `yaml:"interval"`

Retries int `yaml:"retries"`

StartPeriod time.Duration `yaml:"start-period"`

Timeout time.Duration `yaml:"timeout"`

}

const (

containerHasHealthCheckTemplate = `inspect -f "{{.Format}}" {{.Cid}}`

containerHasHealthCheckFormat = `{{if .Config.Healthcheck}}true{{else}}false{{end}}`

containerIsHealthyTemplate = `inspect -f "{{.Format}}" {{.Cid}}`

containerIsHealthyFormat = `{{.State.Health.Status}}`

)

func containerHasHealthCheck(name, cid string, st *state.State, logger *gomol.LogAdapter) (bool, error) {

params := map[string]interface{}{

"Format": containerHasHealthCheckFormat,

"Cid": cid,

}

stdout, stderr, err := DockerOutput(

name+"/healthcheck",

containerHasHealthCheckTemplate,

params,

st,

)

if err != nil {

logger.Error(stderr)

return false, err

}

if strings.TrimSpace(stdout) == "true" {

return true, nil

}

return false, nil

}

func containerIsHealthy(name, cid string, st *state.State, logger *gomol.LogAdapter) (bool, error) {

params := map[string]interface{}{

"Format": containerIsHealthyFormat,