Method runPlanChecks

backend/api/mcp/tool_change.go:349–399

runPlanChecks triggers plan checks and polls for results within the poll budget.

(ctx context.Context, planName string)

// runPlanChecks triggers plan checks and polls for results within the poll budget.
func (s *Server) runPlanChecks(ctx context.Context, planName string) *PlanCheckInfo {
	checkRunName := planName + "/planCheckRun"

	// Trigger plan checks. If this fails, still poll — CreatePlan already
	// initializes plan checks server-side, so they may be running regardless.
	_, _ = s.apiRequest(ctx, "/bytebase.v1.PlanService/RunPlanChecks", map[string]any{
		"name": planName,
	})

	pendingResult := &PlanCheckInfo{
		Status:       planCheckRunning,
		PlanCheckRun: checkRunName,
	}

	// Poll for results. Transient errors (404 before row visible, brief 500)
	// are retried within the budget rather than bailing immediately.
	deadline := time.Now().Add(s.planCheckBudget())
	for time.Now().Before(deadline) {
		if ctx.Err() != nil {
			return pendingResult
		}

		pollResp, err := s.apiRequest(ctx, "/bytebase.v1.PlanService/GetPlanCheckRun", map[string]any{
			"name": checkRunName,
		})
		if err != nil || pollResp.Status >= 400 {
			time.Sleep(planCheckPollInterval)
			continue
		}

		var checkRun planCheckRunResponse
		if err := json.Unmarshal(pollResp.Body, &checkRun); err != nil {
			time.Sleep(planCheckPollInterval)
			continue
		}

		switch checkRun.Status {
		case "DONE":
			return buildPlanCheckInfo(checkRun)
		case "FAILED", "CANCELED":
			return &PlanCheckInfo{Status: planCheckFailed}
		default:
			// RUNNING or unknown — keep polling.
		}

		time.Sleep(planCheckPollInterval)
	}

	// Budget exhausted.
	return pendingResult
}

// planCheckRunResponse mirrors the PlanCheckRun proto response.
type planCheckRunResponse struct {