86 lines
2.4 KiB
TypeScript
86 lines
2.4 KiB
TypeScript
import { SingletonProto } from '@eggjs/tegg';
|
|
import { E500 } from 'egg-errors';
|
|
import { RegistryType } from '../../../common/enum/Registry.js';
|
|
import type { Registry } from '../../../core/entity/Registry.js';
|
|
import {
|
|
AbstractChangeStream,
|
|
RegistryChangesStream,
|
|
} from './AbstractChangesStream.js';
|
|
|
|
const MAX_LIMIT = 10000;
|
|
|
|
type FetchResults = {
|
|
results: {
|
|
seq: number;
|
|
type: string;
|
|
id: string;
|
|
changes: Record<string, string>[];
|
|
gmt_modified: Date;
|
|
}[];
|
|
};
|
|
|
|
@SingletonProto()
|
|
@RegistryChangesStream(RegistryType.Cnpmjsorg)
|
|
export class CnpmjsorgChangesStream extends AbstractChangeStream {
|
|
// cnpmjsorg 未实现 update_seq 字段
|
|
// 默认返回当前时间戳字符串
|
|
async getInitialSince(registry: Registry): Promise<string> {
|
|
const since = String(new Date().getTime());
|
|
this.logger.warn(
|
|
`[CnpmjsorgChangesStream.getInitialSince] since: ${since}, skip query ${registry.changeStream}`
|
|
);
|
|
return since;
|
|
}
|
|
|
|
private async tryFetch(
|
|
registry: Registry,
|
|
since: string,
|
|
limit = 1000
|
|
): Promise<{ data: FetchResults }> {
|
|
if (limit > MAX_LIMIT) {
|
|
throw new E500(
|
|
`limit too large, current since: ${since}, limit: ${limit}`
|
|
);
|
|
}
|
|
const db = this.getChangesStreamUrl(registry, since, limit);
|
|
// json mode
|
|
const res = await this.httpclient.request<FetchResults>(db, {
|
|
followRedirect: true,
|
|
timeout: 30000,
|
|
dataType: 'json',
|
|
gzip: true,
|
|
});
|
|
const { results = [] } = res.data;
|
|
if (results?.length >= limit) {
|
|
const [first] = results;
|
|
const last = results[results.length - 1];
|
|
if (first.gmt_modified === last.gmt_modified) {
|
|
return await this.tryFetch(registry, since, limit + 1000);
|
|
}
|
|
}
|
|
|
|
return res;
|
|
}
|
|
|
|
async *fetchChanges(registry: Registry, since: string) {
|
|
// ref: https://github.com/cnpm/cnpmjs.org/pull/1734
|
|
// 由于 cnpmjsorg 无法计算准确的 seq
|
|
// since 是一个时间戳，需要确保一次返回的结果中首尾两个 gmtModified 不相等
|
|
const { data } = await this.tryFetch(registry, since);
|
|
|
|
if (data.results?.length > 0) {
|
|
for (const change of data.results) {
|
|
const seq = new Date(change.gmt_modified).getTime() + '';
|
|
const fullname = change.id;
|
|
if (seq && fullname && seq !== since) {
|
|
const change = {
|
|
fullname,
|
|
seq,
|
|
};
|
|
yield change;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|