@lssm/lib.metering 0.0.0-canary-20251217083314 → 1.41.0

package/dist/aggregation/index.js

@@ -1,271 +1 @@
-//#region src/aggregation/index.ts
-/**
-* Get the start of a period for a given date.
-*/
-function getPeriodStart(date, periodType) {
-	const d = new Date(date);
-	switch (periodType) {
-		case "HOURLY":
-			d.setMinutes(0, 0, 0);
-			return d;
-		case "DAILY":
-			d.setHours(0, 0, 0, 0);
-			return d;
-		case "WEEKLY":
-			d.setHours(0, 0, 0, 0);
-			const day = d.getDay();
-			d.setDate(d.getDate() - day);
-			return d;
-		case "MONTHLY":
-			d.setHours(0, 0, 0, 0);
-			d.setDate(1);
-			return d;
-		case "YEARLY":
-			d.setHours(0, 0, 0, 0);
-			d.setMonth(0, 1);
-			return d;
-	}
-}
-/**
-* Get the end of a period for a given date.
-*/
-function getPeriodEnd(date, periodType) {
-	const start = getPeriodStart(date, periodType);
-	switch (periodType) {
-		case "HOURLY": return new Date(start.getTime() + 3600 * 1e3);
-		case "DAILY": return new Date(start.getTime() + 1440 * 60 * 1e3);
-		case "WEEKLY": return new Date(start.getTime() + 10080 * 60 * 1e3);
-		case "MONTHLY": {
-			const end = new Date(start);
-			end.setMonth(end.getMonth() + 1);
-			return end;
-		}
-		case "YEARLY": {
-			const end = new Date(start);
-			end.setFullYear(end.getFullYear() + 1);
-			return end;
-		}
-	}
-}
-/**
-* Format a period key for grouping.
-*/
-function formatPeriodKey(date, periodType) {
-	const start = getPeriodStart(date, periodType);
-	const year = start.getFullYear();
-	const month = String(start.getMonth() + 1).padStart(2, "0");
-	const day = String(start.getDate()).padStart(2, "0");
-	const hour = String(start.getHours()).padStart(2, "0");
-	switch (periodType) {
-		case "HOURLY": return `${year}-${month}-${day}T${hour}`;
-		case "DAILY": return `${year}-${month}-${day}`;
-		case "WEEKLY": return `${year}-W${getWeekNumber(start)}`;
-		case "MONTHLY": return `${year}-${month}`;
-		case "YEARLY": return `${year}`;
-	}
-}
-function getWeekNumber(date) {
-	const d = new Date(Date.UTC(date.getFullYear(), date.getMonth(), date.getDate()));
-	const dayNum = d.getUTCDay() || 7;
-	d.setUTCDate(d.getUTCDate() + 4 - dayNum);
-	const yearStart = new Date(Date.UTC(d.getUTCFullYear(), 0, 1));
-	const weekNum = Math.ceil(((d.getTime() - yearStart.getTime()) / 864e5 + 1) / 7);
-	return String(weekNum).padStart(2, "0");
-}
-/**
-* Usage aggregator.
-*
-* Aggregates usage records into summaries based on period type.
-*/
-var UsageAggregator = class {
-	storage;
-	batchSize;
-	constructor(options) {
-		this.storage = options.storage;
-		this.batchSize = options.batchSize || 1e3;
-	}
-	/**
-	* Aggregate usage records for a period.
-	*/
-	async aggregate(params) {
-		const { periodType, periodStart, metricKey } = params;
-		const periodEnd = params.periodEnd || getPeriodEnd(periodStart, periodType);
-		const result = {
-			periodType,
-			periodStart,
-			periodEnd,
-			recordsProcessed: 0,
-			summariesCreated: 0,
-			summariesUpdated: 0,
-			errors: []
-		};
-		const records = await this.storage.getUnaggregatedRecords({
-			metricKey,
-			periodStart,
-			periodEnd,
-			limit: this.batchSize
-		});
-		if (records.length === 0) return result;
-		const groups = this.groupRecords(records, periodType);
-		for (const [groupKey, groupRecords] of groups.entries()) try {
-			await this.aggregateGroup(groupKey, groupRecords, periodType, result);
-		} catch (error) {
-			const [metricKey$1, subjectType, subjectId] = groupKey.split("::");
-			result.errors.push({
-				metricKey: metricKey$1,
-				subjectType,
-				subjectId,
-				error: error instanceof Error ? error.message : String(error)
-			});
-		}
-		const recordIds = records.map((r) => r.id);
-		await this.storage.markRecordsAggregated(recordIds, /* @__PURE__ */ new Date());
-		result.recordsProcessed = records.length;
-		return result;
-	}
-	/**
-	* Group records by metric, subject, and period.
-	*/
-	groupRecords(records, periodType) {
-		const groups = /* @__PURE__ */ new Map();
-		for (const record of records) {
-			const periodKey = formatPeriodKey(record.timestamp, periodType);
-			const groupKey = `${record.metricKey}::${record.subjectType}::${record.subjectId}::${periodKey}`;
-			const existing = groups.get(groupKey) || [];
-			existing.push(record);
-			groups.set(groupKey, existing);
-		}
-		return groups;
-	}
-	/**
-	* Aggregate a group of records into a summary.
-	*/
-	async aggregateGroup(groupKey, records, periodType, result) {
-		const [metricKey, subjectType, subjectId] = groupKey.split("::");
-		if (!metricKey || !subjectType || !subjectId || records.length === 0) return;
-		const firstRecord = records[0];
-		const periodStart = getPeriodStart(firstRecord.timestamp, periodType);
-		const periodEnd = getPeriodEnd(firstRecord.timestamp, periodType);
-		const aggregationType = (await this.storage.getMetric(metricKey))?.aggregationType || "SUM";
-		const quantities = records.map((r) => r.quantity);
-		const aggregated = this.calculateAggregation(quantities, aggregationType);
-		await this.storage.upsertSummary({
-			metricKey,
-			subjectType,
-			subjectId,
-			periodType,
-			periodStart,
-			periodEnd,
-			totalQuantity: aggregated.total,
-			recordCount: records.length,
-			minQuantity: aggregated.min,
-			maxQuantity: aggregated.max,
-			avgQuantity: aggregated.avg
-		});
-		result.summariesCreated++;
-	}
-	/**
-	* Calculate aggregation values.
-	*/
-	calculateAggregation(quantities, aggregationType) {
-		if (quantities.length === 0) return {
-			total: 0,
-			min: 0,
-			max: 0,
-			avg: 0
-		};
-		const min = Math.min(...quantities);
-		const max = Math.max(...quantities);
-		const sum = quantities.reduce((a, b) => a + b, 0);
-		const avg = sum / quantities.length;
-		const count = quantities.length;
-		let total;
-		switch (aggregationType) {
-			case "COUNT":
-				total = count;
-				break;
-			case "SUM":
-				total = sum;
-				break;
-			case "AVG":
-				total = avg;
-				break;
-			case "MAX":
-				total = max;
-				break;
-			case "MIN":
-				total = min;
-				break;
-			case "LAST":
-				total = quantities[quantities.length - 1] ?? 0;
-				break;
-			default: total = sum;
-		}
-		return {
-			total,
-			min,
-			max,
-			avg
-		};
-	}
-};
-/**
-* In-memory usage storage for testing.
-*/
-var InMemoryUsageStorage = class {
-	records = [];
-	summaries = /* @__PURE__ */ new Map();
-	metrics = /* @__PURE__ */ new Map();
-	addRecord(record) {
-		this.records.push(record);
-	}
-	addMetric(metric) {
-		this.metrics.set(metric.key, metric);
-	}
-	async getUnaggregatedRecords(options) {
-		let records = this.records.filter((r) => {
-			const inPeriod = r.timestamp >= options.periodStart && r.timestamp < options.periodEnd;
-			const matchesMetric = !options.metricKey || r.metricKey === options.metricKey;
-			return inPeriod && matchesMetric;
-		});
-		if (options.limit) records = records.slice(0, options.limit);
-		return records;
-	}
-	async markRecordsAggregated(recordIds) {
-		this.records = this.records.filter((r) => !recordIds.includes(r.id));
-	}
-	async upsertSummary(summary) {
-		const key = `${summary.metricKey}::${summary.subjectType}::${summary.subjectId}::${summary.periodType}::${summary.periodStart.toISOString()}`;
-		const existing = this.summaries.get(key);
-		if (existing) {
-			existing.totalQuantity += summary.totalQuantity;
-			existing.recordCount += summary.recordCount;
-			if (summary.minQuantity !== void 0) existing.minQuantity = Math.min(existing.minQuantity ?? Infinity, summary.minQuantity);
-			if (summary.maxQuantity !== void 0) existing.maxQuantity = Math.max(existing.maxQuantity ?? -Infinity, summary.maxQuantity);
-			return existing;
-		}
-		const newSummary = {
-			id: `summary-${Date.now()}-${Math.random().toString(36).slice(2)}`,
-			...summary
-		};
-		this.summaries.set(key, newSummary);
-		return newSummary;
-	}
-	async getMetric(key) {
-		return this.metrics.get(key) || null;
-	}
-	async listMetrics() {
-		return Array.from(this.metrics.values());
-	}
-	getSummaries() {
-		return Array.from(this.summaries.values());
-	}
-	clear() {
-		this.records = [];
-		this.summaries.clear();
-		this.metrics.clear();
-	}
-};
-
-//#endregion
-export { InMemoryUsageStorage, UsageAggregator, formatPeriodKey, getPeriodEnd, getPeriodStart };
+function e(e,t){let n=new Date(e);switch(t){case`HOURLY`:return n.setMinutes(0,0,0),n;case`DAILY`:return n.setHours(0,0,0,0),n;case`WEEKLY`:n.setHours(0,0,0,0);let e=n.getDay();return n.setDate(n.getDate()-e),n;case`MONTHLY`:return n.setHours(0,0,0,0),n.setDate(1),n;case`YEARLY`:return n.setHours(0,0,0,0),n.setMonth(0,1),n}}function t(t,n){let r=e(t,n);switch(n){case`HOURLY`:return new Date(r.getTime()+3600*1e3);case`DAILY`:return new Date(r.getTime()+1440*60*1e3);case`WEEKLY`:return new Date(r.getTime()+10080*60*1e3);case`MONTHLY`:{let e=new Date(r);return e.setMonth(e.getMonth()+1),e}case`YEARLY`:{let e=new Date(r);return e.setFullYear(e.getFullYear()+1),e}}}function n(t,n){let i=e(t,n),a=i.getFullYear(),o=String(i.getMonth()+1).padStart(2,`0`),s=String(i.getDate()).padStart(2,`0`),c=String(i.getHours()).padStart(2,`0`);switch(n){case`HOURLY`:return`${a}-${o}-${s}T${c}`;case`DAILY`:return`${a}-${o}-${s}`;case`WEEKLY`:return`${a}-W${r(i)}`;case`MONTHLY`:return`${a}-${o}`;case`YEARLY`:return`${a}`}}function r(e){let t=new Date(Date.UTC(e.getFullYear(),e.getMonth(),e.getDate())),n=t.getUTCDay()||7;t.setUTCDate(t.getUTCDate()+4-n);let r=new Date(Date.UTC(t.getUTCFullYear(),0,1)),i=Math.ceil(((t.getTime()-r.getTime())/864e5+1)/7);return String(i).padStart(2,`0`)}var i=class{storage;batchSize;constructor(e){this.storage=e.storage,this.batchSize=e.batchSize||1e3}async aggregate(e){let{periodType:n,periodStart:r,metricKey:i}=e,a=e.periodEnd||t(r,n),o={periodType:n,periodStart:r,periodEnd:a,recordsProcessed:0,summariesCreated:0,summariesUpdated:0,errors:[]},s=await this.storage.getUnaggregatedRecords({metricKey:i,periodStart:r,periodEnd:a,limit:this.batchSize});if(s.length===0)return o;let c=this.groupRecords(s,n);for(let[e,t]of c.entries())try{await this.aggregateGroup(e,t,n,o)}catch(t){let[n,r,i]=e.split(`::`);o.errors.push({metricKey:n,subjectType:r,subjectId:i,error:t instanceof Error?t.message:String(t)})}let l=s.map(e=>e.id);return await this.storage.markRecordsAggregated(l,new Date),o.recordsProcessed=s.length,o}groupRecords(e,t){let r=new Map;for(let i of e){let e=n(i.timestamp,t),a=`${i.metricKey}::${i.subjectType}::${i.subjectId}::${e}`,o=r.get(a)||[];o.push(i),r.set(a,o)}return r}async aggregateGroup(n,r,i,a){let[o,s,c]=n.split(`::`);if(!o||!s||!c||r.length===0)return;let l=r[0],u=e(l.timestamp,i),d=t(l.timestamp,i),f=(await this.storage.getMetric(o))?.aggregationType||`SUM`,p=r.map(e=>e.quantity),m=this.calculateAggregation(p,f);await this.storage.upsertSummary({metricKey:o,subjectType:s,subjectId:c,periodType:i,periodStart:u,periodEnd:d,totalQuantity:m.total,recordCount:r.length,minQuantity:m.min,maxQuantity:m.max,avgQuantity:m.avg}),a.summariesCreated++}calculateAggregation(e,t){if(e.length===0)return{total:0,min:0,max:0,avg:0};let n=Math.min(...e),r=Math.max(...e),i=e.reduce((e,t)=>e+t,0),a=i/e.length,o=e.length,s;switch(t){case`COUNT`:s=o;break;case`SUM`:s=i;break;case`AVG`:s=a;break;case`MAX`:s=r;break;case`MIN`:s=n;break;case`LAST`:s=e[e.length-1]??0;break;default:s=i}return{total:s,min:n,max:r,avg:a}}},a=class{records=[];summaries=new Map;metrics=new Map;addRecord(e){this.records.push(e)}addMetric(e){this.metrics.set(e.key,e)}async getUnaggregatedRecords(e){let t=this.records.filter(t=>{let n=t.timestamp>=e.periodStart&&t.timestamp<e.periodEnd,r=!e.metricKey||t.metricKey===e.metricKey;return n&&r});return e.limit&&(t=t.slice(0,e.limit)),t}async markRecordsAggregated(e){this.records=this.records.filter(t=>!e.includes(t.id))}async upsertSummary(e){let t=`${e.metricKey}::${e.subjectType}::${e.subjectId}::${e.periodType}::${e.periodStart.toISOString()}`,n=this.summaries.get(t);if(n)return n.totalQuantity+=e.totalQuantity,n.recordCount+=e.recordCount,e.minQuantity!==void 0&&(n.minQuantity=Math.min(n.minQuantity??1/0,e.minQuantity)),e.maxQuantity!==void 0&&(n.maxQuantity=Math.max(n.maxQuantity??-1/0,e.maxQuantity)),n;let r={id:`summary-${Date.now()}-${Math.random().toString(36).slice(2)}`,...e};return this.summaries.set(t,r),r}async getMetric(e){return this.metrics.get(e)||null}async listMetrics(){return Array.from(this.metrics.values())}getSummaries(){return Array.from(this.summaries.values())}clear(){this.records=[],this.summaries.clear(),this.metrics.clear()}};export{a as InMemoryUsageStorage,i as UsageAggregator,n as formatPeriodKey,t as getPeriodEnd,e as getPeriodStart};