nsgm-cli
Version:
A CLI tool to run Next/Style-components and Graphql/Mysql fullstack project
132 lines (131 loc) • 4.81 kB
JavaScript
;
var __importDefault = (this && this.__importDefault) || function (mod) {
return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.TemplateDataLoader = void 0;
exports.createTemplateDataLoader = createTemplateDataLoader;
const dataloader_1 = __importDefault(require("dataloader"));
const db_1 = __importDefault(require("../db"));
/**
* Template DataLoader
* 针对 template 表的批量数据加载器，解决 N+1 查询问题
*/
class TemplateDataLoader {
constructor() {
// 按 ID 批量加载
this.byId = new dataloader_1.default(async (ids) => {
try {
console.log(`🔍 DataLoader: 批量加载 ${ids.length} 个 template by ID`);
const placeholders = ids.map(() => "?").join(",");
const sql = `SELECT id, name FROM template WHERE id IN (${placeholders})`;
const results = await db_1.default.executeQuery(sql, [...ids]);
// 确保返回顺序与输入 keys 一致，未找到的返回 null
return ids.map((id) => results.find((row) => row.id === id) || null);
}
catch (error) {
console.error("DataLoader byId 批量加载失败:", error);
throw error;
}
}, {
cache: true,
maxBatchSize: 100,
batchScheduleFn: (callback) => setTimeout(callback, 10), // 10ms 内的请求合并
});
// 按名称批量加载
this.byName = new dataloader_1.default(async (names) => {
try {
console.log(`🔍 DataLoader: 批量加载 ${names.length} 个 template by name`);
const placeholders = names.map(() => "?").join(",");
const sql = `SELECT id, name FROM template WHERE name IN (${placeholders})`;
const results = await db_1.default.executeQuery(sql, [...names]);
// 确保返回顺序与输入 keys 一致
return names.map((name) => results.find((row) => row.name === name) || null);
}
catch (error) {
console.error("DataLoader byName 批量加载失败:", error);
throw error;
}
}, {
cache: true,
maxBatchSize: 50,
batchScheduleFn: (callback) => setTimeout(callback, 10),
});
// 按名称模糊搜索（返回数组）
this.searchByName = new dataloader_1.default(async (searchTerms) => {
try {
console.log(`🔍 DataLoader: 批量搜索 ${searchTerms.length} 个关键词`);
// 对于搜索，我们需要为每个搜索词执行独立的查询
const results = await Promise.all(searchTerms.map(async (term) => {
const sql = "SELECT id, name FROM template WHERE name LIKE ?";
return db_1.default.executeQuery(sql, [`%${term}%`]);
}));
return results;
}
catch (error) {
console.error("DataLoader searchByName 批量搜索失败:", error);
throw error;
}
}, {
cache: true,
maxBatchSize: 20, // 搜索请求较少，降低批量大小
batchScheduleFn: (callback) => setTimeout(callback, 20), // 稍长的等待时间
});
}
/**
* 清除所有缓存
*/
clearAll() {
this.byId.clearAll();
this.byName.clearAll();
this.searchByName.clearAll();
console.log("🧹 Template DataLoader 缓存已清空");
}
/**
* 清除特定 ID 的缓存
*/
clearById(id) {
this.byId.clear(id);
}
/**
* 清除特定名称的缓存
*/
clearByName(name) {
this.byName.clear(name);
}
/**
* 预加载数据到缓存
*/
prime(id, data) {
this.byId.prime(id, data);
if (data?.name) {
this.byName.prime(data.name, data);
}
}
/**
* 获取缓存统计信息
*/
getStats() {
return {
byId: {
cacheMap: this.byId.cacheMap?.size || 0,
name: "Template.byId",
},
byName: {
cacheMap: this.byName.cacheMap?.size || 0,
name: "Template.byName",
},
searchByName: {
cacheMap: this.searchByName.cacheMap?.size || 0,
name: "Template.searchByName",
},
};
}
}
exports.TemplateDataLoader = TemplateDataLoader;
/**
* 创建 Template DataLoader 实例
*/
function createTemplateDataLoader() {
return new TemplateDataLoader();
}