126 lines
3.4 KiB
TypeScript
126 lines
3.4 KiB
TypeScript
import type { Context } from 'koa';
|
|
import { Controller, Get, Post } from '../../decorators/route.ts';
|
|
import type { Prisma } from '../../generated/client.ts';
|
|
import { ExecutionQueue } from '../../libs/execution-queue.ts';
|
|
import { prisma } from '../../libs/prisma.ts';
|
|
import { createDeploymentSchema, listDeploymentsQuerySchema } from './dto.ts';
|
|
|
|
@Controller('/deployments')
|
|
export class DeploymentController {
|
|
@Get('')
|
|
async list(ctx: Context) {
|
|
const { page, pageSize, projectId } = listDeploymentsQuerySchema.parse(
|
|
ctx.query,
|
|
);
|
|
const where: Prisma.DeploymentWhereInput = {
|
|
valid: 1,
|
|
};
|
|
|
|
if (projectId) {
|
|
where.projectId = projectId;
|
|
}
|
|
|
|
const isPagination = page !== undefined && pageSize !== undefined;
|
|
|
|
const result = await prisma.deployment.findMany({
|
|
where,
|
|
take: isPagination ? pageSize : undefined,
|
|
skip: isPagination ? (page! - 1) * pageSize! : 0,
|
|
orderBy: {
|
|
createdAt: 'desc',
|
|
},
|
|
});
|
|
const total = await prisma.deployment.count({ where });
|
|
|
|
if (isPagination) {
|
|
return {
|
|
list: result,
|
|
page,
|
|
pageSize,
|
|
total,
|
|
};
|
|
}
|
|
|
|
return result;
|
|
}
|
|
|
|
@Post('')
|
|
async create(ctx: Context) {
|
|
const body = createDeploymentSchema.parse(ctx.request.body);
|
|
|
|
const result = await prisma.deployment.create({
|
|
data: {
|
|
branch: body.branch,
|
|
commitHash: body.commitHash,
|
|
commitMessage: body.commitMessage,
|
|
status: 'pending',
|
|
Project: {
|
|
connect: { id: body.projectId },
|
|
},
|
|
pipelineId: body.pipelineId,
|
|
envVars: body.envVars ? JSON.stringify(body.envVars) : null,
|
|
buildLog: '',
|
|
createdBy: 'system', // TODO: get from user
|
|
updatedBy: 'system',
|
|
valid: 1,
|
|
},
|
|
});
|
|
|
|
// 将新创建的部署任务添加到执行队列
|
|
const executionQueue = ExecutionQueue.getInstance();
|
|
await executionQueue.addTask(result.id, result.pipelineId);
|
|
|
|
return result;
|
|
}
|
|
|
|
// 添加重新执行部署的接口
|
|
@Post('/:id/retry')
|
|
async retry(ctx: Context) {
|
|
const { id } = ctx.params;
|
|
|
|
// 获取原始部署记录
|
|
const originalDeployment = await prisma.deployment.findUnique({
|
|
where: { id: Number(id) },
|
|
});
|
|
|
|
if (!originalDeployment) {
|
|
ctx.status = 404;
|
|
ctx.body = {
|
|
code: 404,
|
|
message: '部署记录不存在',
|
|
data: null,
|
|
timestamp: Date.now(),
|
|
};
|
|
return;
|
|
}
|
|
|
|
// 创建一个新的部署记录，复制原始记录的信息
|
|
const newDeployment = await prisma.deployment.create({
|
|
data: {
|
|
branch: originalDeployment.branch,
|
|
commitHash: originalDeployment.commitHash,
|
|
commitMessage: originalDeployment.commitMessage,
|
|
status: 'pending',
|
|
projectId: originalDeployment.projectId,
|
|
pipelineId: originalDeployment.pipelineId,
|
|
envVars: originalDeployment.envVars,
|
|
buildLog: '',
|
|
createdBy: 'system',
|
|
updatedBy: 'system',
|
|
valid: 1,
|
|
},
|
|
});
|
|
|
|
// 将新创建的部署任务添加到执行队列
|
|
const executionQueue = ExecutionQueue.getInstance();
|
|
await executionQueue.addTask(newDeployment.id, newDeployment.pipelineId);
|
|
|
|
ctx.body = {
|
|
code: 0,
|
|
message: '重新执行任务已创建',
|
|
data: newDeployment,
|
|
timestamp: Date.now(),
|
|
};
|
|
}
|
|
}
|