+ {#snippet header()}
+
+ {#if nconfig.init_bash !== undefined}
+ Enabled
+ {/if}
+
+ {/snippet}
+ {#if (nconfig.init_bash ?? '') !== (config?.init_bash ?? '')}
+
- Workers will get killed upon detecting any changes in this section (scripts or
- interval). It is assumed they are in an environment where the supervisor will restart
- them.
+ Workers will get killed upon detecting changes to the init script. It is assumed they
+ are in an environment where the supervisor will restart them.
+
+
+ {/if}
+
+ Execution logs are visible on the runs page of the admins workspace. Use the script editor
+ with Bash to iterate on your script before setting it here.
+
+
+ {
+ if (config) {
+ const code = e.detail
+ if (code != '') {
+ nconfig.init_bash = code?.replace(/\r\n/g, '\n')
+ } else {
+ nconfig.init_bash = undefined
+ }
+ }
+ }}
+ />
+
+
+
+
+
+
+ {#snippet header()}
+
+ {#if nconfig.periodic_script_bash !== undefined}
+ Enabled
+ {/if}
+
+ {/snippet}
+ {#if (nconfig.periodic_script_bash ?? '') !== (config?.periodic_script_bash ?? '') || (nconfig.periodic_script_interval_seconds ?? 0) !== (config?.periodic_script_interval_seconds ?? 0)}
+
+
+ Workers will get killed upon detecting changes to the periodic script or interval.
+ It is assumed they are in an environment where the supervisor will restart them.
-
-
- Run at start of the workers. More lightweight than requiring custom worker images.
-
-
- {#snippet header()}
-
- {#if nconfig.init_bash !== undefined}
- Enabled
- {/if}
-
- {/snippet}
-
- {
- if (config) {
- const code = e.detail
- if (code != '') {
- nconfig.init_bash = code?.replace(/\r\n/g, '\n')
- } else {
- nconfig.init_bash = undefined
- }
- }
- }}
- />
-
-
-
-
-
- Run periodically at configurable intervals. Useful for maintenance tasks like cleaning
- disk space.
-
-
- {#snippet header()}
-
- {#if nconfig.periodic_script_bash !== undefined}
- Enabled
- {/if}
-
- {/snippet}
+
+
+
-
-
-
-
-
- {
- if (config) {
- const code = e.detail
- if (code != '') {
- nconfig.periodic_script_bash = code?.replace(/\r\n/g, '\n')
- } else {
- nconfig.periodic_script_bash = undefined
- }
- }
- }}
- />
-
-
-
+
+ {
+ if (config) {
+ const code = e.detail
+ if (code != '') {
+ nconfig.periodic_script_bash = code?.replace(/\r\n/g, '\n')
+ } else {
+ nconfig.periodic_script_bash = undefined
+ }
+ }
+ }}
+ />
{#snippet actions()}
+ {#if onOpenYamlEditor}
+
+ {/if}
{#if canEditConfig}
-
- {#if config}
-
- {/if}
{:else}
{
- navigator.clipboard.writeText(YAML.stringify({ name, ...config }))
- sendUserToast('Worker config copied to clipboard as YAML')
- },
- disabled: !config
- },
{
displayName: 'Clean cache',
action: () => {
diff --git a/frontend/src/routes/(root)/(logged)/workers/+page.svelte b/frontend/src/routes/(root)/(logged)/workers/+page.svelte
index 71bbe84b96..03226b0539 100644
--- a/frontend/src/routes/(root)/(logged)/workers/+page.svelte
+++ b/frontend/src/routes/(root)/(logged)/workers/+page.svelte
@@ -29,6 +29,7 @@
import {
ExternalLink,
LineChart,
+ Loader2,
Plus,
Search,
Tags,
@@ -379,6 +380,111 @@
await loadWorkerGroups()
}
+ let yamlConfigDrawer: Drawer | undefined = $state(undefined)
+ let yamlConfigCode = $state('')
+ let yamlConfigOriginal = $state('')
+ let yamlDiffMode = $state(false)
+ let yamlSaving = $state(false)
+
+ function serializeWorkerGroupsAsYaml(groups: Record): string {
+ const priorityGroups = ['default', 'native']
+ const configs = Object.entries(groups)
+ .sort(([a], [b]) => {
+ const ai = priorityGroups.indexOf(a)
+ const bi = priorityGroups.indexOf(b)
+ if (ai !== -1 && bi !== -1) return ai - bi
+ if (ai !== -1) return -1
+ if (bi !== -1) return 1
+ return a.localeCompare(b)
+ })
+ .map(([name, { cache_clear, ...config }]) => {
+ const cleaned: Record = { name }
+ for (const [k, v] of Object.entries(config)) {
+ if (v == null) continue
+ if (Array.isArray(v) && v.length === 0) continue
+ if (typeof v === 'object' && !Array.isArray(v) && Object.keys(v).length === 0) continue
+ cleaned[k] = v
+ }
+ return cleaned
+ })
+ return YAML.stringify(configs)
+ }
+
+ function openYamlDrawer() {
+ if (!workerGroups) {
+ return sendUserToast('No worker groups found', true)
+ }
+ yamlConfigCode = serializeWorkerGroupsAsYaml(workerGroups)
+ yamlConfigOriginal = yamlConfigCode
+ yamlDiffMode = false
+ yamlConfigDrawer?.toggleDrawer?.()
+ }
+
+ async function saveYamlConfig() {
+ yamlSaving = true
+ try {
+ const parsed = YAML.parse(yamlConfigCode)
+ if (!Array.isArray(parsed)) {
+ sendUserToast('YAML must be an array of worker group configs', true)
+ return
+ }
+
+ const newGroups = new Map()
+ for (const c of parsed) {
+ if (typeof c !== 'object' || !c?.name || typeof c.name !== 'string') {
+ sendUserToast('Each entry must have a "name" field', true)
+ return
+ }
+ const { name: _, ...rest } = c
+ newGroups.set(c.name, rest)
+ }
+
+ const oldNames = new Set(Object.keys(workerGroups ?? {}))
+ const newNames = new Set(newGroups.keys())
+
+ // Protect well-known groups from accidental deletion
+ const protectedGroups = ['default', 'native']
+ const deletedProtected = protectedGroups.filter(
+ (g) => oldNames.has(g) && !newNames.has(g)
+ )
+ if (deletedProtected.length > 0) {
+ sendUserToast(
+ `Cannot remove well-known groups: ${deletedProtected.join(', ')}. Add them back to the YAML.`,
+ true
+ )
+ return
+ }
+
+ // Delete removed groups
+ for (const name of oldNames) {
+ if (!newNames.has(name)) {
+ await ConfigService.deleteConfig({ name: 'worker__' + name })
+ }
+ }
+
+ // Create or update groups
+ for (const [name, config] of newGroups) {
+ await ConfigService.updateConfig({
+ name: 'worker__' + name,
+ requestBody: config
+ })
+ }
+
+ yamlDiffMode = false
+ yamlConfigDrawer?.toggleDrawer?.()
+ sendUserToast('Worker group configs saved')
+ await loadWorkerGroups()
+ } catch (err) {
+ if (err instanceof Error) {
+ sendUserToast(err.message, true)
+ } else {
+ sendUserToast('Could not save worker group configs', true)
+ }
+ } finally {
+ yamlSaving = false
+ }
+ }
+
let queueMetricsDrawer: QueueMetricsDrawer | undefined = $state(undefined)
let manageTagsDrawer: ManageTagsDrawer | undefined = $state(undefined)
let selectedTab: string = $state('default')
@@ -523,6 +629,74 @@
+
+ {
+ yamlDiffMode = false
+ yamlConfigDrawer?.toggleDrawer?.()
+ }}
+ >
+
+ Use this YAML to manage worker group configs as code.
+ Learn more
+
+ {#if yamlDiffMode}
+
+ {#await import('$lib/components/DiffEditor.svelte')}
+
+
+
+ {:then Module}
+
+ {/await}
+
No workers currently in this worker group
{/if}