cnpmcore/app/common/adapter/changesStream/CnpmjsorgChangesStream.ts

import { SingletonProto } from 'egg';
import { E500 } from 'egg/errors';

import { RegistryType } from '../../../common/enum/Registry.ts';
import type { Registry } from '../../../core/entity/Registry.ts';
import {
  AbstractChangeStream,
  RegistryChangesStream,
} from './AbstractChangesStream.ts';

const MAX_LIMIT = 10_000;

interface 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(Date.now());
    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: 30_000,
      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;
        }
      }
    }
  }
}