为何调用DataLoader的get方法后会自动触发set方法？

原因解析 #

DataLoader的设计逻辑就是如此：当调用load(key)时，它会先通过自定义CacheMap的get方法检查缓存是否命中。如果缓存中没有该key的结果（即get返回undefined），DataLoader会将该key加入批量处理队列，执行你定义的批量加载函数。当批量加载完成拿到结果后，DataLoader会自动调用CacheMap的set方法将结果存入缓存——这是它内置的缓存优化机制，目的是让后续相同key的load请求直接复用缓存结果，避免重复执行批量查询。

解决方案 #

根据你的需求，有两种合理的处理方式：

1. 完全禁用DataLoader的自动缓存 #

如果不需要DataLoader自动帮你缓存加载结果，在创建DataLoader实例时，将cache选项设置为false即可。这样DataLoader只会调用CacheMap的get方法做缓存检查，不会在加载完成后自动调用set。

修改你的CustomDataloader.ts：

import DataLoader from 'dataloader';
import { CustomCacheMap } from './cacheMap';

export class CustomDataLoader {
  private loader: DataLoader<string, string>;

  constructor() {
    this.loader = new DataLoader(async (keys) => {
      console.log('Executing batch load for keys:', keys);
      return keys.map(key => `Result for ${key}`);
    }, {
      cacheMap: new CustomCacheMap(),
      cache: false // 新增：禁用自动缓存
    });
  }

  async load(key: string) {
    return this.loader.load(key);
  }
}

修改后控制台输出将不再出现set的日志：

CacheMap get called with key: myKey
Executing batch load for keys: [ 'myKey' ]

2. 自定义CacheMap的set逻辑 #

如果需要保留部分缓存行为，可在自定义CacheMap的set方法中添加条件判断，控制何时执行缓存存储。比如只缓存特定key，或者满足某些业务条件时才缓存：

修改你的cacheMap.ts：

export class CustomCacheMap {
  private cache = new Map<string, string>();

  get(key: string) {
    console.log(`CacheMap get called with key: ${key}`);
    return this.cache.get(key);
  }

  set(key: string, value: string) {
    // 示例：仅当key不是'myKey'时才缓存
    if (key !== 'myKey') {
      console.log(`CacheMap set called with key: ${key}, value: ${value}`);
      this.cache.set(key, value);
    }
  }

  delete(key: string) {
    return this.cache.delete(key);
  }

  clear() {
    this.cache.clear();
  }
}

这样调用load('myKey')时，set方法会执行但不会实际存储该key的结果。

内容的提问来源于stack exchange，提问作者hawk_lv