@shipfox/api-runners 2.0.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/.turbo/turbo-build.log +2 -0
- package/.turbo/turbo-type$colon$emit.log +1 -0
- package/.turbo/turbo-type.log +1 -0
- package/CHANGELOG.md +187 -0
- package/LICENSE +21 -0
- package/dist/config.d.ts +30 -0
- package/dist/config.d.ts.map +1 -0
- package/dist/config.js +208 -0
- package/dist/config.js.map +1 -0
- package/dist/core/demand.d.ts +33 -0
- package/dist/core/demand.d.ts.map +1 -0
- package/dist/core/demand.js +169 -0
- package/dist/core/demand.js.map +1 -0
- package/dist/core/entities/ephemeral-registration-token.d.ts +14 -0
- package/dist/core/entities/ephemeral-registration-token.d.ts.map +1 -0
- package/dist/core/entities/ephemeral-registration-token.js +3 -0
- package/dist/core/entities/ephemeral-registration-token.js.map +1 -0
- package/dist/core/entities/manual-registration-token.d.ts +12 -0
- package/dist/core/entities/manual-registration-token.d.ts.map +1 -0
- package/dist/core/entities/manual-registration-token.js +3 -0
- package/dist/core/entities/manual-registration-token.js.map +1 -0
- package/dist/core/entities/provisioned-runner.d.ts +24 -0
- package/dist/core/entities/provisioned-runner.d.ts.map +1 -0
- package/dist/core/entities/provisioned-runner.js +3 -0
- package/dist/core/entities/provisioned-runner.js.map +1 -0
- package/dist/core/entities/provisioner-token.d.ts +18 -0
- package/dist/core/entities/provisioner-token.d.ts.map +1 -0
- package/dist/core/entities/provisioner-token.js +3 -0
- package/dist/core/entities/provisioner-token.js.map +1 -0
- package/dist/core/entities/reservation.d.ts +10 -0
- package/dist/core/entities/reservation.d.ts.map +1 -0
- package/dist/core/entities/reservation.js +3 -0
- package/dist/core/entities/reservation.js.map +1 -0
- package/dist/core/entities/runner-session.d.ts +18 -0
- package/dist/core/entities/runner-session.d.ts.map +1 -0
- package/dist/core/entities/runner-session.js +3 -0
- package/dist/core/entities/runner-session.js.map +1 -0
- package/dist/core/ephemeral-registration-tokens.d.ts +31 -0
- package/dist/core/ephemeral-registration-tokens.d.ts.map +1 -0
- package/dist/core/ephemeral-registration-tokens.js +61 -0
- package/dist/core/ephemeral-registration-tokens.js.map +1 -0
- package/dist/core/errors.d.ts +62 -0
- package/dist/core/errors.d.ts.map +1 -0
- package/dist/core/errors.js +92 -0
- package/dist/core/errors.js.map +1 -0
- package/dist/core/index.d.ts +14 -0
- package/dist/core/index.d.ts.map +1 -0
- package/dist/core/index.js +10 -0
- package/dist/core/index.js.map +1 -0
- package/dist/core/job-executions.d.ts +14 -0
- package/dist/core/job-executions.d.ts.map +1 -0
- package/dist/core/job-executions.js +37 -0
- package/dist/core/job-executions.js.map +1 -0
- package/dist/core/maintenance-policy.d.ts +2 -0
- package/dist/core/maintenance-policy.d.ts.map +1 -0
- package/dist/core/maintenance-policy.js +3 -0
- package/dist/core/maintenance-policy.js.map +1 -0
- package/dist/core/maintenance.d.ts +33 -0
- package/dist/core/maintenance.d.ts.map +1 -0
- package/dist/core/maintenance.js +55 -0
- package/dist/core/maintenance.js.map +1 -0
- package/dist/core/manual-registration-tokens.d.ts +17 -0
- package/dist/core/manual-registration-tokens.d.ts.map +1 -0
- package/dist/core/manual-registration-tokens.js +28 -0
- package/dist/core/manual-registration-tokens.js.map +1 -0
- package/dist/core/provisioned-runners.d.ts +67 -0
- package/dist/core/provisioned-runners.d.ts.map +1 -0
- package/dist/core/provisioned-runners.js +162 -0
- package/dist/core/provisioned-runners.js.map +1 -0
- package/dist/core/provisioner-tokens.d.ts +20 -0
- package/dist/core/provisioner-tokens.d.ts.map +1 -0
- package/dist/core/provisioner-tokens.js +36 -0
- package/dist/core/provisioner-tokens.js.map +1 -0
- package/dist/core/rate-limit.d.ts +35 -0
- package/dist/core/rate-limit.d.ts.map +1 -0
- package/dist/core/rate-limit.js +74 -0
- package/dist/core/rate-limit.js.map +1 -0
- package/dist/core/runner-sessions.d.ts +26 -0
- package/dist/core/runner-sessions.d.ts.map +1 -0
- package/dist/core/runner-sessions.js +41 -0
- package/dist/core/runner-sessions.js.map +1 -0
- package/dist/core/runner-tool-capabilities.d.ts +24 -0
- package/dist/core/runner-tool-capabilities.d.ts.map +1 -0
- package/dist/core/runner-tool-capabilities.js +43 -0
- package/dist/core/runner-tool-capabilities.js.map +1 -0
- package/dist/db/db.d.ts +4478 -0
- package/dist/db/db.d.ts.map +1 -0
- package/dist/db/db.js +36 -0
- package/dist/db/db.js.map +1 -0
- package/dist/db/ephemeral-registration-tokens.d.ts +47 -0
- package/dist/db/ephemeral-registration-tokens.d.ts.map +1 -0
- package/dist/db/ephemeral-registration-tokens.js +170 -0
- package/dist/db/ephemeral-registration-tokens.js.map +1 -0
- package/dist/db/index.d.ts +19 -0
- package/dist/db/index.d.ts.map +1 -0
- package/dist/db/index.js +15 -0
- package/dist/db/index.js.map +1 -0
- package/dist/db/job-executions.d.ts +128 -0
- package/dist/db/job-executions.d.ts.map +1 -0
- package/dist/db/job-executions.js +354 -0
- package/dist/db/job-executions.js.map +1 -0
- package/dist/db/manual-registration-tokens.d.ts +16 -0
- package/dist/db/manual-registration-tokens.d.ts.map +1 -0
- package/dist/db/manual-registration-tokens.js +37 -0
- package/dist/db/manual-registration-tokens.js.map +1 -0
- package/dist/db/provisioned-runners.d.ts +79 -0
- package/dist/db/provisioned-runners.d.ts.map +1 -0
- package/dist/db/provisioned-runners.js +466 -0
- package/dist/db/provisioned-runners.js.map +1 -0
- package/dist/db/provisioner-tokens.d.ts +26 -0
- package/dist/db/provisioner-tokens.d.ts.map +1 -0
- package/dist/db/provisioner-tokens.js +54 -0
- package/dist/db/provisioner-tokens.js.map +1 -0
- package/dist/db/rate-limits.d.ts +18 -0
- package/dist/db/rate-limits.d.ts.map +1 -0
- package/dist/db/rate-limits.js +44 -0
- package/dist/db/rate-limits.js.map +1 -0
- package/dist/db/reservations.d.ts +48 -0
- package/dist/db/reservations.d.ts.map +1 -0
- package/dist/db/reservations.js +162 -0
- package/dist/db/reservations.js.map +1 -0
- package/dist/db/runner-sessions.d.ts +18 -0
- package/dist/db/runner-sessions.d.ts.map +1 -0
- package/dist/db/runner-sessions.js +38 -0
- package/dist/db/runner-sessions.js.map +1 -0
- package/dist/db/schema/common.d.ts +2 -0
- package/dist/db/schema/common.d.ts.map +1 -0
- package/dist/db/schema/common.js +4 -0
- package/dist/db/schema/common.js.map +1 -0
- package/dist/db/schema/ephemeral-registration-tokens.d.ts +199 -0
- package/dist/db/schema/ephemeral-registration-tokens.d.ts.map +1 -0
- package/dist/db/schema/ephemeral-registration-tokens.js +45 -0
- package/dist/db/schema/ephemeral-registration-tokens.js.map +1 -0
- package/dist/db/schema/manual-registration-tokens.d.ts +165 -0
- package/dist/db/schema/manual-registration-tokens.d.ts.map +1 -0
- package/dist/db/schema/manual-registration-tokens.js +40 -0
- package/dist/db/schema/manual-registration-tokens.js.map +1 -0
- package/dist/db/schema/outbox.d.ts +195 -0
- package/dist/db/schema/outbox.d.ts.map +1 -0
- package/dist/db/schema/outbox.js +5 -0
- package/dist/db/schema/outbox.js.map +1 -0
- package/dist/db/schema/pending-job-executions.d.ts +189 -0
- package/dist/db/schema/pending-job-executions.d.ts.map +1 -0
- package/dist/db/schema/pending-job-executions.js +21 -0
- package/dist/db/schema/pending-job-executions.js.map +1 -0
- package/dist/db/schema/provisioned-runners.d.ts +379 -0
- package/dist/db/schema/provisioned-runners.d.ts.map +1 -0
- package/dist/db/schema/provisioned-runners.js +83 -0
- package/dist/db/schema/provisioned-runners.js.map +1 -0
- package/dist/db/schema/provisioner-tokens.d.ts +216 -0
- package/dist/db/schema/provisioner-tokens.d.ts.map +1 -0
- package/dist/db/schema/provisioner-tokens.js +48 -0
- package/dist/db/schema/provisioner-tokens.js.map +1 -0
- package/dist/db/schema/rate-limits.d.ts +145 -0
- package/dist/db/schema/rate-limits.d.ts.map +1 -0
- package/dist/db/schema/rate-limits.js +25 -0
- package/dist/db/schema/rate-limits.js.map +1 -0
- package/dist/db/schema/reservations.d.ts +157 -0
- package/dist/db/schema/reservations.d.ts.map +1 -0
- package/dist/db/schema/reservations.js +34 -0
- package/dist/db/schema/reservations.js.map +1 -0
- package/dist/db/schema/runner-sessions.d.ts +298 -0
- package/dist/db/schema/runner-sessions.d.ts.map +1 -0
- package/dist/db/schema/runner-sessions.js +61 -0
- package/dist/db/schema/runner-sessions.js.map +1 -0
- package/dist/db/schema/running-job-executions.d.ts +334 -0
- package/dist/db/schema/running-job-executions.d.ts.map +1 -0
- package/dist/db/schema/running-job-executions.js +41 -0
- package/dist/db/schema/running-job-executions.js.map +1 -0
- package/dist/index.d.ts +6 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +56 -0
- package/dist/index.js.map +1 -0
- package/dist/metrics/index.d.ts +3 -0
- package/dist/metrics/index.d.ts.map +1 -0
- package/dist/metrics/index.js +4 -0
- package/dist/metrics/index.js.map +1 -0
- package/dist/metrics/instance.d.ts +34 -0
- package/dist/metrics/instance.d.ts.map +1 -0
- package/dist/metrics/instance.js +60 -0
- package/dist/metrics/instance.js.map +1 -0
- package/dist/metrics/service.d.ts +2 -0
- package/dist/metrics/service.d.ts.map +1 -0
- package/dist/metrics/service.js +21 -0
- package/dist/metrics/service.js.map +1 -0
- package/dist/presentation/auth/index.d.ts +3 -0
- package/dist/presentation/auth/index.d.ts.map +1 -0
- package/dist/presentation/auth/index.js +4 -0
- package/dist/presentation/auth/index.js.map +1 -0
- package/dist/presentation/auth/provisioner-token-auth.d.ts +3 -0
- package/dist/presentation/auth/provisioner-token-auth.d.ts.map +1 -0
- package/dist/presentation/auth/provisioner-token-auth.js +81 -0
- package/dist/presentation/auth/provisioner-token-auth.js.map +1 -0
- package/dist/presentation/auth/runner-registration-token-auth.d.ts +17 -0
- package/dist/presentation/auth/runner-registration-token-auth.d.ts.map +1 -0
- package/dist/presentation/auth/runner-registration-token-auth.js +77 -0
- package/dist/presentation/auth/runner-registration-token-auth.js.map +1 -0
- package/dist/presentation/dto/active-provisioner.d.ts +8 -0
- package/dist/presentation/dto/active-provisioner.d.ts.map +1 -0
- package/dist/presentation/dto/active-provisioner.js +10 -0
- package/dist/presentation/dto/active-provisioner.js.map +1 -0
- package/dist/presentation/dto/index.d.ts +7 -0
- package/dist/presentation/dto/index.d.ts.map +1 -0
- package/dist/presentation/dto/index.js +8 -0
- package/dist/presentation/dto/index.js.map +1 -0
- package/dist/presentation/dto/manual-registration-token.d.ts +12 -0
- package/dist/presentation/dto/manual-registration-token.d.ts.map +1 -0
- package/dist/presentation/dto/manual-registration-token.js +14 -0
- package/dist/presentation/dto/manual-registration-token.js.map +1 -0
- package/dist/presentation/dto/mint-registration-tokens.d.ts +5 -0
- package/dist/presentation/dto/mint-registration-tokens.d.ts.map +1 -0
- package/dist/presentation/dto/mint-registration-tokens.js +19 -0
- package/dist/presentation/dto/mint-registration-tokens.js.map +1 -0
- package/dist/presentation/dto/poll-demand.d.ts +4 -0
- package/dist/presentation/dto/poll-demand.d.ts.map +1 -0
- package/dist/presentation/dto/poll-demand.js +19 -0
- package/dist/presentation/dto/poll-demand.js.map +1 -0
- package/dist/presentation/dto/provisioned-runners.d.ts +38 -0
- package/dist/presentation/dto/provisioned-runners.d.ts.map +1 -0
- package/dist/presentation/dto/provisioned-runners.js +44 -0
- package/dist/presentation/dto/provisioned-runners.js.map +1 -0
- package/dist/presentation/dto/provisioner-token.d.ts +15 -0
- package/dist/presentation/dto/provisioner-token.d.ts.map +1 -0
- package/dist/presentation/dto/provisioner-token.js +17 -0
- package/dist/presentation/dto/provisioner-token.js.map +1 -0
- package/dist/presentation/index.d.ts +4 -0
- package/dist/presentation/index.d.ts.map +1 -0
- package/dist/presentation/index.js +5 -0
- package/dist/presentation/index.js.map +1 -0
- package/dist/presentation/routes/create-manual-registration-token.d.ts +2 -0
- package/dist/presentation/routes/create-manual-registration-token.d.ts.map +1 -0
- package/dist/presentation/routes/create-manual-registration-token.js +44 -0
- package/dist/presentation/routes/create-manual-registration-token.js.map +1 -0
- package/dist/presentation/routes/create-provisioner-token.d.ts +2 -0
- package/dist/presentation/routes/create-provisioner-token.d.ts.map +1 -0
- package/dist/presentation/routes/create-provisioner-token.js +42 -0
- package/dist/presentation/routes/create-provisioner-token.js.map +1 -0
- package/dist/presentation/routes/get-provisioner-me.d.ts +2 -0
- package/dist/presentation/routes/get-provisioner-me.d.ts.map +1 -0
- package/dist/presentation/routes/get-provisioner-me.js +22 -0
- package/dist/presentation/routes/get-provisioner-me.js.map +1 -0
- package/dist/presentation/routes/heartbeat.d.ts +2 -0
- package/dist/presentation/routes/heartbeat.d.ts.map +1 -0
- package/dist/presentation/routes/heartbeat.js +64 -0
- package/dist/presentation/routes/heartbeat.js.map +1 -0
- package/dist/presentation/routes/index.d.ts +4 -0
- package/dist/presentation/routes/index.d.ts.map +1 -0
- package/dist/presentation/routes/index.js +97 -0
- package/dist/presentation/routes/index.js.map +1 -0
- package/dist/presentation/routes/list-active-provisioners.d.ts +2 -0
- package/dist/presentation/routes/list-active-provisioners.d.ts.map +1 -0
- package/dist/presentation/routes/list-active-provisioners.js +32 -0
- package/dist/presentation/routes/list-active-provisioners.js.map +1 -0
- package/dist/presentation/routes/list-active-runners.d.ts +2 -0
- package/dist/presentation/routes/list-active-runners.d.ts.map +1 -0
- package/dist/presentation/routes/list-active-runners.js +32 -0
- package/dist/presentation/routes/list-active-runners.js.map +1 -0
- package/dist/presentation/routes/list-manual-registration-tokens.d.ts +2 -0
- package/dist/presentation/routes/list-manual-registration-tokens.d.ts.map +1 -0
- package/dist/presentation/routes/list-manual-registration-tokens.js +32 -0
- package/dist/presentation/routes/list-manual-registration-tokens.js.map +1 -0
- package/dist/presentation/routes/list-provisioner-tokens.d.ts +2 -0
- package/dist/presentation/routes/list-provisioner-tokens.d.ts.map +1 -0
- package/dist/presentation/routes/list-provisioner-tokens.js +32 -0
- package/dist/presentation/routes/list-provisioner-tokens.js.map +1 -0
- package/dist/presentation/routes/mint-registration-tokens.d.ts +2 -0
- package/dist/presentation/routes/mint-registration-tokens.d.ts.map +1 -0
- package/dist/presentation/routes/mint-registration-tokens.js +74 -0
- package/dist/presentation/routes/mint-registration-tokens.js.map +1 -0
- package/dist/presentation/routes/poll-demand.d.ts +2 -0
- package/dist/presentation/routes/poll-demand.d.ts.map +1 -0
- package/dist/presentation/routes/poll-demand.js +53 -0
- package/dist/presentation/routes/poll-demand.js.map +1 -0
- package/dist/presentation/routes/rate-limit.d.ts +4 -0
- package/dist/presentation/routes/rate-limit.d.ts.map +1 -0
- package/dist/presentation/routes/rate-limit.js +94 -0
- package/dist/presentation/routes/rate-limit.js.map +1 -0
- package/dist/presentation/routes/reconcile-provisioned-runners.d.ts +2 -0
- package/dist/presentation/routes/reconcile-provisioned-runners.d.ts.map +1 -0
- package/dist/presentation/routes/reconcile-provisioned-runners.js +27 -0
- package/dist/presentation/routes/reconcile-provisioned-runners.js.map +1 -0
- package/dist/presentation/routes/register.d.ts +2 -0
- package/dist/presentation/routes/register.d.ts.map +1 -0
- package/dist/presentation/routes/register.js +63 -0
- package/dist/presentation/routes/register.js.map +1 -0
- package/dist/presentation/routes/report-provisioned-runners.d.ts +2 -0
- package/dist/presentation/routes/report-provisioned-runners.d.ts.map +1 -0
- package/dist/presentation/routes/report-provisioned-runners.js +37 -0
- package/dist/presentation/routes/report-provisioned-runners.js.map +1 -0
- package/dist/presentation/routes/request-job.d.ts +2 -0
- package/dist/presentation/routes/request-job.d.ts.map +1 -0
- package/dist/presentation/routes/request-job.js +45 -0
- package/dist/presentation/routes/request-job.js.map +1 -0
- package/dist/presentation/routes/revoke-manual-registration-token.d.ts +2 -0
- package/dist/presentation/routes/revoke-manual-registration-token.d.ts.map +1 -0
- package/dist/presentation/routes/revoke-manual-registration-token.js +42 -0
- package/dist/presentation/routes/revoke-manual-registration-token.js.map +1 -0
- package/dist/presentation/routes/revoke-provisioner-token.d.ts +2 -0
- package/dist/presentation/routes/revoke-provisioner-token.d.ts.map +1 -0
- package/dist/presentation/routes/revoke-provisioner-token.js +44 -0
- package/dist/presentation/routes/revoke-provisioner-token.js.map +1 -0
- package/dist/presentation/routes/workspace-membership.d.ts +6 -0
- package/dist/presentation/routes/workspace-membership.d.ts.map +1 -0
- package/dist/presentation/routes/workspace-membership.js +6 -0
- package/dist/presentation/routes/workspace-membership.js.map +1 -0
- package/dist/presentation/subscribers/on-workflows-job-execution-timed-out.d.ts +3 -0
- package/dist/presentation/subscribers/on-workflows-job-execution-timed-out.d.ts.map +1 -0
- package/dist/presentation/subscribers/on-workflows-job-execution-timed-out.js +14 -0
- package/dist/presentation/subscribers/on-workflows-job-execution-timed-out.js.map +1 -0
- package/dist/temporal/activities/index.d.ts +9 -0
- package/dist/temporal/activities/index.d.ts.map +1 -0
- package/dist/temporal/activities/index.js +12 -0
- package/dist/temporal/activities/index.js.map +1 -0
- package/dist/temporal/activities/maintenance-activities.d.ts +31 -0
- package/dist/temporal/activities/maintenance-activities.d.ts.map +1 -0
- package/dist/temporal/activities/maintenance-activities.js +18 -0
- package/dist/temporal/activities/maintenance-activities.js.map +1 -0
- package/dist/temporal/constants.d.ts +2 -0
- package/dist/temporal/constants.d.ts.map +1 -0
- package/dist/temporal/constants.js +3 -0
- package/dist/temporal/constants.js.map +1 -0
- package/dist/temporal/workflows/index.d.ts +2 -0
- package/dist/temporal/workflows/index.d.ts.map +1 -0
- package/dist/temporal/workflows/index.js +3 -0
- package/dist/temporal/workflows/index.js.map +1 -0
- package/dist/temporal/workflows/stuck-job-detector.d.ts +2 -0
- package/dist/temporal/workflows/stuck-job-detector.d.ts.map +1 -0
- package/dist/temporal/workflows/stuck-job-detector.js +61 -0
- package/dist/temporal/workflows/stuck-job-detector.js.map +1 -0
- package/dist/tsconfig.test.tsbuildinfo +1 -0
- package/drizzle/0000_initial.sql +185 -0
- package/drizzle/meta/0000_snapshot.json +1582 -0
- package/drizzle/meta/_journal.json +13 -0
- package/drizzle.config.ts +7 -0
- package/package.json +82 -0
- package/src/config.test.ts +374 -0
- package/src/config.ts +330 -0
- package/src/core/demand.test.ts +301 -0
- package/src/core/demand.ts +239 -0
- package/src/core/entities/ephemeral-registration-token.ts +13 -0
- package/src/core/entities/manual-registration-token.ts +11 -0
- package/src/core/entities/provisioned-runner.ts +30 -0
- package/src/core/entities/provisioner-token.ts +16 -0
- package/src/core/entities/reservation.ts +9 -0
- package/src/core/entities/runner-session.ts +18 -0
- package/src/core/ephemeral-registration-tokens.test.ts +108 -0
- package/src/core/ephemeral-registration-tokens.ts +112 -0
- package/src/core/errors.ts +128 -0
- package/src/core/index.ts +56 -0
- package/src/core/job-executions.ts +47 -0
- package/src/core/maintenance-policy.ts +1 -0
- package/src/core/maintenance.test.ts +251 -0
- package/src/core/maintenance.ts +74 -0
- package/src/core/manual-registration-tokens.test.ts +44 -0
- package/src/core/manual-registration-tokens.ts +51 -0
- package/src/core/provisioned-runners.test.ts +101 -0
- package/src/core/provisioned-runners.ts +301 -0
- package/src/core/provisioner-tokens.test.ts +61 -0
- package/src/core/provisioner-tokens.ts +61 -0
- package/src/core/rate-limit.test.ts +158 -0
- package/src/core/rate-limit.ts +121 -0
- package/src/core/runner-sessions.test.ts +209 -0
- package/src/core/runner-sessions.ts +66 -0
- package/src/core/runner-tool-capabilities.test.ts +141 -0
- package/src/core/runner-tool-capabilities.ts +75 -0
- package/src/db/db.ts +38 -0
- package/src/db/ephemeral-registration-tokens.test.ts +332 -0
- package/src/db/ephemeral-registration-tokens.ts +348 -0
- package/src/db/index.ts +90 -0
- package/src/db/job-executions.test.ts +1500 -0
- package/src/db/job-executions.ts +647 -0
- package/src/db/manual-registration-tokens.test.ts +99 -0
- package/src/db/manual-registration-tokens.ts +87 -0
- package/src/db/provisioned-runners.test.ts +1812 -0
- package/src/db/provisioned-runners.ts +886 -0
- package/src/db/provisioner-tokens.test.ts +133 -0
- package/src/db/provisioner-tokens.ts +142 -0
- package/src/db/rate-limits.test.ts +74 -0
- package/src/db/rate-limits.ts +69 -0
- package/src/db/reservations.test.ts +399 -0
- package/src/db/reservations.ts +296 -0
- package/src/db/runner-sessions.test.ts +153 -0
- package/src/db/runner-sessions.ts +97 -0
- package/src/db/schema/common.ts +3 -0
- package/src/db/schema/ephemeral-registration-tokens.ts +58 -0
- package/src/db/schema/manual-registration-tokens.ts +40 -0
- package/src/db/schema/outbox.ts +4 -0
- package/src/db/schema/pending-job-executions.ts +25 -0
- package/src/db/schema/provisioned-runners.ts +89 -0
- package/src/db/schema/provisioner-tokens.ts +50 -0
- package/src/db/schema/rate-limits.ts +27 -0
- package/src/db/schema/reservations.ts +38 -0
- package/src/db/schema/runner-sessions.ts +77 -0
- package/src/db/schema/running-job-executions.ts +50 -0
- package/src/index.ts +63 -0
- package/src/metrics/index.ts +13 -0
- package/src/metrics/instance.ts +115 -0
- package/src/metrics/service.ts +22 -0
- package/src/presentation/auth/index.ts +6 -0
- package/src/presentation/auth/provisioner-token-auth.ts +76 -0
- package/src/presentation/auth/runner-registration-token-auth.ts +105 -0
- package/src/presentation/dto/active-provisioner.ts +15 -0
- package/src/presentation/dto/index.ts +13 -0
- package/src/presentation/dto/manual-registration-token.ts +23 -0
- package/src/presentation/dto/mint-registration-tokens.ts +35 -0
- package/src/presentation/dto/poll-demand.ts +20 -0
- package/src/presentation/dto/provisioned-runners.ts +88 -0
- package/src/presentation/dto/provisioner-token.ts +29 -0
- package/src/presentation/index.ts +6 -0
- package/src/presentation/routes/create-manual-registration-token.ts +43 -0
- package/src/presentation/routes/create-provisioner-token.ts +41 -0
- package/src/presentation/routes/get-provisioner-me.ts +22 -0
- package/src/presentation/routes/heartbeat.test.ts +325 -0
- package/src/presentation/routes/heartbeat.ts +80 -0
- package/src/presentation/routes/index.ts +86 -0
- package/src/presentation/routes/list-active-provisioners.ts +25 -0
- package/src/presentation/routes/list-active-runners.test.ts +366 -0
- package/src/presentation/routes/list-active-runners.ts +25 -0
- package/src/presentation/routes/list-manual-registration-tokens.ts +25 -0
- package/src/presentation/routes/list-provisioner-tokens.ts +25 -0
- package/src/presentation/routes/manual-registration-tokens.test.ts +213 -0
- package/src/presentation/routes/mint-registration-tokens.test.ts +453 -0
- package/src/presentation/routes/mint-registration-tokens.ts +82 -0
- package/src/presentation/routes/poll-demand-release.test.ts +80 -0
- package/src/presentation/routes/poll-demand.test.ts +323 -0
- package/src/presentation/routes/poll-demand.ts +56 -0
- package/src/presentation/routes/provisioner-me.test.ts +172 -0
- package/src/presentation/routes/provisioner-tokens.test.ts +219 -0
- package/src/presentation/routes/rate-limit.ts +115 -0
- package/src/presentation/routes/reconcile-provisioned-runners.test.ts +380 -0
- package/src/presentation/routes/reconcile-provisioned-runners.ts +30 -0
- package/src/presentation/routes/register.test.ts +453 -0
- package/src/presentation/routes/register.ts +75 -0
- package/src/presentation/routes/report-provisioned-runners.test.ts +249 -0
- package/src/presentation/routes/report-provisioned-runners.ts +40 -0
- package/src/presentation/routes/request-job.test.ts +222 -0
- package/src/presentation/routes/request-job.ts +47 -0
- package/src/presentation/routes/revoke-manual-registration-token.ts +35 -0
- package/src/presentation/routes/revoke-provisioner-token.ts +37 -0
- package/src/presentation/routes/workspace-membership.ts +9 -0
- package/src/presentation/subscribers/on-workflows-job-execution-timed-out.test.ts +147 -0
- package/src/presentation/subscribers/on-workflows-job-execution-timed-out.ts +17 -0
- package/src/temporal/activities/index.ts +17 -0
- package/src/temporal/activities/maintenance-activities.test.ts +104 -0
- package/src/temporal/activities/maintenance-activities.ts +41 -0
- package/src/temporal/constants.ts +1 -0
- package/src/temporal/workflows/index.ts +1 -0
- package/src/temporal/workflows/stuck-job-detector.test.ts +93 -0
- package/src/temporal/workflows/stuck-job-detector.ts +67 -0
- package/test/env.ts +11 -0
- package/test/factories/ephemeral-registration-token.ts +41 -0
- package/test/factories/manual-registration-token.ts +37 -0
- package/test/factories/pending-job.ts +44 -0
- package/test/factories/provisioned-runner.ts +78 -0
- package/test/factories/provisioner-token.ts +41 -0
- package/test/factories/reservation.ts +33 -0
- package/test/factories/runner-session.ts +32 -0
- package/test/globalSetup.ts +23 -0
- package/test/index.ts +16 -0
- package/test/setup.ts +17 -0
- package/tsconfig.build.json +9 -0
- package/tsconfig.build.tsbuildinfo +1 -0
- package/tsconfig.json +3 -0
- package/tsconfig.test.json +8 -0
- package/vitest.config.ts +14 -0
|
@@ -0,0 +1,354 @@
|
|
|
1
|
+
import { RUNNER_JOB_CLAIMED, RUNNER_JOB_LEASE_EXPIRED, RUNNER_JOB_QUEUED } from '@shipfox/api-runners-dto';
|
|
2
|
+
import { logger } from '@shipfox/node-opentelemetry';
|
|
3
|
+
import { writeOutboxEvent, writeOutboxEvents } from '@shipfox/node-outbox';
|
|
4
|
+
import { canonicalizeLabels } from '@shipfox/runner-labels';
|
|
5
|
+
import { and, arrayContained, asc, count, desc, eq, gt, inArray, isNotNull, isNull, lt, lte, or, sql } from 'drizzle-orm';
|
|
6
|
+
import { EmptyRequiredLabelsError, RunnerSessionExhaustedError, RunningJobExecutionNotFoundError } from '#core/errors.js';
|
|
7
|
+
import { jobExecutionEnqueuedCount, jobExecutionLeaseExpiredCount } from '#metrics/instance.js';
|
|
8
|
+
import { db } from './db.js';
|
|
9
|
+
import { runnersOutbox } from './schema/outbox.js';
|
|
10
|
+
import { pendingJobExecutions } from './schema/pending-job-executions.js';
|
|
11
|
+
import { runnerSessions } from './schema/runner-sessions.js';
|
|
12
|
+
import { runningJobExecutions } from './schema/running-job-executions.js';
|
|
13
|
+
// Idempotent while the job execution is still pending: a duplicate jobExecutionId already in
|
|
14
|
+
// `runners_pending_jobs` is a no-op. Temporal retries the enqueue activity
|
|
15
|
+
// at-least-once, so a unique-violation throw on a retry-after-lost-result
|
|
16
|
+
// would permanently fail a healthy job execution's workflow. The guard does not extend
|
|
17
|
+
// past the claim: once the job execution has moved to `runners_running_jobs`, a retry
|
|
18
|
+
// can reinsert an orphan pending row, which a later `claimPendingJobExecution` drops
|
|
19
|
+
// via the running-execution unique constraint (onConflictDoNothing) instead of failing.
|
|
20
|
+
export async function enqueueJobExecution(params) {
|
|
21
|
+
const requiredLabels = [
|
|
22
|
+
...canonicalizeLabels(params.requiredLabels)
|
|
23
|
+
];
|
|
24
|
+
if (requiredLabels.length === 0) throw new EmptyRequiredLabelsError();
|
|
25
|
+
const enqueued = await db().transaction(async (tx)=>{
|
|
26
|
+
const [inserted] = await tx.insert(pendingJobExecutions).values({
|
|
27
|
+
workspaceId: params.workspaceId,
|
|
28
|
+
workflowRunId: params.workflowRunId,
|
|
29
|
+
workflowRunAttemptId: params.workflowRunAttemptId,
|
|
30
|
+
jobId: params.jobId,
|
|
31
|
+
jobExecutionId: params.jobExecutionId,
|
|
32
|
+
projectId: params.projectId,
|
|
33
|
+
requiredLabels
|
|
34
|
+
}).onConflictDoNothing({
|
|
35
|
+
target: pendingJobExecutions.jobExecutionId
|
|
36
|
+
}).returning({
|
|
37
|
+
createdAt: pendingJobExecutions.createdAt
|
|
38
|
+
});
|
|
39
|
+
// A retry that hits the conflict inserts nothing: the first enqueue already
|
|
40
|
+
// emitted the queued event (durably, in the outbox), so re-emitting would
|
|
41
|
+
// only add a redundant row the subscriber coalesces away. Skip it.
|
|
42
|
+
if (!inserted) return false;
|
|
43
|
+
await writeOutboxEvent(tx, runnersOutbox, {
|
|
44
|
+
type: RUNNER_JOB_QUEUED,
|
|
45
|
+
payload: {
|
|
46
|
+
workflowRunId: params.workflowRunId,
|
|
47
|
+
workflowRunAttemptId: params.workflowRunAttemptId,
|
|
48
|
+
jobId: params.jobId,
|
|
49
|
+
jobExecutionId: params.jobExecutionId,
|
|
50
|
+
queuedAt: inserted.createdAt.toISOString()
|
|
51
|
+
}
|
|
52
|
+
});
|
|
53
|
+
return true;
|
|
54
|
+
});
|
|
55
|
+
if (enqueued) jobExecutionEnqueuedCount.add(1);
|
|
56
|
+
}
|
|
57
|
+
export async function claimPendingJobExecution(params) {
|
|
58
|
+
await touchRunnerSessionLiveness({
|
|
59
|
+
workspaceId: params.workspaceId,
|
|
60
|
+
runnerSessionId: params.runnerSessionId,
|
|
61
|
+
throttleSeconds: params.runnerSessionLivenessThrottleSeconds
|
|
62
|
+
});
|
|
63
|
+
if (params.sessionLabels.length === 0) return null;
|
|
64
|
+
return await db().transaction(async (tx)=>{
|
|
65
|
+
let provisionerId = null;
|
|
66
|
+
let provisionedRunnerId = null;
|
|
67
|
+
if (params.maxClaims !== null) {
|
|
68
|
+
const [session] = await tx.select({
|
|
69
|
+
maxClaims: runnerSessions.maxClaims,
|
|
70
|
+
claimsUsed: runnerSessions.claimsUsed,
|
|
71
|
+
provisionerId: runnerSessions.provisionerId,
|
|
72
|
+
provisionedRunnerId: runnerSessions.provisionedRunnerId
|
|
73
|
+
}).from(runnerSessions).where(eq(runnerSessions.id, params.runnerSessionId)).limit(1).for('update');
|
|
74
|
+
if (!session || session.maxClaims === null || session.claimsUsed >= session.maxClaims) {
|
|
75
|
+
throw new RunnerSessionExhaustedError(params.runnerSessionId);
|
|
76
|
+
}
|
|
77
|
+
// Ephemeral sessions are the only capped sessions, and the DB check keeps
|
|
78
|
+
// their provisioned-runner link present as a pair.
|
|
79
|
+
provisionerId = session.provisionerId;
|
|
80
|
+
provisionedRunnerId = session.provisionedRunnerId;
|
|
81
|
+
}
|
|
82
|
+
// `id` is a uuidv7 (time-ordered), so it is a deterministic FIFO tiebreaker
|
|
83
|
+
// for rows sharing a created_at within a batch.
|
|
84
|
+
const [row] = await tx.select().from(pendingJobExecutions).where(and(eq(pendingJobExecutions.workspaceId, params.workspaceId), arrayContained(pendingJobExecutions.requiredLabels, params.sessionLabels))).orderBy(asc(pendingJobExecutions.createdAt), asc(pendingJobExecutions.id)).limit(1).for('update', {
|
|
85
|
+
skipLocked: true
|
|
86
|
+
});
|
|
87
|
+
if (!row) return null;
|
|
88
|
+
await tx.delete(pendingJobExecutions).where(eq(pendingJobExecutions.id, row.id));
|
|
89
|
+
// An enqueue retry that lands after a prior claim can leave an orphan pending
|
|
90
|
+
// row whose jobExecutionId is already in `runners_running_jobs`. Insert-or-skip
|
|
91
|
+
// on the jobExecutionId unique constraint: when the job execution is already running
|
|
92
|
+
// the insert touches no row, so we commit the orphan's deletion and return
|
|
93
|
+
// null rather than let the unique violation roll the claim back into a poison
|
|
94
|
+
// loop. The runner just re-polls for a real job execution.
|
|
95
|
+
const inserted = await tx.insert(runningJobExecutions).values({
|
|
96
|
+
workspaceId: row.workspaceId,
|
|
97
|
+
workflowRunId: row.workflowRunId,
|
|
98
|
+
workflowRunAttemptId: row.workflowRunAttemptId,
|
|
99
|
+
jobId: row.jobId,
|
|
100
|
+
jobExecutionId: row.jobExecutionId,
|
|
101
|
+
projectId: row.projectId,
|
|
102
|
+
runnerSessionId: params.runnerSessionId,
|
|
103
|
+
provisionerId,
|
|
104
|
+
provisionedRunnerId,
|
|
105
|
+
requiredLabels: row.requiredLabels,
|
|
106
|
+
runnerLabels: params.sessionLabels
|
|
107
|
+
}).onConflictDoNothing({
|
|
108
|
+
target: runningJobExecutions.jobExecutionId
|
|
109
|
+
}).returning({
|
|
110
|
+
claimedAt: runningJobExecutions.startedAt
|
|
111
|
+
});
|
|
112
|
+
const claimed = inserted[0];
|
|
113
|
+
if (!claimed) return null;
|
|
114
|
+
if (params.maxClaims !== null) {
|
|
115
|
+
await tx.update(runnerSessions).set({
|
|
116
|
+
claimsUsed: sql`${runnerSessions.claimsUsed} + 1`,
|
|
117
|
+
updatedAt: sql`now()`
|
|
118
|
+
}).where(eq(runnerSessions.id, params.runnerSessionId));
|
|
119
|
+
}
|
|
120
|
+
// The running-row insert is the runner claiming the job execution. Emit in the same tx; the
|
|
121
|
+
// payload carries the row's own claim instant so a consumer records the true time,
|
|
122
|
+
// not the outbox drain time.
|
|
123
|
+
await writeOutboxEvent(tx, runnersOutbox, {
|
|
124
|
+
type: RUNNER_JOB_CLAIMED,
|
|
125
|
+
payload: {
|
|
126
|
+
workflowRunId: row.workflowRunId,
|
|
127
|
+
workflowRunAttemptId: row.workflowRunAttemptId,
|
|
128
|
+
jobId: row.jobId,
|
|
129
|
+
jobExecutionId: row.jobExecutionId,
|
|
130
|
+
claimedAt: claimed.claimedAt.toISOString()
|
|
131
|
+
}
|
|
132
|
+
});
|
|
133
|
+
return {
|
|
134
|
+
workflowRunId: row.workflowRunId,
|
|
135
|
+
workflowRunAttemptId: row.workflowRunAttemptId,
|
|
136
|
+
jobId: row.jobId,
|
|
137
|
+
jobExecutionId: row.jobExecutionId,
|
|
138
|
+
projectId: row.projectId
|
|
139
|
+
};
|
|
140
|
+
});
|
|
141
|
+
}
|
|
142
|
+
async function touchRunnerSessionLiveness(params) {
|
|
143
|
+
await db().update(runnerSessions).set({
|
|
144
|
+
updatedAt: sql`now()`
|
|
145
|
+
}).where(and(eq(runnerSessions.id, params.runnerSessionId), eq(runnerSessions.workspaceId, params.workspaceId), sql`${runnerSessions.updatedAt} < now() - (${params.throttleSeconds} || ' seconds')::interval`));
|
|
146
|
+
}
|
|
147
|
+
/**
|
|
148
|
+
* Releases a job execution's lease when the orchestration workflow finalizes it: deletes the
|
|
149
|
+
* running-job-execution row AND any lingering pending row for the same execution, in one tx.
|
|
150
|
+
* Idempotent (0-row no-op), no token scope (the workflow is authoritative), and
|
|
151
|
+
* emits no event — the workflow already owns the outcome. Sweeping the pending row
|
|
152
|
+
* too closes the at-least-once window where an enqueue retry left an orphan that a
|
|
153
|
+
* later claim would otherwise pick up for an already-finished job execution.
|
|
154
|
+
*/ export async function releaseJobExecution(params) {
|
|
155
|
+
await db().transaction(async (tx)=>{
|
|
156
|
+
// Delete pending before running to match `claimPendingJobExecution`'s lock-acquisition
|
|
157
|
+
// order (it locks the pending row first, then the running row). A concurrent
|
|
158
|
+
// claim picking up an orphan pending row for this same job execution would otherwise
|
|
159
|
+
// deadlock against the reverse order here.
|
|
160
|
+
await tx.delete(pendingJobExecutions).where(eq(pendingJobExecutions.jobExecutionId, params.jobExecutionId));
|
|
161
|
+
await tx.delete(runningJobExecutions).where(eq(runningJobExecutions.jobExecutionId, params.jobExecutionId));
|
|
162
|
+
});
|
|
163
|
+
}
|
|
164
|
+
/**
|
|
165
|
+
* Reaps stale leases (bounded by `limit`), emitting one
|
|
166
|
+
* `runners.job.lease_expired` event per reaped job execution.
|
|
167
|
+
*
|
|
168
|
+
* The cutoff is re-checked in the DELETE, not just the locking subquery, so a
|
|
169
|
+
* heartbeat landing mid-call spares the live row. Each reaped execution also sweeps its
|
|
170
|
+
* pending row: a failed best-effort `releaseJobExecution` would otherwise leave an orphan
|
|
171
|
+
* that a later claim re-runs as an already-finished job execution.
|
|
172
|
+
*
|
|
173
|
+
* Locks running-then-pending, the inverse of `claimPendingJobExecution` / `releaseJobExecution`.
|
|
174
|
+
* That pre-existing asymmetry opens a narrow deadlock window against a concurrent
|
|
175
|
+
* claim of the same orphan-pending job execution; Postgres breaks it and the cron retries.
|
|
176
|
+
*/ export async function expireStuckJobExecutions(params) {
|
|
177
|
+
const reaped = await db().transaction(async (tx)=>{
|
|
178
|
+
const heartbeatCutoff = sql`now() - (${params.thresholdSeconds} || ' seconds')::interval`;
|
|
179
|
+
const firstHeartbeatCutoff = sql`now() - (${params.noFirstHeartbeatGraceSeconds} || ' seconds')::interval`;
|
|
180
|
+
const stalePredicate = or(and(isNull(runningJobExecutions.firstHeartbeatAt), lte(runningJobExecutions.lastHeartbeatAt, runningJobExecutions.startedAt), lt(runningJobExecutions.startedAt, firstHeartbeatCutoff)), and(or(isNotNull(runningJobExecutions.firstHeartbeatAt), gt(runningJobExecutions.lastHeartbeatAt, runningJobExecutions.startedAt)), lt(runningJobExecutions.lastHeartbeatAt, heartbeatCutoff)));
|
|
181
|
+
const staleIds = tx.select({
|
|
182
|
+
id: runningJobExecutions.id
|
|
183
|
+
}).from(runningJobExecutions).where(stalePredicate).orderBy(asc(sql`CASE WHEN ${runningJobExecutions.firstHeartbeatAt} IS NULL AND ${runningJobExecutions.lastHeartbeatAt} <= ${runningJobExecutions.startedAt} THEN ${runningJobExecutions.startedAt} ELSE ${runningJobExecutions.lastHeartbeatAt} END`)).limit(params.limit ?? 100).for('update', {
|
|
184
|
+
skipLocked: true
|
|
185
|
+
});
|
|
186
|
+
const deleted = await tx.delete(runningJobExecutions).where(and(inArray(runningJobExecutions.id, staleIds), stalePredicate)).returning({
|
|
187
|
+
workflowRunId: runningJobExecutions.workflowRunId,
|
|
188
|
+
workflowRunAttemptId: runningJobExecutions.workflowRunAttemptId,
|
|
189
|
+
jobId: runningJobExecutions.jobId,
|
|
190
|
+
jobExecutionId: runningJobExecutions.jobExecutionId
|
|
191
|
+
});
|
|
192
|
+
if (deleted.length === 0) return [];
|
|
193
|
+
await tx.delete(pendingJobExecutions).where(inArray(pendingJobExecutions.jobExecutionId, deleted.map((row)=>row.jobExecutionId)));
|
|
194
|
+
await writeOutboxEvents(tx, runnersOutbox, deleted.map((row)=>({
|
|
195
|
+
type: RUNNER_JOB_LEASE_EXPIRED,
|
|
196
|
+
payload: {
|
|
197
|
+
workflowRunId: row.workflowRunId,
|
|
198
|
+
workflowRunAttemptId: row.workflowRunAttemptId,
|
|
199
|
+
jobId: row.jobId,
|
|
200
|
+
jobExecutionId: row.jobExecutionId
|
|
201
|
+
}
|
|
202
|
+
})));
|
|
203
|
+
return deleted;
|
|
204
|
+
});
|
|
205
|
+
if (reaped.length > 0) jobExecutionLeaseExpiredCount.add(reaped.length);
|
|
206
|
+
return reaped;
|
|
207
|
+
}
|
|
208
|
+
export async function getJobExecutionQueueDepth() {
|
|
209
|
+
const [pending] = await db().select({
|
|
210
|
+
value: count()
|
|
211
|
+
}).from(pendingJobExecutions);
|
|
212
|
+
const [running] = await db().select({
|
|
213
|
+
value: count()
|
|
214
|
+
}).from(runningJobExecutions);
|
|
215
|
+
return {
|
|
216
|
+
pendingJobExecutions: pending?.value ?? 0,
|
|
217
|
+
runningJobExecutions: running?.value ?? 0
|
|
218
|
+
};
|
|
219
|
+
}
|
|
220
|
+
export async function listActiveRunningJobExecutions(params) {
|
|
221
|
+
return await db().select({
|
|
222
|
+
jobId: runningJobExecutions.jobId,
|
|
223
|
+
jobExecutionId: runningJobExecutions.jobExecutionId,
|
|
224
|
+
workflowRunId: runningJobExecutions.workflowRunId,
|
|
225
|
+
workflowRunAttemptId: runningJobExecutions.workflowRunAttemptId,
|
|
226
|
+
projectId: runningJobExecutions.projectId,
|
|
227
|
+
runnerSessionId: runningJobExecutions.runnerSessionId,
|
|
228
|
+
provisionerId: runningJobExecutions.provisionerId,
|
|
229
|
+
provisionedRunnerId: runningJobExecutions.provisionedRunnerId,
|
|
230
|
+
requiredLabels: runningJobExecutions.requiredLabels,
|
|
231
|
+
runnerLabels: runningJobExecutions.runnerLabels,
|
|
232
|
+
startedAt: runningJobExecutions.startedAt,
|
|
233
|
+
lastHeartbeatAt: runningJobExecutions.lastHeartbeatAt
|
|
234
|
+
}).from(runningJobExecutions).where(and(eq(runningJobExecutions.workspaceId, params.workspaceId), sql`${runningJobExecutions.lastHeartbeatAt} > now() - (${params.windowSeconds} || ' seconds')::interval`)).orderBy(desc(runningJobExecutions.lastHeartbeatAt), desc(runningJobExecutions.id)).limit(params.limit ?? 1000);
|
|
235
|
+
}
|
|
236
|
+
export async function listRunningJobExecutionsByProvisionedRunnerTx(tx, params) {
|
|
237
|
+
if (params.provisionedRunnerIds.length === 0) return [];
|
|
238
|
+
const duplicateRows = await tx.select({
|
|
239
|
+
provisionedRunnerId: runningJobExecutions.provisionedRunnerId,
|
|
240
|
+
count: count()
|
|
241
|
+
}).from(runningJobExecutions).where(and(eq(runningJobExecutions.workspaceId, params.workspaceId), eq(runningJobExecutions.provisionerId, params.provisionerId), inArray(runningJobExecutions.provisionedRunnerId, params.provisionedRunnerIds))).groupBy(runningJobExecutions.provisionedRunnerId).having(sql`count(*) > 1`);
|
|
242
|
+
const duplicateProvisionedRunnerIds = duplicateRows.flatMap((row)=>row.provisionedRunnerId ? [
|
|
243
|
+
row.provisionedRunnerId
|
|
244
|
+
] : []);
|
|
245
|
+
if (duplicateProvisionedRunnerIds.length > 0) {
|
|
246
|
+
logger().warn({
|
|
247
|
+
workspaceId: params.workspaceId,
|
|
248
|
+
provisionerId: params.provisionerId,
|
|
249
|
+
provisionedRunnerIds: duplicateProvisionedRunnerIds
|
|
250
|
+
}, 'multiple running job executions are bound to the same provisioned runner');
|
|
251
|
+
}
|
|
252
|
+
const result = await tx.execute(sql`
|
|
253
|
+
SELECT DISTINCT ON (${runningJobExecutions.provisionedRunnerId})
|
|
254
|
+
${runningJobExecutions.workflowRunId} AS "workflowRunId",
|
|
255
|
+
${runningJobExecutions.workflowRunAttemptId} AS "workflowRunAttemptId",
|
|
256
|
+
${runningJobExecutions.jobId} AS "jobId",
|
|
257
|
+
${runningJobExecutions.jobExecutionId} AS "jobExecutionId",
|
|
258
|
+
${runningJobExecutions.provisionedRunnerId} AS "provisionedRunnerId",
|
|
259
|
+
${runningJobExecutions.startedAt} AS "startedAt",
|
|
260
|
+
${runningJobExecutions.lastHeartbeatAt} AS "lastHeartbeatAt",
|
|
261
|
+
${runningJobExecutions.cancellationRequestedAt} AS "cancellationRequestedAt"
|
|
262
|
+
FROM ${runningJobExecutions}
|
|
263
|
+
WHERE
|
|
264
|
+
${runningJobExecutions.workspaceId} = ${params.workspaceId}
|
|
265
|
+
AND ${runningJobExecutions.provisionerId} = ${params.provisionerId}
|
|
266
|
+
AND ${runningJobExecutions.provisionedRunnerId} IN (${sql.join(params.provisionedRunnerIds.map((provisionedRunnerId)=>sql`${provisionedRunnerId}`), sql`, `)})
|
|
267
|
+
ORDER BY ${runningJobExecutions.provisionedRunnerId}, ${runningJobExecutions.startedAt} DESC, ${runningJobExecutions.jobExecutionId} DESC
|
|
268
|
+
`);
|
|
269
|
+
return result.rows.map((row)=>({
|
|
270
|
+
workflowRunId: row.workflowRunId,
|
|
271
|
+
workflowRunAttemptId: row.workflowRunAttemptId,
|
|
272
|
+
jobId: row.jobId,
|
|
273
|
+
jobExecutionId: row.jobExecutionId,
|
|
274
|
+
provisionedRunnerId: row.provisionedRunnerId,
|
|
275
|
+
startedAt: toDate(row.startedAt),
|
|
276
|
+
lastHeartbeatAt: toDate(row.lastHeartbeatAt),
|
|
277
|
+
cancellationRequestedAt: row.cancellationRequestedAt ? toDate(row.cancellationRequestedAt) : null
|
|
278
|
+
}));
|
|
279
|
+
}
|
|
280
|
+
function toDate(value) {
|
|
281
|
+
return value instanceof Date ? value : new Date(value);
|
|
282
|
+
}
|
|
283
|
+
export async function isJobLeaseActive(params) {
|
|
284
|
+
const [row] = await db().select({
|
|
285
|
+
id: runningJobExecutions.id
|
|
286
|
+
}).from(runningJobExecutions).where(and(eq(runningJobExecutions.jobExecutionId, params.jobExecutionId), eq(runningJobExecutions.runnerSessionId, params.runnerSessionId), params.jobId === undefined ? undefined : eq(runningJobExecutions.jobId, params.jobId))).limit(1);
|
|
287
|
+
return row !== undefined;
|
|
288
|
+
}
|
|
289
|
+
export async function recordHeartbeat(params) {
|
|
290
|
+
const result = await db().transaction(async (tx)=>{
|
|
291
|
+
const updated = await tx.update(runningJobExecutions).set({
|
|
292
|
+
firstHeartbeatAt: sql`COALESCE(${runningJobExecutions.firstHeartbeatAt}, now())`,
|
|
293
|
+
lastHeartbeatAt: sql`now()`
|
|
294
|
+
}).where(and(eq(runningJobExecutions.jobExecutionId, params.jobExecutionId), eq(runningJobExecutions.runnerSessionId, params.runnerSessionId))).returning({
|
|
295
|
+
cancellationRequestedAt: runningJobExecutions.cancellationRequestedAt,
|
|
296
|
+
workflowRunId: runningJobExecutions.workflowRunId,
|
|
297
|
+
workflowRunAttemptId: runningJobExecutions.workflowRunAttemptId,
|
|
298
|
+
jobId: runningJobExecutions.jobId,
|
|
299
|
+
jobExecutionId: runningJobExecutions.jobExecutionId,
|
|
300
|
+
projectId: runningJobExecutions.projectId,
|
|
301
|
+
workspaceId: runningJobExecutions.workspaceId,
|
|
302
|
+
runnerSessionId: runningJobExecutions.runnerSessionId
|
|
303
|
+
});
|
|
304
|
+
const row = updated[0];
|
|
305
|
+
if (!row) throw new RunningJobExecutionNotFoundError(params.jobExecutionId);
|
|
306
|
+
const [previous] = await tx.select({
|
|
307
|
+
toolCapabilities: runnerSessions.toolCapabilities
|
|
308
|
+
}).from(runnerSessions).where(eq(runnerSessions.id, params.runnerSessionId)).limit(1).for('update');
|
|
309
|
+
const [session] = await tx.update(runnerSessions).set({
|
|
310
|
+
toolCapabilities: params.toolCapabilities ?? null,
|
|
311
|
+
toolCapabilitiesReportedAt: params.toolCapabilities ? sql`now()` : null,
|
|
312
|
+
updatedAt: sql`now()`
|
|
313
|
+
}).where(eq(runnerSessions.id, params.runnerSessionId)).returning({
|
|
314
|
+
toolCapabilities: runnerSessions.toolCapabilities
|
|
315
|
+
});
|
|
316
|
+
return {
|
|
317
|
+
row,
|
|
318
|
+
previousToolCapabilities: previous?.toolCapabilities ?? null,
|
|
319
|
+
currentToolCapabilities: session?.toolCapabilities ?? null
|
|
320
|
+
};
|
|
321
|
+
});
|
|
322
|
+
const row = result.row;
|
|
323
|
+
if (!row) throw new RunningJobExecutionNotFoundError(params.jobExecutionId);
|
|
324
|
+
return {
|
|
325
|
+
cancellationRequested: row.cancellationRequestedAt !== null,
|
|
326
|
+
previousToolCapabilities: result.previousToolCapabilities,
|
|
327
|
+
currentToolCapabilities: result.currentToolCapabilities,
|
|
328
|
+
runningJobExecution: {
|
|
329
|
+
workflowRunId: row.workflowRunId,
|
|
330
|
+
workflowRunAttemptId: row.workflowRunAttemptId,
|
|
331
|
+
jobId: row.jobId,
|
|
332
|
+
jobExecutionId: row.jobExecutionId,
|
|
333
|
+
projectId: row.projectId,
|
|
334
|
+
workspaceId: row.workspaceId,
|
|
335
|
+
runnerSessionId: row.runnerSessionId
|
|
336
|
+
}
|
|
337
|
+
};
|
|
338
|
+
}
|
|
339
|
+
export async function requestJobExecutionCancellation(params) {
|
|
340
|
+
await db().update(runningJobExecutions).set({
|
|
341
|
+
cancellationRequestedAt: sql`COALESCE(${runningJobExecutions.cancellationRequestedAt}, now())`
|
|
342
|
+
}).where(eq(runningJobExecutions.jobExecutionId, params.jobExecutionId));
|
|
343
|
+
}
|
|
344
|
+
export async function cancelRunnerJobs(params) {
|
|
345
|
+
if (params.jobIds.length === 0) return;
|
|
346
|
+
await db().transaction(async (tx)=>{
|
|
347
|
+
await tx.delete(pendingJobExecutions).where(inArray(pendingJobExecutions.jobId, params.jobIds));
|
|
348
|
+
await tx.update(runningJobExecutions).set({
|
|
349
|
+
cancellationRequestedAt: sql`COALESCE(${runningJobExecutions.cancellationRequestedAt}, now())`
|
|
350
|
+
}).where(inArray(runningJobExecutions.jobId, params.jobIds));
|
|
351
|
+
});
|
|
352
|
+
}
|
|
353
|
+
|
|
354
|
+
//# sourceMappingURL=job-executions.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"sources":["../../src/db/job-executions.ts"],"sourcesContent":["import {\n RUNNER_JOB_CLAIMED,\n RUNNER_JOB_LEASE_EXPIRED,\n RUNNER_JOB_QUEUED,\n type RunnersEventMap,\n type RunnerToolCapabilitiesDto,\n} from '@shipfox/api-runners-dto';\nimport {logger} from '@shipfox/node-opentelemetry';\nimport {writeOutboxEvent, writeOutboxEvents} from '@shipfox/node-outbox';\nimport {canonicalizeLabels} from '@shipfox/runner-labels';\nimport {\n and,\n arrayContained,\n asc,\n count,\n desc,\n eq,\n gt,\n inArray,\n isNotNull,\n isNull,\n lt,\n lte,\n or,\n sql,\n} from 'drizzle-orm';\nimport {\n EmptyRequiredLabelsError,\n RunnerSessionExhaustedError,\n RunningJobExecutionNotFoundError,\n} from '#core/errors.js';\nimport {jobExecutionEnqueuedCount, jobExecutionLeaseExpiredCount} from '#metrics/instance.js';\nimport type {Tx} from './db.js';\nimport {db} from './db.js';\nimport {runnersOutbox} from './schema/outbox.js';\nimport {pendingJobExecutions} from './schema/pending-job-executions.js';\nimport {runnerSessions} from './schema/runner-sessions.js';\nimport {runningJobExecutions} from './schema/running-job-executions.js';\n\nexport interface EnqueueJobExecutionParams {\n workspaceId: string;\n workflowRunId: string;\n workflowRunAttemptId: string;\n jobId: string;\n jobExecutionId: string;\n projectId: string;\n requiredLabels: string[];\n}\n\n// Idempotent while the job execution is still pending: a duplicate jobExecutionId already in\n// `runners_pending_jobs` is a no-op. Temporal retries the enqueue activity\n// at-least-once, so a unique-violation throw on a retry-after-lost-result\n// would permanently fail a healthy job execution's workflow. The guard does not extend\n// past the claim: once the job execution has moved to `runners_running_jobs`, a retry\n// can reinsert an orphan pending row, which a later `claimPendingJobExecution` drops\n// via the running-execution unique constraint (onConflictDoNothing) instead of failing.\nexport async function enqueueJobExecution(params: EnqueueJobExecutionParams): Promise<void> {\n const requiredLabels = [...canonicalizeLabels(params.requiredLabels)];\n if (requiredLabels.length === 0) throw new EmptyRequiredLabelsError();\n\n const enqueued = await db().transaction(async (tx) => {\n const [inserted] = await tx\n .insert(pendingJobExecutions)\n .values({\n workspaceId: params.workspaceId,\n workflowRunId: params.workflowRunId,\n workflowRunAttemptId: params.workflowRunAttemptId,\n jobId: params.jobId,\n jobExecutionId: params.jobExecutionId,\n projectId: params.projectId,\n requiredLabels,\n })\n .onConflictDoNothing({target: pendingJobExecutions.jobExecutionId})\n .returning({createdAt: pendingJobExecutions.createdAt});\n\n // A retry that hits the conflict inserts nothing: the first enqueue already\n // emitted the queued event (durably, in the outbox), so re-emitting would\n // only add a redundant row the subscriber coalesces away. Skip it.\n if (!inserted) return false;\n\n await writeOutboxEvent<RunnersEventMap>(tx, runnersOutbox, {\n type: RUNNER_JOB_QUEUED,\n payload: {\n workflowRunId: params.workflowRunId,\n workflowRunAttemptId: params.workflowRunAttemptId,\n jobId: params.jobId,\n jobExecutionId: params.jobExecutionId,\n queuedAt: inserted.createdAt.toISOString(),\n },\n });\n return true;\n });\n\n if (enqueued) jobExecutionEnqueuedCount.add(1);\n}\n\nexport interface ClaimedJobExecution {\n workflowRunId: string;\n workflowRunAttemptId: string;\n jobId: string;\n jobExecutionId: string;\n projectId: string;\n}\n\nexport interface ActiveRunningJobExecution {\n workflowRunId: string;\n workflowRunAttemptId: string;\n jobId: string;\n jobExecutionId: string;\n projectId: string;\n runnerSessionId: string;\n provisionerId: string | null;\n provisionedRunnerId: string | null;\n requiredLabels: string[];\n runnerLabels: string[];\n startedAt: Date;\n lastHeartbeatAt: Date;\n}\n\nexport interface ProvisionedRunnerBoundJobExecution {\n workflowRunId: string;\n workflowRunAttemptId: string;\n jobId: string;\n jobExecutionId: string;\n provisionedRunnerId: string;\n startedAt: Date;\n lastHeartbeatAt: Date;\n cancellationRequestedAt: Date | null;\n}\n\nexport async function claimPendingJobExecution(params: {\n workspaceId: string;\n runnerSessionId: string;\n sessionLabels: string[];\n maxClaims: number | null;\n runnerSessionLivenessThrottleSeconds: number;\n}): Promise<ClaimedJobExecution | null> {\n await touchRunnerSessionLiveness({\n workspaceId: params.workspaceId,\n runnerSessionId: params.runnerSessionId,\n throttleSeconds: params.runnerSessionLivenessThrottleSeconds,\n });\n\n if (params.sessionLabels.length === 0) return null;\n\n return await db().transaction(async (tx) => {\n let provisionerId: string | null = null;\n let provisionedRunnerId: string | null = null;\n\n if (params.maxClaims !== null) {\n const [session] = await tx\n .select({\n maxClaims: runnerSessions.maxClaims,\n claimsUsed: runnerSessions.claimsUsed,\n provisionerId: runnerSessions.provisionerId,\n provisionedRunnerId: runnerSessions.provisionedRunnerId,\n })\n .from(runnerSessions)\n .where(eq(runnerSessions.id, params.runnerSessionId))\n .limit(1)\n .for('update');\n\n if (!session || session.maxClaims === null || session.claimsUsed >= session.maxClaims) {\n throw new RunnerSessionExhaustedError(params.runnerSessionId);\n }\n\n // Ephemeral sessions are the only capped sessions, and the DB check keeps\n // their provisioned-runner link present as a pair.\n provisionerId = session.provisionerId;\n provisionedRunnerId = session.provisionedRunnerId;\n }\n\n // `id` is a uuidv7 (time-ordered), so it is a deterministic FIFO tiebreaker\n // for rows sharing a created_at within a batch.\n const [row] = await tx\n .select()\n .from(pendingJobExecutions)\n .where(\n and(\n eq(pendingJobExecutions.workspaceId, params.workspaceId),\n arrayContained(pendingJobExecutions.requiredLabels, params.sessionLabels),\n ),\n )\n .orderBy(asc(pendingJobExecutions.createdAt), asc(pendingJobExecutions.id))\n .limit(1)\n .for('update', {skipLocked: true});\n\n if (!row) return null;\n\n await tx.delete(pendingJobExecutions).where(eq(pendingJobExecutions.id, row.id));\n\n // An enqueue retry that lands after a prior claim can leave an orphan pending\n // row whose jobExecutionId is already in `runners_running_jobs`. Insert-or-skip\n // on the jobExecutionId unique constraint: when the job execution is already running\n // the insert touches no row, so we commit the orphan's deletion and return\n // null rather than let the unique violation roll the claim back into a poison\n // loop. The runner just re-polls for a real job execution.\n const inserted = await tx\n .insert(runningJobExecutions)\n .values({\n workspaceId: row.workspaceId,\n workflowRunId: row.workflowRunId,\n workflowRunAttemptId: row.workflowRunAttemptId,\n jobId: row.jobId,\n jobExecutionId: row.jobExecutionId,\n projectId: row.projectId,\n runnerSessionId: params.runnerSessionId,\n provisionerId,\n provisionedRunnerId,\n requiredLabels: row.requiredLabels,\n runnerLabels: params.sessionLabels,\n })\n .onConflictDoNothing({target: runningJobExecutions.jobExecutionId})\n .returning({claimedAt: runningJobExecutions.startedAt});\n\n const claimed = inserted[0];\n if (!claimed) return null;\n\n if (params.maxClaims !== null) {\n await tx\n .update(runnerSessions)\n .set({claimsUsed: sql`${runnerSessions.claimsUsed} + 1`, updatedAt: sql`now()`})\n .where(eq(runnerSessions.id, params.runnerSessionId));\n }\n\n // The running-row insert is the runner claiming the job execution. Emit in the same tx; the\n // payload carries the row's own claim instant so a consumer records the true time,\n // not the outbox drain time.\n await writeOutboxEvent<RunnersEventMap>(tx, runnersOutbox, {\n type: RUNNER_JOB_CLAIMED,\n payload: {\n workflowRunId: row.workflowRunId,\n workflowRunAttemptId: row.workflowRunAttemptId,\n jobId: row.jobId,\n jobExecutionId: row.jobExecutionId,\n claimedAt: claimed.claimedAt.toISOString(),\n },\n });\n\n return {\n workflowRunId: row.workflowRunId,\n workflowRunAttemptId: row.workflowRunAttemptId,\n jobId: row.jobId,\n jobExecutionId: row.jobExecutionId,\n projectId: row.projectId,\n };\n });\n}\n\nasync function touchRunnerSessionLiveness(params: {\n workspaceId: string;\n runnerSessionId: string;\n throttleSeconds: number;\n}): Promise<void> {\n await db()\n .update(runnerSessions)\n .set({updatedAt: sql`now()`})\n .where(\n and(\n eq(runnerSessions.id, params.runnerSessionId),\n eq(runnerSessions.workspaceId, params.workspaceId),\n sql`${runnerSessions.updatedAt} < now() - (${params.throttleSeconds} || ' seconds')::interval`,\n ),\n );\n}\n\n/**\n * Releases a job execution's lease when the orchestration workflow finalizes it: deletes the\n * running-job-execution row AND any lingering pending row for the same execution, in one tx.\n * Idempotent (0-row no-op), no token scope (the workflow is authoritative), and\n * emits no event — the workflow already owns the outcome. Sweeping the pending row\n * too closes the at-least-once window where an enqueue retry left an orphan that a\n * later claim would otherwise pick up for an already-finished job execution.\n */\nexport async function releaseJobExecution(params: {jobExecutionId: string}): Promise<void> {\n await db().transaction(async (tx) => {\n // Delete pending before running to match `claimPendingJobExecution`'s lock-acquisition\n // order (it locks the pending row first, then the running row). A concurrent\n // claim picking up an orphan pending row for this same job execution would otherwise\n // deadlock against the reverse order here.\n await tx\n .delete(pendingJobExecutions)\n .where(eq(pendingJobExecutions.jobExecutionId, params.jobExecutionId));\n await tx\n .delete(runningJobExecutions)\n .where(eq(runningJobExecutions.jobExecutionId, params.jobExecutionId));\n });\n}\n\n/**\n * Reaps stale leases (bounded by `limit`), emitting one\n * `runners.job.lease_expired` event per reaped job execution.\n *\n * The cutoff is re-checked in the DELETE, not just the locking subquery, so a\n * heartbeat landing mid-call spares the live row. Each reaped execution also sweeps its\n * pending row: a failed best-effort `releaseJobExecution` would otherwise leave an orphan\n * that a later claim re-runs as an already-finished job execution.\n *\n * Locks running-then-pending, the inverse of `claimPendingJobExecution` / `releaseJobExecution`.\n * That pre-existing asymmetry opens a narrow deadlock window against a concurrent\n * claim of the same orphan-pending job execution; Postgres breaks it and the cron retries.\n */\nexport async function expireStuckJobExecutions(params: {\n thresholdSeconds: number;\n noFirstHeartbeatGraceSeconds: number;\n limit?: number;\n}): Promise<\n Array<{\n workflowRunId: string;\n workflowRunAttemptId: string;\n jobId: string;\n jobExecutionId: string;\n }>\n> {\n const reaped = await db().transaction(async (tx) => {\n const heartbeatCutoff = sql`now() - (${params.thresholdSeconds} || ' seconds')::interval`;\n const firstHeartbeatCutoff = sql`now() - (${params.noFirstHeartbeatGraceSeconds} || ' seconds')::interval`;\n const stalePredicate = or(\n and(\n isNull(runningJobExecutions.firstHeartbeatAt),\n lte(runningJobExecutions.lastHeartbeatAt, runningJobExecutions.startedAt),\n lt(runningJobExecutions.startedAt, firstHeartbeatCutoff),\n ),\n and(\n or(\n isNotNull(runningJobExecutions.firstHeartbeatAt),\n gt(runningJobExecutions.lastHeartbeatAt, runningJobExecutions.startedAt),\n ),\n lt(runningJobExecutions.lastHeartbeatAt, heartbeatCutoff),\n ),\n );\n\n const staleIds = tx\n .select({id: runningJobExecutions.id})\n .from(runningJobExecutions)\n .where(stalePredicate)\n .orderBy(\n asc(\n sql`CASE WHEN ${runningJobExecutions.firstHeartbeatAt} IS NULL AND ${runningJobExecutions.lastHeartbeatAt} <= ${runningJobExecutions.startedAt} THEN ${runningJobExecutions.startedAt} ELSE ${runningJobExecutions.lastHeartbeatAt} END`,\n ),\n )\n .limit(params.limit ?? 100)\n .for('update', {skipLocked: true});\n\n const deleted = await tx\n .delete(runningJobExecutions)\n .where(and(inArray(runningJobExecutions.id, staleIds), stalePredicate))\n .returning({\n workflowRunId: runningJobExecutions.workflowRunId,\n workflowRunAttemptId: runningJobExecutions.workflowRunAttemptId,\n jobId: runningJobExecutions.jobId,\n jobExecutionId: runningJobExecutions.jobExecutionId,\n });\n\n if (deleted.length === 0) return [];\n\n await tx.delete(pendingJobExecutions).where(\n inArray(\n pendingJobExecutions.jobExecutionId,\n deleted.map((row) => row.jobExecutionId),\n ),\n );\n\n await writeOutboxEvents<RunnersEventMap>(\n tx,\n runnersOutbox,\n deleted.map((row) => ({\n type: RUNNER_JOB_LEASE_EXPIRED,\n payload: {\n workflowRunId: row.workflowRunId,\n workflowRunAttemptId: row.workflowRunAttemptId,\n jobId: row.jobId,\n jobExecutionId: row.jobExecutionId,\n },\n })),\n );\n\n return deleted;\n });\n\n if (reaped.length > 0) jobExecutionLeaseExpiredCount.add(reaped.length);\n\n return reaped;\n}\n\nexport async function getJobExecutionQueueDepth(): Promise<{\n pendingJobExecutions: number;\n runningJobExecutions: number;\n}> {\n const [pending] = await db().select({value: count()}).from(pendingJobExecutions);\n const [running] = await db().select({value: count()}).from(runningJobExecutions);\n return {\n pendingJobExecutions: pending?.value ?? 0,\n runningJobExecutions: running?.value ?? 0,\n };\n}\n\nexport async function listActiveRunningJobExecutions(params: {\n workspaceId: string;\n windowSeconds: number;\n limit?: number;\n}): Promise<ActiveRunningJobExecution[]> {\n return await db()\n .select({\n jobId: runningJobExecutions.jobId,\n jobExecutionId: runningJobExecutions.jobExecutionId,\n workflowRunId: runningJobExecutions.workflowRunId,\n workflowRunAttemptId: runningJobExecutions.workflowRunAttemptId,\n projectId: runningJobExecutions.projectId,\n runnerSessionId: runningJobExecutions.runnerSessionId,\n provisionerId: runningJobExecutions.provisionerId,\n provisionedRunnerId: runningJobExecutions.provisionedRunnerId,\n requiredLabels: runningJobExecutions.requiredLabels,\n runnerLabels: runningJobExecutions.runnerLabels,\n startedAt: runningJobExecutions.startedAt,\n lastHeartbeatAt: runningJobExecutions.lastHeartbeatAt,\n })\n .from(runningJobExecutions)\n .where(\n and(\n eq(runningJobExecutions.workspaceId, params.workspaceId),\n sql`${runningJobExecutions.lastHeartbeatAt} > now() - (${params.windowSeconds} || ' seconds')::interval`,\n ),\n )\n .orderBy(desc(runningJobExecutions.lastHeartbeatAt), desc(runningJobExecutions.id))\n .limit(params.limit ?? 1000);\n}\n\nexport async function listRunningJobExecutionsByProvisionedRunnerTx(\n tx: Tx,\n params: {\n workspaceId: string;\n provisionerId: string;\n provisionedRunnerIds: string[];\n },\n): Promise<ProvisionedRunnerBoundJobExecution[]> {\n if (params.provisionedRunnerIds.length === 0) return [];\n\n const duplicateRows = await tx\n .select({\n provisionedRunnerId: runningJobExecutions.provisionedRunnerId,\n count: count(),\n })\n .from(runningJobExecutions)\n .where(\n and(\n eq(runningJobExecutions.workspaceId, params.workspaceId),\n eq(runningJobExecutions.provisionerId, params.provisionerId),\n inArray(runningJobExecutions.provisionedRunnerId, params.provisionedRunnerIds),\n ),\n )\n .groupBy(runningJobExecutions.provisionedRunnerId)\n .having(sql`count(*) > 1`);\n\n const duplicateProvisionedRunnerIds = duplicateRows.flatMap((row) =>\n row.provisionedRunnerId ? [row.provisionedRunnerId] : [],\n );\n if (duplicateProvisionedRunnerIds.length > 0) {\n logger().warn(\n {\n workspaceId: params.workspaceId,\n provisionerId: params.provisionerId,\n provisionedRunnerIds: duplicateProvisionedRunnerIds,\n },\n 'multiple running job executions are bound to the same provisioned runner',\n );\n }\n\n const result = await tx.execute<{\n workflowRunId: string;\n workflowRunAttemptId: string;\n jobId: string;\n jobExecutionId: string;\n provisionedRunnerId: string;\n startedAt: Date | string;\n lastHeartbeatAt: Date | string;\n cancellationRequestedAt: Date | string | null;\n }>(sql`\n SELECT DISTINCT ON (${runningJobExecutions.provisionedRunnerId})\n ${runningJobExecutions.workflowRunId} AS \"workflowRunId\",\n ${runningJobExecutions.workflowRunAttemptId} AS \"workflowRunAttemptId\",\n ${runningJobExecutions.jobId} AS \"jobId\",\n ${runningJobExecutions.jobExecutionId} AS \"jobExecutionId\",\n ${runningJobExecutions.provisionedRunnerId} AS \"provisionedRunnerId\",\n ${runningJobExecutions.startedAt} AS \"startedAt\",\n ${runningJobExecutions.lastHeartbeatAt} AS \"lastHeartbeatAt\",\n ${runningJobExecutions.cancellationRequestedAt} AS \"cancellationRequestedAt\"\n FROM ${runningJobExecutions}\n WHERE\n ${runningJobExecutions.workspaceId} = ${params.workspaceId}\n AND ${runningJobExecutions.provisionerId} = ${params.provisionerId}\n AND ${runningJobExecutions.provisionedRunnerId} IN (${sql.join(\n params.provisionedRunnerIds.map((provisionedRunnerId) => sql`${provisionedRunnerId}`),\n sql`, `,\n )})\n ORDER BY ${runningJobExecutions.provisionedRunnerId}, ${runningJobExecutions.startedAt} DESC, ${runningJobExecutions.jobExecutionId} DESC\n `);\n\n return result.rows.map((row) => ({\n workflowRunId: row.workflowRunId,\n workflowRunAttemptId: row.workflowRunAttemptId,\n jobId: row.jobId,\n jobExecutionId: row.jobExecutionId,\n provisionedRunnerId: row.provisionedRunnerId,\n startedAt: toDate(row.startedAt),\n lastHeartbeatAt: toDate(row.lastHeartbeatAt),\n cancellationRequestedAt: row.cancellationRequestedAt\n ? toDate(row.cancellationRequestedAt)\n : null,\n }));\n}\n\nfunction toDate(value: Date | string): Date {\n return value instanceof Date ? value : new Date(value);\n}\n\nexport async function isJobLeaseActive(params: {\n jobId?: string;\n jobExecutionId: string;\n runnerSessionId: string;\n}): Promise<boolean> {\n const [row] = await db()\n .select({id: runningJobExecutions.id})\n .from(runningJobExecutions)\n .where(\n and(\n eq(runningJobExecutions.jobExecutionId, params.jobExecutionId),\n eq(runningJobExecutions.runnerSessionId, params.runnerSessionId),\n params.jobId === undefined ? undefined : eq(runningJobExecutions.jobId, params.jobId),\n ),\n )\n .limit(1);\n\n return row !== undefined;\n}\n\nexport async function recordHeartbeat(params: {\n jobExecutionId: string;\n runnerSessionId: string;\n toolCapabilities?: RunnerToolCapabilitiesDto | null;\n}): Promise<{\n cancellationRequested: boolean;\n previousToolCapabilities: RunnerToolCapabilitiesDto | null;\n currentToolCapabilities: RunnerToolCapabilitiesDto | null;\n runningJobExecution: {\n workflowRunId: string;\n workflowRunAttemptId: string;\n jobId: string;\n jobExecutionId: string;\n projectId: string;\n workspaceId: string;\n runnerSessionId: string;\n };\n}> {\n const result = await db().transaction(async (tx) => {\n const updated = await tx\n .update(runningJobExecutions)\n .set({\n firstHeartbeatAt: sql`COALESCE(${runningJobExecutions.firstHeartbeatAt}, now())`,\n lastHeartbeatAt: sql`now()`,\n })\n .where(\n and(\n eq(runningJobExecutions.jobExecutionId, params.jobExecutionId),\n eq(runningJobExecutions.runnerSessionId, params.runnerSessionId),\n ),\n )\n .returning({\n cancellationRequestedAt: runningJobExecutions.cancellationRequestedAt,\n workflowRunId: runningJobExecutions.workflowRunId,\n workflowRunAttemptId: runningJobExecutions.workflowRunAttemptId,\n jobId: runningJobExecutions.jobId,\n jobExecutionId: runningJobExecutions.jobExecutionId,\n projectId: runningJobExecutions.projectId,\n workspaceId: runningJobExecutions.workspaceId,\n runnerSessionId: runningJobExecutions.runnerSessionId,\n });\n\n const row = updated[0];\n if (!row) throw new RunningJobExecutionNotFoundError(params.jobExecutionId);\n\n const [previous] = await tx\n .select({toolCapabilities: runnerSessions.toolCapabilities})\n .from(runnerSessions)\n .where(eq(runnerSessions.id, params.runnerSessionId))\n .limit(1)\n .for('update');\n\n const [session] = await tx\n .update(runnerSessions)\n .set({\n toolCapabilities: params.toolCapabilities ?? null,\n toolCapabilitiesReportedAt: params.toolCapabilities ? sql`now()` : null,\n updatedAt: sql`now()`,\n })\n .where(eq(runnerSessions.id, params.runnerSessionId))\n .returning({toolCapabilities: runnerSessions.toolCapabilities});\n\n return {\n row,\n previousToolCapabilities: previous?.toolCapabilities ?? null,\n currentToolCapabilities: session?.toolCapabilities ?? null,\n };\n });\n\n const row = result.row;\n if (!row) throw new RunningJobExecutionNotFoundError(params.jobExecutionId);\n return {\n cancellationRequested: row.cancellationRequestedAt !== null,\n previousToolCapabilities: result.previousToolCapabilities,\n currentToolCapabilities: result.currentToolCapabilities,\n runningJobExecution: {\n workflowRunId: row.workflowRunId,\n workflowRunAttemptId: row.workflowRunAttemptId,\n jobId: row.jobId,\n jobExecutionId: row.jobExecutionId,\n projectId: row.projectId,\n workspaceId: row.workspaceId,\n runnerSessionId: row.runnerSessionId,\n },\n };\n}\n\nexport async function requestJobExecutionCancellation(params: {\n jobExecutionId: string;\n}): Promise<void> {\n await db()\n .update(runningJobExecutions)\n .set({\n cancellationRequestedAt: sql`COALESCE(${runningJobExecutions.cancellationRequestedAt}, now())`,\n })\n .where(eq(runningJobExecutions.jobExecutionId, params.jobExecutionId));\n}\n\nexport async function cancelRunnerJobs(params: {jobIds: string[]}): Promise<void> {\n if (params.jobIds.length === 0) return;\n\n await db().transaction(async (tx) => {\n await tx.delete(pendingJobExecutions).where(inArray(pendingJobExecutions.jobId, params.jobIds));\n await tx\n .update(runningJobExecutions)\n .set({\n cancellationRequestedAt: sql`COALESCE(${runningJobExecutions.cancellationRequestedAt}, now())`,\n })\n .where(inArray(runningJobExecutions.jobId, params.jobIds));\n });\n}\n"],"names":["RUNNER_JOB_CLAIMED","RUNNER_JOB_LEASE_EXPIRED","RUNNER_JOB_QUEUED","logger","writeOutboxEvent","writeOutboxEvents","canonicalizeLabels","and","arrayContained","asc","count","desc","eq","gt","inArray","isNotNull","isNull","lt","lte","or","sql","EmptyRequiredLabelsError","RunnerSessionExhaustedError","RunningJobExecutionNotFoundError","jobExecutionEnqueuedCount","jobExecutionLeaseExpiredCount","db","runnersOutbox","pendingJobExecutions","runnerSessions","runningJobExecutions","enqueueJobExecution","params","requiredLabels","length","enqueued","transaction","tx","inserted","insert","values","workspaceId","workflowRunId","workflowRunAttemptId","jobId","jobExecutionId","projectId","onConflictDoNothing","target","returning","createdAt","type","payload","queuedAt","toISOString","add","claimPendingJobExecution","touchRunnerSessionLiveness","runnerSessionId","throttleSeconds","runnerSessionLivenessThrottleSeconds","sessionLabels","provisionerId","provisionedRunnerId","maxClaims","session","select","claimsUsed","from","where","id","limit","for","row","orderBy","skipLocked","delete","runnerLabels","claimedAt","startedAt","claimed","update","set","updatedAt","releaseJobExecution","expireStuckJobExecutions","reaped","heartbeatCutoff","thresholdSeconds","firstHeartbeatCutoff","noFirstHeartbeatGraceSeconds","stalePredicate","firstHeartbeatAt","lastHeartbeatAt","staleIds","deleted","map","getJobExecutionQueueDepth","pending","value","running","listActiveRunningJobExecutions","windowSeconds","listRunningJobExecutionsByProvisionedRunnerTx","provisionedRunnerIds","duplicateRows","groupBy","having","duplicateProvisionedRunnerIds","flatMap","warn","result","execute","cancellationRequestedAt","join","rows","toDate","Date","isJobLeaseActive","undefined","recordHeartbeat","updated","previous","toolCapabilities","toolCapabilitiesReportedAt","previousToolCapabilities","currentToolCapabilities","cancellationRequested","runningJobExecution","requestJobExecutionCancellation","cancelRunnerJobs","jobIds"],"mappings":"AAAA,SACEA,kBAAkB,EAClBC,wBAAwB,EACxBC,iBAAiB,QAGZ,2BAA2B;AAClC,SAAQC,MAAM,QAAO,8BAA8B;AACnD,SAAQC,gBAAgB,EAAEC,iBAAiB,QAAO,uBAAuB;AACzE,SAAQC,kBAAkB,QAAO,yBAAyB;AAC1D,SACEC,GAAG,EACHC,cAAc,EACdC,GAAG,EACHC,KAAK,EACLC,IAAI,EACJC,EAAE,EACFC,EAAE,EACFC,OAAO,EACPC,SAAS,EACTC,MAAM,EACNC,EAAE,EACFC,GAAG,EACHC,EAAE,EACFC,GAAG,QACE,cAAc;AACrB,SACEC,wBAAwB,EACxBC,2BAA2B,EAC3BC,gCAAgC,QAC3B,kBAAkB;AACzB,SAAQC,yBAAyB,EAAEC,6BAA6B,QAAO,uBAAuB;AAE9F,SAAQC,EAAE,QAAO,UAAU;AAC3B,SAAQC,aAAa,QAAO,qBAAqB;AACjD,SAAQC,oBAAoB,QAAO,qCAAqC;AACxE,SAAQC,cAAc,QAAO,8BAA8B;AAC3D,SAAQC,oBAAoB,QAAO,qCAAqC;AAYxE,6FAA6F;AAC7F,2EAA2E;AAC3E,0EAA0E;AAC1E,uFAAuF;AACvF,sFAAsF;AACtF,qFAAqF;AACrF,wFAAwF;AACxF,OAAO,eAAeC,oBAAoBC,MAAiC;IACzE,MAAMC,iBAAiB;WAAI3B,mBAAmB0B,OAAOC,cAAc;KAAE;IACrE,IAAIA,eAAeC,MAAM,KAAK,GAAG,MAAM,IAAIb;IAE3C,MAAMc,WAAW,MAAMT,KAAKU,WAAW,CAAC,OAAOC;QAC7C,MAAM,CAACC,SAAS,GAAG,MAAMD,GACtBE,MAAM,CAACX,sBACPY,MAAM,CAAC;YACNC,aAAaT,OAAOS,WAAW;YAC/BC,eAAeV,OAAOU,aAAa;YACnCC,sBAAsBX,OAAOW,oBAAoB;YACjDC,OAAOZ,OAAOY,KAAK;YACnBC,gBAAgBb,OAAOa,cAAc;YACrCC,WAAWd,OAAOc,SAAS;YAC3Bb;QACF,GACCc,mBAAmB,CAAC;YAACC,QAAQpB,qBAAqBiB,cAAc;QAAA,GAChEI,SAAS,CAAC;YAACC,WAAWtB,qBAAqBsB,SAAS;QAAA;QAEvD,4EAA4E;QAC5E,0EAA0E;QAC1E,mEAAmE;QACnE,IAAI,CAACZ,UAAU,OAAO;QAEtB,MAAMlC,iBAAkCiC,IAAIV,eAAe;YACzDwB,MAAMjD;YACNkD,SAAS;gBACPV,eAAeV,OAAOU,aAAa;gBACnCC,sBAAsBX,OAAOW,oBAAoB;gBACjDC,OAAOZ,OAAOY,KAAK;gBACnBC,gBAAgBb,OAAOa,cAAc;gBACrCQ,UAAUf,SAASY,SAAS,CAACI,WAAW;YAC1C;QACF;QACA,OAAO;IACT;IAEA,IAAInB,UAAUX,0BAA0B+B,GAAG,CAAC;AAC9C;AAoCA,OAAO,eAAeC,yBAAyBxB,MAM9C;IACC,MAAMyB,2BAA2B;QAC/BhB,aAAaT,OAAOS,WAAW;QAC/BiB,iBAAiB1B,OAAO0B,eAAe;QACvCC,iBAAiB3B,OAAO4B,oCAAoC;IAC9D;IAEA,IAAI5B,OAAO6B,aAAa,CAAC3B,MAAM,KAAK,GAAG,OAAO;IAE9C,OAAO,MAAMR,KAAKU,WAAW,CAAC,OAAOC;QACnC,IAAIyB,gBAA+B;QACnC,IAAIC,sBAAqC;QAEzC,IAAI/B,OAAOgC,SAAS,KAAK,MAAM;YAC7B,MAAM,CAACC,QAAQ,GAAG,MAAM5B,GACrB6B,MAAM,CAAC;gBACNF,WAAWnC,eAAemC,SAAS;gBACnCG,YAAYtC,eAAesC,UAAU;gBACrCL,eAAejC,eAAeiC,aAAa;gBAC3CC,qBAAqBlC,eAAekC,mBAAmB;YACzD,GACCK,IAAI,CAACvC,gBACLwC,KAAK,CAACzD,GAAGiB,eAAeyC,EAAE,EAAEtC,OAAO0B,eAAe,GAClDa,KAAK,CAAC,GACNC,GAAG,CAAC;YAEP,IAAI,CAACP,WAAWA,QAAQD,SAAS,KAAK,QAAQC,QAAQE,UAAU,IAAIF,QAAQD,SAAS,EAAE;gBACrF,MAAM,IAAI1C,4BAA4BU,OAAO0B,eAAe;YAC9D;YAEA,0EAA0E;YAC1E,mDAAmD;YACnDI,gBAAgBG,QAAQH,aAAa;YACrCC,sBAAsBE,QAAQF,mBAAmB;QACnD;QAEA,4EAA4E;QAC5E,gDAAgD;QAChD,MAAM,CAACU,IAAI,GAAG,MAAMpC,GACjB6B,MAAM,GACNE,IAAI,CAACxC,sBACLyC,KAAK,CACJ9D,IACEK,GAAGgB,qBAAqBa,WAAW,EAAET,OAAOS,WAAW,GACvDjC,eAAeoB,qBAAqBK,cAAc,EAAED,OAAO6B,aAAa,IAG3Ea,OAAO,CAACjE,IAAImB,qBAAqBsB,SAAS,GAAGzC,IAAImB,qBAAqB0C,EAAE,GACxEC,KAAK,CAAC,GACNC,GAAG,CAAC,UAAU;YAACG,YAAY;QAAI;QAElC,IAAI,CAACF,KAAK,OAAO;QAEjB,MAAMpC,GAAGuC,MAAM,CAAChD,sBAAsByC,KAAK,CAACzD,GAAGgB,qBAAqB0C,EAAE,EAAEG,IAAIH,EAAE;QAE9E,8EAA8E;QAC9E,gFAAgF;QAChF,qFAAqF;QACrF,2EAA2E;QAC3E,8EAA8E;QAC9E,2DAA2D;QAC3D,MAAMhC,WAAW,MAAMD,GACpBE,MAAM,CAACT,sBACPU,MAAM,CAAC;YACNC,aAAagC,IAAIhC,WAAW;YAC5BC,eAAe+B,IAAI/B,aAAa;YAChCC,sBAAsB8B,IAAI9B,oBAAoB;YAC9CC,OAAO6B,IAAI7B,KAAK;YAChBC,gBAAgB4B,IAAI5B,cAAc;YAClCC,WAAW2B,IAAI3B,SAAS;YACxBY,iBAAiB1B,OAAO0B,eAAe;YACvCI;YACAC;YACA9B,gBAAgBwC,IAAIxC,cAAc;YAClC4C,cAAc7C,OAAO6B,aAAa;QACpC,GACCd,mBAAmB,CAAC;YAACC,QAAQlB,qBAAqBe,cAAc;QAAA,GAChEI,SAAS,CAAC;YAAC6B,WAAWhD,qBAAqBiD,SAAS;QAAA;QAEvD,MAAMC,UAAU1C,QAAQ,CAAC,EAAE;QAC3B,IAAI,CAAC0C,SAAS,OAAO;QAErB,IAAIhD,OAAOgC,SAAS,KAAK,MAAM;YAC7B,MAAM3B,GACH4C,MAAM,CAACpD,gBACPqD,GAAG,CAAC;gBAACf,YAAY/C,GAAG,CAAC,EAAES,eAAesC,UAAU,CAAC,IAAI,CAAC;gBAAEgB,WAAW/D,GAAG,CAAC,KAAK,CAAC;YAAA,GAC7EiD,KAAK,CAACzD,GAAGiB,eAAeyC,EAAE,EAAEtC,OAAO0B,eAAe;QACvD;QAEA,4FAA4F;QAC5F,mFAAmF;QACnF,6BAA6B;QAC7B,MAAMtD,iBAAkCiC,IAAIV,eAAe;YACzDwB,MAAMnD;YACNoD,SAAS;gBACPV,eAAe+B,IAAI/B,aAAa;gBAChCC,sBAAsB8B,IAAI9B,oBAAoB;gBAC9CC,OAAO6B,IAAI7B,KAAK;gBAChBC,gBAAgB4B,IAAI5B,cAAc;gBAClCiC,WAAWE,QAAQF,SAAS,CAACxB,WAAW;YAC1C;QACF;QAEA,OAAO;YACLZ,eAAe+B,IAAI/B,aAAa;YAChCC,sBAAsB8B,IAAI9B,oBAAoB;YAC9CC,OAAO6B,IAAI7B,KAAK;YAChBC,gBAAgB4B,IAAI5B,cAAc;YAClCC,WAAW2B,IAAI3B,SAAS;QAC1B;IACF;AACF;AAEA,eAAeW,2BAA2BzB,MAIzC;IACC,MAAMN,KACHuD,MAAM,CAACpD,gBACPqD,GAAG,CAAC;QAACC,WAAW/D,GAAG,CAAC,KAAK,CAAC;IAAA,GAC1BiD,KAAK,CACJ9D,IACEK,GAAGiB,eAAeyC,EAAE,EAAEtC,OAAO0B,eAAe,GAC5C9C,GAAGiB,eAAeY,WAAW,EAAET,OAAOS,WAAW,GACjDrB,GAAG,CAAC,EAAES,eAAesD,SAAS,CAAC,YAAY,EAAEnD,OAAO2B,eAAe,CAAC,yBAAyB,CAAC;AAGtG;AAEA;;;;;;;CAOC,GACD,OAAO,eAAeyB,oBAAoBpD,MAAgC;IACxE,MAAMN,KAAKU,WAAW,CAAC,OAAOC;QAC5B,uFAAuF;QACvF,6EAA6E;QAC7E,qFAAqF;QACrF,2CAA2C;QAC3C,MAAMA,GACHuC,MAAM,CAAChD,sBACPyC,KAAK,CAACzD,GAAGgB,qBAAqBiB,cAAc,EAAEb,OAAOa,cAAc;QACtE,MAAMR,GACHuC,MAAM,CAAC9C,sBACPuC,KAAK,CAACzD,GAAGkB,qBAAqBe,cAAc,EAAEb,OAAOa,cAAc;IACxE;AACF;AAEA;;;;;;;;;;;;CAYC,GACD,OAAO,eAAewC,yBAAyBrD,MAI9C;IAQC,MAAMsD,SAAS,MAAM5D,KAAKU,WAAW,CAAC,OAAOC;QAC3C,MAAMkD,kBAAkBnE,GAAG,CAAC,SAAS,EAAEY,OAAOwD,gBAAgB,CAAC,yBAAyB,CAAC;QACzF,MAAMC,uBAAuBrE,GAAG,CAAC,SAAS,EAAEY,OAAO0D,4BAA4B,CAAC,yBAAyB,CAAC;QAC1G,MAAMC,iBAAiBxE,GACrBZ,IACES,OAAOc,qBAAqB8D,gBAAgB,GAC5C1E,IAAIY,qBAAqB+D,eAAe,EAAE/D,qBAAqBiD,SAAS,GACxE9D,GAAGa,qBAAqBiD,SAAS,EAAEU,wBAErClF,IACEY,GACEJ,UAAUe,qBAAqB8D,gBAAgB,GAC/C/E,GAAGiB,qBAAqB+D,eAAe,EAAE/D,qBAAqBiD,SAAS,IAEzE9D,GAAGa,qBAAqB+D,eAAe,EAAEN;QAI7C,MAAMO,WAAWzD,GACd6B,MAAM,CAAC;YAACI,IAAIxC,qBAAqBwC,EAAE;QAAA,GACnCF,IAAI,CAACtC,sBACLuC,KAAK,CAACsB,gBACNjB,OAAO,CACNjE,IACEW,GAAG,CAAC,UAAU,EAAEU,qBAAqB8D,gBAAgB,CAAC,aAAa,EAAE9D,qBAAqB+D,eAAe,CAAC,IAAI,EAAE/D,qBAAqBiD,SAAS,CAAC,MAAM,EAAEjD,qBAAqBiD,SAAS,CAAC,MAAM,EAAEjD,qBAAqB+D,eAAe,CAAC,IAAI,CAAC,GAG3OtB,KAAK,CAACvC,OAAOuC,KAAK,IAAI,KACtBC,GAAG,CAAC,UAAU;YAACG,YAAY;QAAI;QAElC,MAAMoB,UAAU,MAAM1D,GACnBuC,MAAM,CAAC9C,sBACPuC,KAAK,CAAC9D,IAAIO,QAAQgB,qBAAqBwC,EAAE,EAAEwB,WAAWH,iBACtD1C,SAAS,CAAC;YACTP,eAAeZ,qBAAqBY,aAAa;YACjDC,sBAAsBb,qBAAqBa,oBAAoB;YAC/DC,OAAOd,qBAAqBc,KAAK;YACjCC,gBAAgBf,qBAAqBe,cAAc;QACrD;QAEF,IAAIkD,QAAQ7D,MAAM,KAAK,GAAG,OAAO,EAAE;QAEnC,MAAMG,GAAGuC,MAAM,CAAChD,sBAAsByC,KAAK,CACzCvD,QACEc,qBAAqBiB,cAAc,EACnCkD,QAAQC,GAAG,CAAC,CAACvB,MAAQA,IAAI5B,cAAc;QAI3C,MAAMxC,kBACJgC,IACAV,eACAoE,QAAQC,GAAG,CAAC,CAACvB,MAAS,CAAA;gBACpBtB,MAAMlD;gBACNmD,SAAS;oBACPV,eAAe+B,IAAI/B,aAAa;oBAChCC,sBAAsB8B,IAAI9B,oBAAoB;oBAC9CC,OAAO6B,IAAI7B,KAAK;oBAChBC,gBAAgB4B,IAAI5B,cAAc;gBACpC;YACF,CAAA;QAGF,OAAOkD;IACT;IAEA,IAAIT,OAAOpD,MAAM,GAAG,GAAGT,8BAA8B8B,GAAG,CAAC+B,OAAOpD,MAAM;IAEtE,OAAOoD;AACT;AAEA,OAAO,eAAeW;IAIpB,MAAM,CAACC,QAAQ,GAAG,MAAMxE,KAAKwC,MAAM,CAAC;QAACiC,OAAOzF;IAAO,GAAG0D,IAAI,CAACxC;IAC3D,MAAM,CAACwE,QAAQ,GAAG,MAAM1E,KAAKwC,MAAM,CAAC;QAACiC,OAAOzF;IAAO,GAAG0D,IAAI,CAACtC;IAC3D,OAAO;QACLF,sBAAsBsE,SAASC,SAAS;QACxCrE,sBAAsBsE,SAASD,SAAS;IAC1C;AACF;AAEA,OAAO,eAAeE,+BAA+BrE,MAIpD;IACC,OAAO,MAAMN,KACVwC,MAAM,CAAC;QACNtB,OAAOd,qBAAqBc,KAAK;QACjCC,gBAAgBf,qBAAqBe,cAAc;QACnDH,eAAeZ,qBAAqBY,aAAa;QACjDC,sBAAsBb,qBAAqBa,oBAAoB;QAC/DG,WAAWhB,qBAAqBgB,SAAS;QACzCY,iBAAiB5B,qBAAqB4B,eAAe;QACrDI,eAAehC,qBAAqBgC,aAAa;QACjDC,qBAAqBjC,qBAAqBiC,mBAAmB;QAC7D9B,gBAAgBH,qBAAqBG,cAAc;QACnD4C,cAAc/C,qBAAqB+C,YAAY;QAC/CE,WAAWjD,qBAAqBiD,SAAS;QACzCc,iBAAiB/D,qBAAqB+D,eAAe;IACvD,GACCzB,IAAI,CAACtC,sBACLuC,KAAK,CACJ9D,IACEK,GAAGkB,qBAAqBW,WAAW,EAAET,OAAOS,WAAW,GACvDrB,GAAG,CAAC,EAAEU,qBAAqB+D,eAAe,CAAC,YAAY,EAAE7D,OAAOsE,aAAa,CAAC,yBAAyB,CAAC,GAG3G5B,OAAO,CAAC/D,KAAKmB,qBAAqB+D,eAAe,GAAGlF,KAAKmB,qBAAqBwC,EAAE,GAChFC,KAAK,CAACvC,OAAOuC,KAAK,IAAI;AAC3B;AAEA,OAAO,eAAegC,8CACpBlE,EAAM,EACNL,MAIC;IAED,IAAIA,OAAOwE,oBAAoB,CAACtE,MAAM,KAAK,GAAG,OAAO,EAAE;IAEvD,MAAMuE,gBAAgB,MAAMpE,GACzB6B,MAAM,CAAC;QACNH,qBAAqBjC,qBAAqBiC,mBAAmB;QAC7DrD,OAAOA;IACT,GACC0D,IAAI,CAACtC,sBACLuC,KAAK,CACJ9D,IACEK,GAAGkB,qBAAqBW,WAAW,EAAET,OAAOS,WAAW,GACvD7B,GAAGkB,qBAAqBgC,aAAa,EAAE9B,OAAO8B,aAAa,GAC3DhD,QAAQgB,qBAAqBiC,mBAAmB,EAAE/B,OAAOwE,oBAAoB,IAGhFE,OAAO,CAAC5E,qBAAqBiC,mBAAmB,EAChD4C,MAAM,CAACvF,GAAG,CAAC,YAAY,CAAC;IAE3B,MAAMwF,gCAAgCH,cAAcI,OAAO,CAAC,CAACpC,MAC3DA,IAAIV,mBAAmB,GAAG;YAACU,IAAIV,mBAAmB;SAAC,GAAG,EAAE;IAE1D,IAAI6C,8BAA8B1E,MAAM,GAAG,GAAG;QAC5C/B,SAAS2G,IAAI,CACX;YACErE,aAAaT,OAAOS,WAAW;YAC/BqB,eAAe9B,OAAO8B,aAAa;YACnC0C,sBAAsBI;QACxB,GACA;IAEJ;IAEA,MAAMG,SAAS,MAAM1E,GAAG2E,OAAO,CAS5B5F,GAAG,CAAC;wBACe,EAAEU,qBAAqBiC,mBAAmB,CAAC;MAC7D,EAAEjC,qBAAqBY,aAAa,CAAC;MACrC,EAAEZ,qBAAqBa,oBAAoB,CAAC;MAC5C,EAAEb,qBAAqBc,KAAK,CAAC;MAC7B,EAAEd,qBAAqBe,cAAc,CAAC;MACtC,EAAEf,qBAAqBiC,mBAAmB,CAAC;MAC3C,EAAEjC,qBAAqBiD,SAAS,CAAC;MACjC,EAAEjD,qBAAqB+D,eAAe,CAAC;MACvC,EAAE/D,qBAAqBmF,uBAAuB,CAAC;SAC5C,EAAEnF,qBAAqB;;MAE1B,EAAEA,qBAAqBW,WAAW,CAAC,GAAG,EAAET,OAAOS,WAAW,CAAC;UACvD,EAAEX,qBAAqBgC,aAAa,CAAC,GAAG,EAAE9B,OAAO8B,aAAa,CAAC;UAC/D,EAAEhC,qBAAqBiC,mBAAmB,CAAC,KAAK,EAAE3C,IAAI8F,IAAI,CAC5DlF,OAAOwE,oBAAoB,CAACR,GAAG,CAAC,CAACjC,sBAAwB3C,GAAG,CAAC,EAAE2C,oBAAoB,CAAC,GACpF3C,GAAG,CAAC,EAAE,CAAC,EACP;aACK,EAAEU,qBAAqBiC,mBAAmB,CAAC,EAAE,EAAEjC,qBAAqBiD,SAAS,CAAC,OAAO,EAAEjD,qBAAqBe,cAAc,CAAC;EACtI,CAAC;IAED,OAAOkE,OAAOI,IAAI,CAACnB,GAAG,CAAC,CAACvB,MAAS,CAAA;YAC/B/B,eAAe+B,IAAI/B,aAAa;YAChCC,sBAAsB8B,IAAI9B,oBAAoB;YAC9CC,OAAO6B,IAAI7B,KAAK;YAChBC,gBAAgB4B,IAAI5B,cAAc;YAClCkB,qBAAqBU,IAAIV,mBAAmB;YAC5CgB,WAAWqC,OAAO3C,IAAIM,SAAS;YAC/Bc,iBAAiBuB,OAAO3C,IAAIoB,eAAe;YAC3CoB,yBAAyBxC,IAAIwC,uBAAuB,GAChDG,OAAO3C,IAAIwC,uBAAuB,IAClC;QACN,CAAA;AACF;AAEA,SAASG,OAAOjB,KAAoB;IAClC,OAAOA,iBAAiBkB,OAAOlB,QAAQ,IAAIkB,KAAKlB;AAClD;AAEA,OAAO,eAAemB,iBAAiBtF,MAItC;IACC,MAAM,CAACyC,IAAI,GAAG,MAAM/C,KACjBwC,MAAM,CAAC;QAACI,IAAIxC,qBAAqBwC,EAAE;IAAA,GACnCF,IAAI,CAACtC,sBACLuC,KAAK,CACJ9D,IACEK,GAAGkB,qBAAqBe,cAAc,EAAEb,OAAOa,cAAc,GAC7DjC,GAAGkB,qBAAqB4B,eAAe,EAAE1B,OAAO0B,eAAe,GAC/D1B,OAAOY,KAAK,KAAK2E,YAAYA,YAAY3G,GAAGkB,qBAAqBc,KAAK,EAAEZ,OAAOY,KAAK,IAGvF2B,KAAK,CAAC;IAET,OAAOE,QAAQ8C;AACjB;AAEA,OAAO,eAAeC,gBAAgBxF,MAIrC;IAcC,MAAM+E,SAAS,MAAMrF,KAAKU,WAAW,CAAC,OAAOC;QAC3C,MAAMoF,UAAU,MAAMpF,GACnB4C,MAAM,CAACnD,sBACPoD,GAAG,CAAC;YACHU,kBAAkBxE,GAAG,CAAC,SAAS,EAAEU,qBAAqB8D,gBAAgB,CAAC,QAAQ,CAAC;YAChFC,iBAAiBzE,GAAG,CAAC,KAAK,CAAC;QAC7B,GACCiD,KAAK,CACJ9D,IACEK,GAAGkB,qBAAqBe,cAAc,EAAEb,OAAOa,cAAc,GAC7DjC,GAAGkB,qBAAqB4B,eAAe,EAAE1B,OAAO0B,eAAe,IAGlET,SAAS,CAAC;YACTgE,yBAAyBnF,qBAAqBmF,uBAAuB;YACrEvE,eAAeZ,qBAAqBY,aAAa;YACjDC,sBAAsBb,qBAAqBa,oBAAoB;YAC/DC,OAAOd,qBAAqBc,KAAK;YACjCC,gBAAgBf,qBAAqBe,cAAc;YACnDC,WAAWhB,qBAAqBgB,SAAS;YACzCL,aAAaX,qBAAqBW,WAAW;YAC7CiB,iBAAiB5B,qBAAqB4B,eAAe;QACvD;QAEF,MAAMe,MAAMgD,OAAO,CAAC,EAAE;QACtB,IAAI,CAAChD,KAAK,MAAM,IAAIlD,iCAAiCS,OAAOa,cAAc;QAE1E,MAAM,CAAC6E,SAAS,GAAG,MAAMrF,GACtB6B,MAAM,CAAC;YAACyD,kBAAkB9F,eAAe8F,gBAAgB;QAAA,GACzDvD,IAAI,CAACvC,gBACLwC,KAAK,CAACzD,GAAGiB,eAAeyC,EAAE,EAAEtC,OAAO0B,eAAe,GAClDa,KAAK,CAAC,GACNC,GAAG,CAAC;QAEP,MAAM,CAACP,QAAQ,GAAG,MAAM5B,GACrB4C,MAAM,CAACpD,gBACPqD,GAAG,CAAC;YACHyC,kBAAkB3F,OAAO2F,gBAAgB,IAAI;YAC7CC,4BAA4B5F,OAAO2F,gBAAgB,GAAGvG,GAAG,CAAC,KAAK,CAAC,GAAG;YACnE+D,WAAW/D,GAAG,CAAC,KAAK,CAAC;QACvB,GACCiD,KAAK,CAACzD,GAAGiB,eAAeyC,EAAE,EAAEtC,OAAO0B,eAAe,GAClDT,SAAS,CAAC;YAAC0E,kBAAkB9F,eAAe8F,gBAAgB;QAAA;QAE/D,OAAO;YACLlD;YACAoD,0BAA0BH,UAAUC,oBAAoB;YACxDG,yBAAyB7D,SAAS0D,oBAAoB;QACxD;IACF;IAEA,MAAMlD,MAAMsC,OAAOtC,GAAG;IACtB,IAAI,CAACA,KAAK,MAAM,IAAIlD,iCAAiCS,OAAOa,cAAc;IAC1E,OAAO;QACLkF,uBAAuBtD,IAAIwC,uBAAuB,KAAK;QACvDY,0BAA0Bd,OAAOc,wBAAwB;QACzDC,yBAAyBf,OAAOe,uBAAuB;QACvDE,qBAAqB;YACnBtF,eAAe+B,IAAI/B,aAAa;YAChCC,sBAAsB8B,IAAI9B,oBAAoB;YAC9CC,OAAO6B,IAAI7B,KAAK;YAChBC,gBAAgB4B,IAAI5B,cAAc;YAClCC,WAAW2B,IAAI3B,SAAS;YACxBL,aAAagC,IAAIhC,WAAW;YAC5BiB,iBAAiBe,IAAIf,eAAe;QACtC;IACF;AACF;AAEA,OAAO,eAAeuE,gCAAgCjG,MAErD;IACC,MAAMN,KACHuD,MAAM,CAACnD,sBACPoD,GAAG,CAAC;QACH+B,yBAAyB7F,GAAG,CAAC,SAAS,EAAEU,qBAAqBmF,uBAAuB,CAAC,QAAQ,CAAC;IAChG,GACC5C,KAAK,CAACzD,GAAGkB,qBAAqBe,cAAc,EAAEb,OAAOa,cAAc;AACxE;AAEA,OAAO,eAAeqF,iBAAiBlG,MAA0B;IAC/D,IAAIA,OAAOmG,MAAM,CAACjG,MAAM,KAAK,GAAG;IAEhC,MAAMR,KAAKU,WAAW,CAAC,OAAOC;QAC5B,MAAMA,GAAGuC,MAAM,CAAChD,sBAAsByC,KAAK,CAACvD,QAAQc,qBAAqBgB,KAAK,EAAEZ,OAAOmG,MAAM;QAC7F,MAAM9F,GACH4C,MAAM,CAACnD,sBACPoD,GAAG,CAAC;YACH+B,yBAAyB7F,GAAG,CAAC,SAAS,EAAEU,qBAAqBmF,uBAAuB,CAAC,QAAQ,CAAC;QAChG,GACC5C,KAAK,CAACvD,QAAQgB,qBAAqBc,KAAK,EAAEZ,OAAOmG,MAAM;IAC5D;AACF"}
|
|
@@ -0,0 +1,16 @@
|
|
|
1
|
+
import type { ManualRegistrationToken } from '#core/entities/manual-registration-token.js';
|
|
2
|
+
export interface CreateManualRegistrationTokenParams {
|
|
3
|
+
workspaceId: string;
|
|
4
|
+
hashedToken: string;
|
|
5
|
+
prefix: string;
|
|
6
|
+
name?: string | undefined;
|
|
7
|
+
expiresAt?: Date | undefined;
|
|
8
|
+
}
|
|
9
|
+
export declare function createManualRegistrationToken(params: CreateManualRegistrationTokenParams): Promise<ManualRegistrationToken>;
|
|
10
|
+
export declare function resolveManualRegistrationTokenByHash(hashedToken: string): Promise<ManualRegistrationToken | undefined>;
|
|
11
|
+
export declare function listUsableManualRegistrationTokensByWorkspaceId(workspaceId: string): Promise<ManualRegistrationToken[]>;
|
|
12
|
+
export declare function revokeManualRegistrationToken(params: {
|
|
13
|
+
tokenId: string;
|
|
14
|
+
workspaceId: string;
|
|
15
|
+
}): Promise<ManualRegistrationToken | undefined>;
|
|
16
|
+
//# sourceMappingURL=manual-registration-tokens.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"manual-registration-tokens.d.ts","sourceRoot":"","sources":["../../src/db/manual-registration-tokens.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAC,uBAAuB,EAAC,MAAM,6CAA6C,CAAC;AAOzF,MAAM,WAAW,mCAAmC;IAClD,WAAW,EAAE,MAAM,CAAC;IACpB,WAAW,EAAE,MAAM,CAAC;IACpB,MAAM,EAAE,MAAM,CAAC;IACf,IAAI,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC1B,SAAS,CAAC,EAAE,IAAI,GAAG,SAAS,CAAC;CAC9B;AAED,wBAAsB,6BAA6B,CACjD,MAAM,EAAE,mCAAmC,GAC1C,OAAO,CAAC,uBAAuB,CAAC,CAelC;AAED,wBAAsB,oCAAoC,CACxD,WAAW,EAAE,MAAM,GAClB,OAAO,CAAC,uBAAuB,GAAG,SAAS,CAAC,CAU9C;AAED,wBAAsB,+CAA+C,CACnE,WAAW,EAAE,MAAM,GAClB,OAAO,CAAC,uBAAuB,EAAE,CAAC,CAepC;AAED,wBAAsB,6BAA6B,CAAC,MAAM,EAAE;IAC1D,OAAO,EAAE,MAAM,CAAC;IAChB,WAAW,EAAE,MAAM,CAAC;CACrB,GAAG,OAAO,CAAC,uBAAuB,GAAG,SAAS,CAAC,CAe/C"}
|
|
@@ -0,0 +1,37 @@
|
|
|
1
|
+
import { and, desc, eq, gt, isNull, or } from 'drizzle-orm';
|
|
2
|
+
import { db } from './db.js';
|
|
3
|
+
import { manualRegistrationTokens, toManualRegistrationToken } from './schema/manual-registration-tokens.js';
|
|
4
|
+
export async function createManualRegistrationToken(params) {
|
|
5
|
+
const rows = await db().insert(manualRegistrationTokens).values({
|
|
6
|
+
workspaceId: params.workspaceId,
|
|
7
|
+
hashedToken: params.hashedToken,
|
|
8
|
+
prefix: params.prefix,
|
|
9
|
+
name: params.name ?? null,
|
|
10
|
+
expiresAt: params.expiresAt ?? null
|
|
11
|
+
}).returning();
|
|
12
|
+
const row = rows[0];
|
|
13
|
+
if (!row) throw new Error('Insert returned no rows');
|
|
14
|
+
return toManualRegistrationToken(row);
|
|
15
|
+
}
|
|
16
|
+
export async function resolveManualRegistrationTokenByHash(hashedToken) {
|
|
17
|
+
const rows = await db().select().from(manualRegistrationTokens).where(eq(manualRegistrationTokens.hashedToken, hashedToken)).limit(1);
|
|
18
|
+
const row = rows[0];
|
|
19
|
+
if (!row) return undefined;
|
|
20
|
+
return toManualRegistrationToken(row);
|
|
21
|
+
}
|
|
22
|
+
export async function listUsableManualRegistrationTokensByWorkspaceId(workspaceId) {
|
|
23
|
+
const now = new Date();
|
|
24
|
+
const rows = await db().select().from(manualRegistrationTokens).where(and(eq(manualRegistrationTokens.workspaceId, workspaceId), isNull(manualRegistrationTokens.revokedAt), or(isNull(manualRegistrationTokens.expiresAt), gt(manualRegistrationTokens.expiresAt, now)))).orderBy(desc(manualRegistrationTokens.createdAt), desc(manualRegistrationTokens.id));
|
|
25
|
+
return rows.map(toManualRegistrationToken);
|
|
26
|
+
}
|
|
27
|
+
export async function revokeManualRegistrationToken(params) {
|
|
28
|
+
const rows = await db().update(manualRegistrationTokens).set({
|
|
29
|
+
revokedAt: new Date(),
|
|
30
|
+
updatedAt: new Date()
|
|
31
|
+
}).where(and(eq(manualRegistrationTokens.id, params.tokenId), eq(manualRegistrationTokens.workspaceId, params.workspaceId))).returning();
|
|
32
|
+
const row = rows[0];
|
|
33
|
+
if (!row) return undefined;
|
|
34
|
+
return toManualRegistrationToken(row);
|
|
35
|
+
}
|
|
36
|
+
|
|
37
|
+
//# sourceMappingURL=manual-registration-tokens.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"sources":["../../src/db/manual-registration-tokens.ts"],"sourcesContent":["import {and, desc, eq, gt, isNull, or} from 'drizzle-orm';\nimport type {ManualRegistrationToken} from '#core/entities/manual-registration-token.js';\nimport {db} from './db.js';\nimport {\n manualRegistrationTokens,\n toManualRegistrationToken,\n} from './schema/manual-registration-tokens.js';\n\nexport interface CreateManualRegistrationTokenParams {\n workspaceId: string;\n hashedToken: string;\n prefix: string;\n name?: string | undefined;\n expiresAt?: Date | undefined;\n}\n\nexport async function createManualRegistrationToken(\n params: CreateManualRegistrationTokenParams,\n): Promise<ManualRegistrationToken> {\n const rows = await db()\n .insert(manualRegistrationTokens)\n .values({\n workspaceId: params.workspaceId,\n hashedToken: params.hashedToken,\n prefix: params.prefix,\n name: params.name ?? null,\n expiresAt: params.expiresAt ?? null,\n })\n .returning();\n\n const row = rows[0];\n if (!row) throw new Error('Insert returned no rows');\n return toManualRegistrationToken(row);\n}\n\nexport async function resolveManualRegistrationTokenByHash(\n hashedToken: string,\n): Promise<ManualRegistrationToken | undefined> {\n const rows = await db()\n .select()\n .from(manualRegistrationTokens)\n .where(eq(manualRegistrationTokens.hashedToken, hashedToken))\n .limit(1);\n\n const row = rows[0];\n if (!row) return undefined;\n return toManualRegistrationToken(row);\n}\n\nexport async function listUsableManualRegistrationTokensByWorkspaceId(\n workspaceId: string,\n): Promise<ManualRegistrationToken[]> {\n const now = new Date();\n const rows = await db()\n .select()\n .from(manualRegistrationTokens)\n .where(\n and(\n eq(manualRegistrationTokens.workspaceId, workspaceId),\n isNull(manualRegistrationTokens.revokedAt),\n or(isNull(manualRegistrationTokens.expiresAt), gt(manualRegistrationTokens.expiresAt, now)),\n ),\n )\n .orderBy(desc(manualRegistrationTokens.createdAt), desc(manualRegistrationTokens.id));\n\n return rows.map(toManualRegistrationToken);\n}\n\nexport async function revokeManualRegistrationToken(params: {\n tokenId: string;\n workspaceId: string;\n}): Promise<ManualRegistrationToken | undefined> {\n const rows = await db()\n .update(manualRegistrationTokens)\n .set({revokedAt: new Date(), updatedAt: new Date()})\n .where(\n and(\n eq(manualRegistrationTokens.id, params.tokenId),\n eq(manualRegistrationTokens.workspaceId, params.workspaceId),\n ),\n )\n .returning();\n\n const row = rows[0];\n if (!row) return undefined;\n return toManualRegistrationToken(row);\n}\n"],"names":["and","desc","eq","gt","isNull","or","db","manualRegistrationTokens","toManualRegistrationToken","createManualRegistrationToken","params","rows","insert","values","workspaceId","hashedToken","prefix","name","expiresAt","returning","row","Error","resolveManualRegistrationTokenByHash","select","from","where","limit","undefined","listUsableManualRegistrationTokensByWorkspaceId","now","Date","revokedAt","orderBy","createdAt","id","map","revokeManualRegistrationToken","update","set","updatedAt","tokenId"],"mappings":"AAAA,SAAQA,GAAG,EAAEC,IAAI,EAAEC,EAAE,EAAEC,EAAE,EAAEC,MAAM,EAAEC,EAAE,QAAO,cAAc;AAE1D,SAAQC,EAAE,QAAO,UAAU;AAC3B,SACEC,wBAAwB,EACxBC,yBAAyB,QACpB,yCAAyC;AAUhD,OAAO,eAAeC,8BACpBC,MAA2C;IAE3C,MAAMC,OAAO,MAAML,KAChBM,MAAM,CAACL,0BACPM,MAAM,CAAC;QACNC,aAAaJ,OAAOI,WAAW;QAC/BC,aAAaL,OAAOK,WAAW;QAC/BC,QAAQN,OAAOM,MAAM;QACrBC,MAAMP,OAAOO,IAAI,IAAI;QACrBC,WAAWR,OAAOQ,SAAS,IAAI;IACjC,GACCC,SAAS;IAEZ,MAAMC,MAAMT,IAAI,CAAC,EAAE;IACnB,IAAI,CAACS,KAAK,MAAM,IAAIC,MAAM;IAC1B,OAAOb,0BAA0BY;AACnC;AAEA,OAAO,eAAeE,qCACpBP,WAAmB;IAEnB,MAAMJ,OAAO,MAAML,KAChBiB,MAAM,GACNC,IAAI,CAACjB,0BACLkB,KAAK,CAACvB,GAAGK,yBAAyBQ,WAAW,EAAEA,cAC/CW,KAAK,CAAC;IAET,MAAMN,MAAMT,IAAI,CAAC,EAAE;IACnB,IAAI,CAACS,KAAK,OAAOO;IACjB,OAAOnB,0BAA0BY;AACnC;AAEA,OAAO,eAAeQ,gDACpBd,WAAmB;IAEnB,MAAMe,MAAM,IAAIC;IAChB,MAAMnB,OAAO,MAAML,KAChBiB,MAAM,GACNC,IAAI,CAACjB,0BACLkB,KAAK,CACJzB,IACEE,GAAGK,yBAAyBO,WAAW,EAAEA,cACzCV,OAAOG,yBAAyBwB,SAAS,GACzC1B,GAAGD,OAAOG,yBAAyBW,SAAS,GAAGf,GAAGI,yBAAyBW,SAAS,EAAEW,QAGzFG,OAAO,CAAC/B,KAAKM,yBAAyB0B,SAAS,GAAGhC,KAAKM,yBAAyB2B,EAAE;IAErF,OAAOvB,KAAKwB,GAAG,CAAC3B;AAClB;AAEA,OAAO,eAAe4B,8BAA8B1B,MAGnD;IACC,MAAMC,OAAO,MAAML,KAChB+B,MAAM,CAAC9B,0BACP+B,GAAG,CAAC;QAACP,WAAW,IAAID;QAAQS,WAAW,IAAIT;IAAM,GACjDL,KAAK,CACJzB,IACEE,GAAGK,yBAAyB2B,EAAE,EAAExB,OAAO8B,OAAO,GAC9CtC,GAAGK,yBAAyBO,WAAW,EAAEJ,OAAOI,WAAW,IAG9DK,SAAS;IAEZ,MAAMC,MAAMT,IAAI,CAAC,EAAE;IACnB,IAAI,CAACS,KAAK,OAAOO;IACjB,OAAOnB,0BAA0BY;AACnC"}
|
|
@@ -0,0 +1,79 @@
|
|
|
1
|
+
import type { ProvisionedRunner, ProvisionedRunnerState } from '#core/entities/provisioned-runner.js';
|
|
2
|
+
import type { Tx } from './db.js';
|
|
3
|
+
import { type ProvisionedRunnerBoundJobExecution } from './job-executions.js';
|
|
4
|
+
export declare const terminalStates: readonly ["stopped", "failed", "terminated"];
|
|
5
|
+
export declare const activeStates: readonly ["starting", "running", "stopping"];
|
|
6
|
+
export declare const divergenceCountStates: readonly ["starting", "running"];
|
|
7
|
+
export type ProvisionedRunnerTerminateIntentReason = 'job-cancelled';
|
|
8
|
+
export interface ProvisionedRunnerTerminateIntent {
|
|
9
|
+
provisionedRunnerId: string;
|
|
10
|
+
reason: ProvisionedRunnerTerminateIntentReason;
|
|
11
|
+
}
|
|
12
|
+
export interface ActiveProvisionedRunnerTemplateCount {
|
|
13
|
+
templateKey: string;
|
|
14
|
+
state: (typeof divergenceCountStates)[number];
|
|
15
|
+
count: number;
|
|
16
|
+
}
|
|
17
|
+
export interface ProvisionedRunnerReportEvent {
|
|
18
|
+
provisionedRunnerId: string;
|
|
19
|
+
reservationId: string | null;
|
|
20
|
+
templateKey: string | null;
|
|
21
|
+
labels: string[];
|
|
22
|
+
state: ProvisionedRunnerState;
|
|
23
|
+
reason: string | null;
|
|
24
|
+
runnerSessionId: string | null;
|
|
25
|
+
providerKind: string | null;
|
|
26
|
+
reportedAt: Date;
|
|
27
|
+
}
|
|
28
|
+
export interface ReportProvisionedRunnersParams {
|
|
29
|
+
workspaceId: string;
|
|
30
|
+
provisionerId: string;
|
|
31
|
+
events: ProvisionedRunnerReportEvent[];
|
|
32
|
+
}
|
|
33
|
+
export interface ReconcileProvisionedRunnersParams {
|
|
34
|
+
workspaceId: string;
|
|
35
|
+
provisionerId: string;
|
|
36
|
+
observedProvisionedRunnerIds: string[];
|
|
37
|
+
terminateGraceSeconds: number;
|
|
38
|
+
}
|
|
39
|
+
export interface ReconcileProvisionedRunnersDbResult {
|
|
40
|
+
observedRows: ProvisionedRunner[];
|
|
41
|
+
boundJobExecutionsByProvisionedRunnerId: Map<string, ProvisionedRunnerBoundJobExecution>;
|
|
42
|
+
absentIds: string[];
|
|
43
|
+
reservationsReleased: number;
|
|
44
|
+
}
|
|
45
|
+
export interface ReapStaleProvisionedRunnersResult {
|
|
46
|
+
reaped: number;
|
|
47
|
+
reservationsReleased: number;
|
|
48
|
+
}
|
|
49
|
+
export declare function reportProvisionedRunners(params: ReportProvisionedRunnersParams): Promise<{
|
|
50
|
+
accepted: number;
|
|
51
|
+
reservationsReleased: number;
|
|
52
|
+
terminateIntentsHonored: ProvisionedRunnerTerminateIntent[];
|
|
53
|
+
}>;
|
|
54
|
+
export declare function listActiveProvisionedRunnerCountsByTemplateTx(tx: Tx, params: {
|
|
55
|
+
workspaceId: string;
|
|
56
|
+
provisionerId: string;
|
|
57
|
+
}): Promise<ActiveProvisionedRunnerTemplateCount[]>;
|
|
58
|
+
export declare function listActiveProvisionedRunners(params: {
|
|
59
|
+
workspaceId: string;
|
|
60
|
+
windowSeconds: number;
|
|
61
|
+
limit?: number;
|
|
62
|
+
}): Promise<ProvisionedRunner[]>;
|
|
63
|
+
export declare function listProvisionerTerminateIntents(params: {
|
|
64
|
+
workspaceId: string;
|
|
65
|
+
provisionerId: string;
|
|
66
|
+
limit: number;
|
|
67
|
+
}): Promise<string[]>;
|
|
68
|
+
export declare function listProvisionerTerminateIntentRowsTx(tx: Tx, params: {
|
|
69
|
+
workspaceId: string;
|
|
70
|
+
provisionerId: string;
|
|
71
|
+
limit: number;
|
|
72
|
+
}): Promise<ProvisionedRunnerTerminateIntent[]>;
|
|
73
|
+
export declare function reconcileProvisionedRunners(params: ReconcileProvisionedRunnersParams): Promise<ReconcileProvisionedRunnersDbResult>;
|
|
74
|
+
export declare function reapStaleProvisionedRunners(params: {
|
|
75
|
+
thresholdSeconds: number;
|
|
76
|
+
limit: number;
|
|
77
|
+
}): Promise<ReapStaleProvisionedRunnersResult>;
|
|
78
|
+
export declare function isTerminalState(state: ProvisionedRunnerState): boolean;
|
|
79
|
+
//# sourceMappingURL=provisioned-runners.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"provisioned-runners.d.ts","sourceRoot":"","sources":["../../src/db/provisioned-runners.ts"],"names":[],"mappings":"AAoBA,OAAO,KAAK,EAAC,iBAAiB,EAAE,sBAAsB,EAAC,MAAM,sCAAsC,CAAC;AACpG,OAAO,KAAK,EAAC,EAAE,EAAC,MAAM,SAAS,CAAC;AAEhC,OAAO,EAEL,KAAK,kCAAkC,EACxC,MAAM,qBAAqB,CAAC;AAQ7B,eAAO,MAAM,cAAc,8CAI2B,CAAC;AACvD,eAAO,MAAM,YAAY,8CAI6B,CAAC;AACvD,eAAO,MAAM,qBAAqB,kCAG/B,CAAC;AAEJ,MAAM,MAAM,sCAAsC,GAAG,eAAe,CAAC;AAErE,MAAM,WAAW,gCAAgC;IAC/C,mBAAmB,EAAE,MAAM,CAAC;IAC5B,MAAM,EAAE,sCAAsC,CAAC;CAChD;AAED,MAAM,WAAW,oCAAoC;IACnD,WAAW,EAAE,MAAM,CAAC;IACpB,KAAK,EAAE,CAAC,OAAO,qBAAqB,CAAC,CAAC,MAAM,CAAC,CAAC;IAC9C,KAAK,EAAE,MAAM,CAAC;CACf;AAED,MAAM,WAAW,4BAA4B;IAC3C,mBAAmB,EAAE,MAAM,CAAC;IAC5B,aAAa,EAAE,MAAM,GAAG,IAAI,CAAC;IAC7B,WAAW,EAAE,MAAM,GAAG,IAAI,CAAC;IAC3B,MAAM,EAAE,MAAM,EAAE,CAAC;IACjB,KAAK,EAAE,sBAAsB,CAAC;IAC9B,MAAM,EAAE,MAAM,GAAG,IAAI,CAAC;IACtB,eAAe,EAAE,MAAM,GAAG,IAAI,CAAC;IAC/B,YAAY,EAAE,MAAM,GAAG,IAAI,CAAC;IAC5B,UAAU,EAAE,IAAI,CAAC;CAClB;AAED,MAAM,WAAW,8BAA8B;IAC7C,WAAW,EAAE,MAAM,CAAC;IACpB,aAAa,EAAE,MAAM,CAAC;IACtB,MAAM,EAAE,4BAA4B,EAAE,CAAC;CACxC;AAED,MAAM,WAAW,iCAAiC;IAChD,WAAW,EAAE,MAAM,CAAC;IACpB,aAAa,EAAE,MAAM,CAAC;IACtB,4BAA4B,EAAE,MAAM,EAAE,CAAC;IACvC,qBAAqB,EAAE,MAAM,CAAC;CAC/B;AAED,MAAM,WAAW,mCAAmC;IAClD,YAAY,EAAE,iBAAiB,EAAE,CAAC;IAClC,uCAAuC,EAAE,GAAG,CAAC,MAAM,EAAE,kCAAkC,CAAC,CAAC;IACzF,SAAS,EAAE,MAAM,EAAE,CAAC;IACpB,oBAAoB,EAAE,MAAM,CAAC;CAC9B;AAED,MAAM,WAAW,iCAAiC;IAChD,MAAM,EAAE,MAAM,CAAC;IACf,oBAAoB,EAAE,MAAM,CAAC;CAC9B;AAiBD,wBAAsB,wBAAwB,CAAC,MAAM,EAAE,8BAA8B,GAAG,OAAO,CAAC;IAC9F,QAAQ,EAAE,MAAM,CAAC;IACjB,oBAAoB,EAAE,MAAM,CAAC;IAC7B,uBAAuB,EAAE,gCAAgC,EAAE,CAAC;CAC7D,CAAC,CA+ED;AAED,wBAAsB,6CAA6C,CACjE,EAAE,EAAE,EAAE,EACN,MAAM,EAAE;IAAC,WAAW,EAAE,MAAM,CAAC;IAAC,aAAa,EAAE,MAAM,CAAA;CAAC,GACnD,OAAO,CAAC,oCAAoC,EAAE,CAAC,CAuBjD;AAED,wBAAsB,4BAA4B,CAAC,MAAM,EAAE;IACzD,WAAW,EAAE,MAAM,CAAC;IACpB,aAAa,EAAE,MAAM,CAAC;IACtB,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB,GAAG,OAAO,CAAC,iBAAiB,EAAE,CAAC,CAe/B;AAED,wBAAsB,+BAA+B,CAAC,MAAM,EAAE;IAC5D,WAAW,EAAE,MAAM,CAAC;IACpB,aAAa,EAAE,MAAM,CAAC;IACtB,KAAK,EAAE,MAAM,CAAC;CACf,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,CAKpB;AAED,wBAAsB,oCAAoC,CACxD,EAAE,EAAE,EAAE,EACN,MAAM,EAAE;IACN,WAAW,EAAE,MAAM,CAAC;IACpB,aAAa,EAAE,MAAM,CAAC;IACtB,KAAK,EAAE,MAAM,CAAC;CACf,GACA,OAAO,CAAC,gCAAgC,EAAE,CAAC,CAoB7C;AAkFD,wBAAsB,2BAA2B,CAC/C,MAAM,EAAE,iCAAiC,GACxC,OAAO,CAAC,mCAAmC,CAAC,CA2F9C;AAED,wBAAsB,2BAA2B,CAAC,MAAM,EAAE;IACxD,gBAAgB,EAAE,MAAM,CAAC;IACzB,KAAK,EAAE,MAAM,CAAC;CACf,GAAG,OAAO,CAAC,iCAAiC,CAAC,CA0D7C;AAiWD,wBAAgB,eAAe,CAAC,KAAK,EAAE,sBAAsB,GAAG,OAAO,CAEtE"}
|