浏览代码

perf: auto import mock file

vben 4 年之前
父节点
当前提交
df6b5e926f
共有 1 个文件被更改,包括 11 次插入12 次删除
  1. 11 12
      mock/_createProductionServer.ts

+ 11 - 12
mock/_createProductionServer.ts

@@ -1,19 +1,18 @@
 import { createProdMockServer } from 'vite-plugin-mock/es/createProdMockServer';
 import { createProdMockServer } from 'vite-plugin-mock/es/createProdMockServer';
-import userMock from './sys/user';
-import menuMock from './sys/menu';
-import tableDemoMock from './demo/table-demo';
-import accountDemoMock from './demo/account';
-import selectDemoMock from './demo/select-demo';
+
+const modules = import.meta.globEager('./**/*.ts');
+
+const mockModules: any[] = [];
+Object.keys(modules).forEach((key) => {
+  if (key.includes('/_')) {
+    return;
+  }
+  mockModules.push(...modules[key].default);
+});
 
 
 /**
 /**
  * Used in a production environment. Need to manually import all modules
  * Used in a production environment. Need to manually import all modules
  */
  */
 export function setupProdMockServer() {
 export function setupProdMockServer() {
-  createProdMockServer([
-    ...userMock,
-    ...menuMock,
-    ...tableDemoMock,
-    ...accountDemoMock,
-    ...selectDemoMock,
-  ]);
+  createProdMockServer(mockModules);
 }
 }