wehub-resource-sync
59 lines
1.9 KiB
TypeScript
59 lines
1.9 KiB
TypeScript
import { type AuthenticatedEnvironment } from "~/services/apiAuth.server";
|
|
import { marqs } from "~/v3/marqs/index.server";
|
|
import { engine } from "~/v3/runEngine.server";
|
|
import { getQueueSizeLimit } from "~/v3/utils/queueLimits.server";
|
|
import { BasePresenter } from "./basePresenter.server";
|
|
|
|
export type Environment = {
|
|
running: number;
|
|
queued: number;
|
|
concurrencyLimit: number;
|
|
burstFactor: number;
|
|
runsEnabled: boolean;
|
|
queueSizeLimit: number | null;
|
|
};
|
|
|
|
export class EnvironmentQueuePresenter extends BasePresenter {
|
|
async call(environment: AuthenticatedEnvironment): Promise<Environment> {
|
|
const [engineV1Executing, engineV2Executing, engineV1Queued, engineV2Queued] =
|
|
await Promise.all([
|
|
marqs.currentConcurrencyOfEnvironment(environment),
|
|
engine.concurrencyOfEnvQueue(environment),
|
|
marqs.lengthOfEnvQueue(environment),
|
|
engine.lengthOfEnvQueue(environment),
|
|
]);
|
|
|
|
const running = (engineV1Executing ?? 0) + (engineV2Executing ?? 0);
|
|
const queued = (engineV1Queued ?? 0) + (engineV2Queued ?? 0);
|
|
|
|
const organization = await this._replica.organization.findFirst({
|
|
where: {
|
|
id: environment.organizationId,
|
|
},
|
|
select: {
|
|
runsEnabled: true,
|
|
maximumDevQueueSize: true,
|
|
maximumDeployedQueueSize: true,
|
|
},
|
|
});
|
|
|
|
if (!organization) {
|
|
throw new Error("Organization not found");
|
|
}
|
|
|
|
const queueSizeLimit = getQueueSizeLimit(environment.type, organization);
|
|
|
|
return {
|
|
running,
|
|
queued,
|
|
concurrencyLimit: environment.maximumConcurrencyLimit,
|
|
burstFactor:
|
|
typeof environment.concurrencyLimitBurstFactor === "number"
|
|
? environment.concurrencyLimitBurstFactor
|
|
: environment.concurrencyLimitBurstFactor.toNumber(),
|
|
runsEnabled: environment.type === "DEVELOPMENT" || organization.runsEnabled,
|
|
queueSizeLimit,
|
|
};
|
|
}
|
|
}
|