Function WaitForRestore

dgraphapi/cluster.go:523–567

WaitForRestore waits for restore to complete on all alphas

(c Cluster)

// WaitForRestore waits for restore to complete on all alphas
func WaitForRestore(c Cluster) error {
loop:
	for {
		time.Sleep(waitDurBeforeRetry)

		resp, err := c.AlphasHealth()
		if err != nil && strings.Contains(err.Error(), "the server is in draining mode") {
			continue loop
		} else if err != nil {
			return err
		}
		for _, hr := range resp {
			if strings.Contains(hr, "opRestore") {
				continue loop
			}
		}
		break
	}

	// sometimes it is possible that one alpha hasn't even started the restore process.
	// In this case, the alpha will not show "opRestore" in the response to health endpoint
	// and the "loop" will not be able to detect this case. The "loop2" below ensures
	// by looking into the logs that the restore process has in fact been started and completed.
loop2:
	for range 60 {
		time.Sleep(waitDurBeforeRetry)

		alphasLogs, err := c.AlphasLogs()
		if err != nil {
			return err
		}

		for _, alphaLogs := range alphasLogs {
			// It is possible that the alpha didn't do a restore but streamed snapshot from any other alpha
			if !strings.Contains(alphaLogs, "Operation completed with id: opRestore") &&
				!strings.Contains(alphaLogs, "Operation completed with id: opSnapshot") {

				continue loop2
			}
		}
		return nil
	}

	return errors.Errorf("restore wasn't started on at least 1 alpha")
}

func (hc *HTTPClient) Export(dest, format string, namespace int) error {
	const exportRequest = `mutation export($dest: String!, $f: String!, $ns: Int) {