swagger #33

Merged
VladislavD merged 2 commits from swagger into rc 2025-08-07 14:58:34 +03:00
18 changed files with 393 additions and 74 deletions

21
.env
View File

@ -1,9 +1,9 @@
#Прометеус
# Прометеус
#PROMETHEUS_API=http://192.168.2.34:9090/api/v1
#FRONTEND_URL=192.168.2.39:5173
#Постгресс
# Постгресс
#DB_HOST=192.168.2.37
#DB_PORT=5432
#DB_USER=trust
@ -11,19 +11,28 @@
#DB_NAME=trust-db
#JWT
# JWT
#JWT_SECRET=x7F!2p9L#q1$z0*8R5vYgMnBk
#JWT_SECRET=x7Fcdp9L#q1$z0*8R5vYgMnBk
#COOKIE
# COOKIE
# Для production
#COOKIE_SECURE=true
#COOKIE_SAME_SITE=strict
# Для development
# COOKIE_SECURE=false
# COOKIE_SAME_SITE=lax
#COOKIE_SECURE=false
#COOKIE_SAME_SITE=lax
# Для меню
#RANGES_API_URL=http://192.168.2.39:9999
#RANGES_API_ENDPOINT=/api/ranges/9999
# ClickHouse
#CLICKHOUSE_HOST=http://192.168.2.37:8123
#CLICKHOUSE_USER=vlad
#CLICKHOUSE_PASSWORD=vlad
#CLICKHOUSE_DB=zvks
# Для ai api
#AI_SERVICE_URL=http://192.168.2.39:5134

View File

