wehub-resource-sync
108 lines
3.1 KiB
TypeScript
108 lines
3.1 KiB
TypeScript
import type { ActionFunctionArgs, LoaderFunctionArgs } from "@remix-run/server-runtime";
|
|
import { json } from "@remix-run/server-runtime";
|
|
import { z } from "zod";
|
|
import { prisma } from "~/db.server";
|
|
import { requireAdminApiRequest } from "~/services/personalAccessToken.server";
|
|
import { controlPlaneResolver } from "~/v3/runOpsMigration/controlPlaneResolver.server";
|
|
import { engine } from "~/v3/runEngine.server";
|
|
import { updateEnvConcurrencyLimits } from "~/v3/runQueue.server";
|
|
|
|
const ParamsSchema = z.object({
|
|
environmentId: z.string(),
|
|
});
|
|
|
|
const RequestBodySchema = z.object({
|
|
envMaximumConcurrencyLimit: z.number(),
|
|
orgMaximumConcurrencyLimit: z.number(),
|
|
});
|
|
|
|
export async function action({ request, params }: ActionFunctionArgs) {
|
|
await requireAdminApiRequest(request);
|
|
|
|
const parsedParams = ParamsSchema.parse(params);
|
|
|
|
const rawBody = await request.json();
|
|
const body = RequestBodySchema.parse(rawBody);
|
|
|
|
const environment = await prisma.runtimeEnvironment.update({
|
|
where: {
|
|
id: parsedParams.environmentId,
|
|
},
|
|
data: {
|
|
maximumConcurrencyLimit: body.envMaximumConcurrencyLimit,
|
|
organization: {
|
|
update: {
|
|
data: {
|
|
maximumConcurrencyLimit: body.orgMaximumConcurrencyLimit,
|
|
},
|
|
},
|
|
},
|
|
},
|
|
include: {
|
|
organization: true,
|
|
project: true,
|
|
},
|
|
});
|
|
|
|
await updateEnvConcurrencyLimits(environment);
|
|
|
|
// Org max-concurrency changed too, which is embedded in every env of the org; invalidating
|
|
// the org drops the env/authEnv rows for all of them (including this env).
|
|
controlPlaneResolver.invalidateOrganization(environment.organizationId);
|
|
|
|
return json({ success: true });
|
|
}
|
|
|
|
const SearchParamsSchema = z.object({
|
|
queue: z.string().optional(),
|
|
});
|
|
|
|
export async function loader({ request, params }: LoaderFunctionArgs) {
|
|
await requireAdminApiRequest(request);
|
|
|
|
const parsedParams = ParamsSchema.parse(params);
|
|
|
|
const environment = await prisma.runtimeEnvironment.findFirst({
|
|
where: {
|
|
id: parsedParams.environmentId,
|
|
},
|
|
include: {
|
|
organization: true,
|
|
project: true,
|
|
},
|
|
});
|
|
|
|
if (!environment) {
|
|
return json({ error: "Environment not found" }, { status: 404 });
|
|
}
|
|
|
|
const requestUrl = new URL(request.url);
|
|
const searchParams = SearchParamsSchema.parse(
|
|
Object.fromEntries(requestUrl.searchParams.entries())
|
|
);
|
|
|
|
const concurrencyLimit = await engine.runQueue.getEnvConcurrencyLimit(environment);
|
|
const currentConcurrency = await engine.runQueue.currentConcurrencyOfEnvironment(environment);
|
|
|
|
if (searchParams.queue) {
|
|
const queueConcurrencyLimit = await engine.runQueue.getQueueConcurrencyLimit(
|
|
environment,
|
|
searchParams.queue
|
|
);
|
|
const queueCurrentConcurrency = await engine.runQueue.currentConcurrencyOfQueue(
|
|
environment,
|
|
searchParams.queue
|
|
);
|
|
|
|
return json({
|
|
id: environment.id,
|
|
concurrencyLimit,
|
|
currentConcurrency,
|
|
queueConcurrencyLimit,
|
|
queueCurrentConcurrency,
|
|
});
|
|
}
|
|
|
|
return json({ id: environment.id, concurrencyLimit, currentConcurrency });
|
|
}
|