Skip to content

Commit

Permalink
More logs in databricks plugin
Browse files Browse the repository at this point in the history
Signed-off-by: Kevin Su <[email protected]>
  • Loading branch information
pingsutw committed Oct 12, 2023
1 parent 0ca2d22 commit 234d7aa
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions flyteplugins/go/tasks/plugins/webapi/databricks/plugin.go
Original file line number Diff line number Diff line change
Expand Up @@ -152,13 +152,17 @@ func (p Plugin) Create(ctx context.Context, taskCtx webapi.TaskExecutionContextR

func (p Plugin) Get(ctx context.Context, taskCtx webapi.GetContext) (latest webapi.Resource, err error) {
exec := taskCtx.ResourceMeta().(ResourceMetaWrapper)
logger.Info(ctx, "Get databricks job status", "runID", exec.RunID)
logger.Info(ctx, "Get databricks job token", "token", exec.Token)
req, err := buildRequest(get, nil, p.cfg.databricksEndpoint,
p.cfg.DatabricksInstance, exec.Token, exec.RunID, false)
if err != nil {
logger.Errorf(ctx, "Failed to build databricks job request [%v]", err)
return nil, err
}
logger.Info(ctx, "Get databricks job request", "req", req)
resp, err := p.client.Do(req)
logger.Info(ctx, "Get databricks job response", "resp", resp)
if err != nil {
logger.Errorf(ctx, "Failed to get databricks job status [%v]", resp)
return nil, err
Expand Down

0 comments on commit 234d7aa

Please sign in to comment.