From 234d7aa7c9726af0c2f1c77db094e3421c47ec5e Mon Sep 17 00:00:00 2001 From: Kevin Su Date: Wed, 11 Oct 2023 23:39:30 -0700 Subject: [PATCH] More logs in databricks plugin Signed-off-by: Kevin Su --- flyteplugins/go/tasks/plugins/webapi/databricks/plugin.go | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/flyteplugins/go/tasks/plugins/webapi/databricks/plugin.go b/flyteplugins/go/tasks/plugins/webapi/databricks/plugin.go index 238932d638..04b169c2de 100644 --- a/flyteplugins/go/tasks/plugins/webapi/databricks/plugin.go +++ b/flyteplugins/go/tasks/plugins/webapi/databricks/plugin.go @@ -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