Class LimiterFileSystem

packages/filesystem/limiter.ts:87–141

// 文件系统限速器，防止并发请求过多达到服务器限制
// 也防止上传/下载带宽占用过多超时导致失败/数据不全的问题
export default class LimiterFileSystem implements FileSystem {
  private fs: FileSystem;

  private limiter: RateLimiter;

  constructor(fs: FileSystem, limiter?: RateLimiter) {
    this.fs = fs;
    this.limiter = limiter || new RateLimiter();
  }

  verify(): Promise<void> {
    return this.limiter.execute(() => this.fs.verify(), "verify");
  }

  async open(file: FileInfo): Promise<FileReader> {
    return this.limiter.execute(async () => {
      const reader = await this.fs.open(file);
      return {
        read: (type) => this.limiter.execute(() => reader.read(type), "read"),
      };
    }, "open");
  }

  async openDir(path: string): Promise<FileSystem> {
    return this.limiter.execute(async () => {
      const fs = await this.fs.openDir(path);
      return new LimiterFileSystem(fs, this.limiter);
    }, "openDir");
  }

  async create(path: string, opts?: FileCreateOptions): Promise<FileWriter> {
    return this.limiter.execute(async () => {
      const writer = await this.fs.create(path, opts);
      return {
        write: (content) => this.limiter.execute(() => writer.write(content), "write"),
      };
    }, "create");
  }

  createDir(dir: string, opts?: FileCreateOptions): Promise<void> {
    return this.limiter.execute(() => this.fs.createDir(dir, opts), "createDir");
  }

  delete(path: string): Promise<void> {
    return this.limiter.execute(() => this.fs.delete(path), "delete");
  }

  list(): Promise<FileInfo[]> {
    return this.limiter.execute(() => this.fs.list(), "list");
  }

  getDirUrl(): Promise<string> {
    return this.limiter.execute(() => this.fs.getDirUrl(), "getDirUrl");
  }
}