Method HealthCheck

dgraphtest/local_cluster.go:729–782

(zeroOnly bool)

}

func (c *LocalCluster) HealthCheck(zeroOnly bool) error {
	log.Printf("[INFO] checking health of containers")
	var wg sync.WaitGroup
	errChan := make(chan error, len(c.zeros)+len(c.alphas))

	for _, zo := range c.zeros {
		if !zo.isRunning {
			break
		}
		wg.Add(1)
		go func(z *zero) {
			defer wg.Done()
			if err := c.containerHealthCheck(z.healthURL); err != nil {
				errChan <- err
				return
			}
			log.Printf("[INFO] container [%v] passed health check", z.containerName)

			if err := c.checkDgraphVersion(z.containerName); err != nil {
				errChan <- err
			}
		}(zo)
	}

	if !zeroOnly {
		for _, aa := range c.alphas {
			if !aa.isRunning {
				break
			}
			wg.Add(1)
			go func(a *alpha) {
				defer wg.Done()
				if err := c.containerHealthCheck(a.healthURL); err != nil {
					errChan <- err
					return
				}
				log.Printf("[INFO] container [%v] passed health check", a.containerName)

				if err := c.checkDgraphVersion(a.containerName); err != nil {
					errChan <- err
				}
			}(aa)
		}
	}

	wg.Wait()
	close(errChan)

	for err := range errChan {
		return err
	}

	return nil
}

func (c *LocalCluster) containerHealthCheck(url func(c *LocalCluster) (string, error)) error {
	endpoint, err := url(c)