highperfocused/public-pool
1
0
Fork 0
mirror of https://github.com/benjamin-wilson/public-pool.git synced 2025-04-04 01:48:21 +02:00
Code Issues Packages Projects Releases Wiki Activity

wrap save in a transaction

Browse Source
This commit is contained in:
Ben Wilson 2023-12-10 17:23:08 -05:00
parent 06acad6770
commit ee5fea1df5
1 changed files with 27 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
src/ORM/client-statistics/client-statistics.service.ts
View File

@ -1,6 +1,6 @@
import { Injectable } from '@nestjs/common';
import { InjectRepository } from '@nestjs/typeorm';
import { Repository } from 'typeorm';
import { InjectDataSource, InjectRepository } from '@nestjs/typeorm';
import { DataSource, EntityManager, Repository } from 'typeorm';
import { ClientStatisticsEntity } from './client-statistics.entity';
@ -10,7 +10,8 @@ export class ClientStatisticsService {
constructor(
@InjectDataSource()
private dataSource: DataSource,
@InjectRepository(ClientStatisticsEntity)
private clientStatisticsRepository: Repository<ClientStatisticsEntity>,
) {
@ -18,18 +19,30 @@ export class ClientStatisticsService {
}
public async save(clientStatistic: Partial<ClientStatisticsEntity>) {
const res1 = await this.clientStatisticsRepository.createQueryBuilder()
.update(ClientStatisticsEntity)
.set({
shares: () => `"shares" + ${clientStatistic.shares}`, // Use the actual value of shares here
acceptedCount: () => `"acceptedCount" + 1`
})
.where('address = :address AND clientName = :clientName AND sessionId = :sessionId AND time = :time', { address: clientStatistic.address, clientName: clientStatistic.clientName, sessionId: clientStatistic.sessionId, time: clientStatistic.time })
.execute();
// Use transaction to ensure atomicity
await this.dataSource.transaction(async (entityManager: EntityManager) => {
// Attempt to update the existing record
const updateResult = await entityManager
.createQueryBuilder()
.update(ClientStatisticsEntity)
.set({
shares: () => `"shares" + ${clientStatistic.shares}`,
acceptedCount: () => `"acceptedCount" + 1`
})
.where('address = :address AND clientName = :clientName AND sessionId = :sessionId AND time = :time', {
address: clientStatistic.address,
clientName: clientStatistic.clientName,
sessionId: clientStatistic.sessionId,
time: clientStatistic.time
})
.execute();
if (res1.affected == 0) {
await this.clientStatisticsRepository.insert(clientStatistic);
}
// Check if the update affected any rows
if (updateResult.affected === 0) {
// If no rows were updated, insert a new record
await entityManager.insert(ClientStatisticsEntity, clientStatistic);
}
});
}
public async deleteOldStatistics() {