nsgm-cli 2.1.21 → 2.1.23

package/server/apis/sso.js

@@ -8,8 +8,12 @@ const getLoginCredentials = () => {
   const username = process.env.LOGIN_USERNAME || 'admin'
   let passwordHash = process.env.LOGIN_PASSWORD_HASH
   
+  console.warn('[SSO] getLoginCredentials - username:', username)
+  console.warn('[SSO] getLoginCredentials - passwordHash from env:', passwordHash ? `${passwordHash.substring(0, 20)}...` : 'undefined')
+  
   // 如果环境变量被截断，尝试手动读取 .env 文件
   if (!passwordHash || passwordHash.length < 60) {
+    console.warn('[SSO] passwordHash too short or missing, trying to read from .env')
     try {
       const fs = require('fs')
       const path = require('path')
@@ -21,16 +25,17 @@ const getLoginCredentials = () => {
         if (line.startsWith('LOGIN_PASSWORD_HASH=')) {
           // 移除引号和前缀
           passwordHash = line.replace('LOGIN_PASSWORD_HASH=', '').replace(/"/g, '').trim()
+          console.warn('[SSO] Found passwordHash in .env:', `${passwordHash.substring(0, 20)}...`)
           break
         }
       }
     } catch (error) {
-      console.error('读取 .env 文件失败:', error.message)
+      console.warn('[SSO] 读取 .env 文件失败:', error.message)
     }
   }
   
   if (!passwordHash) {
-    console.warn('⚠️  警告: LOGIN_PASSWORD_HASH 环境变量未设置，使用默认密码哈希')
+    console.warn('[SSO] ⚠️  警告: LOGIN_PASSWORD_HASH 环境变量未设置，使用默认密码哈希')
     // 默认密码 "admin123" 的哈希值（仅用于开发环境）
     return {
       username,
@@ -38,6 +43,7 @@ const getLoginCredentials = () => {
     }
   }
   
+  console.warn('[SSO] getLoginCredentials - final passwordHash:', `${passwordHash.substring(0, 20)}...`)
   return { username, passwordHash }
 }
 
@@ -75,15 +81,27 @@ router.get('/ticketCheck', async (req, res) => {
   const { query } = req
   const { name } = query
   
+  console.warn('[SSO] ticketCheck called, name:', name)
+  
   try {
     // 使用 Buffer 解码 Base64 字符串，然后使用 decodeURIComponent 处理特殊字符
     const decodedBase64 = Buffer.from(name, 'base64').toString('utf-8')
+    console.warn('[SSO] Decoded Base64:', decodedBase64)
+    
     const decodedName = decodeURIComponent(decodedBase64)
+    console.warn('[SSO] Decoded name:', decodedName)
+    
     const [inputUsername, inputPassword] = decodedName.split(',')
+    console.warn('[SSO] Input username:', inputUsername, 'password length:', inputPassword?.length)
+    
+    const { username: expectedUsername, passwordHash } = getLoginCredentials()
+    console.warn('[SSO] Expected username:', expectedUsername, 'passwordHash length:', passwordHash?.length)
     
     const isValid = await validateCredentials(inputUsername, inputPassword)
+    console.warn('[SSO] Credentials valid:', isValid)
     
     if (isValid) {
+      console.warn('[SSO] Login successful for user:', inputUsername)
       res.json({
         name: 'ticketCheck',
         query,
@@ -97,7 +115,7 @@ router.get('/ticketCheck', async (req, res) => {
       })
     } else {
       // 记录失败的登录尝试（用于安全审计）
-      console.log(`登录失败 - 用户名: ${inputUsername}, 时间: ${new Date().toISOString()}, IP: ${req.ip}`)
+      console.warn(`[SSO] 登录失败 - 用户名: ${inputUsername}, 时间: ${new Date().toISOString()}, IP: ${req.ip}`)
       
       res.json({
         name: 'ticketCheck',
@@ -107,7 +125,7 @@ router.get('/ticketCheck', async (req, res) => {
       })
     }
   } catch (error) {
-    console.error('登录验证出错:', error)
+    console.warn('[SSO] 登录验证出错:', error)
     res.status(500).json({
       name: 'ticketCheck',
       returnCode: -1,

package/server/modules/template/resolver.js

@@ -46,35 +46,86 @@ module.exports = {
             throw error;
         }
     }, 
-    // 根据ID获取模板
-    templateGet: async ({ id }) => {
+    
+    // 根据ID获取模板 - 使用 DataLoader 优化
+    templateGet: async ({ id }, context) => {
         try {
             validateId(id);
             
-            const sql = 'SELECT id, name FROM template WHERE id = ?';
-            const values = [id];
-
-            console.log('根据ID查询模板:', { sql, values });
+            console.log('🚀 使用 DataLoader 根据ID查询模板:', { id });
             
-            const results = await executeQuery(sql, values);
+            // 使用 DataLoader 批量加载，自动去重和缓存
+            const result = await context.dataloaders.template.byId.load(Number(id));
             
-            if (results.length === 0) {
+            if (!result) {
                 throw new Error(`ID为 ${id} 的模板不存在`);
             }
             
-            return results[0];
+            return result;
         } catch (error) {
             console.error('获取模板失败:', error.message);
             throw error;
         }
     },
 
-    // 搜索模板（分页）
-    templateSearch: async ({ page = 0, pageSize = 10, data = {} }) => {
+    // 批量获取模板 - 新增方法，展示 DataLoader 批量能力
+    templateBatchGet: async ({ ids }, context) => {
+        try {
+            if (!Array.isArray(ids) || ids.length === 0) {
+                throw new Error('ID列表不能为空');
+            }
+            
+            // 验证所有ID
+            const validIds = ids.map(id => {
+                validateId(id);
+                return Number(id);
+            });
+            
+            console.log('🚀 使用 DataLoader 批量查询模板:', { ids: validIds });
+            
+            // DataLoader 自动批量处理，一次查询获取所有数据
+            const results = await context.dataloaders.template.byId.loadMany(validIds);
+            
+            // 过滤掉 null 结果（未找到的记录）
+            return results.filter(result => result !== null && !(result instanceof Error));
+        } catch (error) {
+            console.error('批量获取模板失败:', error.message);
+            throw error;
+        }
+    },
+
+    // 搜索模板（分页）- 使用 DataLoader 优化搜索
+    templateSearch: async ({ page = 0, pageSize = 10, data = {} }, context) => {
         try {
             validatePagination(page, pageSize);
             
             const { name } = data;
+            
+            // 如果有名称搜索，尝试使用 DataLoader 搜索缓存
+            if (name && name.trim() !== '') {
+                console.log('🚀 使用 DataLoader 搜索模板:', { searchTerm: name.trim() });
+                
+                try {
+                    // 使用 DataLoader 进行搜索（这里会缓存搜索结果）
+                    const searchResults = await context.dataloaders.template.searchByName.load(name.trim());
+                    
+                    // 手动分页处理
+                    const totalCounts = searchResults.length;
+                    const startIndex = page * pageSize;
+                    const endIndex = startIndex + pageSize;
+                    const items = searchResults.slice(startIndex, endIndex);
+                    
+                    return {
+                        totalCounts,
+                        items
+                    };
+                } catch (dataLoaderError) {
+                    console.warn('DataLoader 搜索失败，回退到直接查询:', dataLoaderError.message);
+                    // 如果 DataLoader 失败，回退到原始查询方式
+                }
+            }
+            
+            // 原始查询方式（作为备用）
             const values = [];
             const countValues = [];
             
@@ -91,7 +142,7 @@ module.exports = {
             
             values.push(pageSize, page * pageSize);
             
-            console.log('搜索模板:', { sql, values, countSql, countValues });
+            console.log('搜索模板（备用查询）:', { sql, values, countSql, countValues });
             
             return await executePaginatedQuery(sql, countSql, values, countValues);
         } catch (error) {
@@ -100,8 +151,8 @@ module.exports = {
         }
     },
 
-    // 添加模板
-    templateAdd: async ({ data }) => {
+    // 添加模板 - 添加 DataLoader 缓存清理
+    templateAdd: async ({ data }, context) => {
         try {
             const { name } = data || {};
             validateName(name);
@@ -112,7 +163,16 @@ module.exports = {
             console.log('添加模板:', { sql, values });
             
             const results = await executeQuery(sql, values);
-            return results.insertId;
+            const insertId = results.insertId;
+            
+            // 预加载新数据到 DataLoader 缓存
+            if (insertId && context?.dataloaders?.template) {
+                const newTemplate = { id: insertId, name: name.trim() };
+                context.dataloaders.template.prime(insertId, newTemplate);
+                console.log('🚀 新模板已预加载到 DataLoader 缓存:', newTemplate);
+            }
+            
+            return insertId;
         } catch (error) {
             console.error('添加模板失败:', error.message);
             throw error;
@@ -146,8 +206,8 @@ module.exports = {
         }
     },
 
-    // 更新模板
-    templateUpdate: async ({ id, data }) => {
+    // 更新模板 - 添加 DataLoader 缓存清理
+    templateUpdate: async ({ id, data }, context) => {
         try {
             validateId(id);
             const { name } = data || {};
@@ -169,6 +229,12 @@ module.exports = {
                 throw new Error(`ID为 ${id} 的模板不存在`);
             }
             
+            // 清除 DataLoader 缓存，确保下次查询获取最新数据
+            if (context?.dataloaders?.template) {
+                context.dataloaders.template.clearById(Number(id));
+                console.log('🧹 已清除 DataLoader 缓存:', { id });
+            }
+            
             return true;
         } catch (error) {
             console.error('更新模板失败:', error.message);
@@ -176,8 +242,8 @@ module.exports = {
         }
     },
 
-    // 删除模板
-    templateDelete: async ({ id }) => {
+    // 删除模板 - 添加 DataLoader 缓存清理
+    templateDelete: async ({ id }, context) => {
         try {
             validateId(id);
             
@@ -192,6 +258,12 @@ module.exports = {
                 throw new Error(`ID为 ${id} 的模板不存在`);
             }
             
+            // 清除 DataLoader 缓存
+            if (context?.dataloaders?.template) {
+                context.dataloaders.template.clearById(Number(id));
+                console.log('🧹 已清除 DataLoader 缓存:', { id });
+            }
+            
             return true;
         } catch (error) {
             console.error('删除模板失败:', error.message);
@@ -199,8 +271,8 @@ module.exports = {
         }
     },
 
-    // 批量删除模板
-    templateBatchDelete: async ({ ids }) => {
+    // 批量删除模板 - 添加 DataLoader 缓存清理
+    templateBatchDelete: async ({ ids }, context) => {
         try {
             if (!Array.isArray(ids) || ids.length === 0) {
                 throw new Error('批量删除的ID列表不能为空');
@@ -220,6 +292,14 @@ module.exports = {
                 throw new Error('没有找到要删除的模板');
             }
             
+            // 批量清除 DataLoader 缓存
+            if (context?.dataloaders?.template) {
+                ids.forEach(id => {
+                    context.dataloaders.template.clearById(Number(id));
+                });
+                console.log('🧹 已批量清除 DataLoader 缓存:', { ids });
+            }
+            
             return true;
         } catch (error) {
             console.error('批量删除模板失败:', error.message);

package/server/modules/template/schema.js

@@ -2,6 +2,7 @@ module.exports = {
     query: `
         template(page: Int, pageSize: Int): Templates
         templateGet(id: Int): Template
+        templateBatchGet(ids: [Int]): [Template]
         templateSearch(page: Int, pageSize: Int, data: TemplateSearchInput): Templates
     `,
     mutation: `