@datastax/astra-db-ts
Version:
Data API TypeScript client
125 lines (124 loc) • 4.93 kB
JavaScript
"use strict";
// Copyright Datastax, Inc
// SPDX-License-Identifier: Apache-2.0
Object.defineProperty(exports, "__esModule", { value: true });
exports.FindCursor = void 0;
const vector_js_1 = require("../../documents/datatypes/vector.js");
const abstract_cursor_js_1 = require("../../documents/cursors/abstract-cursor.js");
const constants_js_1 = require("../../lib/constants.js");
const ctx_js_1 = require("../../lib/api/ser-des/ctx.js");
const utils_js_1 = require("../../lib/utils.js");
const common_js_1 = require("../../documents/cursors/common.js");
class FindCursor extends abstract_cursor_js_1.AbstractCursor {
constructor(parent, serdes, filter, options, mapping) {
super(options ?? {}, mapping);
Object.defineProperty(this, "_httpClient", {
enumerable: true,
configurable: true,
writable: true,
value: void 0
});
Object.defineProperty(this, "_serdes", {
enumerable: true,
configurable: true,
writable: true,
value: void 0
});
Object.defineProperty(this, "_parent", {
enumerable: true,
configurable: true,
writable: true,
value: void 0
});
Object.defineProperty(this, "_filter", {
enumerable: true,
configurable: true,
writable: true,
value: void 0
});
Object.defineProperty(this, "_sortVector", {
enumerable: true,
configurable: true,
writable: true,
value: new utils_js_1.QueryState()
});
this._parent = parent;
this._httpClient = parent._httpClient;
this._serdes = serdes;
this._filter = filter;
}
[constants_js_1.$CustomInspect]() {
return `${this.constructor.name}(source="${this._parent.keyspace}.${this._parent.name}",state="${this._state}",consumed=${this.consumed()},buffered=${this.buffered()})`;
}
filter(filter) {
return (0, common_js_1.buildFLCFilter)(this, filter);
}
sort(sort) {
return (0, common_js_1.buildFLCSort)(this, sort);
}
limit(limit) {
return (0, common_js_1.buildFLCOption)(this, 'limit', limit || undefined);
}
skip(skip) {
return (0, common_js_1.buildFLCOption)(this, 'skip', skip);
}
includeSortVector(includeSortVector) {
return (0, common_js_1.buildFLCOption)(this, 'includeSortVector', includeSortVector ?? true);
}
project(projection) {
return (0, common_js_1.buildFLCPreMapOption)(this, 'projection', structuredClone(projection));
}
includeSimilarity(includeSimilarity) {
return (0, common_js_1.buildFLCPreMapOption)(this, 'includeSimilarity', includeSimilarity ?? true);
}
map(map) {
return (0, common_js_1.buildFLCMap)(this, map);
}
async getSortVector() {
if (this._sortVector.state === utils_js_1.QueryState.Unattempted && this._options.includeSortVector) {
const reset2idle = this._state === 'idle';
await this._next(true, '.getSortVector');
if (reset2idle) {
this._state = 'idle';
}
}
return this._sortVector.unwrap();
}
clone() {
return (0, common_js_1.cloneFLC)(this, this._filter, this._options, this._mapping);
}
async _nextPage(extra, tm) {
const command = {
find: {
filter: this._filter[0],
projection: this._options.projection,
sort: this._options.sort,
options: {
includeSimilarity: this._options.includeSimilarity,
includeSortVector: this._options.includeSortVector,
limit: this._options.limit,
skip: this._options.skip,
pageState: this._nextPageState.unwrap(),
},
},
};
const raw = await this._httpClient.executeCommand(command, {
timeoutManager: tm ?? this._httpClient.tm.single('generalMethodTimeoutMs', this._options),
bigNumsPresent: this._filter[1],
extraLogInfo: extra,
});
this._nextPageState.swap(raw.data?.nextPageState);
const buffer = raw.data?.documents ?? [];
for (let i = 0, n = buffer.length; i < n; i++) {
buffer[i] = this._serdes.deserialize(buffer[i], raw, ctx_js_1.SerDesTarget.Record);
}
const sortVector = raw.status?.sortVector;
this._sortVector.swap(sortVector ? (0, vector_js_1.vector)(sortVector) : sortVector);
this._options.includeSortVector = false;
return buffer;
}
_tm() {
return this._httpClient.tm;
}
}
exports.FindCursor = FindCursor;