From f6761ddd00762ef19729d818c1872c4ad0a5bfa4 Mon Sep 17 00:00:00 2001 From: Nolan Brubaker Date: Wed, 16 May 2018 15:47:29 -0400 Subject: [PATCH] Check restore Phase before downloading logs Signed-off-by: Nolan Brubaker --- pkg/cmd/cli/restore/logs.go | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/pkg/cmd/cli/restore/logs.go b/pkg/cmd/cli/restore/logs.go index 64ffff1f56..c7281c03b4 100644 --- a/pkg/cmd/cli/restore/logs.go +++ b/pkg/cmd/cli/restore/logs.go @@ -20,6 +20,7 @@ import ( "os" "time" + "github.com/pkg/errors" "github.com/spf13/cobra" metav1 "k8s.io/apimachinery/pkg/apis/meta/v1" @@ -82,6 +83,12 @@ func (l *LogsOptions) Validate(f client.Factory) error { } l.client = c - _, err = l.client.ArkV1().Restores(f.Namespace()).Get(l.RestoreName, metav1.GetOptions{}) - return err + r, err := l.client.ArkV1().Restores(f.Namespace()).Get(l.RestoreName, metav1.GetOptions{}) + if err != nil { + return err + } + if r.Status.Phase != v1.RestorePhaseCompleted { + return errors.Errorf("unable to retrieve logs because restore is not complete") + } + return nil }