couchbase
Version:
The official Couchbase Node.js Client Library.
170 lines (169 loc) • 6.82 kB
JavaScript
;
Object.defineProperty(exports, "__esModule", { value: true });
exports.QueryExecutor = void 0;
const bindingutilities_1 = require("./bindingutilities");
const observability_1 = require("./observability");
const observabilityhandler_1 = require("./observabilityhandler");
const observabilitytypes_1 = require("./observabilitytypes");
const querytypes_1 = require("./querytypes");
const streamablepromises_1 = require("./streamablepromises");
const utilities_1 = require("./utilities");
/**
* @internal
*/
class QueryExecutor {
/**
* @internal
*/
constructor(cluster) {
this._cluster = cluster;
}
/**
* @internal
*/
get observabilityInstruments() {
return this._cluster.observabilityInstruments;
}
/**
* @internal
*/
static _processQueryResponse(emitter, err, resp, obsReqHandler) {
if (err) {
obsReqHandler === null || obsReqHandler === void 0 ? void 0 : obsReqHandler.endWithError(err);
emitter.emit('error', err);
emitter.emit('end');
return;
}
resp.rows.forEach((row) => {
emitter.emit('row', JSON.parse(row));
});
{
const metaData = resp.meta;
let warnings;
if (metaData.warnings) {
warnings = metaData.warnings.map((warningData) => new querytypes_1.QueryWarning({
code: warningData.code,
message: warningData.message,
}));
}
else {
warnings = [];
}
let metrics;
if (metaData.metrics) {
const metricsData = metaData.metrics;
metrics = new querytypes_1.QueryMetrics({
elapsedTime: metricsData.elapsed_time,
executionTime: metricsData.execution_time,
sortCount: metricsData.sort_count || 0,
resultCount: metricsData.result_count || 0,
resultSize: metricsData.result_size || 0,
mutationCount: metricsData.mutation_count || 0,
errorCount: metricsData.error_count || 0,
warningCount: metricsData.warning_count || 0,
});
}
else {
metrics = undefined;
}
const meta = new querytypes_1.QueryMetaData({
requestId: metaData.request_id,
clientContextId: metaData.client_context_id,
status: metaData.status,
signature: metaData.signature
? JSON.parse(metaData.signature)
: undefined,
warnings: warnings,
metrics: metrics,
profile: metaData.profile ? JSON.parse(metaData.profile) : undefined,
});
emitter.emit('meta', meta);
}
obsReqHandler === null || obsReqHandler === void 0 ? void 0 : obsReqHandler.end();
emitter.emit('end');
}
/**
* @internal
*
* Used by transactions (observability currently not available for transactions)
*/
static execute(exec) {
const emitter = new streamablepromises_1.StreamableRowPromise((rows, meta) => {
return new querytypes_1.QueryResult({
rows: rows,
meta: meta,
});
});
exec((cppErr, resp) => {
let err = null;
if (cppErr) {
err = (0, bindingutilities_1.errorFromCpp)(cppErr);
}
QueryExecutor._processQueryResponse(emitter, err, resp);
});
return emitter;
}
/**
* @internal
*/
static executePromise(queryPromise, obsReqHandler) {
const emitter = new streamablepromises_1.StreamableRowPromise((rows, meta) => {
return new querytypes_1.QueryResult({
rows: rows,
meta: meta,
});
});
utilities_1.PromiseHelper.wrapAsync(async () => {
const [err, resp] = await queryPromise;
QueryExecutor._processQueryResponse(emitter, err, resp, obsReqHandler);
});
return emitter;
}
/**
* @internal
*/
query(query, options) {
const timeout = options.timeout || this._cluster.queryTimeout;
const obsReqHandler = new observabilityhandler_1.ObservableRequestHandler(observabilitytypes_1.StreamingOp.Query, this.observabilityInstruments, options === null || options === void 0 ? void 0 : options.parentSpan);
obsReqHandler.setRequestHttpAttributes({
statement: query,
queryContext: options.queryContext,
queryOptions: options,
});
return QueryExecutor.executePromise((0, observability_1.wrapObservableBindingCall)(this._cluster.conn.query.bind(this._cluster.conn), {
statement: query,
client_context_id: options.clientContextId,
adhoc: options.adhoc === false ? false : true,
metrics: options.metrics || false,
readonly: options.readOnly || false,
flex_index: options.flexIndex || false,
preserve_expiry: options.preserveExpiry || false,
use_replica: options.useReplica,
max_parallelism: options.maxParallelism,
scan_cap: options.scanCap,
scan_wait: options.scanWait,
pipeline_batch: options.pipelineBatch,
pipeline_cap: options.pipelineCap,
scan_consistency: (0, bindingutilities_1.queryScanConsistencyToCpp)(options.scanConsistency),
mutation_state: (0, bindingutilities_1.mutationStateToCpp)(options.consistentWith).tokens,
timeout: timeout,
query_context: options.queryContext,
profile: (0, bindingutilities_1.queryProfileToCpp)(options.profile),
raw: options.raw
? Object.fromEntries(Object.entries(options.raw)
.filter(([, v]) => v !== undefined)
.map(([k, v]) => [k, JSON.stringify(v)]))
: {},
positional_parameters: options.parameters && Array.isArray(options.parameters)
? options.parameters.map((v) => JSON.stringify(v !== null && v !== void 0 ? v : null))
: [],
named_parameters: options.parameters && !Array.isArray(options.parameters)
? Object.fromEntries(Object.entries(options.parameters)
.filter(([, v]) => v !== undefined)
.map(([k, v]) => [k, JSON.stringify(v)]))
: {},
body_str: '',
}, obsReqHandler), obsReqHandler);
}
}
exports.QueryExecutor = QueryExecutor;