@azure/cosmos
Version:
Microsoft Azure Cosmos DB Service Node.js SDK for NOSQL API
97 lines • 4.29 kB
JavaScript
"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.GroupByValueEndpointComponent = void 0;
const hashObject_js_1 = require("../../utils/hashObject.js");
const index_js_1 = require("../Aggregators/index.js");
const headerUtils_js_1 = require("../headerUtils.js");
const emptyGroup_js_1 = require("./emptyGroup.js");
/** @hidden */
class GroupByValueEndpointComponent {
constructor(executionContext, queryInfo) {
this.executionContext = executionContext;
this.queryInfo = queryInfo;
this.aggregators = new Map();
this.aggregateResultArray = [];
this.completed = false;
// VALUE queries will only every have a single grouping
this.aggregateType = this.queryInfo.aggregates[0];
}
hasMoreResults() {
return this.executionContext.hasMoreResults();
}
async fetchMore(diagnosticNode) {
if (this.completed) {
return {
result: undefined,
headers: (0, headerUtils_js_1.getInitialHeader)(),
};
}
const aggregateHeaders = (0, headerUtils_js_1.getInitialHeader)();
const response = await this.executionContext.fetchMore(diagnosticNode);
(0, headerUtils_js_1.mergeHeaders)(aggregateHeaders, response.headers);
if (response === undefined || response.result === undefined) {
if (this.aggregators.size > 0) {
return this.generateAggregateResponse(aggregateHeaders);
}
return { result: undefined, headers: aggregateHeaders };
}
for (const item of response.result) {
if (item) {
let grouping = emptyGroup_js_1.emptyGroup;
let payload = item;
if (item.groupByItems) {
// If the query contains a GROUP BY clause, it will have a payload property and groupByItems
payload = item.payload;
grouping = await (0, hashObject_js_1.hashObject)(item.groupByItems);
}
const aggregator = this.aggregators.get(grouping);
if (!aggregator) {
// This is the first time we have seen a grouping so create a new aggregator
this.aggregators.set(grouping, (0, index_js_1.createAggregator)(this.aggregateType));
}
if (this.aggregateType) {
const aggregateResult = (0, emptyGroup_js_1.extractAggregateResult)(payload[0]);
// if aggregate result is null, we need to short circuit aggregation and return undefined
if (aggregateResult === null) {
this.completed = true;
}
this.aggregators.get(grouping).aggregate(aggregateResult);
}
else {
// Queries with no aggregates pass the payload directly to the aggregator
// Example: SELECT VALUE c.team FROM c GROUP BY c.team
this.aggregators.get(grouping).aggregate(payload);
}
}
}
// We bail early since we got an undefined result back `[{}]`
if (this.completed) {
return {
result: undefined,
headers: aggregateHeaders,
};
}
if (this.executionContext.hasMoreResults()) {
return { result: [], headers: aggregateHeaders };
}
else {
// If no results are left in the underlying execution context, convert our aggregate results to an array
return this.generateAggregateResponse(aggregateHeaders);
}
}
generateAggregateResponse(aggregateHeaders) {
for (const aggregator of this.aggregators.values()) {
const result = aggregator.getResult();
if (result !== undefined) {
this.aggregateResultArray.push(result);
}
}
this.completed = true;
return {
result: this.aggregateResultArray,
headers: aggregateHeaders,
};
}
}
exports.GroupByValueEndpointComponent = GroupByValueEndpointComponent;
//# sourceMappingURL=GroupByValueEndpointComponent.js.map