jsm-utilities
Version:
A utilities library.
125 lines (124 loc) • 5.29 kB
JavaScript
;
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
if (k2 === undefined) k2 = k;
var desc = Object.getOwnPropertyDescriptor(m, k);
if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
desc = { enumerable: true, get: function() { return m[k]; } };
}
Object.defineProperty(o, k2, desc);
}) : (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 () {
var ownKeys = function(o) {
ownKeys = Object.getOwnPropertyNames || function (o) {
var ar = [];
for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
return ar;
};
return ownKeys(o);
};
return function (mod) {
if (mod && mod.__esModule) return mod;
var result = {};
if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
__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.isObjectIdValid = void 0;
exports.upgradeIndexes = upgradeIndexes;
exports.ensureIndexes = ensureIndexes;
const jsm_logger_1 = __importStar(require("jsm-logger"));
const lodash_1 = require("lodash");
const mongodb_1 = require("mongodb");
const exceptions_1 = require("../../common/exceptions");
const logger = (0, jsm_logger_1.default)(jsm_logger_1.LoggerContext.UTILITY, `MongoDB-helpers`, { library: 'jsm-utilities' });
const isObjectIdValid = (id) => {
try {
if (!id)
return false;
if (mongodb_1.ObjectId.isValid(id.toString()) &&
String(new mongodb_1.ObjectId(id.toString())) === id)
return true;
return false;
}
catch (error) {
logger.save.error({
name: "isObjectIdValid",
payload: {
related_to: id,
error: (0, exceptions_1.errorToObject)(error),
},
});
return false;
}
};
exports.isObjectIdValid = isObjectIdValid;
function upgradeIndexes(model, cache) {
return __awaiter(this, void 0, void 0, function* () {
try {
if (!model)
return;
if (typeof (cache === null || cache === void 0 ? void 0 : cache.generateForeignKey) !== "function") {
logger.warn("upgradeIndexes", "Cache is not available");
return;
}
const collectionKey = cache.generateForeignKey(`dbCollectionsPerformance:${model.modelName}`);
const entries = yield cache.getCollectionEntries(collectionKey);
const indexes = (0, lodash_1.sortBy)(Object.values(entries || {}), (o) => -o.value.count);
for (const index of indexes) {
const { data } = index.value;
const { fields, key } = data;
if (Object.keys(fields).some((k) => k.includes("$")))
continue;
const keys = Object.keys(fields);
const indexes = yield model.listIndexes();
const indexKeys = indexes.map((i) => i.name);
if (!indexKeys.includes(key) && !keys.includes("$")) {
try {
logger.info("upgradeIndexes", model.collection.name, "Creating index", key);
yield model.schema.index(fields, { name: key });
}
catch (error) {
logger.error("upgradeIndexes", model.collection.name, error);
}
}
}
}
catch (error) {
logger.error(error.message, error);
}
});
}
function ensureIndexes(model, cache) {
upgradeIndexes(model, cache).then(() => {
model
.createIndexes()
.then(() => {
// logger.success("ensure-indexes", model.collection.name);
})
.catch((error) => {
var _a;
logger.error("ensure-indexes", model.collection.name, ((_a = error.message) === null || _a === void 0 ? void 0 : _a.includes("Index already exists"))
? error.message
: error);
});
});
}