cnpmcore
Version:
Private NPM Registry for Enterprise
158 lines • 14.9 kB
JavaScript
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);
};
import { Inject } from 'egg';
import { Schedule, ScheduleType } from 'egg/schedule';
import { GLOBAL_WORKER } from "../../common/constants.js";
import dayjs from "../../common/dayjs.js";
import { TaskType } from "../../common/enum/Task.js";
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 = dayjs();
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,
packageCount: Number(packageTotal.packageCount),
packageVersionCount: Number(packageTotal.packageVersionCount),
download,
lastChangeId: (lastChange && lastChange.id) || 0,
cacheTime: new Date().toISOString(),
changesStream: {},
upstreamRegistries: [],
};
const tasks = await this.taskRepository.findTasksByCondition({
type: 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 === GLOBAL_WORKER) {
totalData.changesStream = data;
}
}
await this.cacheService.saveTotalData(totalData);
this.logger.info('[UpdateTotalData.subscribe] total data: %j', totalData);
}
};
__decorate([
Inject(),
__metadata("design:type", Function)
], UpdateTotalData.prototype, "logger", void 0);
__decorate([
Inject(),
__metadata("design:type", Function)
], UpdateTotalData.prototype, "packageRepository", void 0);
__decorate([
Inject(),
__metadata("design:type", Function)
], UpdateTotalData.prototype, "taskRepository", void 0);
__decorate([
Inject(),
__metadata("design:type", Function)
], UpdateTotalData.prototype, "changeRepository", void 0);
__decorate([
Inject(),
__metadata("design:type", Function)
], UpdateTotalData.prototype, "packageVersionDownloadRepository", void 0);
__decorate([
Inject(),
__metadata("design:type", Function)
], UpdateTotalData.prototype, "cacheService", void 0);
__decorate([
Inject(),
__metadata("design:type", Function)
], UpdateTotalData.prototype, "registryManagerService", void 0);
UpdateTotalData = __decorate([
Schedule({
type: ScheduleType.WORKER,
scheduleData: {
interval: 60_000,
},
}, {
// immediate = false on unittest env
immediate: process.env.NODE_ENV !== 'test',
})
], UpdateTotalData);
export { UpdateTotalData };
//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiVXBkYXRlVG90YWxEYXRhLmpzIiwic291cmNlUm9vdCI6IiIsInNvdXJjZXMiOlsiLi4vLi4vLi4vLi4vYXBwL3BvcnQvc2NoZWR1bGUvVXBkYXRlVG90YWxEYXRhLnRzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiI7Ozs7Ozs7OztBQUFBLE9BQU8sRUFBRSxNQUFNLEVBQVUsTUFBTSxLQUFLLENBQUM7QUFDckMsT0FBTyxFQUFFLFFBQVEsRUFBRSxZQUFZLEVBQXVCLE1BQU0sY0FBYyxDQUFDO0FBRTNFLE9BQU8sRUFBRSxhQUFhLEVBQUUsTUFBTSwyQkFBMkIsQ0FBQztBQUMxRCxPQUFPLEtBQUssTUFBTSx1QkFBdUIsQ0FBQztBQUMxQyxPQUFPLEVBQUUsUUFBUSxFQUFFLE1BQU0sMkJBQTJCLENBQUM7QUFxQjlDLElBQU0sZUFBZSxHQUFyQixNQUFNLGVBQWU7SUFzQjFCLGdDQUFnQztJQUN4QixLQUFLLENBQUMscUJBQXFCO1FBQ2pDLE1BQU0sUUFBUSxHQUFpQjtZQUM3QixLQUFLLEVBQUUsQ0FBQztZQUNSLFNBQVMsRUFBRSxDQUFDO1lBQ1osZUFBZSxFQUFFLENBQUM7WUFDbEIsUUFBUSxFQUFFLENBQUM7WUFDWCxTQUFTLEVBQUUsQ0FBQztZQUNaLFFBQVEsRUFBRSxDQUFDO1lBQ1gsUUFBUSxFQUFFLENBQUM7WUFDWCxTQUFTLEVBQUUsQ0FBQztZQUNaLFFBQVEsRUFBRSxDQUFDO1NBQ1osQ0FBQztRQUNGLE1BQU0sS0FBSyxHQUFHLEtBQUssRUFBRSxDQUFDO1FBQ3RCLE1BQU0sZ0JBQWdCLEdBQUcsS0FBSyxDQUFDLFFBQVEsQ0FBQyxDQUFDLEVBQUUsTUFBTSxDQUFDLENBQUMsT0FBTyxDQUFDLE1BQU0sQ0FBQyxDQUFDO1FBQ25FLE1BQU0sSUFBSSxHQUFHLE1BQU0sSUFBSSxDQUFDLGdDQUFnQyxDQUFDLEtBQUssQ0FBQyxPQUFPLEVBQUUsZ0JBQWdCLENBQUMsTUFBTSxFQUFFLEVBQUUsS0FBSyxDQUFDLE1BQU0sRUFBRSxDQUFDLENBQUM7UUFDbkgsSUFBSSxJQUFJLENBQUMsTUFBTSxHQUFHLENBQUMsRUFBRSxDQUFDO1lBQ3BCLE1BQU0sUUFBUSxHQUFHLE1BQU0sQ0FBQyxLQUFLLENBQUMsTUFBTSxDQUFDLFVBQVUsQ0FBQyxDQUFDLENBQUM7WUFDbEQsTUFBTSxZQUFZLEdBQUcsTUFBTSxDQUFDLEtBQUssQ0FBQyxRQUFRLENBQUMsQ0FBQyxFQUFFLEtBQUssQ0FBQyxDQUFDLE1BQU0sQ0FBQyxVQUFVLENBQUMsQ0FBQyxDQUFDO1lBQ3pFLE1BQU0sa0JBQWtCLEdBQUcsTUFBTSxDQUFDLEtBQUssQ0FBQyxRQUFRLENBQUMsQ0FBQyxFQUFFLE1BQU0sQ0FBQyxDQUFDLE9BQU8sQ0FBQyxNQUFNLENBQUMsQ0FBQyxNQUFNLENBQUMsVUFBVSxDQUFDLENBQUMsQ0FBQztZQUNoRyxNQUFNLG1CQUFtQixHQUFHLE1BQU0sQ0FBQyxLQUFLLENBQUMsT0FBTyxDQUFDLE1BQU0sQ0FBQyxDQUFDLE1BQU0sQ0FBQyxVQUFVLENBQUMsQ0FBQyxDQUFDO1lBQzdFLE1BQU0saUJBQWlCLEdBQUcsTUFBTSxDQUFDLEtBQUssQ0FBQyxLQUFLLENBQUMsTUFBTSxDQUFDLENBQUMsTUFBTSxDQUFDLFVBQVUsQ0FBQyxDQUFDLENBQUM7WUFDekUsTUFBTSxvQkFBb0IsR0FBRyxNQUFNLENBQUMsS0FBSyxDQUFDLE9BQU8sQ0FBQyxPQUFPLENBQUMsQ0FBQyxNQUFNLENBQUMsVUFBVSxDQUFDLENBQUMsQ0FBQztZQUMvRSxNQUFNLGtCQUFrQixHQUFHLE1BQU0sQ0FBQyxLQUFLLENBQUMsS0FBSyxDQUFDLE9BQU8sQ0FBQyxDQUFDLE1BQU0sQ0FBQyxVQUFVLENBQUMsQ0FBQyxDQUFDO1lBQzNFLE1BQU0sbUJBQW1CLEdBQUcsTUFBTSxDQUFDLEtBQUssQ0FBQyxPQUFPLENBQUMsTUFBTSxDQUFDLENBQUMsTUFBTSxDQUFDLFVBQVUsQ0FBQyxDQUFDLENBQUM7WUFDN0UsTUFBTSxpQkFBaUIsR0FBRyxNQUFNLENBQUMsS0FBSyxDQUFDLEtBQUssQ0FBQyxNQUFNLENBQUMsQ0FBQyxNQUFNLENBQUMsVUFBVSxDQUFDLENBQUMsQ0FBQztZQUN6RSxNQUFNLG1CQUFtQixHQUFHLE1BQU0sQ0FBQyxLQUFLLENBQUMsUUFBUSxDQUFDLENBQUMsRUFBRSxNQUFNLENBQUMsQ0FBQyxPQUFPLENBQUMsTUFBTSxDQUFDLENBQUMsTUFBTSxDQUFDLFVBQVUsQ0FBQyxDQUFDLENBQUM7WUFDakcsTUFBTSxpQkFBaUIsR0FBRyxNQUFNLENBQUMsS0FBSyxDQUFDLFFBQVEsQ0FBQyxDQUFDLEVBQUUsTUFBTSxDQUFDLENBQUMsS0FBSyxDQUFDLE1BQU0sQ0FBQyxDQUFDLE1BQU0sQ0FBQyxVQUFVLENBQUMsQ0FBQyxDQUFDO1lBQzdGLE1BQU0sb0JBQW9CLEdBQUcsTUFBTSxDQUFDLEtBQUssQ0FBQyxRQUFRLENBQUMsQ0FBQyxFQUFFLE9BQU8sQ0FBQyxDQUFDLE9BQU8sQ0FBQyxPQUFPLENBQUMsQ0FBQyxNQUFNLENBQUMsVUFBVSxDQUFDLENBQUMsQ0FBQztZQUNwRyxNQUFNLGtCQUFrQixHQUFHLE1BQU0sQ0FBQyxLQUFLLENBQUMsUUFBUSxDQUFDLENBQUMsRUFBRSxPQUFPLENBQUMsQ0FBQyxLQUFLLENBQUMsT0FBTyxDQUFDLENBQUMsTUFBTSxDQUFDLFVBQVUsQ0FBQyxDQUFDLENBQUM7WUFDaEcsTUFBTSxtQkFBbUIsR0FBRyxNQUFNLENBQUMsS0FBSyxDQUFDLFFBQVEsQ0FBQyxDQUFDLEVBQUUsTUFBTSxDQUFDLENBQUMsT0FBTyxDQUFDLE1BQU0sQ0FBQyxDQUFDLE1BQU0sQ0FBQyxVQUFVLENBQUMsQ0FBQyxDQUFDO1lBQ2pHLE1BQU0saUJBQWlCLEdBQUcsTUFBTSxDQUFDLEtBQUssQ0FBQyxRQUFRLENBQUMsQ0FBQyxFQUFFLE1BQU0sQ0FBQyxDQUFDLEtBQUssQ0FBQyxNQUFNLENBQUMsQ0FBQyxNQUFNLENBQUMsVUFBVSxDQUFDLENBQUMsQ0FBQztZQUU3RixLQUFLLE1BQU0sR0FBRyxJQUFJLElBQUksRUFBRSxDQUFDO2dCQUN2QixLQUFLLElBQUksQ0FBQyxHQUFHLENBQUMsRUFBRSxDQUFDLElBQUksRUFBRSxFQUFFLENBQUMsRUFBRSxFQUFFLENBQUM7b0JBQzdCLE1BQU0sR0FBRyxHQUFHLE1BQU0sQ0FBQyxDQUFDLENBQUMsQ0FBQyxRQUFRLENBQUMsQ0FBQyxFQUFFLEdBQUcsQ0FBQyxDQUFDO29CQUN2QyxNQUFNLEtBQUssR0FBRyxJQUFJLEdBQUcsRUFBc0IsQ0FBQztvQkFDNUMsTUFBTSxPQUFPLEdBQUcsR0FBRyxDQUFDLEtBQUssQ0FBVyxDQUFDO29CQUNyQyxJQUFJLENBQUMsT0FBTzt3QkFBRSxTQUFTO29CQUN2QixNQUFNLE1BQU0sR0FBRyxHQUFHLENBQUMsU0FBUyxHQUFHLEdBQUcsR0FBRyxDQUFDLENBQUM7b0JBQ3ZDLElBQUksTUFBTSxLQUFLLFFBQVE7d0JBQUUsUUFBUSxDQUFDLEtBQUssSUFBSSxPQUFPLENBQUM7b0JBQ25ELElBQUksTUFBTSxLQUFLLFlBQVk7d0JBQUUsUUFBUSxDQUFDLFNBQVMsSUFBSSxPQUFPLENBQUM7b0JBQzNELElBQUksTUFBTSxLQUFLLGtCQUFrQjt3QkFBRSxRQUFRLENBQUMsZUFBZSxJQUFJLE9BQU8sQ0FBQztvQkFDdkUsSUFBSSxNQUFNLElBQUksbUJBQW1CLElBQUksTUFBTSxJQUFJLGlCQUFpQjt3QkFBRSxRQUFRLENBQUMsUUFBUSxJQUFJLE9BQU8sQ0FBQztvQkFDL0YsSUFBSSxNQUFNLElBQUksb0JBQW9CLElBQUksTUFBTSxJQUFJLGtCQUFrQjt3QkFBRSxRQUFRLENBQUMsU0FBUyxJQUFJLE9BQU8sQ0FBQztvQkFDbEcsSUFBSSxNQUFNLElBQUksbUJBQW1CLElBQUksTUFBTSxJQUFJLGlCQUFpQjt3QkFBRSxRQUFRLENBQUMsUUFBUSxJQUFJLE9BQU8sQ0FBQztvQkFDL0YsSUFBSSxNQUFNLElBQUksbUJBQW1CLElBQUksTUFBTSxJQUFJLGlCQUFpQjt3QkFBRSxRQUFRLENBQUMsUUFBUSxJQUFJLE9BQU8sQ0FBQztvQkFDL0YsSUFBSSxNQUFNLElBQUksb0JBQW9CLElBQUksTUFBTSxJQUFJLGtCQUFrQjt3QkFBRSxRQUFRLENBQUMsU0FBUyxJQUFJLE9BQU8sQ0FBQztvQkFDbEcsSUFBSSxNQUFNLElBQUksbUJBQW1CLElBQUksTUFBTSxJQUFJLGlCQUFpQjt3QkFBRSxRQUFRLENBQUMsUUFBUSxJQUFJLE9BQU8sQ0FBQztnQkFDakcsQ0FBQztZQUNILENBQUM7UUFDSCxDQUFDO1FBQ0QsT0FBTyxRQUFRLENBQUM7SUFDbEIsQ0FBQztJQUVELEtBQUssQ0FBQyxTQUFTO1FBQ2IsTUFBTSxZQUFZLEdBQUcsTUFBTSxJQUFJLENBQUMsaUJBQWlCLENBQUMsVUFBVSxFQUFFLENBQUM7UUFDL0QsTUFBTSxRQUFRLEdBQUcsTUFBTSxJQUFJLENBQUMscUJBQXFCLEVBQUUsQ0FBQztRQUVwRCxNQUFNLFVBQVUsR0FBRyxNQUFNLElBQUksQ0FBQyxnQkFBZ0IsQ0FBQyxhQUFhLEVBQUUsQ0FBQztRQUMvRCxNQUFNLFNBQVMsR0FBYztZQUMzQixHQUFHLFlBQVk7WUFDZixZQUFZLEVBQUUsTUFBTSxDQUFDLFlBQVksQ0FBQyxZQUFZLENBQUM7WUFDL0MsbUJBQW1CLEVBQUUsTUFBTSxDQUFDLFlBQVksQ0FBQyxtQkFBbUIsQ0FBQztZQUM3RCxRQUFRO1lBQ1IsWUFBWSxFQUFFLENBQUMsVUFBVSxJQUFJLFVBQVUsQ0FBQyxFQUFFLENBQUMsSUFBSSxDQUFDO1lBQ2hELFNBQVMsRUFBRSxJQUFJLElBQUksRUFBRSxDQUFDLFdBQVcsRUFBRTtZQUNuQyxhQUFhLEVBQUUsRUFBc0M7WUFDckQsa0JBQWtCLEVBQUUsRUFBRTtTQUN2QixDQUFDO1FBRUYsTUFBTSxLQUFLLEdBQUcsTUFBTSxJQUFJLENBQUMsY0FBYyxDQUFDLG9CQUFvQixDQUFDO1lBQzNELElBQUksRUFBRSxRQUFRLENBQUMsYUFBYTtTQUM3QixDQUFDLENBQUM7UUFDSCxLQUFLLE1BQU0sSUFBSSxJQUFJLEtBQUssRUFBRSxDQUFDO1lBQ3pCLG1CQUFtQjtZQUNuQixNQUFNLElBQUksR0FBRyxJQUFJLENBQUMsSUFBNkIsQ0FBQztZQUNoRCwwQkFBMEI7WUFDMUIsTUFBTSxRQUFRLEdBQUcsTUFBTSxJQUFJLENBQUMsc0JBQXNCLENBQUMsZ0JBQWdCLENBQUMsSUFBSSxDQUFDLFVBQW9CLENBQUMsQ0FBQztZQUMvRixJQUFJLFFBQVEsRUFBRSxDQUFDO2dCQUNiLFNBQVMsQ0FBQyxrQkFBa0IsQ0FBQyxJQUFJLENBQUM7b0JBQ2hDLEdBQUcsSUFBSTtvQkFDUCxlQUFlLEVBQUUsUUFBUSxFQUFFLElBQUk7b0JBQy9CLGtCQUFrQixFQUFFLFFBQVEsRUFBRSxZQUFZO29CQUMxQyxhQUFhLEVBQUUsUUFBUSxFQUFFLElBQUk7aUJBQzlCLENBQUMsQ0FBQztZQUNMLENBQUM7WUFFRCxtQkFBbUI7WUFDbkIsSUFBSSxJQUFJLENBQUMsVUFBVSxLQUFLLGFBQWEsRUFBRSxDQUFDO2dCQUN0QyxTQUFTLENBQUMsYUFBYSxHQUFHLElBQUksQ0FBQztZQUNqQyxDQUFDO1FBQ0gsQ0FBQztRQUVELE1BQU0sSUFBSSxDQUFDLFlBQVksQ0FBQyxhQUFhLENBQUMsU0FBUyxDQUFDLENBQUM7UUFDakQsSUFBSSxDQUFDLE1BQU0sQ0FBQyxJQUFJLENBQUMsNENBQTRDLEVBQUUsU0FBUyxDQUFDLENBQUM7SUFDNUUsQ0FBQztDQUNGLENBQUE7QUFySGtCO0lBRGhCLE1BQU0sRUFBRTs7K0NBQ3VCO0FBR2Y7SUFEaEIsTUFBTSxFQUFFOzswREFDNkM7QUFHckM7SUFEaEIsTUFBTSxFQUFFOzt1REFDdUM7QUFHL0I7SUFEaEIsTUFBTSxFQUFFOzt5REFDMkM7QUFHbkM7SUFEaEIsTUFBTSxFQUFFOzt5RUFDMkU7QUFHbkU7SUFEaEIsTUFBTSxFQUFFOztxREFDbUM7QUFHM0I7SUFEaEIsTUFBTSxFQUFFOzsrREFDdUQ7QUFwQnJELGVBQWU7SUFaM0IsUUFBUSxDQUNQO1FBQ0UsSUFBSSxFQUFFLFlBQVksQ0FBQyxNQUFNO1FBQ3pCLFlBQVksRUFBRTtZQUNaLFFBQVEsRUFBRSxNQUFNO1NBQ2pCO0tBQ0YsRUFDRDtRQUNFLG9DQUFvQztRQUNwQyxTQUFTLEVBQUUsT0FBTyxDQUFDLEdBQUcsQ0FBQyxRQUFRLEtBQUssTUFBTTtLQUMzQyxDQUNGO0dBQ1ksZUFBZSxDQXVIM0IifQ==