fix delete post data bug, add PENDING message info

Signed-off-by: Future Outlier <[email protected]>
flyteorg · Nov 5, 2023 · 1596536 · 1596536
1 parent 814b697
commit 1596536
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/flyteplugins/go/tasks/plugins/webapi/databricks/plugin.go b/flyteplugins/go/tasks/plugins/webapi/databricks/plugin.go
@@ -236,6 +236,11 @@ func (p Plugin) Status(ctx context.Context, taskCtx webapi.StatusContext) (phase
 			}
 			return pluginsCore.PhaseInfoFailure(string(rune(statusCode)), message, taskInfo), nil
 		}
+
+		if lifeCycleState == "PENDING" {
+			return core.PhaseInfoInitializing(time.Now(), core.DefaultPhaseVersion, message, taskInfo), nil
+		}
+
 		return core.PhaseInfoRunning(pluginsCore.DefaultPhaseVersion, taskInfo), nil
 	case http.StatusBadRequest:
 		fallthrough
@@ -283,7 +288,7 @@ func buildRequest(
 	var err error
 	if isCancel {
 		databricksURL += "/cancel"
-		data = []byte(fmt.Sprintf("{ run_id: %v }", runID))
+		data = []byte(fmt.Sprintf("{ \"run_id\": %v }", runID))
 	} else if method == post {
 		databricksURL += "/submit"
 		mJSON, err := json.Marshal(databricksJob)