Skip to content

Commit f2ce384

Browse files
committed
move last_node_id to sequencer
1 parent c15e230 commit f2ce384

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

src/databricks/labs/ucx/sequencing/sequencing.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,6 @@ class MigrationStep:
2323

2424
@dataclass
2525
class MigrationNode:
26-
last_node_id = 0
2726
node_id: int
2827
object_type: str
2928
object_id: str
@@ -69,6 +68,7 @@ class MigrationSequencer:
6968

7069
def __init__(self, ws: WorkspaceClient):
7170
self._ws = ws
71+
self._last_node_id = 0
7272
self._root = MigrationNode(
7373
node_id=0, object_type="ROOT", object_id="ROOT", object_name="ROOT", object_owner="NONE"
7474
)
@@ -79,9 +79,9 @@ def register_workflow_task(self, task: jobs.Task, job: jobs.Job, _graph: Depende
7979
if task_node:
8080
return task_node
8181
job_node = self.register_workflow_job(job)
82-
MigrationNode.last_node_id += 1
82+
self._last_node_id += 1
8383
task_node = MigrationNode(
84-
node_id=MigrationNode.last_node_id,
84+
node_id=self._last_node_id,
8585
object_type="TASK",
8686
object_id=task_id,
8787
object_name=task.task_key,
@@ -100,10 +100,10 @@ def register_workflow_job(self, job: jobs.Job) -> MigrationNode:
100100
job_node = self._find_node(object_type="JOB", object_id=str(job.job_id))
101101
if job_node:
102102
return job_node
103-
MigrationNode.last_node_id += 1
103+
self._last_node_id += 1
104104
job_name = job.settings.name if job.settings and job.settings.name else str(job.job_id)
105105
job_node = MigrationNode(
106-
node_id=MigrationNode.last_node_id,
106+
node_id=self._last_node_id,
107107
object_type="JOB",
108108
object_id=str(job.job_id),
109109
object_name=job_name,
@@ -132,9 +132,9 @@ def register_cluster(self, cluster_key: str) -> MigrationNode:
132132
details = self._ws.clusters.get(cluster_key)
133133
object_name = details.cluster_name if details and details.cluster_name else cluster_key
134134
object_owner = details.creator_user_name if details and details.creator_user_name else "<UNKNOWN>"
135-
MigrationNode.last_node_id += 1
135+
self._last_node_id += 1
136136
cluster_node = MigrationNode(
137-
node_id=MigrationNode.last_node_id,
137+
node_id=self._last_node_id,
138138
object_type="CLUSTER",
139139
object_id=cluster_key,
140140
object_name=object_name,

0 commit comments

Comments
 (0)