@ -44,7 +44,10 @@
"@types/cookie-parser": "^1.4.8",
"@nestjs/jwt": "^11.0.0",
"@nestjs/passport": "^11.0.5",
"@nestjs/swagger": "11.1.4"
"@nestjs/swagger": "11.1.4",
"@clickhouse/client": "^1.11.2",
"date-fns": "4.1.0",
"@clickhouse/client-web": "^1.11.2"
},
"devDependencies": {
"@eslint/eslintrc": "^3.2.0",

0
src/ai/ai.service.ts Normal file
View File

View File

@ -4,7 +4,9 @@ import { HttpModule } from '@nestjs/axios';
import { ConfigModule } from '@nestjs/config';
import { AuthModule } from './auth/auth.module';
import { MenuModule } from './menu/menu.module';
import { PrometheusModule } from './prometheus.module';
import { PrometheusModule } from './prometheus/prometheus.module';
import { ClickHouseModule } from './clickhouse/clickhouse.module';
import { ClickHouseController } from './clickhouse/clickhouse.controller';
@Module({
imports: [
@ -27,6 +29,8 @@ import { PrometheusModule } from './prometheus.module';
AuthModule,
PrometheusModule,
MenuModule,
ClickHouseModule,
],
controllers: [ClickHouseController],
})
export class AppModule {}
export class AppModule { }

View File

@ -0,0 +1,62 @@
import { Controller, Get, Post } from '@nestjs/common';
import { ClickHouseService } from './clickhouse.service';
import { ApiOperation, ApiResponse, ApiTags } from '@nestjs/swagger';
import { HttpService } from '@nestjs/axios';
import { firstValueFrom } from 'rxjs';
import { ConfigService } from '@nestjs/config';
@ApiTags('Clickhouse')
@Controller('clickhouse')
export class ClickHouseController {
constructor(
private readonly clickhouseService: ClickHouseService,
private readonly httpService: HttpService,
private readonly configService: ConfigService,
) { }
@Get()
@ApiOperation({ summary: 'Get metrics from ClickHouse' })
@ApiResponse({
status: 200,
description: 'Metrics data',
schema: {
type: 'array',
items: {
type: 'object',
properties: {
description: { type: 'string' },
device: { type: 'number' },
id: { type: 'string' },
name: { type: 'string' },
source: { type: 'string' },
status: { type: 'number' },
timestamp: { type: 'number' },
value: { type: 'string' },
},
},
},
})
async getClckhouse() {
return this.clickhouseService.getClckhouse();
}
@Post('send-to-ai')
@ApiOperation({ summary: 'Send metrics to AI service' })
@ApiResponse({
status: 200,
description: 'AI service response',
})
async sendToAI() {
const metrics = await this.clickhouseService.getClckhouse();
const aiServiceUrl = this.configService.get('AI_SERVICE_URL/api/metrics/rest');
try {
const response = await firstValueFrom(
this.httpService.post(aiServiceUrl, metrics)
);
return response.data;
} catch (error) {
throw new Error(`Failed to send data to AI: ${error.message}`);
}
}
}

View File

@ -0,0 +1,25 @@
import { Module, Global } from '@nestjs/common';
import { createClient, ClickHouseClient } from '@clickhouse/client';
import { ClickHouseService } from './clickhouse.service';
import { HttpModule } from '@nestjs/axios';
@Global()
@Module({
imports: [HttpModule],
providers: [
{
provide: 'CLICKHOUSE_CLIENT',
useFactory: (): ClickHouseClient => {
return createClient({
host: process.env.CLICKHOUSE_HOST || 'http://localhost:8123',
username: process.env.CLICKHOUSE_USER || 'default',
password: process.env.CLICKHOUSE_PASSWORD || '',
database: process.env.CLICKHOUSE_DB || 'default',
});
},
},
ClickHouseService,
],
exports: ['CLICKHOUSE_CLIENT', ClickHouseService],
})
export class ClickHouseModule { }

View File

@ -0,0 +1,72 @@
import { Injectable, Inject } from '@nestjs/common';
import { ClickHouseClient } from '@clickhouse/client';
interface ClickHouseRow {
EventDataTime: string;
ParameterBody: string;
CreateDataTime: string;
}
interface MetricData {
id: string;
name: string;
type: string;
addr?: string;
value: number | string | null;
description: string;
status: number;
device: number;
source: string;
}
interface ParameterBody {
service_name: string;
metrics: MetricData[];
}
@Injectable()
export class ClickHouseService {
constructor(
@Inject('CLICKHOUSE_CLIENT')
private readonly clickhouseClient: ClickHouseClient,
) { }
async getClckhouse() {
const query = `
SELECT
EventDataTime,
ParameterBody,
CreateDataTime
FROM zvks.complex_parameters
ORDER BY EventDataTime DESC
LIMIT 100
`;
const result = await this.clickhouseClient.query({
query,
format: 'JSONEachRow',
});
const rows = await result.json<ClickHouseRow>();
// Парсинг данных
return rows.flatMap((row: ClickHouseRow) => {
try {
const parameterBody: ParameterBody = JSON.parse(row.ParameterBody);
return parameterBody.metrics.map((metric: MetricData) => ({
id: metric.id,
name: metric.name,
value: metric.value !== null ? metric.value.toString() : 'null',
description: metric.description,
status: metric.status,
device: metric.device,
source: metric.source,
timestamp: new Date(row.EventDataTime).getTime(),
}));
} catch (e) {
console.error('Error parsing metric:', e);
return [];
}
});
}
}

View File

@ -42,7 +42,7 @@ async function bootstrap() {
});
// Настройка CORS
app.enableCors({
app.enableCors({//ПОСТАВИТЬ ПРОКСИ, ЧТОБЫ КОРС НЕ РУГАЛСЯ, ИЗМЕНЕНИЕ ПОЛИТИКИ СЕТЕВЫХ ПАКЕТОВ. ПИШУ IP СВОЙ, А ПОРТ ПРОКСИ. REVERSE PROXY.
origin: [process.env.FRONTEND_URL, "http://dev.msf.enode"],
credentials: true,
methods: 'GET,HEAD,PUT,PATCH,POST,DELETE,OPTIONS',

View File

@ -6,26 +6,33 @@ import { MenuItem } from './menu.interface';
export class MenuController {
constructor(private readonly menuService: MenuService) { }
@Get('full')
async getFullMenu(@Headers('if-modified-since') ifModifiedSince?: string) {
try {
const result = await this.menuService.getFullMenuWithCache(ifModifiedSince);
@Get('full')
async getFullMenu(@Headers('if-modified-since') ifModifiedSince?: string) {
console.log('GET /menu/full requested');
try {
const result = await this.menuService.getFullMenuWithCache(ifModifiedSince);
if (!result.fresh && ifModifiedSince) {
throw new HttpException('Not Modified', HttpStatus.NOT_MODIFIED);
}
if (!result.fresh && ifModifiedSince) {
throw new HttpException('Not Modified', HttpStatus.NOT_MODIFIED);
}
return result.menu;
} catch (error) {
if (error.status === HttpStatus.NOT_MODIFIED) {
throw error;
return result.menu;
} catch (error) {
if (error.status === HttpStatus.NOT_MODIFIED) {
throw error;
}
throw new HttpException(
error.message || 'Failed to load menu',
HttpStatus.INTERNAL_SERVER_ERROR
);
}
throw new HttpException(
error.message || 'Failed to load menu',
HttpStatus.INTERNAL_SERVER_ERROR
);
}
}
/*
@Get('full')
async getFullMenu() {
console.log('Simplified endpoint called');
return { test: 'OK' }; // Простейший ответ
} */
@Get('check-updates')
async checkUpdates(@Headers('if-modified-since') ifModifiedSince: string) {

View File

@ -2,7 +2,7 @@ import { Module } from '@nestjs/common';
import { MenuController } from './menu.controller';
import { HttpModule } from '@nestjs/axios';
import { MenuService } from './menu.service';
import { PrometheusModule } from '../prometheus.module';
import { PrometheusModule } from '../prometheus/prometheus.module';
import { RangeService } from './range.service';
import { RangeController } from './range.controller';

View File

@ -1,5 +1,5 @@
import { Injectable, HttpException, HttpStatus } from '@nestjs/common';
import { PrometheusService } from '../prometheus.service';
import { PrometheusService } from '../prometheus/prometheus.service';
import { MenuItem } from './menu.interface';
import * as fs from 'fs/promises';
import * as path from 'path';

View File

View File

@ -26,6 +26,7 @@ export class MetricsGateway implements OnGatewayInit, OnGatewayConnection, OnGat
stopUpdates: () => void;
clients: Set<string>;
}>();
private lastSentData = new Map<string, any>(); // Кэш последних отправленных данных
constructor(private readonly prometheusService: PrometheusService) { }
@ -48,6 +49,7 @@ export class MetricsGateway implements OnGatewayInit, OnGatewayConnection, OnGat
if (subscription.clients.size === 0) {
subscription.stopUpdates();
this.metricSubscriptions.delete(metric);
this.lastSentData.delete(metric);
}
}
}
@ -59,6 +61,7 @@ export class MetricsGateway implements OnGatewayInit, OnGatewayConnection, OnGat
if (subscription.clients.size === 0) {
subscription.stopUpdates();
this.metricSubscriptions.delete(metric);
this.lastSentData.delete(metric);
}
}
}
@ -90,17 +93,28 @@ export class MetricsGateway implements OnGatewayInit, OnGatewayConnection, OnGat
}
try {
const subscriptionKey = this.getSubscriptionKey(metric, filters);
// Отправляем текущие данные сразу при запросе
const initialData = await this.prometheusService.fetchMetricsWithFilters(metric, filters);
client.emit('metrics-data', { metric, data: initialData, requestId });
this.lastSentData.set(subscriptionKey, initialData);
const stopUpdates = await this.sendPeriodicUpdates(
metric,
step || 5000,
(data) => {
client.emit('metrics-data', { metric, data, requestId });
const lastData = this.lastSentData.get(subscriptionKey);
if (!this.isDataEqual(lastData, data)) {
client.emit('metrics-data', { metric, data, requestId });
this.lastSentData.set(subscriptionKey, data);
}
},
filters
);
const cleanup = () => {
stopUpdates();
this.lastSentData.delete(subscriptionKey);
client.off('disconnect', cleanup);
client.off('unsubscribe-metric', cleanup);
};
@ -120,7 +134,21 @@ export class MetricsGateway implements OnGatewayInit, OnGatewayConnection, OnGat
const filterKeys = Object.keys(filters).sort();
const filterString = filterKeys.map(k => `${k}=${encodeURIComponent(filters[k])}`).join('&');
return `${metric}${filterString ? `?${filterString}` : ''}`;
}
}
// Сравниваем данные, чтобы избежать лишних отправок
private isDataEqual(oldData: any[], newData: any[]): boolean {
if (!oldData || !newData || oldData.length !== newData.length) return false;
return oldData.every((oldItem, index) => {
const newItem = newData[index];
return (
oldItem.value === newItem.value &&
oldItem.status === newItem.status &&
oldItem.timestamp === newItem.timestamp
);
});
}
@SubscribeMessage('subscribe-metric')
async handleSubscribeMetric(
@ -131,18 +159,35 @@ export class MetricsGateway implements OnGatewayInit, OnGatewayConnection, OnGat
filters?: Record<string, string>;
}
) {
const { metric, interval = 5000, filters = {} } = payload;
const { metric, interval = 60000, filters = {} } = payload; // По умолчанию 60 секунд
const subscriptionKey = this.getSubscriptionKey(metric, filters);
if (!this.metricSubscriptions.has(subscriptionKey)) {
// Отправляем текущие данные сразу при подписке
try {
const initialData = await this.prometheusService.fetchMetricsWithFilters(metric, filters);
client.emit('metrics-data', {
metric: subscriptionKey,
data: initialData
});
this.lastSentData.set(subscriptionKey, initialData);
} catch (error) {
this.logger.error(`Error fetching initial data for ${metric}:`, error.message);
}
const stopUpdates = await this.sendPeriodicUpdates(
metric,
interval,
(data) => {
this.server.emit('metrics-data', {
metric: subscriptionKey,
data
});
// Отправляем только если данные изменились
const lastData = this.lastSentData.get(subscriptionKey);
if (!this.isDataEqual(lastData, data)) {
this.server.emit('metrics-data', {
metric: subscriptionKey,
data
});
this.lastSentData.set(subscriptionKey, data);
}
},
filters
);
@ -153,6 +198,14 @@ export class MetricsGateway implements OnGatewayInit, OnGatewayConnection, OnGat
});
} else {
this.metricSubscriptions.get(subscriptionKey)?.clients.add(client.id);
// Отправляем кэшированные данные новому клиенту
const cachedData = this.lastSentData.get(subscriptionKey);
if (cachedData) {
client.emit('metrics-data', {
metric: subscriptionKey,
data: cachedData
});
}
}
const unsubscribe = () => {
@ -162,6 +215,7 @@ export class MetricsGateway implements OnGatewayInit, OnGatewayConnection, OnGat
if (subscription.clients.size === 0) {
subscription.stopUpdates();
this.metricSubscriptions.delete(subscriptionKey);
this.lastSentData.delete(subscriptionKey);
}
}
};
@ -176,6 +230,11 @@ export class MetricsGateway implements OnGatewayInit, OnGatewayConnection, OnGat
callback: (data: any) => void,
filters: Record<string, string> = {}
) {
// Добавляем небольшую случайную задержку, чтобы избежать пиковой нагрузки
const initialDelay = Math.floor(Math.random() * 5000);
await new Promise(resolve => setTimeout(resolve, initialDelay));
const timer = setInterval(async () => {
try {
const data = await this.prometheusService.fetchMetricsWithFilters(metric, filters);
@ -187,6 +246,7 @@ export class MetricsGateway implements OnGatewayInit, OnGatewayConnection, OnGat
return () => {
clearInterval(timer);
this.lastSentData.delete(this.getSubscriptionKey(metric, filters));
this.logger.log(`Stopped updates for ${metric}`);
};
}

View File

@ -3,11 +3,13 @@ import { HttpService } from '@nestjs/axios';
import { ConfigService } from '@nestjs/config';
import { lastValueFrom } from 'rxjs';
import { PrometheusMetric } from './prometheus-metric.interface';
import { MenuItem } from './menu/menu.interface';
import { MenuItem } from '../menu/menu.interface';
@Injectable()
export class PrometheusService {
private readonly prometheusUrl: string;
private metricCache = new Map<string, { data: any; timestamp: number }>();
private metadataCache = new Map<string, { type: string | null; description: string | undefined; timestamp: number }>();
constructor(
private readonly httpService: HttpService,
@ -18,6 +20,13 @@ export class PrometheusService {
}
async fetchMetricType(metric: string): Promise<string | null> {
const cacheKey = `metadata-type-${metric}`;
const cacheEntry = this.metadataCache.get(cacheKey);
if (cacheEntry && Date.now() - cacheEntry.timestamp < 30000) {
return cacheEntry.type;
}
try {
const response = await lastValueFrom(
this.httpService.get(`${this.prometheusUrl}/metadata`, {
@ -25,14 +34,29 @@ export class PrometheusService {
})
);
const metadata = response.data.data[metric];
return metadata?.length ? metadata[0].type : null;
const result = metadata?.length ? metadata[0].type : null;
this.metadataCache.set(cacheKey, {
type: result,
description: cacheEntry?.description,
timestamp: Date.now()
});
return result;
} catch (error) {
console.error(`Ошибка при получении типа метрики ${metric}:`, error);
return null;
return cacheEntry?.type || null;
}
}
async fetchMetricDescription(metric: string): Promise<string | undefined> {
const cacheKey = `metadata-description-${metric}`;
const cacheEntry = this.metadataCache.get(cacheKey);
if (cacheEntry && Date.now() - cacheEntry.timestamp < 30000) {
return cacheEntry.description;
}
try {
const response = await lastValueFrom(
this.httpService.get(`${this.prometheusUrl}/metadata`, {
@ -40,14 +64,29 @@ export class PrometheusService {
})
);
const metadata = response.data.data[metric];
return metadata?.length ? metadata[0].help : undefined;
const result = metadata?.length ? metadata[0].help : undefined;
this.metadataCache.set(cacheKey, {
type: cacheEntry?.type ?? null,
description: result,
timestamp: Date.now()
});
return result;
} catch (error) {
console.error(`Ошибка при получении описания метрики ${metric}:`, error);
return undefined;
return cacheEntry?.description;
}
}
async fetchMetrics(metric: string): Promise<PrometheusMetric[]> {
const cacheKey = `${metric}:{}`;
const cacheEntry = this.metricCache.get(cacheKey);
if (cacheEntry && Date.now() - cacheEntry.timestamp < 5000) {
return cacheEntry.data;
}
try {
const response = await lastValueFrom(
this.httpService.get(`${this.prometheusUrl}/query`, {
@ -58,7 +97,7 @@ export class PrometheusService {
const metricType = await this.fetchMetricType(metric);
const metricDescription = await this.fetchMetricDescription(metric);
return response.data.data.result.map((entry): PrometheusMetric => ({
const result = response.data.data.result.map((entry): PrometheusMetric => ({
__name__: entry.metric.__name__ || metric,
device: entry.metric.device,
instance: entry.metric.instance,
@ -71,13 +110,24 @@ export class PrometheusService {
description: metricDescription,
...entry.metric
}));
this.metricCache.set(cacheKey, { data: result, timestamp: Date.now() });
return result;
} catch (error) {
console.error(`Error fetching metrics for ${metric}:`, error);
if (cacheEntry) return cacheEntry.data;
throw error;
}
}
async fetchMetricsWithFilters(metric: string, filters: Record<string, string>): Promise<PrometheusMetric[]> {
const cacheKey = `${metric}:${JSON.stringify(filters)}`;
const cacheEntry = this.metricCache.get(cacheKey);
if (cacheEntry && Date.now() - cacheEntry.timestamp < 5000) {
return cacheEntry.data;
}
try {
const query = this.buildFilteredQuery(metric, filters);
const response = await lastValueFrom(
@ -89,7 +139,7 @@ export class PrometheusService {
const metricType = await this.fetchMetricType(metric);
const metricDescription = await this.fetchMetricDescription(metric);
return response.data.data.result.map((entry): PrometheusMetric => ({
const result = response.data.data.result.map((entry): PrometheusMetric => ({
__name__: entry.metric.__name__ || metric,
device: entry.metric.device,
instance: entry.metric.instance,
@ -102,30 +152,38 @@ export class PrometheusService {
description: metricDescription,
...entry.metric
}));
this.metricCache.set(cacheKey, { data: result, timestamp: Date.now() });
return result;
} catch (error) {
console.error(`Error fetching metrics with filters for ${metric}:`, error);
if (cacheEntry) return cacheEntry.data;
throw error;
}
}
private buildFilteredQuery(metric: string, filters: Record<string, string>): string {
const filterParts = Object.entries(filters)
.filter(([_, value]) => value !== undefined && value !== null && value !== "")
.map(([key, value]) => {
// Убираем автоматическое добавление "module$" для source_id
return `${key}="${value}"`;
});
const filterParts = Object.entries(filters)
.filter(([_, value]) => value !== undefined && value !== null && value !== "")
.map(([key, value]) => {
return `${key}="${value}"`;
});
return filterParts.length > 0
? `${metric}{${filterParts.join(',')}}`
: metric;
}
return filterParts.length > 0
? `${metric}{${filterParts.join(',')}}`
: metric;
}
async fetchMetricsRange(metric: string, start: number, end: number, step: number, filters: Record<string, string> = {}): Promise<PrometheusMetric[]> {
// Рассчитываем оптимальный шаг, если не указан
const duration = end - start;
const optimalStep = Math.max(Math.floor(duration / 1000), 15); // Минимум 15 секунд
const query = this.buildFilteredQuery(metric, {
...filters,
instance: '192.168.2.34:9050'
});
try {
const response = await lastValueFrom(
this.httpService.get(`${this.prometheusUrl}/query_range`, {
@ -133,7 +191,7 @@ export class PrometheusService {
query,
start,
end,
step: step.toString()
step: optimalStep.toString()
},
})
);
@ -217,21 +275,40 @@ export class PrometheusService {
}
async fetchAllMetrics(): Promise<string[]> {
const response = await lastValueFrom(
this.httpService.get(`${this.prometheusUrl}/label/__name__/values`)
);
return response.data.data;
try {
const response = await lastValueFrom(
this.httpService.get(`${this.prometheusUrl}/label/__name__/values`)
);
return response.data.data;
} catch (error) {
console.error('Error fetching all metrics:', error);
return [];
}
}
async fetchAllMetricsWithValues(): Promise<any[]> {
const metricNames = await this.fetchAllMetrics();
const zvksMetrics = metricNames.filter(metric => metric.startsWith('zvks'));
const metricNames = await this.fetchAllMetrics();
const zvksMetrics = metricNames.filter(metric =>
metric.startsWith('zvks') ||
metric.includes('server_li') ||
metric.includes('application_li')
);
const promises = zvksMetrics.map(async (metric) => {
const data = await this.fetchMetrics(metric);
return { metric, data };
});
const promises = zvksMetrics.map(async (metric) => {
try {
const data = await this.fetchMetrics(metric);
return { metric, data };
} catch (error) {
console.error(`Error fetching data for metric ${metric}:`, error);
return { metric, data: [] };
}
});
return Promise.all(promises);
}
return Promise.all(promises);
}
clearCache(): void {
this.metricCache.clear();
this.metadataCache.clear();
}
}