Bump flytestdlib

Signed-off-by: Kevin Su <[email protected]>
flyteorg · Sep 15, 2023 · a417872 · a417872
1 parent ac38261
commit a417872
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 0 deletions.
diff --git a/go/tasks/plugins/array/awsbatch/jobs_store.go b/go/tasks/plugins/array/awsbatch/jobs_store.go
@@ -49,6 +49,10 @@ type Job struct {
 	SubJobs        []*Job               `json:"array,omitempty"`
 }
 
+func (j Job) IsTerminal() bool {
+	return j.Status.Phase.IsTerminal()
+}
+
 type Attempt struct {
 	LogStream string    `json:"logStream,omitempty"`
 	StartedAt time.Time `json:"startedAt,omitempty"`

diff --git a/go/tasks/plugins/hive/execution_state.go b/go/tasks/plugins/hive/execution_state.go
@@ -36,6 +36,10 @@ const (
 	PhaseQueryFailed
 )
 
+func (p ExecutionPhase) IsTerminal() bool {
+	return p == PhaseQuerySucceeded || p == PhaseQueryFailed
+}
+
 func (p ExecutionPhase) String() string {
 	switch p {
 	case PhaseNotStarted:

diff --git a/go/tasks/plugins/hive/executions_cache.go b/go/tasks/plugins/hive/executions_cache.go
@@ -64,6 +64,10 @@ func (e ExecutionStateCacheItem) ID() string {
 	return e.Identifier
 }
 
+func (e ExecutionStateCacheItem) IsTerminal() bool {
+	return e.ExecutionState.Phase.IsTerminal()
+}
+
 // This basically grab an updated status from the Qubole API and store it in the cache
 // All other handling should be in the synchronous loop.
 func (q *QuboleHiveExecutionsCache) SyncQuboleQuery(ctx context.Context, batch cache.Batch) (