91 lines
2.5 KiB
TypeScript
Executable File
91 lines
2.5 KiB
TypeScript
Executable File
import { Controller, Get, Res } from "@nestjs/common";
|
|
import { ApiTags, ApiOperation } from "@nestjs/swagger";
|
|
import type { Response } from "express";
|
|
import { Public } from "../../common/decorators";
|
|
import { PrismaService } from "../../database/prisma.service";
|
|
import { PredictionsService } from "../predictions/predictions.service";
|
|
|
|
@ApiTags("Health")
|
|
@Controller("health")
|
|
export class HealthController {
|
|
constructor(
|
|
private prisma: PrismaService,
|
|
private readonly predictionsService: PredictionsService,
|
|
) {}
|
|
|
|
@Get()
|
|
@Public()
|
|
@ApiOperation({ summary: "Basic health check" })
|
|
async check(@Res() response: Response) {
|
|
const database = await this.getDatabaseHealth();
|
|
const aiEngine = await this.predictionsService.checkHealth();
|
|
const ok = database.status === "up" && aiEngine.predictionServiceReady;
|
|
|
|
return response.status(ok ? 200 : 503).json({
|
|
status: ok ? "ok" : "degraded",
|
|
timestamp: new Date().toISOString(),
|
|
checks: {
|
|
database,
|
|
aiEngine,
|
|
},
|
|
});
|
|
}
|
|
|
|
@Get("ready")
|
|
@Public()
|
|
@ApiOperation({ summary: "Readiness check (includes database)" })
|
|
async readiness(@Res() response: Response) {
|
|
const database = await this.getDatabaseHealth();
|
|
const aiEngine = await this.predictionsService.checkHealth();
|
|
const ready = database.status === "up" && aiEngine.predictionServiceReady;
|
|
|
|
return response.status(ready ? 200 : 503).json({
|
|
status: ready ? "ready" : "not_ready",
|
|
timestamp: new Date().toISOString(),
|
|
checks: {
|
|
database,
|
|
aiEngine,
|
|
},
|
|
});
|
|
}
|
|
|
|
@Get("live")
|
|
@Public()
|
|
@ApiOperation({ summary: "Liveness check" })
|
|
liveness(@Res() response: Response) {
|
|
return response
|
|
.status(200)
|
|
.json({ status: "ok", timestamp: new Date().toISOString() });
|
|
}
|
|
|
|
@Get("dependencies")
|
|
@Public()
|
|
@ApiOperation({ summary: "Dependency-level health details" })
|
|
async dependencies(@Res() response: Response) {
|
|
const database = await this.getDatabaseHealth();
|
|
const aiEngine = await this.predictionsService.checkHealth();
|
|
|
|
return response.status(200).json({
|
|
timestamp: new Date().toISOString(),
|
|
checks: {
|
|
database,
|
|
aiEngine,
|
|
},
|
|
});
|
|
}
|
|
|
|
private async getDatabaseHealth() {
|
|
try {
|
|
await this.prisma.$queryRaw`SELECT 1`;
|
|
return {
|
|
status: "up",
|
|
};
|
|
} catch (error: unknown) {
|
|
return {
|
|
status: "down",
|
|
detail: error instanceof Error ? error.message : "Unknown database error",
|
|
};
|
|
}
|
|
}
|
|
}
|