Revert "SP-1778-be-fix-time-out-connections-in-the-db"

This commit is contained in:
faljawhary
2025-06-23 06:58:57 -06:00
committed by GitHub
parent 52cb69cc84
commit 75d03366c2
4 changed files with 38 additions and 97 deletions

View File

@ -24,7 +24,6 @@ import { PowerClampEnergyEnum } from '@app/common/constants/power.clamp.enargy.e
import { PresenceSensorEnum } from '@app/common/constants/presence.sensor.enum'; import { PresenceSensorEnum } from '@app/common/constants/presence.sensor.enum';
import { OccupancyService } from '@app/common/helper/services/occupancy.service'; import { OccupancyService } from '@app/common/helper/services/occupancy.service';
import { AqiDataService } from '@app/common/helper/services/aqi.data.service'; import { AqiDataService } from '@app/common/helper/services/aqi.data.service';
import { DataSource, QueryRunner } from 'typeorm';
@Injectable() @Injectable()
export class DeviceStatusFirebaseService { export class DeviceStatusFirebaseService {
private tuya: TuyaContext; private tuya: TuyaContext;
@ -36,7 +35,6 @@ export class DeviceStatusFirebaseService {
private readonly occupancyService: OccupancyService, private readonly occupancyService: OccupancyService,
private readonly aqiDataService: AqiDataService, private readonly aqiDataService: AqiDataService,
private deviceStatusLogRepository: DeviceStatusLogRepository, private deviceStatusLogRepository: DeviceStatusLogRepository,
private readonly dataSource: DataSource,
) { ) {
const accessKey = this.configService.get<string>('auth-config.ACCESS_KEY'); const accessKey = this.configService.get<string>('auth-config.ACCESS_KEY');
const secretKey = this.configService.get<string>('auth-config.SECRET_KEY'); const secretKey = this.configService.get<string>('auth-config.SECRET_KEY');
@ -81,46 +79,28 @@ export class DeviceStatusFirebaseService {
async addDeviceStatusToFirebase( async addDeviceStatusToFirebase(
addDeviceStatusDto: AddDeviceStatusDto, addDeviceStatusDto: AddDeviceStatusDto,
): Promise<AddDeviceStatusDto | null> { ): Promise<AddDeviceStatusDto | null> {
const queryRunner = this.dataSource.createQueryRunner();
await queryRunner.connect();
await queryRunner.startTransaction();
try { try {
const device = await this.getDeviceByDeviceTuyaUuid( const device = await this.getDeviceByDeviceTuyaUuid(
addDeviceStatusDto.deviceTuyaUuid, addDeviceStatusDto.deviceTuyaUuid,
queryRunner,
); );
if (device?.uuid) { if (device?.uuid) {
const result = await this.createDeviceStatusFirebase( return await this.createDeviceStatusFirebase({
{ deviceUuid: device.uuid,
deviceUuid: device.uuid, ...addDeviceStatusDto,
...addDeviceStatusDto, productType: device.productDevice.prodType,
productType: device.productDevice.prodType, });
},
queryRunner,
);
await queryRunner.commitTransaction();
return result;
} }
// Return null if device not found or no UUID // Return null if device not found or no UUID
await queryRunner.rollbackTransaction();
return null; return null;
} catch (error) { } catch (error) {
await queryRunner.rollbackTransaction(); // Handle the error silently, perhaps log it internally or ignore it
return null; return null;
} finally {
await queryRunner.release();
} }
} }
async getDeviceByDeviceTuyaUuid(
deviceTuyaUuid: string,
queryRunner?: QueryRunner,
) {
const repo = queryRunner
? queryRunner.manager.getRepository(this.deviceRepository.target)
: this.deviceRepository;
return await repo.findOne({ async getDeviceByDeviceTuyaUuid(deviceTuyaUuid: string) {
return await this.deviceRepository.findOne({
where: { where: {
deviceTuyaUuid, deviceTuyaUuid,
isActive: true, isActive: true,
@ -128,7 +108,6 @@ export class DeviceStatusFirebaseService {
relations: ['productDevice'], relations: ['productDevice'],
}); });
} }
async getDevicesInstructionStatus(deviceUuid: string) { async getDevicesInstructionStatus(deviceUuid: string) {
try { try {
const deviceDetails = await this.getDeviceByDeviceUuid(deviceUuid); const deviceDetails = await this.getDeviceByDeviceUuid(deviceUuid);
@ -174,14 +153,9 @@ export class DeviceStatusFirebaseService {
} }
async getDeviceByDeviceUuid( async getDeviceByDeviceUuid(
deviceUuid: string, deviceUuid: string,
withProductDevice = true, withProductDevice: boolean = true,
queryRunner?: QueryRunner,
) { ) {
const repo = queryRunner return await this.deviceRepository.findOne({
? queryRunner.manager.getRepository(this.deviceRepository.target)
: this.deviceRepository;
return await repo.findOne({
where: { where: {
uuid: deviceUuid, uuid: deviceUuid,
isActive: true, isActive: true,
@ -189,20 +163,21 @@ export class DeviceStatusFirebaseService {
...(withProductDevice && { relations: ['productDevice'] }), ...(withProductDevice && { relations: ['productDevice'] }),
}); });
} }
async createDeviceStatusFirebase( async createDeviceStatusFirebase(
addDeviceStatusDto: AddDeviceStatusDto, addDeviceStatusDto: AddDeviceStatusDto,
queryRunner?: QueryRunner,
): Promise<any> { ): Promise<any> {
const dataRef = ref( const dataRef = ref(
this.firebaseDb, this.firebaseDb,
`device-status/${addDeviceStatusDto.deviceUuid}`, `device-status/${addDeviceStatusDto.deviceUuid}`,
); );
// Step 1: Update Firebase Realtime Database // Use a transaction to handle concurrent updates
await runTransaction(dataRef, (existingData) => { await runTransaction(dataRef, (existingData) => {
if (!existingData) existingData = {}; if (!existingData) {
existingData = {};
}
// Assign default values if fields are not present
if (!existingData.deviceTuyaUuid) { if (!existingData.deviceTuyaUuid) {
existingData.deviceTuyaUuid = addDeviceStatusDto.deviceTuyaUuid; existingData.deviceTuyaUuid = addDeviceStatusDto.deviceTuyaUuid;
} }
@ -216,15 +191,18 @@ export class DeviceStatusFirebaseService {
existingData.status = []; existingData.status = [];
} }
// Merge incoming status with existing status // Create a map to track existing status codes
const statusMap = new Map( const statusMap = new Map(
existingData.status.map((item) => [item.code, item.value]), existingData.status.map((item) => [item.code, item.value]),
); );
// Update or add status codes
for (const statusItem of addDeviceStatusDto.status) { for (const statusItem of addDeviceStatusDto.status) {
statusMap.set(statusItem.code, statusItem.value); statusMap.set(statusItem.code, statusItem.value);
} }
// Convert the map back to an array format
existingData.status = Array.from(statusMap, ([code, value]) => ({ existingData.status = Array.from(statusMap, ([code, value]) => ({
code, code,
value, value,
@ -233,9 +211,9 @@ export class DeviceStatusFirebaseService {
return existingData; return existingData;
}); });
// Step 2: Save device status log entries // Save logs to your repository
const newLogs = addDeviceStatusDto.log.properties.map((property) => const newLogs = addDeviceStatusDto.log.properties.map((property) => {
this.deviceStatusLogRepository.create({ return this.deviceStatusLogRepository.create({
deviceId: addDeviceStatusDto.deviceUuid, deviceId: addDeviceStatusDto.deviceUuid,
deviceTuyaId: addDeviceStatusDto.deviceTuyaUuid, deviceTuyaId: addDeviceStatusDto.deviceTuyaUuid,
productId: addDeviceStatusDto.log.productId, productId: addDeviceStatusDto.log.productId,
@ -244,19 +222,10 @@ export class DeviceStatusFirebaseService {
value: property.value, value: property.value,
eventId: addDeviceStatusDto.log.dataId, eventId: addDeviceStatusDto.log.dataId,
eventTime: new Date(property.time).toISOString(), eventTime: new Date(property.time).toISOString(),
}), });
); });
await this.deviceStatusLogRepository.save(newLogs);
if (queryRunner) {
const repo = queryRunner.manager.getRepository(
this.deviceStatusLogRepository.target,
);
await repo.save(newLogs);
} else {
await this.deviceStatusLogRepository.save(newLogs);
}
// Step 3: Trigger additional data services
if (addDeviceStatusDto.productType === ProductType.PC) { if (addDeviceStatusDto.productType === ProductType.PC) {
const energyCodes = new Set([ const energyCodes = new Set([
PowerClampEnergyEnum.ENERGY_CONSUMED, PowerClampEnergyEnum.ENERGY_CONSUMED,
@ -300,8 +269,7 @@ export class DeviceStatusFirebaseService {
addDeviceStatusDto.deviceUuid, addDeviceStatusDto.deviceUuid,
); );
} }
// Return the updated data
// Step 4: Return updated Firebase status
const snapshot: DataSnapshot = await get(dataRef); const snapshot: DataSnapshot = await get(dataRef);
return snapshot.val(); return snapshot.val();
} }

View File

@ -36,18 +36,9 @@ export class AqiDataService {
procedureFileName: string, procedureFileName: string,
params: (string | number | null)[], params: (string | number | null)[],
): Promise<void> { ): Promise<void> {
const queryRunner = this.dataSource.createQueryRunner(); const query = this.loadQuery(procedureFolderName, procedureFileName);
await queryRunner.connect(); await this.dataSource.query(query, params);
try { console.log(`Procedure ${procedureFileName} executed successfully.`);
const query = this.loadQuery(procedureFolderName, procedureFileName);
await queryRunner.query(query, params);
console.log(`Procedure ${procedureFileName} executed successfully.`);
} catch (err) {
console.error(`Failed to execute procedure ${procedureFileName}:`, err);
throw err;
} finally {
await queryRunner.release();
}
} }
private loadQuery(folderName: string, fileName: string): string { private loadQuery(folderName: string, fileName: string): string {

View File

@ -57,18 +57,9 @@ export class OccupancyService {
procedureFileName: string, procedureFileName: string,
params: (string | number | null)[], params: (string | number | null)[],
): Promise<void> { ): Promise<void> {
const queryRunner = this.dataSource.createQueryRunner(); const query = this.loadQuery(procedureFolderName, procedureFileName);
await queryRunner.connect(); await this.dataSource.query(query, params);
try { console.log(`Procedure ${procedureFileName} executed successfully.`);
const query = this.loadQuery(procedureFolderName, procedureFileName);
await queryRunner.query(query, params);
console.log(`Procedure ${procedureFileName} executed successfully.`);
} catch (err) {
console.error(`Failed to execute procedure ${procedureFileName}:`, err);
throw err;
} finally {
await queryRunner.release();
}
} }
private loadQuery(folderName: string, fileName: string): string { private loadQuery(folderName: string, fileName: string): string {

View File

@ -46,21 +46,12 @@ export class PowerClampService {
procedureFileName: string, procedureFileName: string,
params: (string | number | null)[], params: (string | number | null)[],
): Promise<void> { ): Promise<void> {
const queryRunner = this.dataSource.createQueryRunner(); const query = this.loadQuery(
await queryRunner.connect(); 'fact_device_energy_consumed',
try { procedureFileName,
const query = this.loadQuery( );
'fact_device_energy_consumed', await this.dataSource.query(query, params);
procedureFileName, console.log(`Procedure ${procedureFileName} executed successfully.`);
);
await queryRunner.query(query, params);
console.log(`Procedure ${procedureFileName} executed successfully.`);
} catch (err) {
console.error(`Failed to execute procedure ${procedureFileName}:`, err);
throw err;
} finally {
await queryRunner.release();
}
} }
private loadQuery(folderName: string, fileName: string): string { private loadQuery(folderName: string, fileName: string): string {