Mirrors/unleash.unleash
1
0
Fork 0
mirror of https://github.com/Unleash/unleash.git synced 2025-04-15 01:16:22 +02:00
Code Issues Projects Releases Wiki Activity

chore(1-3293): verify that data is end of day/start of day

Browse Source
This commit is contained in:
Thomas Heartman 2025-01-24 14:48:14 +01:00
parent 6079114217
commit 3ad10be1b2
No known key found for this signature in database
GPG Key ID: BD1F880DAED1EE78
1 changed files with 22 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
src/lib/features/traffic-data-usage/traffic-data-usage-store.ts
View File

@ -1,4 +1,10 @@
import { endOfMonth, startOfMonth, subMonths } from 'date-fns';
import {
endOfDay,
endOfMonth,
startOfDay,
startOfMonth,
subMonths,
} from 'date-fns';
import type { Db } from '../../db/db';
import type { Logger, LogProvider } from '../../logger';
import type {
@ -95,23 +101,12 @@ export class TrafficDataUsageStore implements ITrafficDataUsageStore {
to: Date,
): Promise<IStatTrafficUsage[]> {
const rows = await this.db<IStatTrafficUsage>(TABLE)
.where('day', '>=', from)
.andWhere('day', '<=', to);
.where('day', '>=', startOfDay(from))
.andWhere('day', '<=', endOfDay(to));
return rows.map(mapRow);
}
// @deprecated: remove with flag `dataUsageMultiMonthView`
async getTrafficDataUsageForPeriod(
period: string,
): Promise<IStatTrafficUsage[]> {
const month = new Date(period);
return this.getDailyTrafficDataUsageForPeriod(
startOfMonth(month),
endOfMonth(month),
);
}
async getMonthlyTrafficDataUsageForPeriod(
from: Date,
to: Date,
@ -123,8 +118,8 @@ export class TrafficDataUsageStore implements ITrafficDataUsageStore {
this.db.raw(`to_char(day, 'YYYY-MM') AS month`),
this.db.raw(`SUM(count) AS count`),
)
.where('day', '>=', from)
.andWhere('day', '<=', to)
.where('day', '>=', startOfDay(from))
.andWhere('day', '<=', endOfDay(to))
.groupBy([
'traffic_group',
this.db.raw(`to_char(day, 'YYYY-MM')`),
@ -144,6 +139,17 @@ export class TrafficDataUsageStore implements ITrafficDataUsageStore {
);
}
async getTrafficDataUsageForPeriod(
period: string,
): Promise<IStatTrafficUsage[]> {
const month = new Date(period);
return this.getDailyTrafficDataUsageForPeriod(
startOfMonth(month),
endOfMonth(month),
);
}
// @deprecated: remove with flag `dataUsageMultiMonthView`
async getTrafficDataForMonthRange(
monthsBack: number,
): Promise<IStatMonthlyTrafficUsage[]> {