深入 DataLoader：基于请求合并与缓存失效的自定义批处理策略

本教程将带你超越 DataLoader 的基本用法，实现针对不同查询模式的动态批处理策略，并讲解如何管理缓存生命周期以避免脏数据。 · 难度：入门 · +10XP

深入 DataLoader：基于请求合并与缓存失效的自定义批处理策略

大多数教程只介绍 DataLoader 的单次批量加载，但实际项目中需要处理跨请求的缓存复用、批量大小控制以及根据字段动态切换加载逻辑。本教程将实现一个可配置的 DataLoader 工厂，支持批处理窗口延迟、手动缓存清除和基于上下文的键选择器。同时，你将学会如何利用 DataLoader 的缓存机制解决 N+1 问题，并避免因长连接导致的缓存过期问题。

class CustomDataLoader {
  constructor(batchFn, options = {}) {
    this.batchFn = batchFn;
    this.cache = new Map();
    this.batchWindowMs = options.batchWindowMs || 100;
    this.maxBatchSize = options.maxBatchSize || 100;
    this._queue = [];
    this._timer = null;
  }
  load(key) {
    if (this.cache.has(key)) return Promise.resolve(this.cache.get(key));
    return new Promise(resolve => {
      this._queue.push({ key, resolve });
      this._scheduleBatch();
    });
  }
  _scheduleBatch() {
    if (this._timer) clearTimeout(this._timer);
    this._timer = setTimeout(() => {
      const batch = this._queue.splice(0, this.maxBatchSize);
      this.batchFn(batch.map(b => b.key)).then(values => {
        batch.forEach((b, i) => {
          this.cache.set(b.key, values[i]);
          b.resolve(values[i]);
        });
      });
    }, this.batchWindowMs);
  }
  clearCache(key) { this.cache.delete(key); }
}