示例#1
0
func retriveHealthCheckPort(service *bridge.Service) int {
	if portMappings := service.PortMappings(); len(portMappings) > 0 {
		if checkPort, ok := service.Attrs["check_port"]; ok {
			if p, err := strconv.Atoi(checkPort); err == nil {
				if port, ok := portMappings[p]; ok {
					return port
				}
			}
		}
	}
	return service.Port
}
示例#2
0
func (r *ConsulAdapter) buildCheck(service *bridge.Service) *consulapi.AgentServiceCheck {
	check := new(consulapi.AgentServiceCheck)

	if healthcheck := service.HealthCheck(); healthcheck != nil {
		if healthcheck["protocol"] == "http" {
			check.HTTP = fmt.Sprintf("http://%s:%s%s", service.IP, healthcheck["port"], healthcheck["path"])
			check.Timeout = healthcheck["timeoutInSeconds"] + "s"
			check.Interval = healthcheck["checkIntervalInSeconds"] + "s"
		} else if healthcheck["protocol"] == "tcp" {
			check.TCP = fmt.Sprintf("%s:%s", service.IP, healthcheck["port"])
			check.Timeout = healthcheck["timeoutInSeconds"] + "s"
			check.Interval = healthcheck["checkIntervalInSeconds"] + "s"
		} else {
			return nil
		}
	} else if path := service.Attrs["check_http"]; path != "" {
		check.HTTP = fmt.Sprintf("http://%s:%d%s", service.IP, retriveHealthCheckPort(service), path)
		if timeout := service.Attrs["check_timeout"]; timeout != "" {
			check.Timeout = timeout
		}
	} else if path := service.Attrs["check_https"]; path != "" {
		check.HTTP = fmt.Sprintf("https://%s:%d%s", service.IP, retriveHealthCheckPort(service), path)
		if timeout := service.Attrs["check_timeout"]; timeout != "" {
			check.Timeout = timeout
		}
	} else if cmd := service.Attrs["check_cmd"]; cmd != "" {
		check.Script = fmt.Sprintf("check-cmd %s %s %s", service.Origin.ContainerID[:12], service.Origin.ExposedPort, cmd)
	} else if script := service.Attrs["check_script"]; script != "" {
		check.Script = r.interpolateService(script, service)
	} else if ttl := service.Attrs["check_ttl"]; ttl != "" {
		check.TTL = ttl
	} else if tcp := service.Attrs["check_tcp"]; tcp != "" {
		check.TCP = fmt.Sprintf("%s:%d", service.IP, retriveHealthCheckPort(service))
		if timeout := service.Attrs["check_timeout"]; timeout != "" {
			check.Timeout = timeout
		}
	} else {
		return nil
	}

	if check.Interval == "" && (check.Script != "" || check.HTTP != "" || check.TCP != "") {
		if interval := service.Attrs["check_interval"]; interval != "" {
			check.Interval = interval
		} else {
			check.Interval = DefaultInterval
		}
	}
	return check
}