access-db/lib/elasticsearch/find.js

[统一数据库的连接]Unified access to all kinds of databases

"use strict";
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());
    });
};
var __importDefault = (this && this.__importDefault) || function (mod) {
    return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
const dbElasticsearch_1 = require("../utils/dbElasticsearch");
const findTrans_1 = __importDefault(require("../utils/findTrans"));
const constants_1 = require("../constants/constants");
const utils_1 = require("../utils/utils");
const error_1 = require("../constants/error");
function fetchFind(table, params) {
    return new Promise((resolve, reject) => __awaiter(this, void 0, void 0, function* () {
        try {
            if (!dbElasticsearch_1.client)
                return;
            let QQ = (0, findTrans_1.default)(params, 1, null, constants_1.PLATFORM_NAME.ELASTICSEARCH);
            /** 筛选哪些字段要返回，那些不返回 */
            let tempSelect = {
                _source_excludes: [],
                _source_includes: []
            };
            if (params.select) {
                if (!(0, utils_1.isArray)(params.select))
                    throw new Error(error_1.SELECT_NOT_ARRAY);
                params.select.forEach(item => {
                    if (item[0] === '-') {
                        tempSelect._source_excludes.push(item.substring(1));
                    }
                    else {
                        tempSelect._source_includes.push(item);
                    }
                });
            }
            /** 排序 */
            let sort = [];
            if (params.orderBy) {
                if (!(0, utils_1.isArray)(params.orderBy))
                    throw new Error(error_1.ORDER_BY_NOT_ARRAY);
                params.orderBy.forEach(item => {
                    if (item[0] === '-') {
                        sort.push({
                            [`${item.substring(1)}`]: 'desc'
                        });
                    }
                    else {
                        sort.push({
                            [`${item}`]: 'asc'
                        });
                    }
                });
            }
            let res = yield dbElasticsearch_1.client.search(Object.assign(Object.assign({ index: (0, dbElasticsearch_1.esTable)(table), query: QQ === 2 ? { match_all: {} } : QQ, size: params.limit || 20, from: (params.limit || 20) * ((params.page || 1) - 1) }, tempSelect), { sort }));
            let tempTotal = 0;
            if (typeof (res.hits.total) === 'number') {
                tempTotal = res.hits.total;
            }
            else {
                tempTotal = res.hits.total.value;
            }
            let tempObjs = [];
            for (let i = 0; i < res.hits.hits.length; i++) {
                let tempSource = res.hits.hits[i]._source || {};
                tempObjs.push(Object.assign({ _index: res.hits.hits[i]._index, _id: res.hits.hits[i]._id, _score: res.hits.hits[i]._score }, tempSource));
            }
            resolve({ data: {
                    objects: tempObjs,
                    took: res.took,
                    max_score: res.hits.max_score,
                    total: tempTotal
                } });
        }
        catch (err) {
            reject(err);
        }
    }));
}
exports.default = fetchFind;