Add workflowId and workflowExecutionId information

flyteorg · Oct 28, 2020 · 077d75d · 077d75d
1 parent 5a41bd9
commit 077d75d
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 11 deletions.
diff --git a/go/tasks/plugins/hive/client/qubole_client.go b/go/tasks/plugins/hive/client/qubole_client.go
@@ -44,12 +44,14 @@ type QuboleCommandDetails struct {
 }
 
 type CommandMetadata struct {
-	TaskName      string
-	Domain        string
-	Project       string
-	Labels        map[string]string
-	AttemptNumber uint32
-	MaxAttempts   uint32
+	TaskName            string
+	Domain              string
+	Project             string
+	Labels              map[string]string
+	AttemptNumber       uint32
+	MaxAttempts         uint32
+	WorkflowID          string
+	WorkflowExecutionID string
 }
 
 // QuboleClient API Request Body, meant to be passed into JSON.marshal

diff --git a/go/tasks/plugins/hive/execution_state.go b/go/tasks/plugins/hive/execution_state.go
@@ -7,6 +7,7 @@ import (
 	"time"
 
 	"github.com/lyft/flytestdlib/cache"
+	"github.com/lyft/flytestdlib/contextutils"
 
 	idlCore "github.com/lyft/flyteidl/gen/pb-go/flyteidl/core"
 	"github.com/lyft/flyteidl/gen/pb-go/flyteidl/plugins"
@@ -344,11 +345,13 @@ func KickOffQuery(ctx context.Context, tCtx core.TaskExecutionContext, currentSt
 
 	taskExecutionIdentifier := tCtx.TaskExecutionMetadata().GetTaskExecutionID().GetID()
 	commandMetadata := client.CommandMetadata{TaskName: taskName,
-		Domain:        taskExecutionIdentifier.GetTaskId().GetDomain(),
-		Project:       taskExecutionIdentifier.GetNodeExecutionId().GetExecutionId().GetProject(),
-		Labels:        tCtx.TaskExecutionMetadata().GetLabels(),
-		AttemptNumber: taskExecutionIdentifier.GetRetryAttempt(),
-		MaxAttempts:   tCtx.TaskExecutionMetadata().GetMaxAttempts(),
+		Domain:              taskExecutionIdentifier.GetNodeExecutionId().GetExecutionId().GetDomain(),
+		Project:             taskExecutionIdentifier.GetNodeExecutionId().GetExecutionId().GetProject(),
+		Labels:              tCtx.TaskExecutionMetadata().GetLabels(),
+		AttemptNumber:       taskExecutionIdentifier.GetRetryAttempt(),
+		MaxAttempts:         tCtx.TaskExecutionMetadata().GetMaxAttempts(),
+		WorkflowExecutionID: taskExecutionIdentifier.GetNodeExecutionId().GetExecutionId().GetName(),
+		WorkflowID:          contextutils.Value(ctx, contextutils.WorkflowIDKey),
 	}
 
 	cmdDetails, err := quboleClient.ExecuteHiveCommand(ctx, query, timeoutSec,