marketdata-api/lib/models/DateRange.js

OData server. Get market data from crypto exchanges open REST API.

"use strict";
var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
    else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
    return c > 3 && r && Object.defineProperty(target, key, r), r;
};
var __param = (this && this.__param) || function (paramIndex, decorator) {
    return function (target, key) { decorator(target, key, paramIndex); }
};
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 };
};
var __importStar = (this && this.__importStar) || function (mod) {
    if (mod && mod.__esModule) return mod;
    var result = {};
    if (mod != null) for (var k in mod) if (Object.hasOwnProperty.call(mod, k)) result[k] = mod[k];
    result["default"] = mod;
    return result;
};
Object.defineProperty(exports, "__esModule", { value: true });
const event_stream_1 = __importDefault(require("event-stream"));
const get_candles_1 = require("get-candles");
const lodash_1 = __importDefault(require("lodash"));
const Moment = __importStar(require("moment"));
const moment_range_1 = require("moment-range");
const mongodb_1 = require("mongodb");
const odata_v4_server_1 = require("odata-v4-server");
const connect_1 = __importDefault(require("../connect"));
const moment = moment_range_1.extendMoment(Moment);
const collectionName = "dateRange";
class DateRange {
    constructor(data) {
        Object.assign(this, data);
    }
    importData(result) {
        return __awaiter(this, void 0, void 0, function* () {
            // tslint:disable-next-line: variable-name
            const _id = new mongodb_1.ObjectID(result._id);
            const db = yield connect_1.default();
            const collection = db.collection(collectionName);
            const { begin, end, parentId, status } = yield collection.findOne({ _id });
            // запускать только если !status
            if (!status) {
                return 0;
            }
            // получить список всех диапазонов набора со статусом true
            const ranges = lodash_1.default.reduce(collection
                .find({ $and: [{ parentId }, { status: true }] }) // TODO добавить условие по дате
                .map(e => {
                return moment.range(e.begin, e.end);
            }), (accumulator, value) => lodash_1.default.flatten(accumulator
                .map(e => e.subtract(value))
                .filter(e => e)), [moment.range(begin, end)]);
            // из текущего диапазона исключить все пересечения
            // для каждого полученного элемента если исходный массив не нулевой
            // из каждого элемента исходного массива исключить полученный
            // для каждого элемента исходного массива выполнить запрос к источнику данных
            // после успешного получения сохранить в базе данных, изменить статус на true
            // вернуть положительный результат
            const { exchange, currency, asset, period } = yield collection.findOne({ _id: parentId });
            const collectionCandle = db.collection("candle");
            for (const r of ranges) {
                yield new Promise(resolve => {
                    const rs = get_candles_1.streamCandle({
                        exchange,
                        currency,
                        asset,
                        period,
                        start: moment.utc(r.start).toISOString(),
                        end: moment.utc(r.end).toISOString()
                    }).pipe(event_stream_1.default.map((chunk, next) => {
                        const candle = JSON.parse(chunk);
                        collectionCandle.insertOne(candle, next);
                    }));
                    rs.on("end", () => resolve());
                });
            }
            // сохранить в бд
            return collection.updateOne({ _id }, { $set: { status: true } })
                .then(r => r.modifiedCount);
        });
    }
}
__decorate([
    odata_v4_server_1.Edm.Key,
    odata_v4_server_1.Edm.Computed,
    odata_v4_server_1.Edm.String
], DateRange.prototype, "_id", void 0);
__decorate([
    odata_v4_server_1.Edm.String
], DateRange.prototype, "parentId", void 0);
__decorate([
    odata_v4_server_1.Edm.String
], DateRange.prototype, "begin", void 0);
__decorate([
    odata_v4_server_1.Edm.String
], DateRange.prototype, "end", void 0);
__decorate([
    odata_v4_server_1.Edm.Boolean
], DateRange.prototype, "status", void 0);
__decorate([
    odata_v4_server_1.Edm.Action,
    __param(0, odata_v4_server_1.odata.result)
], DateRange.prototype, "importData", null);
exports.DateRange = DateRange;
//# sourceMappingURL=DateRange.js.map