UNPKG

cnpmcore

Version:
166 lines 15.7 kB
"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 __metadata = (this && this.__metadata) || function (k, v) { if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v); }; var __importDefault = (this && this.__importDefault) || function (mod) { return (mod && mod.__esModule) ? mod : { "default": mod }; }; Object.defineProperty(exports, "__esModule", { value: true }); exports.UpdateTotalData = void 0; const schedule_1 = require("@eggjs/tegg/schedule"); const tegg_1 = require("@eggjs/tegg"); const RegistryManagerService_1 = require("../../core/service/RegistryManagerService"); const PackageVersionDownloadRepository_1 = require("../../repository/PackageVersionDownloadRepository"); const PackageRepository_1 = require("../../repository/PackageRepository"); const TaskRepository_1 = require("../../repository/TaskRepository"); const ChangeRepository_1 = require("../../repository/ChangeRepository"); const CacheService_1 = require("../../core/service/CacheService"); const Task_1 = require("../../common/enum/Task"); const constants_1 = require("../../common/constants"); const dayjs_1 = __importDefault(require("../../common/dayjs")); let UpdateTotalData = class UpdateTotalData { // 计算下载量相关信息,不区分不同 changesStream async calculateDownloadInfo() { const download = { today: 0, yesterday: 0, samedayLastweek: 0, thisweek: 0, thismonth: 0, thisyear: 0, lastweek: 0, lastmonth: 0, lastyear: 0, }; const today = (0, dayjs_1.default)(); const lastYearStartDay = today.subtract(1, 'year').startOf('year'); const rows = await this.packageVersionDownloadRepository.query('total', lastYearStartDay.toDate(), today.toDate()); if (rows.length > 0) { const todayInt = Number(today.format('YYYYMMDD')); const yesterdayInt = Number(today.subtract(1, 'day').format('YYYYMMDD')); const samedayLastweekInt = Number(today.subtract(1, 'week').startOf('week').format('YYYYMMDD')); const thisWeekStartDayInt = Number(today.startOf('week').format('YYYYMMDD')); const thisWeekEndDayInt = Number(today.endOf('week').format('YYYYMMDD')); const thisMonthStartDayInt = Number(today.startOf('month').format('YYYYMMDD')); const thisMonthEndDayInt = Number(today.endOf('month').format('YYYYMMDD')); const thisYearStartDayInt = Number(today.startOf('year').format('YYYYMMDD')); const thisYearEndDayInt = Number(today.endOf('year').format('YYYYMMDD')); const lastWeekStartDayInt = Number(today.subtract(1, 'week').startOf('week').format('YYYYMMDD')); const lastWeekEndDayInt = Number(today.subtract(1, 'week').endOf('week').format('YYYYMMDD')); const lastMonthStartDayInt = Number(today.subtract(1, 'month').startOf('month').format('YYYYMMDD')); const lastMonthEndDayInt = Number(today.subtract(1, 'month').endOf('month').format('YYYYMMDD')); const lastYearStartDayInt = Number(today.subtract(1, 'year').startOf('year').format('YYYYMMDD')); const lastYearEndDayInt = Number(today.subtract(1, 'year').endOf('year').format('YYYYMMDD')); for (const row of rows) { for (let i = 1; i <= 31; i++) { const day = String(i).padStart(2, '0'); const field = `d${day}`; const counter = row[field]; if (!counter) continue; const dayInt = row.yearMonth * 100 + i; if (dayInt === todayInt) download.today += counter; if (dayInt === yesterdayInt) download.yesterday += counter; if (dayInt === samedayLastweekInt) download.samedayLastweek += counter; if (dayInt >= thisWeekStartDayInt && dayInt <= thisWeekEndDayInt) download.thisweek += counter; if (dayInt >= thisMonthStartDayInt && dayInt <= thisMonthEndDayInt) download.thismonth += counter; if (dayInt >= thisYearStartDayInt && dayInt <= thisYearEndDayInt) download.thisyear += counter; if (dayInt >= lastWeekStartDayInt && dayInt <= lastWeekEndDayInt) download.lastweek += counter; if (dayInt >= lastMonthStartDayInt && dayInt <= lastMonthEndDayInt) download.lastmonth += counter; if (dayInt >= lastYearStartDayInt && dayInt <= lastYearEndDayInt) download.lastyear += counter; } } } return download; } async subscribe() { const packageTotal = await this.packageRepository.queryTotal(); const download = await this.calculateDownloadInfo(); const lastChange = await this.changeRepository.getLastChange(); const totalData = { ...packageTotal, download, lastChangeId: lastChange && lastChange.id || 0, cacheTime: new Date().toISOString(), changesStream: {}, upstreamRegistries: [], }; const tasks = await this.taskRepository.findTasksByCondition({ type: Task_1.TaskType.ChangesStream }); for (const task of tasks) { // 全局 changesStream const data = task.data; // 补充录入 upstreamRegistries const registry = await this.registryManagerService.findByRegistryId(data.registryId); if (registry) { totalData.upstreamRegistries.push({ ...data, source_registry: registry?.host, changes_stream_url: registry?.changeStream, registry_name: registry?.name, }); } // 兼容 LegacyInfo 字段 if (task.targetName === constants_1.GLOBAL_WORKER) { totalData.changesStream = data; } } await this.cacheService.saveTotalData(totalData); this.logger.info('[UpdateTotalData.subscribe] total data: %j', totalData); } }; exports.UpdateTotalData = UpdateTotalData; __decorate([ (0, tegg_1.Inject)(), __metadata("design:type", Object) ], UpdateTotalData.prototype, "logger", void 0); __decorate([ (0, tegg_1.Inject)(), __metadata("design:type", PackageRepository_1.PackageRepository) ], UpdateTotalData.prototype, "packageRepository", void 0); __decorate([ (0, tegg_1.Inject)(), __metadata("design:type", TaskRepository_1.TaskRepository) ], UpdateTotalData.prototype, "taskRepository", void 0); __decorate([ (0, tegg_1.Inject)(), __metadata("design:type", ChangeRepository_1.ChangeRepository) ], UpdateTotalData.prototype, "changeRepository", void 0); __decorate([ (0, tegg_1.Inject)(), __metadata("design:type", PackageVersionDownloadRepository_1.PackageVersionDownloadRepository) ], UpdateTotalData.prototype, "packageVersionDownloadRepository", void 0); __decorate([ (0, tegg_1.Inject)(), __metadata("design:type", CacheService_1.CacheService) ], UpdateTotalData.prototype, "cacheService", void 0); __decorate([ (0, tegg_1.Inject)(), __metadata("design:type", RegistryManagerService_1.RegistryManagerService) ], UpdateTotalData.prototype, "registryManagerService", void 0); exports.UpdateTotalData = UpdateTotalData = __decorate([ (0, schedule_1.Schedule)({ type: schedule_1.ScheduleType.WORKER, scheduleData: { interval: 60000, }, }, { // immediate = false on unittest env immediate: process.env.NODE_ENV !== 'test', }) ], UpdateTotalData); //# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiVXBkYXRlVG90YWxEYXRhLmpzIiwic291cmNlUm9vdCI6IiIsInNvdXJjZXMiOlsiLi4vLi4vLi4vLi4vYXBwL3BvcnQvc2NoZWR1bGUvVXBkYXRlVG90YWxEYXRhLnRzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiI7Ozs7Ozs7Ozs7Ozs7OztBQUNBLG1EQUE4RTtBQUM5RSxzQ0FBcUM7QUFFckMsc0ZBQW1GO0FBQ25GLHdHQUFxRztBQUNyRywwRUFBdUU7QUFDdkUsb0VBQWlFO0FBQ2pFLHdFQUFxRTtBQUNyRSxrRUFBd0Y7QUFDeEYsaURBQWtEO0FBQ2xELHNEQUF1RDtBQUN2RCwrREFBdUM7QUFXaEMsSUFBTSxlQUFlLEdBQXJCLE1BQU0sZUFBZTtJQXNCMUIsZ0NBQWdDO0lBQ3hCLEtBQUssQ0FBQyxxQkFBcUI7UUFDakMsTUFBTSxRQUFRLEdBQWlCO1lBQzdCLEtBQUssRUFBRSxDQUFDO1lBQ1IsU0FBUyxFQUFFLENBQUM7WUFDWixlQUFlLEVBQUUsQ0FBQztZQUNsQixRQUFRLEVBQUUsQ0FBQztZQUNYLFNBQVMsRUFBRSxDQUFDO1lBQ1osUUFBUSxFQUFFLENBQUM7WUFDWCxRQUFRLEVBQUUsQ0FBQztZQUNYLFNBQVMsRUFBRSxDQUFDO1lBQ1osUUFBUSxFQUFFLENBQUM7U0FDWixDQUFDO1FBQ0YsTUFBTSxLQUFLLEdBQUcsSUFBQSxlQUFLLEdBQUUsQ0FBQztRQUN0QixNQUFNLGdCQUFnQixHQUFHLEtBQUssQ0FBQyxRQUFRLENBQUMsQ0FBQyxFQUFFLE1BQU0sQ0FBQyxDQUFDLE9BQU8sQ0FBQyxNQUFNLENBQUMsQ0FBQztRQUNuRSxNQUFNLElBQUksR0FBRyxNQUFNLElBQUksQ0FBQyxnQ0FBZ0MsQ0FBQyxLQUFLLENBQUMsT0FBTyxFQUFFLGdCQUFnQixDQUFDLE1BQU0sRUFBRSxFQUFFLEtBQUssQ0FBQyxNQUFNLEVBQUUsQ0FBQyxDQUFDO1FBQ25ILElBQUksSUFBSSxDQUFDLE1BQU0sR0FBRyxDQUFDLEVBQUU7WUFDbkIsTUFBTSxRQUFRLEdBQUcsTUFBTSxDQUFDLEtBQUssQ0FBQyxNQUFNLENBQUMsVUFBVSxDQUFDLENBQUMsQ0FBQztZQUNsRCxNQUFNLFlBQVksR0FBRyxNQUFNLENBQUMsS0FBSyxDQUFDLFFBQVEsQ0FBQyxDQUFDLEVBQUUsS0FBSyxDQUFDLENBQUMsTUFBTSxDQUFDLFVBQVUsQ0FBQyxDQUFDLENBQUM7WUFDekUsTUFBTSxrQkFBa0IsR0FBRyxNQUFNLENBQUMsS0FBSyxDQUFDLFFBQVEsQ0FBQyxDQUFDLEVBQUUsTUFBTSxDQUFDLENBQUMsT0FBTyxDQUFDLE1BQU0sQ0FBQyxDQUFDLE1BQU0sQ0FBQyxVQUFVLENBQUMsQ0FBQyxDQUFDO1lBQ2hHLE1BQU0sbUJBQW1CLEdBQUcsTUFBTSxDQUFDLEtBQUssQ0FBQyxPQUFPLENBQUMsTUFBTSxDQUFDLENBQUMsTUFBTSxDQUFDLFVBQVUsQ0FBQyxDQUFDLENBQUM7WUFDN0UsTUFBTSxpQkFBaUIsR0FBRyxNQUFNLENBQUMsS0FBSyxDQUFDLEtBQUssQ0FBQyxNQUFNLENBQUMsQ0FBQyxNQUFNLENBQUMsVUFBVSxDQUFDLENBQUMsQ0FBQztZQUN6RSxNQUFNLG9CQUFvQixHQUFHLE1BQU0sQ0FBQyxLQUFLLENBQUMsT0FBTyxDQUFDLE9BQU8sQ0FBQyxDQUFDLE1BQU0sQ0FBQyxVQUFVLENBQUMsQ0FBQyxDQUFDO1lBQy9FLE1BQU0sa0JBQWtCLEdBQUcsTUFBTSxDQUFDLEtBQUssQ0FBQyxLQUFLLENBQUMsT0FBTyxDQUFDLENBQUMsTUFBTSxDQUFDLFVBQVUsQ0FBQyxDQUFDLENBQUM7WUFDM0UsTUFBTSxtQkFBbUIsR0FBRyxNQUFNLENBQUMsS0FBSyxDQUFDLE9BQU8sQ0FBQyxNQUFNLENBQUMsQ0FBQyxNQUFNLENBQUMsVUFBVSxDQUFDLENBQUMsQ0FBQztZQUM3RSxNQUFNLGlCQUFpQixHQUFHLE1BQU0sQ0FBQyxLQUFLLENBQUMsS0FBSyxDQUFDLE1BQU0sQ0FBQyxDQUFDLE1BQU0sQ0FBQyxVQUFVLENBQUMsQ0FBQyxDQUFDO1lBQ3pFLE1BQU0sbUJBQW1CLEdBQUcsTUFBTSxDQUFDLEtBQUssQ0FBQyxRQUFRLENBQUMsQ0FBQyxFQUFFLE1BQU0sQ0FBQyxDQUFDLE9BQU8sQ0FBQyxNQUFNLENBQUMsQ0FBQyxNQUFNLENBQUMsVUFBVSxDQUFDLENBQUMsQ0FBQztZQUNqRyxNQUFNLGlCQUFpQixHQUFHLE1BQU0sQ0FBQyxLQUFLLENBQUMsUUFBUSxDQUFDLENBQUMsRUFBRSxNQUFNLENBQUMsQ0FBQyxLQUFLLENBQUMsTUFBTSxDQUFDLENBQUMsTUFBTSxDQUFDLFVBQVUsQ0FBQyxDQUFDLENBQUM7WUFDN0YsTUFBTSxvQkFBb0IsR0FBRyxNQUFNLENBQUMsS0FBSyxDQUFDLFFBQVEsQ0FBQyxDQUFDLEVBQUUsT0FBTyxDQUFDLENBQUMsT0FBTyxDQUFDLE9BQU8sQ0FBQyxDQUFDLE1BQU0sQ0FBQyxVQUFVLENBQUMsQ0FBQyxDQUFDO1lBQ3BHLE1BQU0sa0JBQWtCLEdBQUcsTUFBTSxDQUFDLEtBQUssQ0FBQyxRQUFRLENBQUMsQ0FBQyxFQUFFLE9BQU8sQ0FBQyxDQUFDLEtBQUssQ0FBQyxPQUFPLENBQUMsQ0FBQyxNQUFNLENBQUMsVUFBVSxDQUFDLENBQUMsQ0FBQztZQUNoRyxNQUFNLG1CQUFtQixHQUFHLE1BQU0sQ0FBQyxLQUFLLENBQUMsUUFBUSxDQUFDLENBQUMsRUFBRSxNQUFNLENBQUMsQ0FBQyxPQUFPLENBQUMsTUFBTSxDQUFDLENBQUMsTUFBTSxDQUFDLFVBQVUsQ0FBQyxDQUFDLENBQUM7WUFDakcsTUFBTSxpQkFBaUIsR0FBRyxNQUFNLENBQUMsS0FBSyxDQUFDLFFBQVEsQ0FBQyxDQUFDLEVBQUUsTUFBTSxDQUFDLENBQUMsS0FBSyxDQUFDLE1BQU0sQ0FBQyxDQUFDLE1BQU0sQ0FBQyxVQUFVLENBQUMsQ0FBQyxDQUFDO1lBRTdGLEtBQUssTUFBTSxHQUFHLElBQUksSUFBSSxFQUFFO2dCQUN0QixLQUFLLElBQUksQ0FBQyxHQUFHLENBQUMsRUFBRSxDQUFDLElBQUksRUFBRSxFQUFFLENBQUMsRUFBRSxFQUFFO29CQUM1QixNQUFNLEdBQUcsR0FBRyxNQUFNLENBQUMsQ0FBQyxDQUFDLENBQUMsUUFBUSxDQUFDLENBQUMsRUFBRSxHQUFHLENBQUMsQ0FBQztvQkFDdkMsTUFBTSxLQUFLLEdBQUcsSUFBSSxHQUFHLEVBQXNCLENBQUM7b0JBQzVDLE1BQU0sT0FBTyxHQUFHLEdBQUcsQ0FBQyxLQUFLLENBQVcsQ0FBQztvQkFDckMsSUFBSSxDQUFDLE9BQU87d0JBQUUsU0FBUztvQkFDdkIsTUFBTSxNQUFNLEdBQUcsR0FBRyxDQUFDLFNBQVMsR0FBRyxHQUFHLEdBQUcsQ0FBQyxDQUFDO29CQUN2QyxJQUFJLE1BQU0sS0FBSyxRQUFRO3dCQUFFLFFBQVEsQ0FBQyxLQUFLLElBQUksT0FBTyxDQUFDO29CQUNuRCxJQUFJLE1BQU0sS0FBSyxZQUFZO3dCQUFFLFFBQVEsQ0FBQyxTQUFTLElBQUksT0FBTyxDQUFDO29CQUMzRCxJQUFJLE1BQU0sS0FBSyxrQkFBa0I7d0JBQUUsUUFBUSxDQUFDLGVBQWUsSUFBSSxPQUFPLENBQUM7b0JBQ3ZFLElBQUksTUFBTSxJQUFJLG1CQUFtQixJQUFJLE1BQU0sSUFBSSxpQkFBaUI7d0JBQUUsUUFBUSxDQUFDLFFBQVEsSUFBSSxPQUFPLENBQUM7b0JBQy9GLElBQUksTUFBTSxJQUFJLG9CQUFvQixJQUFJLE1BQU0sSUFBSSxrQkFBa0I7d0JBQUUsUUFBUSxDQUFDLFNBQVMsSUFBSSxPQUFPLENBQUM7b0JBQ2xHLElBQUksTUFBTSxJQUFJLG1CQUFtQixJQUFJLE1BQU0sSUFBSSxpQkFBaUI7d0JBQUUsUUFBUSxDQUFDLFFBQVEsSUFBSSxPQUFPLENBQUM7b0JBQy9GLElBQUksTUFBTSxJQUFJLG1CQUFtQixJQUFJLE1BQU0sSUFBSSxpQkFBaUI7d0JBQUUsUUFBUSxDQUFDLFFBQVEsSUFBSSxPQUFPLENBQUM7b0JBQy9GLElBQUksTUFBTSxJQUFJLG9CQUFvQixJQUFJLE1BQU0sSUFBSSxrQkFBa0I7d0JBQUUsUUFBUSxDQUFDLFNBQVMsSUFBSSxPQUFPLENBQUM7b0JBQ2xHLElBQUksTUFBTSxJQUFJLG1CQUFtQixJQUFJLE1BQU0sSUFBSSxpQkFBaUI7d0JBQUUsUUFBUSxDQUFDLFFBQVEsSUFBSSxPQUFPLENBQUM7aUJBQ2hHO2FBQ0Y7U0FDRjtRQUNELE9BQU8sUUFBUSxDQUFDO0lBQ2xCLENBQUM7SUFFRCxLQUFLLENBQUMsU0FBUztRQUNiLE1BQU0sWUFBWSxHQUFHLE1BQU0sSUFBSSxDQUFDLGlCQUFpQixDQUFDLFVBQVUsRUFBRSxDQUFDO1FBQy9ELE1BQU0sUUFBUSxHQUFHLE1BQU0sSUFBSSxDQUFDLHFCQUFxQixFQUFFLENBQUM7UUFFcEQsTUFBTSxVQUFVLEdBQUcsTUFBTSxJQUFJLENBQUMsZ0JBQWdCLENBQUMsYUFBYSxFQUFFLENBQUM7UUFDL0QsTUFBTSxTQUFTLEdBQWM7WUFDM0IsR0FBRyxZQUFZO1lBQ2YsUUFBUTtZQUNSLFlBQVksRUFBRSxVQUFVLElBQUksVUFBVSxDQUFDLEVBQUUsSUFBSSxDQUFDO1lBQzlDLFNBQVMsRUFBRSxJQUFJLElBQUksRUFBRSxDQUFDLFdBQVcsRUFBRTtZQUNuQyxhQUFhLEVBQUUsRUFBc0M7WUFDckQsa0JBQWtCLEVBQUUsRUFBRTtTQUN2QixDQUFDO1FBRUYsTUFBTSxLQUFLLEdBQUcsTUFBTSxJQUFJLENBQUMsY0FBYyxDQUFDLG9CQUFvQixDQUFDLEVBQUUsSUFBSSxFQUFFLGVBQVEsQ0FBQyxhQUFhLEVBQUUsQ0FBQyxDQUFDO1FBQy9GLEtBQUssTUFBTSxJQUFJLElBQUksS0FBSyxFQUFFO1lBQ3hCLG1CQUFtQjtZQUNuQixNQUFNLElBQUksR0FBRyxJQUFJLENBQUMsSUFBNkIsQ0FBQztZQUNoRCwwQkFBMEI7WUFDMUIsTUFBTSxRQUFRLEdBQUcsTUFBTSxJQUFJLENBQUMsc0JBQXNCLENBQUMsZ0JBQWdCLENBQUMsSUFBSSxDQUFDLFVBQW9CLENBQUMsQ0FBQztZQUMvRixJQUFJLFFBQVEsRUFBRTtnQkFDWixTQUFTLENBQUMsa0JBQWtCLENBQUMsSUFBSSxDQUFDO29CQUNoQyxHQUFHLElBQUk7b0JBQ1AsZUFBZSxFQUFFLFFBQVEsRUFBRSxJQUFJO29CQUMvQixrQkFBa0IsRUFBRSxRQUFRLEVBQUUsWUFBWTtvQkFDMUMsYUFBYSxFQUFFLFFBQVEsRUFBRSxJQUFJO2lCQUM5QixDQUFDLENBQUM7YUFDSjtZQUVELG1CQUFtQjtZQUNuQixJQUFJLElBQUksQ0FBQyxVQUFVLEtBQUsseUJBQWEsRUFBRTtnQkFDckMsU0FBUyxDQUFDLGFBQWEsR0FBRyxJQUFJLENBQUM7YUFDaEM7U0FDRjtRQUVELE1BQU0sSUFBSSxDQUFDLFlBQVksQ0FBQyxhQUFhLENBQUMsU0FBUyxDQUFDLENBQUM7UUFDakQsSUFBSSxDQUFDLE1BQU0sQ0FBQyxJQUFJLENBQUMsNENBQTRDLEVBQUUsU0FBUyxDQUFDLENBQUM7SUFDNUUsQ0FBQztDQUNGLENBQUE7QUFuSFksMENBQWU7QUFFVDtJQURoQixJQUFBLGFBQU0sR0FBRTs7K0NBQzBCO0FBR2xCO0lBRGhCLElBQUEsYUFBTSxHQUFFOzhCQUMyQixxQ0FBaUI7MERBQUM7QUFHckM7SUFEaEIsSUFBQSxhQUFNLEdBQUU7OEJBQ3dCLCtCQUFjO3VEQUFDO0FBRy9CO0lBRGhCLElBQUEsYUFBTSxHQUFFOzhCQUMwQixtQ0FBZ0I7eURBQUM7QUFHbkM7SUFEaEIsSUFBQSxhQUFNLEdBQUU7OEJBQzBDLG1FQUFnQzt5RUFBQztBQUduRTtJQURoQixJQUFBLGFBQU0sR0FBRTs4QkFDc0IsMkJBQVk7cURBQUM7QUFHM0I7SUFEaEIsSUFBQSxhQUFNLEdBQUU7OEJBQ2dDLCtDQUFzQjsrREFBQzswQkFwQnJELGVBQWU7SUFUM0IsSUFBQSxtQkFBUSxFQUFpQjtRQUN4QixJQUFJLEVBQUUsdUJBQVksQ0FBQyxNQUFNO1FBQ3pCLFlBQVksRUFBRTtZQUNaLFFBQVEsRUFBRSxLQUFLO1NBQ2hCO0tBQ0YsRUFBRTtRQUNELG9DQUFvQztRQUNwQyxTQUFTLEVBQUUsT0FBTyxDQUFDLEdBQUcsQ0FBQyxRQUFRLEtBQUssTUFBTTtLQUMzQyxDQUFDO0dBQ1csZUFBZSxDQW1IM0IifQ==