Skip to content

Commit 25c6e3a

Browse files
committed
back to single background_job.yaml, just pass scale as param
1 parent 533f78c commit 25c6e3a

File tree

3 files changed

+10
-69
lines changed

3 files changed

+10
-69
lines changed

backend/btrixcloud/crawlmanager.py

+3-9
Original file line numberDiff line numberDiff line change
@@ -168,8 +168,7 @@ async def run_re_add_org_pages_job(
168168
)
169169

170170
async def run_optimize_pages_job(
171-
self,
172-
existing_job_id: Optional[str] = None,
171+
self, existing_job_id: Optional[str] = None, scale=3
173172
) -> str:
174173
"""run job to recalculate storage stats for the org"""
175174

@@ -179,17 +178,14 @@ async def run_optimize_pages_job(
179178
job_id = f"optimize-pages-{secrets.token_hex(5)}"
180179

181180
return await self._run_bg_job_with_ops_classes(
182-
job_id,
183-
job_type=BgJobType.OPTIMIZE_PAGES.value,
184-
migration_job=True,
181+
job_id, job_type=BgJobType.OPTIMIZE_PAGES.value, scale=scale
185182
)
186183

187184
async def _run_bg_job_with_ops_classes(
188185
self,
189186
job_id: str,
190187
job_type: str,
191188
oid: Optional[str] = None,
192-
migration_job: bool = False,
193189
**kwargs,
194190
) -> str:
195191
"""run background job with access to ops classes"""
@@ -204,9 +200,7 @@ async def _run_bg_job_with_ops_classes(
204200
if oid:
205201
params["oid"] = oid
206202

207-
template = "migration_job.yaml" if migration_job else "background_job.yaml"
208-
209-
data = self.templates.env.get_template(template).render(params)
203+
data = self.templates.env.get_template("background_job.yaml").render(params)
210204

211205
await self.create_from_yaml(data, namespace=DEFAULT_NAMESPACE)
212206

chart/app-templates/background_job.yaml

+7-1
Original file line numberDiff line numberDiff line change
@@ -5,13 +5,18 @@ metadata:
55
labels:
66
role: "background-job"
77
job_type: {{ job_type }}
8+
{% if oid %}
89
btrix.org: {{ oid }}
10+
{% endif %}
911

1012
spec:
1113
ttlSecondsAfterFinished: 90
1214
backoffLimit: 3
1315
template:
1416
spec:
17+
{% if scale %}
18+
parallelism: {{ scale }}
19+
{% endif %}
1520
restartPolicy: Never
1621
priorityClassName: bg-job
1722
podFailurePolicy:
@@ -35,9 +40,10 @@ spec:
3540
- name: BG_JOB_TYPE
3641
value: {{ job_type }}
3742

43+
{% if oid %}
3844
- name: OID
3945
value: {{ oid }}
40-
46+
{% endif %}
4147
- name: CRAWL_TYPE
4248
value: {{ crawl_type }}
4349

chart/app-templates/migration_job.yaml

-59
This file was deleted.

0 commit comments

Comments
 (0)