slonik-trpc
Version:
Slonik tRPC loader
95 lines (94 loc) • 4.57 kB
JavaScript
;
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
if (k2 === undefined) k2 = k;
Object.defineProperty(o, k2, { enumerable: true, get: function() { return m[k]; } });
}) : (function(o, m, k, k2) {
if (k2 === undefined) k2 = k;
o[k2] = m[k];
}));
var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
Object.defineProperty(o, "default", { enumerable: true, value: v });
}) : function(o, v) {
o["default"] = v;
});
var __importStar = (this && this.__importStar) || function (mod) {
if (mod && mod.__esModule) return mod;
var result = {};
if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
__setModuleDefault(result, mod);
return result;
};
var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
return new (P || (P = Promise))(function (resolve, reject) {
function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
step((generator = generator.apply(thisArg, _arguments || [])).next());
});
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.makeQueryAnalyzer = void 0;
const slonik_1 = require("slonik");
const zod_1 = require("./zod");
function makeQueryAnalyzer(db) {
const generateMockFilters = (queryLoader, options) => __awaiter(this, void 0, void 0, function* () {
const args = yield queryLoader.getLoadArgs({
sortableColumns: [],
disabledFilters: {
AND: true,
OR: true,
NOT: true,
}
}).pick({
where: true,
});
const mocked = (0, zod_1.mockZod)(args, options);
return mocked.where;
});
const self = {
analyzeQuery: (query) => {
return db
.any(slonik_1.sql.unsafe `EXPLAIN (ANALYZE, FORMAT JSON) ${query}`)
.then((a) => {
const result = a[0]['QUERY PLAN'][0];
return {
execution: result['Execution Time'],
planning: result['Planning Time'],
plan: result['Plan'],
};
});
},
/** Runs each field as a separate query, to try and find which fields take more time to resolve.
* This is useful when you have sub-queries as fields, and you want to know which sub-queries are heavy.
* @returns A number in milliseonds for each field
* */
benchmarkQueryLoaderFields: (queryLoader, options) => __awaiter(this, void 0, void 0, function* () {
const { iterations = 1, concurrency = 1, args } = options || {};
const { default: Nativebird } = yield Promise.resolve().then(() => __importStar(require('nativebird')));
const selectableFields = queryLoader.getSelectableFields();
// Analyze every selectable field separately, and then compare which fields take the most time
const promises = yield Nativebird.map(selectableFields, (field) => __awaiter(this, void 0, void 0, function* () {
const query = yield queryLoader.getQuery(Object.assign(Object.assign({ take: 100 }, args), { select: [field] }));
let totalTime = 0;
for (let i = 0; i < iterations; i++) {
totalTime += yield self.analyzeQuery(query).then((a) => a.execution + a.planning);
}
return [field, totalTime];
}), { concurrency });
return Object.fromEntries(promises);
}),
/**
* Runs the query with mock filters, to see if all the filters are valid.
* */
testAllFilters: (queryLoader, options) => __awaiter(this, void 0, void 0, function* () {
const where = yield generateMockFilters(queryLoader, options);
const query = yield queryLoader.getQuery({
where,
});
return self.analyzeQuery(query);
}),
};
return self;
}
exports.makeQueryAnalyzer = makeQueryAnalyzer;