swagger #33
21
.env
21
.env
|
|
@ -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
|
||||
|
|
@ -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",
|
||||
|
|
|
|||
|
|
@ -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 { }
|
||||
|
|
|
|||
|
|
@ -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}`);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
|
@ -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 { }
|
||||
|
|
@ -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 [];
|
||||
}
|
||||
});
|
||||
}
|
||||
}
|
||||
|
|
@ -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',
|
||||
|
|
|
|||
|
|
@ -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) {
|
||||
|
|
|
|||
|
|
@ -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';
|
||||
|
||||
|
|
|
|||
|
|
@ -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';
|
||||
|
|
|
|||
|
|
@ -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}`);
|
||||
};
|
||||
}
|
||||
|
|
@ -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();
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue